aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_core.py
diff options
context:
space:
mode:
author苏向夜 <fu050409@163.com>2024-02-28 22:45:42 +0800
committer苏向夜 <fu050409@163.com>2024-02-28 22:45:42 +0800
commit318a30667b03a3dce5deedaadd7a71785edd4088 (patch)
tree0d5affae3c2afcd032d96c9cc1b5540ca8e7a239 /tests/test_core.py
parentb30be75b29d8320da812ecf7accd96af03424ad4 (diff)
downloadinfini-318a30667b03a3dce5deedaadd7a71785edd4088.tar.gz
infini-318a30667b03a3dce5deedaadd7a71785edd4088.zip
test(injector): fix tests after injector added
Diffstat (limited to 'tests/test_core.py')
-rw-r--r--tests/test_core.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/test_core.py b/tests/test_core.py
index 11a8c57d..5f3f12f1 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -1,6 +1,7 @@
from infini.core import Core
from infini.generator import TextGenerator
from infini.handler import Handler
+from infini.injector import Injector
from infini.input import Input
from infini.interceptor import Interceptor
from infini.output import Output
@@ -60,6 +61,7 @@ def test_core():
core.interceptor = interceptor
core.pre_interceptor = interceptor
core.generator = generator
+ core.injector = Injector()
outputs = set()
for output in core.input(command_input):