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

test: Fix windows tests. First try of potential fixes.

Browse source
This commit is contained in:
Andrej Milicevic 2025-10-09 10:48:21 +02:00
parent 2932a627bb
commit d1d8e33471
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tests/test_library.py
View file

@ -82,6 +82,11 @@ async def main():
data_root_directory = get_storage_config()["data_root_directory"] data_root_directory = get_storage_config()["data_root_directory"]
assert not os.path.isdir(data_root_directory), "Local data files are not deleted" 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 # Assert relational, vector and graph databases have been cleaned properly
await cognee.prune.prune_system(metadata=True) await cognee.prune.prune_system(metadata=True)
@ -89,7 +94,7 @@ async def main():
collection_names = await connection.table_names() collection_names = await connection.table_names()
assert len(collection_names) == 0, "LanceDB vector database is not empty" 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 db_path = get_relational_engine().db_path
dir_path = os.path.dirname(db_path) dir_path = os.path.dirname(db_path)