From c3cec818d7509f0d6fd85593a884e070a41a49a2 Mon Sep 17 00:00:00 2001 From: lxobr <122801072+lxobr@users.noreply.github.com> Date: Fri, 19 Dec 2025 16:22:47 +0100 Subject: [PATCH] fix: update tests --- cognee/tests/unit/modules/graph/cognee_graph_test.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cognee/tests/unit/modules/graph/cognee_graph_test.py b/cognee/tests/unit/modules/graph/cognee_graph_test.py index 0458fca76..41f12e73a 100644 --- a/cognee/tests/unit/modules/graph/cognee_graph_test.py +++ b/cognee/tests/unit/modules/graph/cognee_graph_test.py @@ -468,7 +468,7 @@ async def test_map_vector_distances_no_edge_matches(setup_graph): @pytest.mark.asyncio async def test_map_vector_distances_none_returns_early(setup_graph): - """Test that edge_distances=None returns early without error and vector_distance stays None.""" + """Test that edge_distances=None returns early without error and vector_distance is set to default penalty.""" graph = setup_graph graph.add_node(Node("1")) graph.add_node(Node("2")) @@ -476,12 +476,12 @@ async def test_map_vector_distances_none_returns_early(setup_graph): await graph.map_vector_distances_to_graph_edges(edge_distances=None) - assert graph.edges[0].attributes.get("vector_distance") is None + assert graph.edges[0].attributes.get("vector_distance") == [3.5] @pytest.mark.asyncio async def test_map_vector_distances_empty_nodes_returns_early(setup_graph): - """Test that node_distances={} returns early without error and vector_distance stays None.""" + """Test that node_distances={} returns early without error and vector_distance is set to default penalty.""" graph = setup_graph node1 = Node("1") node2 = Node("2") @@ -490,8 +490,8 @@ async def test_map_vector_distances_empty_nodes_returns_early(setup_graph): await graph.map_vector_distances_to_graph_nodes({}) - assert node1.attributes.get("vector_distance") is None - assert node2.attributes.get("vector_distance") is None + assert node1.attributes.get("vector_distance") == [3.5] + assert node2.attributes.get("vector_distance") == [3.5] @pytest.mark.asyncio