Add file and retrieve content.
(self)
| 87 | self.wm = WorkingMemory(max_tokens=MAX_FILE_CONTEXT_TOKENS) |
| 88 | |
| 89 | def test_add_and_get(self): |
| 90 | """Add file and retrieve content.""" |
| 91 | self.wm.add("test.py", "print('hello')", tokens=5) |
| 92 | content = self.wm.get("test.py") |
| 93 | self.assertEqual(content, "print('hello')") |
| 94 | |
| 95 | def test_get_nonexistent_returns_none(self): |
| 96 | """Getting nonexistent file returns None.""" |