diff options
Diffstat (limited to 'tests/test_generator.py')
| -rw-r--r-- | tests/test_generator.py | 68 |
1 files changed, 0 insertions, 68 deletions
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 文本: 变量测试" - ) |
