diff options
| author | 2024-03-02 11:35:29 +0800 | |
|---|---|---|
| committer | 2024-03-02 11:35:29 +0800 | |
| commit | c681a3d355089a91ec83a692664fb4cf6ef43f26 (patch) | |
| tree | 79698c194de640b6d565d8e0ea929872d25b31f9 /tests/test_injector.py | |
| parent | b18c32da9df9673cc1acee78c251151f7e7411ec (diff) | |
| download | infini-c681a3d355089a91ec83a692664fb4cf6ef43f26.tar.gz infini-c681a3d355089a91ec83a692664fb4cf6ef43f26.zip | |
test(injector): update injector tests
Diffstat (limited to 'tests/test_injector.py')
| -rw-r--r-- | tests/test_injector.py | 5 |
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 |
