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-10 18:18:24 +01:00
parent 0329777290
commit ff1add1af0
1 changed files with 18 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
cognee/tests/unit/modules/retrieval/triplet_retriever_test.py
View file

@ -92,15 +92,17 @@ async def test_get_completion_without_session(mock_vector_engine):
retriever = TripletRetriever() retriever = TripletRetriever()
with patch( with (
patch(
"cognee.modules.retrieval.triplet_retriever.get_vector_engine", "cognee.modules.retrieval.triplet_retriever.get_vector_engine",
return_value=mock_vector_engine, return_value=mock_vector_engine,
), patch( ),
patch(
"cognee.modules.retrieval.triplet_retriever.generate_completion", "cognee.modules.retrieval.triplet_retriever.generate_completion",
return_value="Generated answer", return_value="Generated answer",
), patch( ),
"cognee.modules.retrieval.triplet_retriever.CacheConfig" patch("cognee.modules.retrieval.triplet_retriever.CacheConfig") as mock_cache_config,
) as mock_cache_config: ):
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = False mock_config.caching = False
mock_cache_config.return_value = mock_config mock_cache_config.return_value = mock_config
@ -117,12 +119,13 @@ async def test_get_completion_with_provided_context(mock_vector_engine):
"""Test get_completion with provided context.""" """Test get_completion with provided context."""
retriever = TripletRetriever() retriever = TripletRetriever()
with patch( with (
patch(
"cognee.modules.retrieval.triplet_retriever.generate_completion", "cognee.modules.retrieval.triplet_retriever.generate_completion",
return_value="Generated answer", return_value="Generated answer",
), patch( ),
"cognee.modules.retrieval.triplet_retriever.CacheConfig" patch("cognee.modules.retrieval.triplet_retriever.CacheConfig") as mock_cache_config,
) as mock_cache_config: ):
mock_config = MagicMock() mock_config = MagicMock()
mock_config.caching = False mock_config.caching = False
mock_cache_config.return_value = mock_config mock_cache_config.return_value = mock_config