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

ruff

Browse source
This commit is contained in:
hajdul88 2025-12-11 13:13:12 +01:00
parent ca593bba15
commit 4d0dca0b6c
2 changed files with 8 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
cognee/tests/unit/modules/retrieval/conversation_history_test.py
View file

@ -158,9 +158,7 @@ class TestConversationHistoryUtils:
"""Test save_conversation_history returns False when user_id is None.""" """Test save_conversation_history returns False when user_id is None."""
session_user.set(None) session_user.set(None)
with patch( with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = True mock_config.caching = True
MockCacheConfig.return_value = mock_config MockCacheConfig.return_value = mock_config
@ -183,9 +181,7 @@ class TestConversationHistoryUtils:
user = create_mock_user() user = create_mock_user()
session_user.set(user) session_user.set(user)
with patch( with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = False mock_config.caching = False
MockCacheConfig.return_value = mock_config MockCacheConfig.return_value = mock_config
@ -305,9 +301,7 @@ class TestConversationHistoryUtils:
"""Test get_conversation_history returns empty string when user_id is None.""" """Test get_conversation_history returns empty string when user_id is None."""
session_user.set(None) session_user.set(None)
with patch( with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = True mock_config.caching = True
MockCacheConfig.return_value = mock_config MockCacheConfig.return_value = mock_config
@ -326,9 +320,7 @@ class TestConversationHistoryUtils:
user = create_mock_user() user = create_mock_user()
session_user.set(user) session_user.set(user)
with patch( with patch("cognee.modules.retrieval.utils.session_cache.CacheConfig") as MockCacheConfig:
"cognee.modules.retrieval.utils.session_cache.CacheConfig"
) as MockCacheConfig:
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = False mock_config.caching = False
MockCacheConfig.return_value = mock_config MockCacheConfig.return_value = mock_config
@ -367,9 +359,7 @@ class TestConversationHistoryUtils:
await get_conversation_history(session_id=None) await get_conversation_history(session_id=None)
mock_cache.get_latest_qa.assert_called_once_with( mock_cache.get_latest_qa.assert_called_once_with(str(user.id), "default_session")
str(user.id), "default_session"
)
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_get_conversation_history_cache_engine_none(self): async def test_get_conversation_history_cache_engine_none(self):
@ -406,9 +396,7 @@ class TestConversationHistoryUtils:
from cognee.infrastructure.databases.exceptions import CacheConnectionError from cognee.infrastructure.databases.exceptions import CacheConnectionError
mock_cache = create_mock_cache_engine([]) mock_cache = create_mock_cache_engine([])
mock_cache.get_latest_qa = AsyncMock( mock_cache.get_latest_qa = AsyncMock(side_effect=CacheConnectionError("Connection failed"))
side_effect=CacheConnectionError("Connection failed")
)
cache_module = importlib.import_module( cache_module = importlib.import_module(
"cognee.infrastructure.databases.cache.get_cache_engine" "cognee.infrastructure.databases.cache.get_cache_engine"

3
cognee/tests/unit/modules/retrieval/test_completion.py
View file

@ -227,7 +227,8 @@ class TestGenerateCompletion:
) )
mock_render.assert_called_once_with( mock_render.assert_called_once_with(
"user_prompt.txt", {"question": "What is AI?", "context": "AI is artificial intelligence"} "user_prompt.txt",
{"question": "What is AI?", "context": "AI is artificial intelligence"},
) )