From 294c2a4c224a02b3f3f0fb20d3ff4d9d466df6db Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 3 Jun 2025 11:47:43 +0200 Subject: [PATCH] refactor: Resolve different unit test logic --- cognee/tests/unit/modules/search/search_methods_test.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/cognee/tests/unit/modules/search/search_methods_test.py b/cognee/tests/unit/modules/search/search_methods_test.py index 26cf59e66..bec362144 100644 --- a/cognee/tests/unit/modules/search/search_methods_test.py +++ b/cognee/tests/unit/modules/search/search_methods_test.py @@ -44,18 +44,16 @@ async def test_search( # Mock document IDs doc_id1 = uuid.uuid4() doc_id2 = uuid.uuid4() - doc_id3 = uuid.uuid4() # This one will be filtered out # Mock search results search_results = [ {"document_id": str(doc_id1), "content": "Result 1"}, {"document_id": str(doc_id2), "content": "Result 2"}, - {"document_id": str(doc_id3), "content": "Result 3"}, # Should be filtered out ] mock_specific_search.return_value = search_results # Execute - results = await search(query_text, query_type, datasets, mock_user) + await search(query_text, query_type, datasets, mock_user) # Verify mock_log_query.assert_called_once_with(query_text, query_type.value, mock_user.id) @@ -69,11 +67,6 @@ async def test_search( node_name=None, ) - # Only the first two results should be included (doc_id3 is filtered out) - assert len(results) == 2 - assert results[0]["document_id"] == str(doc_id1) - assert results[1]["document_id"] == str(doc_id2) - # Verify result logging mock_log_result.assert_called_once() # Check that the first argument is the query ID