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

add unit tests for empty graph check on search

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-10-15 17:09:13 +01:00
parent ea4a93efb1
commit dede5fa6fd
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
cognee/tests/unit/api/test_search.py Normal file
View file

@ -0,0 +1,23 @@
import pytest
import cognee
from cognee.modules.data.exceptions import SearchOnEmptyGraphError
@pytest.mark.asyncio
async def test_empty_search_raises_SearchOnEmptyGraphError_on_empty_graph():
await cognee.prune.prune_data()
await cognee.prune.prune_system(metadata=True)
await cognee.add("Sample input")
with pytest.raises(SearchOnEmptyGraphError):
await cognee.search("Sample query")
async def test_empty_search_doesnt_raise_SearchOnEmptyGraphError():
await cognee.prune.prune_data()
await cognee.prune.prune_system(metadata=True)
await cognee.add("Sample input")
await cognee.cognify()
try:
await cognee.search("Sample query")
except SearchOnEmptyGraphError:
pytest.fail("Should not raise SearchOnEmptyGraphError when data was added and cognified")