From 58dd518690d61632dd92814ebc2c43a4b6932612 Mon Sep 17 00:00:00 2001 From: lxobr <122801072+lxobr@users.noreply.github.com> Date: Fri, 9 Jan 2026 14:54:10 +0100 Subject: [PATCH] chore: update tests --- .../test_brute_force_triplet_search.py | 44 +++++++++---------- 1 file changed, 22 insertions(+), 22 deletions(-) diff --git a/cognee/tests/unit/modules/retrieval/test_brute_force_triplet_search.py b/cognee/tests/unit/modules/retrieval/test_brute_force_triplet_search.py index b7cbe08d7..00db1e794 100644 --- a/cognee/tests/unit/modules/retrieval/test_brute_force_triplet_search.py +++ b/cognee/tests/unit/modules/retrieval/test_brute_force_triplet_search.py @@ -57,7 +57,7 @@ async def test_brute_force_triplet_search_wide_search_limit_global_search(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search( @@ -79,7 +79,7 @@ async def test_brute_force_triplet_search_wide_search_limit_filtered_search(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search( @@ -101,7 +101,7 @@ async def test_brute_force_triplet_search_wide_search_default(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search(query="test", node_name=None) @@ -119,7 +119,7 @@ async def test_brute_force_triplet_search_default_collections(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search(query="test") @@ -149,7 +149,7 @@ async def test_brute_force_triplet_search_custom_collections(): custom_collections = ["CustomCol1", "CustomCol2"] with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search(query="test", collections=custom_collections) @@ -171,7 +171,7 @@ async def test_brute_force_triplet_search_always_includes_edge_collection(): collections_without_edge = ["Entity_name", "TextSummary_text"] with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search(query="test", collections=collections_without_edge) @@ -194,7 +194,7 @@ async def test_brute_force_triplet_search_all_collections_empty(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): results = await brute_force_triplet_search(query="test") @@ -216,7 +216,7 @@ async def test_brute_force_triplet_search_embeds_query(): mock_vector_engine.search = AsyncMock(return_value=[]) with patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ): await brute_force_triplet_search(query=query_text) @@ -249,7 +249,7 @@ async def test_brute_force_triplet_search_extracts_node_ids_global_search(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -279,7 +279,7 @@ async def test_brute_force_triplet_search_reuses_provided_fragment(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -311,7 +311,7 @@ async def test_brute_force_triplet_search_creates_fragment_when_not_provided(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -340,7 +340,7 @@ async def test_brute_force_triplet_search_passes_top_k_to_importance_calculation with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -430,7 +430,7 @@ async def test_brute_force_triplet_search_deduplicates_node_ids(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -471,7 +471,7 @@ async def test_brute_force_triplet_search_excludes_edge_collection(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -523,7 +523,7 @@ async def test_brute_force_triplet_search_skips_nodes_without_ids(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -564,7 +564,7 @@ async def test_brute_force_triplet_search_handles_tuple_results(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -606,7 +606,7 @@ async def test_brute_force_triplet_search_mixed_empty_collections(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -689,7 +689,7 @@ async def test_brute_force_triplet_search_vector_engine_init_error(): """Test brute_force_triplet_search handles vector engine initialization error (lines 145-147).""" with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine" + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine" ) as mock_get_vector_engine, ): mock_get_vector_engine.side_effect = Exception("Initialization error") @@ -716,7 +716,7 @@ async def test_brute_force_triplet_search_collection_not_found_error(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -743,7 +743,7 @@ async def test_brute_force_triplet_search_generic_exception(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), ): @@ -769,7 +769,7 @@ async def test_brute_force_triplet_search_with_node_name_sets_relevant_ids_to_no with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch( @@ -804,7 +804,7 @@ async def test_brute_force_triplet_search_collection_not_found_at_top_level(): with ( patch( - "cognee.modules.retrieval.utils.brute_force_triplet_search.get_vector_engine", + "cognee.modules.retrieval.utils.node_edge_vector_search.get_vector_engine", return_value=mock_vector_engine, ), patch(