Create tests/test_planner.py

tests/test_planner.py

@@ -0,0 +1,23 @@
+import unittest
+from planner import Planner, PlanStep
+
+class PlannerTest(unittest.TestCase):
+    def setUp(self):
+        self.planner = Planner(tool_registry={
+            "web_search": lambda query: {"results": ["res1", "res2"]},
+            "calculator": lambda expression: {"result": eval(expression)}
+        })
+
+    def test_generate_plan_and_execution(self):
+        plan = self.planner.generate_plan("search AI papers")
+        self.assertIsInstance(plan, list)
+        for step in plan:
+            result = self.planner.execute_step(step)
+            self.assertIsNotNone(result)
+
+    def test_interrupt_replan(self):
+        new_plan = self.planner.interrupt_and_replan("new context", "calculate 2 + 2")
+        self.assertIsInstance(new_plan, list)
+
+if __name__ == "__main__":
+    unittest.main()