more tests for WorkingMemory

cheshire-cat-ai · Jan 27, 2025 · 356b822 · 356b822
1 parent 94d5acc
commit 356b822
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/core/tests/memory/test_class_working_memory.py b/core/tests/memory/test_class_working_memory.py
@@ -65,4 +65,16 @@ def test_langchainfy_chat_history():
     assert langchain_convo[1].name == "AI"
     assert langchain_convo[1].content == "Meow"
 
+
+def test_working_memory_as_dictionary_object():
+
+    wm = WorkingMemory()
+    wm.a = "a"
+    wm["b"] = "b"
+    assert wm.a == "a"
+    assert wm["a"] == "a"
+    assert wm.b == "b"
+    assert wm["b"] == "b"
+    # assert wm.c is None # too dangerous
+
 # TODOV2: add tests for multimodal messages!