fix: fix lookup test, refactor search params test

qdrant · Jul 1, 2024 · d9a4320 · d9a4320
1 parent 0fc3341
commit d9a4320
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/tests/congruence_tests/test_query.py b/tests/congruence_tests/test_query.py
@@ -522,7 +522,7 @@ def no_query_no_prefetch(cls, client: QdrantBase) -> models.QueryResponse:
 
 
 def test_dense_query_lookup_from_another_collection():
-    fixture_points = generate_fixtures()
+    fixture_points = generate_fixtures(10)
 
     secondary_collection_points = generate_fixtures(10)
 
@@ -717,19 +717,11 @@ def test_dense_query_parametrized():
         local_client,
         remote_client,
         searcher.dense_queries_parametrized,
-        search_params={"hnsw_ef": 128},
-    )
-    compare_client_results(
-        local_client,
-        remote_client,
-        searcher.dense_queries_parametrized,
-        search_params={"indexed_only": True},
-    )
-    compare_client_results(
-        local_client,
-        remote_client,
-        searcher.dense_queries_parametrized,
-        search_params={"quantization": {"ignore": True, "rescore": True, "oversampling": 2.0}},
+        search_params={
+            "hnsw_ef": 128,
+            "indexed_only": True,
+            "quantization": {"ignore": True, "rescore": True, "oversampling": 2.0},
+        },
     )