diff --git a/lucene/core/src/test/org/apache/lucene/search/BaseKnnVectorQueryTestCase.java b/lucene/core/src/test/org/apache/lucene/search/BaseKnnVectorQueryTestCase.java index d45902d3bb3..fb4775712a0 100644 --- a/lucene/core/src/test/org/apache/lucene/search/BaseKnnVectorQueryTestCase.java +++ b/lucene/core/src/test/org/apache/lucene/search/BaseKnnVectorQueryTestCase.java @@ -281,12 +281,19 @@ abstract class BaseKnnVectorQueryTestCase extends LuceneTestCase { DocIdSetIterator it = scorer.iterator(); assertEquals(3, it.cost()); - assertEquals(2, it.nextDoc()); - assertEquals(1 / 2f, scorer.score(), 0); - assertEquals(4, it.advance(4)); - assertEquals(1 / 6f, scorer.score(), 0); - - assertEquals(NO_MORE_DOCS, it.advance(5)); + int firstDoc = it.nextDoc(); + if (firstDoc == 1) { + assertEquals(1 / 6f, scorer.score(), 0); + assertEquals(3, it.advance(3)); + assertEquals(1 / 2f, scorer.score(), 0); + assertEquals(NO_MORE_DOCS, it.advance(4)); + } else { + assertEquals(2, firstDoc); + assertEquals(1 / 2f, scorer.score(), 0); + assertEquals(4, it.advance(4)); + assertEquals(1 / 6f, scorer.score(), 0); + assertEquals(NO_MORE_DOCS, it.advance(5)); + } expectThrows(ArrayIndexOutOfBoundsException.class, scorer::score); } } @@ -384,7 +391,7 @@ abstract class BaseKnnVectorQueryTestCase extends LuceneTestCase { assertEquals(0, matched.getDetails().length); assertEquals("within top 3 docs", matched.getDescription()); - Explanation nomatch = searcher.explain(query, 1); + Explanation nomatch = searcher.explain(query, 5); assertFalse(nomatch.isMatch()); assertEquals(0f, nomatch.getValue()); assertEquals(0, matched.getDetails().length);