From 2f5d0e01071cd31fa90e5fc5d1b07f464f4cb87e Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Fri, 12 Dec 2025 12:22:12 +0100
Subject: [PATCH] lint fix + adds comments

---
 cognee/tests/test_search_db.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cognee/tests/test_search_db.py b/cognee/tests/test_search_db.py
index 6d71ce86c..9ca16c890 100644
--- a/cognee/tests/test_search_db.py
+++ b/cognee/tests/test_search_db.py
@@ -318,12 +318,13 @@ async def feedback_state():
 
 @pytest.mark.asyncio
 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"]
 
 
 @pytest.mark.asyncio
 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"]
 
     for name in [
@@ -367,6 +368,7 @@ async def test_e2e_retriever_contexts(e2e_state):
 
 @pytest.mark.asyncio
 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():
         assert isinstance(triplets, list), f"{name}: Triplets should be a list"
         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
 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
 
     sr = e2e_state["search_results"]
@@ -436,6 +439,7 @@ async def test_e2e_search_results_and_wrappers(e2e_state):
 
 @pytest.mark.asyncio
 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"]
     nodes, edges = graph
 
@@ -468,6 +472,7 @@ async def test_e2e_graph_side_effects_and_node_fields(e2e_state):
 
 @pytest.mark.asyncio
 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"]
     for _from_node, _to_node, relationship_name, properties in edges:
         if relationship_name == "used_graph_element_to_answer":