From 443133ffbbfa97c2c58f252f41aab8d936af62ce Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 21 Nov 2024 18:37:09 +0100 Subject: [PATCH] test: Add database deletion test for Weaviate vector database Added database deletion test for Weaviate vector database Test COG-488 --- cognee/tests/test_weaviate.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cognee/tests/test_weaviate.py b/cognee/tests/test_weaviate.py index c93dc036a..c352df13e 100644 --- a/cognee/tests/test_weaviate.py +++ b/cognee/tests/test_weaviate.py @@ -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())