aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_injector.py
diff options
context:
space:
mode:
author苏向夜 <fu050409@163.com>2024-03-02 11:35:29 +0800
committer苏向夜 <fu050409@163.com>2024-03-02 11:35:29 +0800
commitc681a3d355089a91ec83a692664fb4cf6ef43f26 (patch)
tree79698c194de640b6d565d8e0ea929872d25b31f9 /tests/test_injector.py
parentb18c32da9df9673cc1acee78c251151f7e7411ec (diff)
downloadinfini-c681a3d355089a91ec83a692664fb4cf6ef43f26.tar.gz
infini-c681a3d355089a91ec83a692664fb4cf6ef43f26.zip
test(injector): update injector tests
Diffstat (limited to 'tests/test_injector.py')
-rw-r--r--tests/test_injector.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/test_injector.py b/tests/test_injector.py
index 0d320a2c..426e5684 100644
--- a/tests/test_injector.py
+++ b/tests/test_injector.py
@@ -2,10 +2,13 @@ from infini.injector import Injector
def test_injector():
+ def name(name: str):
+ return name
+
def add(a: int, b: int = 0):
return a + b
injector = Injector()
- injector.parameters = {"a": 12, "b": 20, "c": 0}
+ injector.parameters = {"a": 12, "b": 20, "c": 0, "card_name": name}
assert injector.inject(add)() == 32
assert injector.output(add) == 32