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

Update test_chunks_retriever.py

Browse source
This commit is contained in:
hajdul88 2025-12-10 14:03:40 +01:00
parent b4fb4ce49b
commit a14dacdc0f
1 changed files with 0 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/tests/integration/retrieval/test_chunks_retriever.py
View file

@ -200,18 +200,6 @@ async def setup_test_environment_empty():
pass pass
@pytest.mark.asyncio
async def test_chunks_retriever_context_simple(setup_test_environment_with_chunks_simple):
"""Integration test: verify ChunksRetriever can retrieve chunk context (simple)."""
retriever = ChunksRetriever()
context = await retriever.get_context("Mike")
assert isinstance(context, list), "Context should be a list"
assert len(context) > 0, "Context should not be empty"
assert context[0]["text"] == "Mike Broski", "Failed to get Mike Broski"
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_chunks_retriever_context_multiple_chunks(setup_test_environment_with_chunks_simple): async def test_chunks_retriever_context_multiple_chunks(setup_test_environment_with_chunks_simple):
"""Integration test: verify ChunksRetriever can retrieve multiple chunks.""" """Integration test: verify ChunksRetriever can retrieve multiple chunks."""