From d1d8e334716d81fe6e8b1f1b185743197f27d79e Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Thu, 9 Oct 2025 10:48:21 +0200 Subject: [PATCH] test: Fix windows tests. First try of potential fixes. --- cognee/tests/test_library.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py index c5e6cc64b..fe1a0bdfa 100755 --- a/cognee/tests/test_library.py +++ b/cognee/tests/test_library.py @@ -82,6 +82,11 @@ async def main(): data_root_directory = get_storage_config()["data_root_directory"] assert not os.path.isdir(data_root_directory), "Local data files are not deleted" + from cognee.infrastructure.databases.relational import get_relational_engine + + get_relational_engine().get_session().close() + await get_relational_engine().engine.dispose() + # Assert relational, vector and graph databases have been cleaned properly await cognee.prune.prune_system(metadata=True) @@ -89,7 +94,7 @@ async def main(): collection_names = await connection.table_names() assert len(collection_names) == 0, "LanceDB vector database is not empty" - from cognee.infrastructure.databases.relational import get_relational_engine + db_path = get_relational_engine().db_path dir_path = os.path.dirname(db_path)