Inital Commit

tests/test_memory.py

@@ -0,0 +1,27 @@
+from traderai.memory import MemoryStore
+
+
+def test_memory_store_recalls_saved_fact(tmp_path):
+    store = MemoryStore(str(tmp_path / "memory.sqlite3"))
+    store.remember("preference", "The user prefers Polaris Bits searches to include barter listings.", importance=5)
+
+    results = store.recall("Polaris barter")
+
+    assert results
+    assert "Polaris Bits" in results[0]["content"]
+
+
+def test_memory_store_clear_selected_sections(tmp_path):
+    store = MemoryStore(str(tmp_path / "memory.sqlite3"))
+    store.remember("note", "Forgettable note")
+    store.add_conversation("user", "hello")
+    store.set_profile("configured_name", "Hudson")
+
+    deleted = store.clear(include_profile=False)
+    snapshot = store.inspect()
+
+    assert deleted["memories"] == 1
+    assert deleted["conversations"] == 1
+    assert snapshot["memories"] == []
+    assert snapshot["conversations"] == []
+    assert snapshot["profile"][0]["key"] == "configured_name"