Initial commit

tests/test_input_source.py

@@ -0,0 +1,26 @@
+import unittest
+
+from agentos.input_source import TextInputSource
+
+
+class TextInputSourceTests(unittest.TestCase):
+    def test_skips_blank_input_and_returns_trimmed_transcript(self) -> None:
+        values = iter(["  ", "  中文输入  "])
+        source = TextInputSource(reader=lambda _prompt: next(values))
+
+        event = source.read()
+
+        self.assertIsNotNone(event)
+        assert event is not None
+        self.assertEqual(event.text, "中文输入")
+        self.assertEqual(event.source, "simulated_speech")
+
+    def test_eof_stops_input(self) -> None:
+        def end_input(_prompt: str) -> str:
+            raise EOFError
+
+        self.assertIsNone(TextInputSource(reader=end_input).read())
+
+
+if __name__ == "__main__":
+    unittest.main()