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

Fix TestHnswGraph rare failure

This commit is contained in:
Michael Sokolov 2022-08-19 16:18:24 -04:00
parent 0a58318e16
commit daa56d30f0
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lucene/core/src/test/org/apache/lucene/util/hnsw/TestHnswGraph.java
View File

@ -206,7 +206,7 @@ public class TestHnswGraph extends LuceneTestCase {
IndexReader reader2 = DirectoryReader.open(dir2)) { IndexReader reader2 = DirectoryReader.open(dir2)) {
IndexSearcher searcher = new IndexSearcher(reader); IndexSearcher searcher = new IndexSearcher(reader);
IndexSearcher searcher2 = new IndexSearcher(reader2); IndexSearcher searcher2 = new IndexSearcher(reader2);
OUTER:
for (int i = 0; i < 10; i++) { for (int i = 0; i < 10; i++) {
// ask to explore a lot of candidates to ensure the same returned hits, // ask to explore a lot of candidates to ensure the same returned hits,
// as graphs of 2 indices are organized differently // as graphs of 2 indices are organized differently
@ -217,7 +217,14 @@ public class TestHnswGraph extends LuceneTestCase {
List<Integer> docs2 = new ArrayList<>(); List<Integer> docs2 = new ArrayList<>();
TopDocs topDocs = searcher.search(query, 5); TopDocs topDocs = searcher.search(query, 5);
float lastScore = -1;
for (ScoreDoc scoreDoc : topDocs.scoreDocs) { for (ScoreDoc scoreDoc : topDocs.scoreDocs) {
if (scoreDoc.score == lastScore) {
// if we have repeated score this test is invalid
continue OUTER;
} else {
lastScore = scoreDoc.score;
}
Document doc = reader.document(scoreDoc.doc, Set.of("id")); Document doc = reader.document(scoreDoc.doc, Set.of("id"));
ids1.add(doc.get("id")); ids1.add(doc.get("id"));
docs1.add(scoreDoc.doc); docs1.add(scoreDoc.doc);