Formatting

tests/test_memory.py

@@ -1,18 +1,17 @@
 """Tests for the Memory system."""
-import pytest
-import json
+
 from datetime import datetime
-from pathlib import Path
+
+import pytest
 
 from infrastructure.persistence import (
-    Memory,
-    LongTermMemory,
-    ShortTermMemory,
     EpisodicMemory,
-    init_memory,
+    LongTermMemory,
+    Memory,
+    ShortTermMemory,
     get_memory,
-    set_memory,
     has_memory,
+    init_memory,
 )
 from infrastructure.persistence.context import _memory_ctx
 
@@ -23,11 +22,12 @@ def is_iso_format(s: str) -> bool:
         return False
     try:
         # Attempt to parse the string as an ISO 8601 timestamp
-        datetime.fromisoformat(s.replace('Z', '+00:00'))
+        datetime.fromisoformat(s.replace("Z", "+00:00"))
         return True
     except (ValueError, TypeError):
         return False
 
+
 class TestLongTermMemory:
     """Tests for LongTermMemory."""
 
@@ -116,12 +116,18 @@ class TestLongTermMemory:
         assert data["config"]["key"] == "value"
 
     def test_from_dict(self):
-        data = {"config": {"download_folder": "/downloads"}, "preferences": {"preferred_quality": "4K"}, "library": {"movies": [{"imdb_id": "tt1", "title": "Test"}]}, "following": []}
+        data = {
+            "config": {"download_folder": "/downloads"},
+            "preferences": {"preferred_quality": "4K"},
+            "library": {"movies": [{"imdb_id": "tt1", "title": "Test"}]},
+            "following": [],
+        }
         ltm = LongTermMemory.from_dict(data)
         assert ltm.get_config("download_folder") == "/downloads"
         assert ltm.preferences["preferred_quality"] == "4K"
         assert len(ltm.library["movies"]) == 1
 
+
 class TestShortTermMemory:
     """Tests for ShortTermMemory."""
 
@@ -162,6 +168,7 @@ class TestShortTermMemory:
         assert stm.conversation_history == []
         assert stm.language == "en"
 
+
 class TestEpisodicMemory:
     """Tests for EpisodicMemory."""
 
@@ -192,6 +199,7 @@ class TestEpisodicMemory:
         assert result is not None
         assert result["name"] == "Result 2"
 
+
 class TestMemory:
     """Tests for the Memory manager."""
 
@@ -217,11 +225,10 @@ class TestMemory:
         assert memory.stm.conversation_history == []
         assert memory.episodic.recent_errors == []
 
+
 class TestMemoryContext:
     """Tests for memory context functions."""
 
-
-
     def test_get_memory_not_initialized(self):
         _memory_ctx.set(None)
         with pytest.raises(RuntimeError, match="Memory not initialized"):