You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@lucene.apache.org by be...@apache.org on 2023/11/28 19:50:55 UTC
(lucene) branch main updated: Fix *HnswVectorsFormat.testIndexedValueNotAliased test flakiness (#12848)
This is an automated email from the ASF dual-hosted git repository.
benwtrent pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/lucene.git
The following commit(s) were added to refs/heads/main by this push:
new 2bb69f32462 Fix *HnswVectorsFormat.testIndexedValueNotAliased test flakiness (#12848)
2bb69f32462 is described below
commit 2bb69f3246218dd8176cf92d8064623688c5272c
Author: Benjamin Trent <be...@gmail.com>
AuthorDate: Tue Nov 28 14:50:48 2023 -0500
Fix *HnswVectorsFormat.testIndexedValueNotAliased test flakiness (#12848)
---
.../lucene/tests/index/BaseKnnVectorsFormatTestCase.java | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/lucene/test-framework/src/java/org/apache/lucene/tests/index/BaseKnnVectorsFormatTestCase.java b/lucene/test-framework/src/java/org/apache/lucene/tests/index/BaseKnnVectorsFormatTestCase.java
index d513c6361be..6172d8eb140 100644
--- a/lucene/test-framework/src/java/org/apache/lucene/tests/index/BaseKnnVectorsFormatTestCase.java
+++ b/lucene/test-framework/src/java/org/apache/lucene/tests/index/BaseKnnVectorsFormatTestCase.java
@@ -42,6 +42,7 @@ import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriterConfig;
import org.apache.lucene.index.LeafReader;
import org.apache.lucene.index.LeafReaderContext;
+import org.apache.lucene.index.NoMergePolicy;
import org.apache.lucene.index.StoredFields;
import org.apache.lucene.index.Term;
import org.apache.lucene.index.VectorEncoding;
@@ -732,7 +733,13 @@ public abstract class BaseKnnVectorsFormatTestCase extends BaseIndexFileFormatTe
String fieldName = "field";
float[] v = {0};
try (Directory dir = newDirectory();
- IndexWriter iw = new IndexWriter(dir, newIndexWriterConfig())) {
+ IndexWriter iw =
+ new IndexWriter(
+ dir,
+ newIndexWriterConfig()
+ .setMergePolicy(NoMergePolicy.INSTANCE)
+ .setMaxBufferedDocs(3)
+ .setRAMBufferSizeMB(-1))) {
Document doc1 = new Document();
doc1.add(new KnnFloatVectorField(fieldName, v, VectorSimilarityFunction.EUCLIDEAN));
v[0] = 1;
@@ -748,6 +755,7 @@ public abstract class BaseKnnVectorsFormatTestCase extends BaseIndexFileFormatTe
try (IndexReader reader = DirectoryReader.open(iw)) {
LeafReader r = getOnlyLeafReader(reader);
FloatVectorValues vectorValues = r.getFloatVectorValues(fieldName);
+ assertEquals(3, vectorValues.size());
vectorValues.nextDoc();
assertEquals(1, vectorValues.vectorValue()[0], 0);
vectorValues.nextDoc();