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

test: Add database deletion test for Weaviate vector database

Browse source 
Added database deletion test for Weaviate vector database

Test COG-488
This commit is contained in:
Igor Ilic 2024-11-21 18:37:09 +01:00
parent 0f8baeeb0d
commit 443133ffbb
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tests/test_weaviate.py
View file

@ -57,9 +57,15 @@ async def main():
print(f"{result}\n")
history = await cognee.get_search_history()
assert len(history) == 6, "Search history is not correct."
await cognee.prune.prune_data()
assert not os.path.isdir(data_directory_path), "Local data files are not deleted"
await cognee.prune.prune_system(metadata=True)
collections = get_vector_engine().client.collections.list_all()
assert len(collections) == 0, "Weaviate vector database is not empty"
if __name__ == "__main__":
import asyncio
asyncio.run(main())