From b815aeaa2afa5ccd13ff4d06194a154a0c787317 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 21 Nov 2024 13:44:09 +0100 Subject: [PATCH] test: Add test for database deletion for LanceDB and SQLite Added database deletion test for LanceDB and SQLite Test COG-488 --- cognee/tests/test_library.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py index 66d218c3b..9a49206a1 100755 --- a/cognee/tests/test_library.py +++ b/cognee/tests/test_library.py @@ -57,6 +57,16 @@ async def main(): 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) + connection = await vector_engine.get_connection() + collection_names = await connection.table_names() + assert len(collection_names) == 0, "The vector database is not empty" + from cognee.infrastructure.databases.relational import get_relational_engine + assert not os.path.exists(get_relational_engine().db_path), "The relational database is not empty" + if __name__ == "__main__": import asyncio asyncio.run(main(), debug=True)