Set NoMergePolicy for codec test (#754)

Sets NoMerge as merge policy for codec test. This allows us to reliably predict how many segments will be created. Signed-off-by: John Mazanec <[email protected]>
opensearch-project · Feb 14, 2023 · b8f2deb · b8f2deb
1 parent 5b06cfe
commit b8f2deb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/test/java/org/opensearch/knn/index/codec/KNNCodecTestCase.java b/src/test/java/org/opensearch/knn/index/codec/KNNCodecTestCase.java
@@ -9,6 +9,7 @@
 import com.google.common.collect.ImmutableSet;
 import org.apache.lucene.codecs.perfield.PerFieldKnnVectorsFormat;
 import org.apache.lucene.document.KnnVectorField;
+import org.apache.lucene.index.NoMergePolicy;
 import org.apache.lucene.index.VectorSimilarityFunction;
 import org.apache.lucene.search.Query;
 import org.apache.lucene.search.TopDocs;
@@ -121,6 +122,8 @@ public void testMultiFieldsKnnIndex(Codec codec) throws Exception {
         IndexWriterConfig iwc = newIndexWriterConfig();
         iwc.setMergeScheduler(new SerialMergeScheduler());
         iwc.setCodec(codec);
+        // Set merge policy to no merges so that we create a predictable number of segments.
+        iwc.setMergePolicy(NoMergePolicy.INSTANCE);
 
         /**
          * Add doc with field "test_vector"