From f3b426b6911c6925c6d0b165ed998e818b9f021b Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Sat, 27 Jan 2024 16:24:18 +0800
Subject: :sparkles: test(loader): refact loader tests

---
 tests/test_loader.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'tests/test_loader.py')

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()
-- 
cgit v1.2.3-70-g09d2