(self)
| 74 | self.assertEqual(len(memory_content), 2) |
| 75 | |
| 76 | def test_reset(self): |
| 77 | self.memory.push("user", "Hello") |
| 78 | new_memory = [{"role": "system", "content": "New prompt"}] |
| 79 | self.memory.reset(new_memory) |
| 80 | self.assertEqual(self.memory.memory, new_memory) |
| 81 | |
| 82 | def test_save_and_load_memory(self): |
| 83 | self.memory.push("user", "Hello") |