From ec4566c3d17370c778a1e6cd6b22ed85263731a0 Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Sat, 2 Aug 2025 11:30:27 +0800 Subject: refactor(v3): publish v3 branch --- tests/test_generator.py | 68 ------------------------------------------------- 1 file changed, 68 deletions(-) delete mode 100644 tests/test_generator.py (limited to 'tests/test_generator.py') diff --git a/tests/test_generator.py b/tests/test_generator.py deleted file mode 100644 index 30ad01bd..00000000 --- a/tests/test_generator.py +++ /dev/null @@ -1,68 +0,0 @@ -from infini.generator import Generator, TextGenerator -from infini.injector import Injector -from infini.output import Output - - -def test_generator(): - generator = TextGenerator() - generator.events = { - "test.event1": "Event1 文本", - } - generator.match(Output("text", "test.event1")) - assert generator.output(Output("text", "test.event1"), Injector()) == "Event1 文本" - - -def test_generator_with_var(): - generator = TextGenerator() - generator.events = { - "test.event1": "Event1 文本: {{ var }}", - } - - assert ( - generator.output( - Output("text", "test.event1", variables={"var": "变量测试"}), Injector() - ) - == "Event1 文本: 变量测试" - ) - - -def test_generator_injector(): - def name(nickname: str = "苏向夜"): - return nickname - - injector = Injector() - injector.parameters = {"a": 12, "b": 20, "c": 0} - - generator = TextGenerator() - generator.events = { - "test.event1": "[{{ card_name }}]Event1 文本: {{ var }}", - } - generator.global_variables = {"card_name": name} - assert ( - generator.output( - Output("text", "test.event1", variables={"var": "变量测试"}), injector - ) - == "[苏向夜]Event1 文本: 变量测试" - ) - -def test_register_generator(): - def name(nickname: str = "苏向夜"): - return nickname - - custom = TextGenerator() - custom.type = "custom_text" - - generator = Generator() - generator.events = { - "test.event1": "[{{ card_name }}]Event1 文本: {{ var }}", - } - generator.generators.update({"custom_text": custom}) - - generator.global_variables = {"card_name": name} - assert ( - generator.output( - Output("custom_text", "test.event1", variables={"var": "变量测试"}), - Injector(), - ) - == "[苏向夜]Event1 文本: 变量测试" - ) -- cgit v1.2.3-70-g09d2