aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test_injector.py
blob: 426e568451b9a943adf2151f05d44fd584fe31ef (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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, "card_name": name}
    assert injector.inject(add)() == 32
    assert injector.output(add) == 32