diff options
| author |   苏向夜 <fu050409@163.com> | 2024-03-04 11:48:35 +0800 |
|---|---|---|
| committer | 苏向夜 <fu050409@163.com> | 2024-03-04 11:48:35 +0800 |
| commit | c5943be36fcdf442a92e2a4a4fc7fa03fbefc268 (patch) | |
| tree | 70bb997e6ed9efa6be7a0e3d916e575a81186a49 /tests/test_interceptor.py | |
| parent | 0f32d6f3520e03a48f9143f2a4f7e48be7742e9b (diff) | |
| download | infini-c5943be36fcdf442a92e2a4a4fc7fa03fbefc268.tar.gz infini-c5943be36fcdf442a92e2a4a4fc7fa03fbefc268.zip | |
feat(test): update tests for injector changes
Diffstat (limited to 'tests/test_interceptor.py')
| -rw-r--r-- | tests/test_interceptor.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_interceptor.py b/tests/test_interceptor.py index ffb03d56..ded63471 100644 --- a/tests/test_interceptor.py +++ b/tests/test_interceptor.py @@ -27,10 +27,12 @@ def test_interceptor(): assert isinstance(valid_output, Input) assert valid_output.get_plain_text() == "这个叫苏向夜." - for output in interceptor.output("简律纯"): + for output in interceptor.output(Output("text", "none", block=True), "简律纯"): assert isinstance(output, Output) assert output.name == "block.jianlvchun" - for output in interceptor.output("这个叫苏向夜."): + for output in interceptor.output( + Output("text", "none", block=True), "这个叫苏向夜." + ): assert isinstance(output, str) assert output == "这个叫苏向夜." |