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

lint: ruff format fix

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-08-19 11:59:44 +01:00
parent 9d41bc9573
commit e8b7fe7d0e
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/tests/unit/infrastructure/databases/graph/kuzu/test_adapter_s3_migration.py
View file

@ -77,11 +77,11 @@ def _load_adapter_with_stubs(monkeypatch):
kuzu_db_mod.Database = _PlaceholderDB kuzu_db_mod.Database = _PlaceholderDB
# Create minimal stub tree for required cognee imports to avoid executing package __init__ # Create minimal stub tree for required cognee imports to avoid executing package __init__
root = _install_stub("cognee") _install_stub("cognee")
infra = _install_stub("cognee.infrastructure") _install_stub("cognee.infrastructure")
databases = _install_stub("cognee.infrastructure.databases") _install_stub("cognee.infrastructure.databases")
graph = _install_stub("cognee.infrastructure.databases.graph") _install_stub("cognee.infrastructure.databases.graph")
kuzu_pkg = _install_stub("cognee.infrastructure.databases.graph.kuzu") _install_stub("cognee.infrastructure.databases.graph.kuzu")
# graph_db_interface stub # graph_db_interface stub
gdi_mod = _install_stub("cognee.infrastructure.databases.graph.graph_db_interface") gdi_mod = _install_stub("cognee.infrastructure.databases.graph.graph_db_interface")
@ -182,5 +182,5 @@ def test_adapter_s3_auto_migration(monkeypatch):
monkeypatch.setattr(mod.KuzuAdapter, "pull_from_s3", lambda self: None) monkeypatch.setattr(mod.KuzuAdapter, "pull_from_s3", lambda self: None)
monkeypatch.setattr(mod.KuzuAdapter, "push_to_s3", lambda self: None) monkeypatch.setattr(mod.KuzuAdapter, "push_to_s3", lambda self: None)
adapter = mod.KuzuAdapter("s3://bucket/db") mod.KuzuAdapter("s3://bucket/db")
assert calls["migrated"] is True assert calls["migrated"] is True