1 files changed, 11 insertions, 0 deletions
diff --git a/tests/test_generator.py b/tests/test_generator.py
new file mode 100644
index 00000000..32f1d626
--- /dev/null
+++ b/tests/test_generator.py
@@ -0,0 +1,11 @@
+from infini.generator import Generator
+from infini.output import Output
+
+
+def test_generator():
+    generator = Generator()
+    generator.events = {
+        "test.event1": "Event1 文本",
+    }
+    generator.match(Output("text", "test.event1"))
+    assert generator.output(Output("text", "test.event1")) == "Event1 文本"