Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Test fix: make float vector dims even for SQ testing

This commit is contained in:
ChrisHegarty 2024-09-20 15:27:38 +01:00
parent 73d71acedd
commit 53d1c2bd2f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lucene/test-framework/src/java/org/apache/lucene/tests/index/BaseKnnVectorsFormatTestCase.java
View File

@ -1817,7 +1817,7 @@ public abstract class BaseKnnVectorsFormatTestCase extends BaseIndexFileFormatTe
Directory dir1 = newDirectory();
IndexWriter w1 = new IndexWriter(dir1, newIndexWriterConfig());
Document doc = new Document();
doc.add(new KnnFloatVectorField("float", new float[] {1f}));
doc.add(new KnnFloatVectorField("float", new float[] {1f, 2f}));
doc.add(new KnnByteVectorField("byte", new byte[] {42}));
w1.addDocument(doc);
@ -1849,12 +1849,14 @@ public abstract class BaseKnnVectorsFormatTestCase extends BaseIndexFileFormatTe
assertNotNull(floatVectors);
assertEquals(0, floatVectors.nextDoc());
float[] vector = floatVectors.vectorValue();
assertEquals(1, vector.length);
assertEquals(2, vector.length);
assertEquals(1f, vector[0], 0f);
assertEquals(2f, vector[1], 0f);
assertEquals(1, floatVectors.nextDoc());
vector = floatVectors.vectorValue();
assertEquals(1, vector.length);
assertEquals(2, vector.length);
assertEquals(1f, vector[0], 0f);
assertEquals(2f, vector[1], 0f);
assertEquals(DocIdSetIterator.NO_MORE_DOCS, floatVectors.nextDoc());
IOUtils.close(reader, w2, dir1, dir2);