From 0d0fe3aa30c6eb8bb6d8021fc7942357c5fde37f Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Thu, 11 Dec 2025 12:27:27 +0100
Subject: [PATCH] ruff

---
 .../retrieval/temporal_retriever_test.py      | 44 +++++++++++++++----
 1 file changed, 35 insertions(+), 9 deletions(-)

diff --git a/cognee/tests/unit/modules/retrieval/temporal_retriever_test.py b/cognee/tests/unit/modules/retrieval/temporal_retriever_test.py
index 03433c379..1d2f4c84d 100644
--- a/cognee/tests/unit/modules/retrieval/temporal_retriever_test.py
+++ b/cognee/tests/unit/modules/retrieval/temporal_retriever_test.py
@@ -616,9 +616,15 @@ async def test_extract_time_from_query_relative_path():
     with (
         patch("cognee.modules.retrieval.temporal_retriever.os.path.isabs", return_value=False),
         patch("cognee.modules.retrieval.temporal_retriever.datetime") as mock_datetime,
-        patch("cognee.modules.retrieval.temporal_retriever.render_prompt", return_value="System prompt"),
+        patch(
+            "cognee.modules.retrieval.temporal_retriever.render_prompt",
+            return_value="System prompt",
+        ),
         patch.object(
-            LLMGateway, "acreate_structured_output", new_callable=AsyncMock, return_value=mock_interval
+            LLMGateway,
+            "acreate_structured_output",
+            new_callable=AsyncMock,
+            return_value=mock_interval,
         ),
     ):
         mock_datetime.now.return_value.strftime.return_value = "11-12-2024"
@@ -632,7 +638,9 @@ async def test_extract_time_from_query_relative_path():
 @pytest.mark.asyncio
 async def test_extract_time_from_query_absolute_path():
     """Test extract_time_from_query with absolute prompt path."""
-    retriever = TemporalRetriever(time_extraction_prompt_path="/absolute/path/to/extract_query_time.txt")
+    retriever = TemporalRetriever(
+        time_extraction_prompt_path="/absolute/path/to/extract_query_time.txt"
+    )
 
     mock_timestamp_from = Timestamp(year=2024, month=1, day=1)
     mock_timestamp_to = Timestamp(year=2024, month=12, day=31)
@@ -640,12 +648,24 @@ async def test_extract_time_from_query_absolute_path():
 
     with (
         patch("cognee.modules.retrieval.temporal_retriever.os.path.isabs", return_value=True),
-        patch("cognee.modules.retrieval.temporal_retriever.os.path.dirname", return_value="/absolute/path/to"),
-        patch("cognee.modules.retrieval.temporal_retriever.os.path.basename", return_value="extract_query_time.txt"),
+        patch(
+            "cognee.modules.retrieval.temporal_retriever.os.path.dirname",
+            return_value="/absolute/path/to",
+        ),
+        patch(
+            "cognee.modules.retrieval.temporal_retriever.os.path.basename",
+            return_value="extract_query_time.txt",
+        ),
         patch("cognee.modules.retrieval.temporal_retriever.datetime") as mock_datetime,
-        patch("cognee.modules.retrieval.temporal_retriever.render_prompt", return_value="System prompt"),
+        patch(
+            "cognee.modules.retrieval.temporal_retriever.render_prompt",
+            return_value="System prompt",
+        ),
         patch.object(
-            LLMGateway, "acreate_structured_output", new_callable=AsyncMock, return_value=mock_interval
+            LLMGateway,
+            "acreate_structured_output",
+            new_callable=AsyncMock,
+            return_value=mock_interval,
         ),
     ):
         mock_datetime.now.return_value.strftime.return_value = "11-12-2024"
@@ -666,9 +686,15 @@ async def test_extract_time_from_query_with_none_values():
     with (
         patch("cognee.modules.retrieval.temporal_retriever.os.path.isabs", return_value=False),
         patch("cognee.modules.retrieval.temporal_retriever.datetime") as mock_datetime,
-        patch("cognee.modules.retrieval.temporal_retriever.render_prompt", return_value="System prompt"),
+        patch(
+            "cognee.modules.retrieval.temporal_retriever.render_prompt",
+            return_value="System prompt",
+        ),
         patch.object(
-            LLMGateway, "acreate_structured_output", new_callable=AsyncMock, return_value=mock_interval
+            LLMGateway,
+            "acreate_structured_output",
+            new_callable=AsyncMock,
+            return_value=mock_interval,
         ),
     ):
         mock_datetime.now.return_value.strftime.return_value = "11-12-2024"