(self)
| 80 | self.assertEqual(self.memory.memory, new_memory) |
| 81 | |
| 82 | def test_save_and_load_memory(self): |
| 83 | self.memory.push("user", "Hello") |
| 84 | self.memory.push("assistant", "Hi") |
| 85 | self.memory.save_memory() |
| 86 | |
| 87 | new_memory = Memory(self.system_prompt, recover_last_session=True) |
| 88 | new_memory.load_memory() |
| 89 | self.assertEqual(len(new_memory.memory), 3) # System + messages |
| 90 | self.assertEqual(new_memory.memory[1]['content'], "Hello") |
| 91 | |
| 92 | if __name__ == '__main__': |
| 93 | unittest.main() |
nothing calls this directly
no test coverage detected