From a44ab88519a784ef22ca9d133200124373db8dc7 Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Thu, 9 Oct 2025 17:31:27 +0200 Subject: [PATCH] test: try calling gc to fix windows issue --- .../databases/relational/sqlalchemy/SqlAlchemyAdapter.py | 4 +++- cognee/tests/test_library.py | 7 ++----- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py b/cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py index 88d2abc7e..4908295ca 100644 --- a/cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py +++ b/cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py @@ -507,7 +507,9 @@ class SQLAlchemyAdapter: if self.engine.dialect.name == "sqlite": await self.engine.dispose(close=True) # Wait for the database connections to close and release the file (Windows) - await asyncio.sleep(2) + import gc + gc.collect() + # await asyncio.sleep(2) db_directory = path.dirname(self.db_path) file_name = path.basename(self.db_path) file_storage = get_file_storage(db_directory) diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py index 2933c77ba..c5e6cc64b 100755 --- a/cognee/tests/test_library.py +++ b/cognee/tests/test_library.py @@ -82,11 +82,6 @@ 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) @@ -94,6 +89,8 @@ 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) file_path = os.path.basename(db_path)