Disable KNN format randomization in some KNN block join tests

apache · Dec 9, 2024 · 4f8035c · 4f8035c
1 parent 6b0112c
commit 4f8035c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/...ne/join/src/test/org/apache/lucene/search/join/ParentBlockJoinKnnVectorQueryTestCase.java b/...ne/join/src/test/org/apache/lucene/search/join/ParentBlockJoinKnnVectorQueryTestCase.java
@@ -50,6 +50,7 @@
 import org.apache.lucene.store.Directory;
 import org.apache.lucene.tests.index.RandomIndexWriter;
 import org.apache.lucene.tests.util.LuceneTestCase;
+import org.apache.lucene.tests.util.TestUtil;
 
 abstract class ParentBlockJoinKnnVectorQueryTestCase extends LuceneTestCase {
 
@@ -185,7 +186,10 @@ public void testScoringWithMultipleChildren() throws IOException {
     try (Directory d = newDirectory()) {
       try (IndexWriter w =
           new IndexWriter(
-              d, newIndexWriterConfig().setMergePolicy(newMergePolicy(random(), false)))) {
+              d,
+              newIndexWriterConfig()
+                  .setCodec(TestUtil.getDefaultCodec())
+                  .setMergePolicy(newMergePolicy(random(), false)))) {
         List<Document> toAdd = new ArrayList<>();
         for (int j = 1; j <= 5; j++) {
           Document doc = new Document();
@@ -240,7 +244,8 @@ public void testSkewedIndex() throws IOException {
      * randomly fail to find one).
      */
     try (Directory d = newDirectory()) {
-      try (IndexWriter w = new IndexWriter(d, new IndexWriterConfig())) {
+      try (IndexWriter w =
+          new IndexWriter(d, new IndexWriterConfig().setCodec(TestUtil.getDefaultCodec()))) {
         int r = 0;
         for (int i = 0; i < 5; i++) {
           for (int j = 0; j < 5; j++) {

diff --git a/...e/join/src/test/org/apache/lucene/search/join/TestParentBlockJoinFloatKnnVectorQuery.java b/...e/join/src/test/org/apache/lucene/search/join/TestParentBlockJoinFloatKnnVectorQuery.java
@@ -35,6 +35,7 @@
 import org.apache.lucene.search.Query;
 import org.apache.lucene.search.TermQuery;
 import org.apache.lucene.store.Directory;
+import org.apache.lucene.tests.util.TestUtil;
 
 public class TestParentBlockJoinFloatKnnVectorQuery extends ParentBlockJoinKnnVectorQueryTestCase {
 
@@ -76,7 +77,10 @@ public void testScoreCosine() throws IOException {
     try (Directory d = newDirectory()) {
       try (IndexWriter w =
           new IndexWriter(
-              d, new IndexWriterConfig().setMergePolicy(newMergePolicy(random(), false)))) {
+              d,
+              new IndexWriterConfig()
+                  .setCodec(TestUtil.getDefaultCodec())
+                  .setMergePolicy(newMergePolicy(random(), false)))) {
         for (int j = 1; j <= 5; j++) {
           List<Document> toAdd = new ArrayList<>();
           Document doc = new Document();