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

feat: adds asserts to summaries retriever int test

Browse source
This commit is contained in:
hajdul88 2025-12-10 11:47:51 +01:00
parent b0d806526a
commit 2896bd10ee
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/tests/integration/retrieval/test_summaries_retriever.py
View file

@ -164,6 +164,8 @@ async def test_summaries_retriever_context(setup_test_environment_with_summaries
context = await retriever.get_context("Christina") context = await retriever.get_context("Christina")
assert isinstance(context, list), "Context should be a list"
assert len(context) > 0, "Context should not be empty"
assert context[0]["text"] == "C.M.", "Failed to get Christina Mayer" assert context[0]["text"] == "C.M.", "Failed to get Christina Mayer"