| 12345678910111213141516171819202122232425 |
- import pytest
- from app.memory.store import MemoryManager
- @pytest.mark.asyncio
- async def test_memory_manager_init():
- mgr = MemoryManager()
- assert mgr is not None
- assert mgr.memory_vectorstore is not None
- @pytest.mark.asyncio
- async def test_working_memory():
- mgr = MemoryManager()
- mem = mgr.get_working_memory()
- assert mem is not None
- assert mem.memory_key == "history"
- @pytest.mark.asyncio
- async def test_longterm_memory():
- mgr = MemoryManager()
- mem = mgr.get_longterm_memory(user_id=1)
- assert mem is not None
- assert mem.memory_key == "long_term_memory"
|
