From 1eb15c9f9c00f00ecd17dc729b1ab0034743a8a3 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 11 Dec 2025 19:07:39 +0100
Subject: [PATCH] Update test_search_db.py

---
 cognee/tests/test_search_db.py | 42 ----------------------------------
 1 file changed, 42 deletions(-)

diff --git a/cognee/tests/test_search_db.py b/cognee/tests/test_search_db.py
index 1c4941b55..5a82b5c8d 100644
--- a/cognee/tests/test_search_db.py
+++ b/cognee/tests/test_search_db.py
@@ -156,13 +156,6 @@ async def test_retriever_contexts():
         f"TripletRetriever: Context did not contain 'germany' or 'netherlands'; got: {context_triplet!r}"
     )
 
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass
-
 
 @pytest.mark.asyncio
 async def test_retriever_triplets():
@@ -202,13 +195,6 @@ async def test_retriever_triplets():
                 f"{name}: node_2 vector_distance {distance} out of [0,1], this shouldn't happen"
             )
 
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass
-
 
 @pytest.mark.asyncio
 async def test_search_operations():
@@ -276,13 +262,6 @@ async def test_search_operations():
             f"{name}: expected 'netherlands' in result, got: {text!r}"
         )
 
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass
-
 
 @pytest.mark.asyncio
 async def test_graph_node_and_edge_counts():
@@ -360,13 +339,6 @@ async def test_graph_node_and_edge_counts():
         f"Expected at least six 'belongs_to_set' edges, but found {edge_type_counts.get('belongs_to_set', 0)}"
     )
 
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass
-
 
 @pytest.mark.asyncio
 async def test_node_field_validation():
@@ -438,13 +410,6 @@ async def test_node_field_validation():
                     f"Node {node_id} has invalid value for '{field}': {value!r}"
                 )
 
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass
-
 
 @pytest.mark.asyncio
 async def test_feedback_weight_calculation():
@@ -478,10 +443,3 @@ async def test_feedback_weight_calculation():
             assert properties["feedback_weight"] >= 6, (
                 "Feedback weight calculation is not correct, it should be more then 6."
             )
-
-    # Cleanup
-    try:
-        await cognee.prune.prune_data()
-        await cognee.prune.prune_system(metadata=True)
-    except Exception:
-        pass