forgot to push updated test

nmslib · Oct 16, 2023 · c4f7019 · c4f7019
1 parent 141b76a
commit c4f7019
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/bindings_test_returnlists.py b/tests/python/bindings_test_returnlists.py
@@ -42,8 +42,8 @@ def testRandomSelf(self):
         bf_index.add_items(data)
 
         # filter_function designed to return different number of elements for each query
-        filter_function = lambda index: index > (num_elements-2)  if random.random() < 0.5 else index > (num_elements-1)
-        labels, distances = hnsw_index.knn_query_return_lists(data, k=1, num_threads=1, filter=filter_function)
+        filter_function = lambda index: index > (num_elements-3)  if random.random() < 0.5 else index >= (num_elements-1)
+        labels, distances = hnsw_index.knn_query_return_lists(data, k=2, num_threads=1, filter=filter_function)
 
         # Assert the return type for both labels and distances is a list
         self.assertTrue(isinstance(labels, list))