(self)
| 58 | self.assertEqual(len(self.memory.memory), 1) # doesn't clear sys message |
| 59 | |
| 60 | def test_clear_section(self): |
| 61 | self.memory.clear() |
| 62 | mem_begin_idx = self.memory.push("user", "Hi i want you to make...") |
| 63 | self.memory.push("assistant", "<code>") |
| 64 | self.memory.push("user", "sys feedback: error") |
| 65 | self.memory.push("assistant", "<corrected code>") |
| 66 | mem_end_idx = self.memory.push("user", "according to search...") |
| 67 | self.memory.clear_section(mem_begin_idx+1, mem_end_idx-1) |
| 68 | self.assertEqual(len(self.memory.memory), 3) # 3 msg with sys msg |
| 69 | self.assertEqual(self.memory.memory[0]['role'], "system") |
| 70 | |
| 71 | def test_get(self): |
| 72 | self.memory.push("user", "Hello") |
nothing calls this directly
no test coverage detected