(self)
| 46 | self.assertTrue(os.path.exists(os.path.join(save_path, filename))) |
| 47 | |
| 48 | def test_push(self): |
| 49 | index = self.memory.push("user", "Hello") |
| 50 | self.assertEqual(index, 0) |
| 51 | self.assertEqual(len(self.memory.memory), 2) |
| 52 | self.assertEqual(self.memory.memory[1]['role'], "user") |
| 53 | self.assertEqual(self.memory.memory[1]['content'], "Hello") |
| 54 | |
| 55 | def test_clear(self): |
| 56 | self.memory.push("user", "Hello") |