test(injector): fix tests after injector added

1 files changed, 5 insertions, 17 deletions

diff --git a/tests/test_generator.py b/tests/test_generator.py
index 0eed5756..e796efc7 100644
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -1,4 +1,5 @@
 from infini.generator import TextGenerator
+from infini.injector import Injector
 from infini.output import Output
 
 
@@ -8,7 +9,7 @@ def test_generator():
         "test.event1": "Event1 文本",
     }
     generator.match(Output("text", "test.event1"))
-    assert generator.output(Output("text", "test.event1")) == "Event1 文本"
+    assert generator.output(Output("text", "test.event1"), Injector()) == "Event1 文本"
 
 
 def test_generator_with_var():
@@ -18,21 +19,8 @@ def test_generator_with_var():
     }
 
     assert (
-        generator.output(Output("text", "test.event1", variables={"var": "变量测试"}))
+        generator.output(
+            Output("text", "test.event1", variables={"var": "变量测试"}), Injector()
+        )
         == "Event1 文本: 变量测试"
     )
-
-
-def test_generator_with_function():
-    def add(a, b):
-        return a + b
-
-    generator = TextGenerator()
-    generator.events = {
-        "test.event": "{{ func(1, 2) }}",
-    }
-
-    assert (
-        generator.output(Output("text", "test.event", variables={"func": add}))
-        == "3"
-    )