From ce5fc60131bb5c11823705aa2e8d0c6e1066bb0c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Fri, 19 Sep 2025 16:08:22 +0100 Subject: [PATCH] Fix test --- tests/unit/test_query/test_doc_embeddings_qdrant_query.py | 8 ++++---- .../unit/test_query/test_graph_embeddings_qdrant_query.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/unit/test_query/test_doc_embeddings_qdrant_query.py b/tests/unit/test_query/test_doc_embeddings_qdrant_query.py index b9a306c1..f4a1d977 100644 --- a/tests/unit/test_query/test_doc_embeddings_qdrant_query.py +++ b/tests/unit/test_query/test_doc_embeddings_qdrant_query.py @@ -104,7 +104,7 @@ class TestQdrantDocEmbeddingsQuery(IsolatedAsyncioTestCase): # Assert # Verify query was called with correct parameters - expected_collection = 'd_test_user_test_collection_3' + expected_collection = 'd_test_user_test_collection' mock_qdrant_instance.query_points.assert_called_once_with( collection_name=expected_collection, query=[0.1, 0.2, 0.3], @@ -166,7 +166,7 @@ class TestQdrantDocEmbeddingsQuery(IsolatedAsyncioTestCase): assert mock_qdrant_instance.query_points.call_count == 2 # Verify both collections were queried - expected_collection = 'd_multi_user_multi_collection_2' + expected_collection = 'd_multi_user_multi_collection' calls = mock_qdrant_instance.query_points.call_args_list assert calls[0][1]['collection_name'] == expected_collection assert calls[1][1]['collection_name'] == expected_collection @@ -303,11 +303,11 @@ class TestQdrantDocEmbeddingsQuery(IsolatedAsyncioTestCase): calls = mock_qdrant_instance.query_points.call_args_list # First call should use 2D collection - assert calls[0][1]['collection_name'] == 'd_dim_user_dim_collection_2' + assert calls[0][1]['collection_name'] == 'd_dim_user_dim_collection' assert calls[0][1]['query'] == [0.1, 0.2] # Second call should use 3D collection - assert calls[1][1]['collection_name'] == 'd_dim_user_dim_collection_3' + assert calls[1][1]['collection_name'] == 'd_dim_user_dim_collection' assert calls[1][1]['query'] == [0.3, 0.4, 0.5] # Verify results diff --git a/tests/unit/test_query/test_graph_embeddings_qdrant_query.py b/tests/unit/test_query/test_graph_embeddings_qdrant_query.py index 11d11d35..0dd0e94e 100644 --- a/tests/unit/test_query/test_graph_embeddings_qdrant_query.py +++ b/tests/unit/test_query/test_graph_embeddings_qdrant_query.py @@ -176,7 +176,7 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase): # Assert # Verify query was called with correct parameters - expected_collection = 't_test_user_test_collection_3' + expected_collection = 't_test_user_test_collection' mock_qdrant_instance.query_points.assert_called_once_with( collection_name=expected_collection, query=[0.1, 0.2, 0.3], @@ -236,7 +236,7 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase): assert mock_qdrant_instance.query_points.call_count == 2 # Verify both collections were queried - expected_collection = 't_multi_user_multi_collection_2' + expected_collection = 't_multi_user_multi_collection' calls = mock_qdrant_instance.query_points.call_args_list assert calls[0][1]['collection_name'] == expected_collection assert calls[1][1]['collection_name'] == expected_collection @@ -374,11 +374,11 @@ class TestQdrantGraphEmbeddingsQuery(IsolatedAsyncioTestCase): calls = mock_qdrant_instance.query_points.call_args_list # First call should use 2D collection - assert calls[0][1]['collection_name'] == 't_dim_user_dim_collection_2' + assert calls[0][1]['collection_name'] == 't_dim_user_dim_collection' assert calls[0][1]['query'] == [0.1, 0.2] # Second call should use 3D collection - assert calls[1][1]['collection_name'] == 't_dim_user_dim_collection_3' + assert calls[1][1]['collection_name'] == 't_dim_user_dim_collection' assert calls[1][1]['query'] == [0.3, 0.4, 0.5] # Verify results