diff options
| author | 2025-08-02 11:30:27 +0800 | |
|---|---|---|
| committer | 2025-08-02 11:30:27 +0800 | |
| commit | ec4566c3d17370c778a1e6cd6b22ed85263731a0 (patch) | |
| tree | 469258084d3ae0e72db37dd98593731ff20131e4 /tests/test_loader.py | |
| parent | c68a18ca6440f6460b3b5fce311901229f8f50a9 (diff) | |
| download | infini-ec4566c3d17370c778a1e6cd6b22ed85263731a0.tar.gz infini-ec4566c3d17370c778a1e6cd6b22ed85263731a0.zip | |
refactor(v3): publish v3 branch
Diffstat (limited to 'tests/test_loader.py')
| -rw-r--r-- | tests/test_loader.py | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/tests/test_loader.py b/tests/test_loader.py deleted file mode 100644 index 82c1dc7b..00000000 --- a/tests/test_loader.py +++ /dev/null @@ -1,47 +0,0 @@ -from infini.input import Input -from infini.loader import Loader -from infini.output import Output -from infini.register import Register - - -def test_loader(): - blocked_god_input = Input("这是苏向夜的杰作.") - snh_input = Input("撅少年狐!") - - register = Register() - - @register.pre_interceptor("苏向夜", priority=0) - def test_pre_interceptor(_: Input): - return Output("text", "block.sxy", block=True) - - @register.handler("撅少年狐") - def test_handler(_: Input): - return Output("text", "block.snh", block=True) - - register.register_textevent("block.sxy", "不可直呼{{ sxy_id }}的ID") - register.register_textevent("block.snh", "不许撅{{ get_snh_id }}") - - register.register_variable("sxy_id", "苏向夜") - - @register.dynamic_variable() - def get_snh_id(): - return "少年狐" - - @register.interceptor("苏向夜", priority=0) - def test_interceptor(_: Input): - return Output("text", "block.sxy", block=True) - - loader = Loader() - loader.load_from_register(register) - core = loader.into_core() - - for output in core.input(blocked_god_input): - assert output == "不可直呼苏向夜的ID" - - for output in core.input(snh_input): - assert output == "不许撅少年狐" - - -def test_with_loader(): - with Loader() as loader: - loader.into_core() |
