mirror of
https://github.com/kennethreitz/langchain.git
synced 2026-06-05 23:00:18 +00:00
Harrison Chase
12 lines
293 B
Python
12 lines
293 B
Python
from langchain.chains.base import SimpleMemory
|
|
|
|
|
|
def test_simple_memory() -> None:
|
|
"""Test SimpleMemory."""
|
|
memory = SimpleMemory(memories={"baz": "foo"})
|
|
|
|
output = memory.load_memory_variables({})
|
|
|
|
assert output == {"baz": "foo"}
|
|
assert ["baz"] == memory.memory_variables
|