1 files changed, 27 insertions, 0 deletions
diff --git a/tests/test_generator.py b/tests/test_generator.py
index 32f1d626..bcab31ed 100644
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -9,3 +9,30 @@ def test_generator():
     }
     generator.match(Output("text", "test.event1"))
     assert generator.output(Output("text", "test.event1")) == "Event1 文本"
+
+
+def test_generator_with_var():
+    generator = Generator()
+    generator.events = {
+        "test.event1": "Event1 文本: {{ var }}",
+    }
+
+    assert (
+        generator.output(Output("text", "test.event1", variables={"var": "变量测试"}))
+        == "Event1 文本: 变量测试"
+    )
+
+
+def test_generator_with_function():
+    def add(a, b):
+        return a + b
+
+    generator = Generator()
+    generator.events = {
+        "test.event": "{{ func(1, 2) }}",
+    }
+
+    assert (
+        generator.output(Output("text", "test.event", variables={"func": add}))
+        == "3"
+    )