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

feat: adds missing checks to graph completion retriever integration test

Browse source
This commit is contained in:
hajdul88 2025-12-10 11:46:16 +01:00
parent 49f4938e11
commit 723bcd70a2
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cognee/tests/integration/retrieval/test_graph_completion_retriever.py
View file

@ -255,3 +255,14 @@ async def test_get_graph_completion_context_on_empty_graph(setup_test_environmen
context = await retriever.get_context("Who works at Figma?")
assert context == [], "Context should be empty on an empty graph"
@pytest.mark.asyncio
async def test_graph_completion_get_triplets_empty(setup_test_environment_empty):
"""Integration test: verify GraphCompletionRetriever get_triplets handles empty graph."""
retriever = GraphCompletionRetriever()
triplets = await retriever.get_triplets("Who works at Figma?")
assert isinstance(triplets, list), "Triplets should be a list"
assert len(triplets) == 0, "Should return empty list on empty graph"