diff options
Diffstat (limited to 'tests/test_loader.py')
| -rw-r--r-- | tests/test_loader.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/test_loader.py b/tests/test_loader.py index 552223d3..439700e5 100644 --- a/tests/test_loader.py +++ b/tests/test_loader.py @@ -33,7 +33,7 @@ def test_loader(): loader = Loader() loader.load_from_register(register) - core = loader.output() + core = loader.into_core() for output in core.input(blocked_god_input): assert output == "不可直呼苏向夜的ID" @@ -42,5 +42,6 @@ def test_loader(): assert output == "不许撅少年狐" -if __name__ == "__main__": - test_loader() +def test_with_loader(): + with Loader() as loader: + loader.into_core() |