From 318a30667b03a3dce5deedaadd7a71785edd4088 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Wed, 28 Feb 2024 22:45:42 +0800 Subject: test(injector): fix tests after injector added --- tests/test_injector.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 tests/test_injector.py (limited to 'tests/test_injector.py') diff --git a/tests/test_injector.py b/tests/test_injector.py new file mode 100644 index 00000000..0d320a2c --- /dev/null +++ b/tests/test_injector.py @@ -0,0 +1,11 @@ +from infini.injector import Injector + + +def test_injector(): + def add(a: int, b: int = 0): + return a + b + + injector = Injector() + injector.parameters = {"a": 12, "b": 20, "c": 0} + assert injector.inject(add)() == 32 + assert injector.output(add) == 32 -- cgit v1.2.3-70-g09d2