gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lint fix + adds comments

Browse source
This commit is contained in:
hajdul88 2025-12-12 12:22:12 +01:00
parent eaf29f2e52
commit 2f5d0e0107
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tests/test_search_db.py
View file

@ -318,12 +318,13 @@ async def feedback_state():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_graph_vector_consistency(e2e_state): async def test_e2e_graph_vector_consistency(e2e_state):
"""Graph and vector stores contain the same triplet edges."""
assert e2e_state["graph_edges_count"] == e2e_state["vector_collection_edges_count"] assert e2e_state["graph_edges_count"] == e2e_state["vector_collection_edges_count"]
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_retriever_contexts(e2e_state): async def test_e2e_retriever_contexts(e2e_state):
query = "Next to which country is Germany located?" """All retrievers return non-empty, well-typed contexts."""
contexts = e2e_state["contexts"] contexts = e2e_state["contexts"]
for name in [ for name in [
@ -367,6 +368,7 @@ async def test_e2e_retriever_contexts(e2e_state):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_retriever_triplets_have_vector_distances(e2e_state): async def test_e2e_retriever_triplets_have_vector_distances(e2e_state):
"""Graph retriever triplets include sane vector_distance metadata."""
for name, triplets in e2e_state["triplets"].items(): for name, triplets in e2e_state["triplets"].items():
assert isinstance(triplets, list), f"{name}: Triplets should be a list" assert isinstance(triplets, list), f"{name}: Triplets should be a list"
assert triplets, f"{name}: Triplets list should not be empty" assert triplets, f"{name}: Triplets list should not be empty"
@ -383,6 +385,7 @@ async def test_e2e_retriever_triplets_have_vector_distances(e2e_state):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_search_results_and_wrappers(e2e_state): async def test_e2e_search_results_and_wrappers(e2e_state):
"""Search returns expected shapes across search types and access modes."""
from cognee.context_global_variables import backend_access_control_enabled from cognee.context_global_variables import backend_access_control_enabled
sr = e2e_state["search_results"] sr = e2e_state["search_results"]
@ -436,6 +439,7 @@ async def test_e2e_search_results_and_wrappers(e2e_state):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_graph_side_effects_and_node_fields(e2e_state): async def test_e2e_graph_side_effects_and_node_fields(e2e_state):
"""Search interactions create expected graph nodes/edges and required fields."""
graph = e2e_state["graph_snapshot"] graph = e2e_state["graph_snapshot"]
nodes, edges = graph nodes, edges = graph
@ -468,6 +472,7 @@ async def test_e2e_graph_side_effects_and_node_fields(e2e_state):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_e2e_feedback_weight_calculation(feedback_state): async def test_e2e_feedback_weight_calculation(feedback_state):
"""Positive feedback increases used_graph_element_to_answer feedback_weight."""
_nodes, edges = feedback_state["graph_snapshot"] _nodes, edges = feedback_state["graph_snapshot"]
for _from_node, _to_node, relationship_name, properties in edges: for _from_node, _to_node, relationship_name, properties in edges:
if relationship_name == "used_graph_element_to_answer": if relationship_name == "used_graph_element_to_answer":