diff --git a/cognee/tests/unit/infrastructure/databases/cache/redis/test_redis_adapter.py b/cognee/tests/unit/infrastructure/databases/cache/redis/test_redis_adapter.py index 7e24f3754..642bd9993 100644 --- a/cognee/tests/unit/infrastructure/databases/cache/redis/test_redis_adapter.py +++ b/cognee/tests/unit/infrastructure/databases/cache/redis/test_redis_adapter.py @@ -207,7 +207,9 @@ def test_multiple_acquire_release_cycles(redis_adapter): def test_redis_adapter_redis_client_initialization(): """Test that Redis client is initialized with correct connection parameters.""" - with patch("cognee.infrastructure.databases.cache.redis.RedisAdapter.redis.Redis") as mock_redis_class: + with patch( + "cognee.infrastructure.databases.cache.redis.RedisAdapter.redis.Redis" + ) as mock_redis_class: mock_redis_instance = MagicMock() mock_redis_class.return_value = mock_redis_instance diff --git a/cognee/tests/unit/infrastructure/databases/cache/test_cache_engine.py b/cognee/tests/unit/infrastructure/databases/cache/test_cache_engine.py index 53fe72f22..7bd34c10d 100644 --- a/cognee/tests/unit/infrastructure/databases/cache/test_cache_engine.py +++ b/cognee/tests/unit/infrastructure/databases/cache/test_cache_engine.py @@ -211,7 +211,9 @@ def test_full_workflow_with_context_manager(): mock_config.agentic_lock_expire = 240 mock_config.agentic_lock_timeout = 300 - with patch("cognee.infrastructure.databases.cache.redis.RedisAdapter.redis.Redis") as mock_redis_class: + with patch( + "cognee.infrastructure.databases.cache.redis.RedisAdapter.redis.Redis" + ) as mock_redis_class: mock_redis_instance = MagicMock() mock_redis_class.return_value = mock_redis_instance