diff --git a/lucene/core/src/java/org/apache/lucene/util/hnsw/NeighborQueue.java b/lucene/core/src/java/org/apache/lucene/util/hnsw/NeighborQueue.java index bcc43435872..582467cd976 100644 --- a/lucene/core/src/java/org/apache/lucene/util/hnsw/NeighborQueue.java +++ b/lucene/core/src/java/org/apache/lucene/util/hnsw/NeighborQueue.java @@ -156,6 +156,7 @@ public class NeighborQueue { public void clear() { heap.clear(); visitedCount = 0; + incomplete = false; } public int visitedCount() { diff --git a/lucene/core/src/test/org/apache/lucene/util/hnsw/TestNeighborQueue.java b/lucene/core/src/test/org/apache/lucene/util/hnsw/TestNeighborQueue.java index ef432b3bcb3..274deec41cb 100644 --- a/lucene/core/src/test/org/apache/lucene/util/hnsw/TestNeighborQueue.java +++ b/lucene/core/src/test/org/apache/lucene/util/hnsw/TestNeighborQueue.java @@ -72,10 +72,12 @@ public class TestNeighborQueue extends LuceneTestCase { nn.add(1, 1.1f); nn.add(2, -2.2f); nn.setVisitedCount(42); + nn.markIncomplete(); nn.clear(); assertEquals(0, nn.size()); assertEquals(0, nn.visitedCount()); + assertFalse(nn.incomplete()); } public void testMaxSizeQueue() {