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

Update test_search_db.py

Browse source
This commit is contained in:
hajdul88 2025-12-11 19:07:39 +01:00
parent 7e0c9f0c91
commit 1eb15c9f9c
1 changed files with 0 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
cognee/tests/test_search_db.py
View file

@ -156,13 +156,6 @@ async def test_retriever_contexts():
f"TripletRetriever: Context did not contain 'germany' or 'netherlands'; got: {context_triplet!r}" 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 @pytest.mark.asyncio
async def test_retriever_triplets(): 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" 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 @pytest.mark.asyncio
async def test_search_operations(): async def test_search_operations():
@ -276,13 +262,6 @@ async def test_search_operations():
f"{name}: expected 'netherlands' in result, got: {text!r}" 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 @pytest.mark.asyncio
async def test_graph_node_and_edge_counts(): 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)}" 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 @pytest.mark.asyncio
async def test_node_field_validation(): 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}" 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 @pytest.mark.asyncio
async def test_feedback_weight_calculation(): async def test_feedback_weight_calculation():
@ -478,10 +443,3 @@ async def test_feedback_weight_calculation():
assert properties["feedback_weight"] >= 6, ( assert properties["feedback_weight"] >= 6, (
"Feedback weight calculation is not correct, it should be more then 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