tests/test_generator.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

from infini.generator import TextGenerator
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")) == "Event1 文本"


def test_generator_with_var():
    generator = TextGenerator()
    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 = TextGenerator()
    generator.events = {
        "test.event": "{{ func(1, 2) }}",
    }

    assert (
        generator.output(Output("text", "test.event", variables={"func": add}))
        == "3"
    )