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

feat: Adds bruteforce retriever test for neo4j

Browse source
This commit is contained in:
hajdul88 2024-11-26 16:42:35 +01:00
parent 59f8ec665f
commit 0441e19bc9
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/tests/test_neo4j.py
View file

@ -62,6 +62,9 @@ async def main():
assert len(history) == 6, "Search history is not correct."
results = await brute_force_triplet_search('Who has the most experience with graphic design?')
assert len(results) > 0
await cognee.prune.prune_data()
assert not os.path.isdir(data_directory_path), "Local data files are not deleted"