참고소스 수정본

참고/neo4j-graphrag-python-main/tests/unit/experimental/pipeline/test_store.py

@@ -0,0 +1,15 @@
+import pytest
+from neo4j_graphrag.experimental.pipeline.stores import InMemoryStore
+
+
+@pytest.mark.asyncio
+async def test_memory_store() -> None:
+    store = InMemoryStore()
+    await store.add("key", "value")
+    res = await store.get("key")
+    assert res == "value"
+
+    with pytest.raises(KeyError):
+        await store.add("key", "value", overwrite=False)
+
+    assert store.all() == {"key": "value"}