Skip to content

Commit 9c9773f

Browse files
committed
Fix test to use just floats
1 parent 863b6e6 commit 9c9773f

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

server/src/test/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldTypeTests.java

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,7 @@ public void testRescoreOversampleUsedWithoutQuantization() {
412412
DenseVectorFieldType nonQuantizedField = new DenseVectorFieldType(
413413
"f",
414414
IndexVersion.current(),
415-
FLOAT,
415+
randomFrom(FLOAT, BYTE),
416416
3,
417417
true,
418418
VectorSimilarity.COSINE,
@@ -439,7 +439,7 @@ public void testRescoreOversampleModifiesKnnParams() {
439439
DenseVectorFieldType fieldType = new DenseVectorFieldType(
440440
"f",
441441
IndexVersion.current(),
442-
randomFrom(BYTE, FLOAT),
442+
FLOAT,
443443
3,
444444
true,
445445
VectorSimilarity.COSINE,
@@ -465,7 +465,15 @@ private static void checkRescoreQueryParameters(
465465
int expectedK,
466466
int expectedCandidates
467467
) {
468-
Query query = fieldType.createKnnQuery(new VectorData(null, new byte[] { 1, 4, 10 }), k, candidates, oversample, null, null, null);
468+
Query query = fieldType.createKnnQuery(
469+
VectorData.fromFloats(new float[] { 1, 4, 10 }),
470+
k,
471+
candidates,
472+
oversample,
473+
null,
474+
null,
475+
null
476+
);
469477
RescoreKnnVectorQuery rescoreQuery = (RescoreKnnVectorQuery) query;
470478
ESKnnFloatVectorQuery esKnnQuery = (ESKnnFloatVectorQuery) rescoreQuery.innerQuery();
471479
assertThat("Unexpected total results", rescoreQuery.k(), equalTo(expectedResults));

0 commit comments

Comments
 (0)