From b2574ffd2aa6d2583ee6e9ccb12f414e8add0c43 Mon Sep 17 00:00:00 2001 From: 简律纯 Date: Mon, 15 Apr 2024 05:44:37 +0800 Subject: chore(tests): fix injector test --- tests/test_generator.py | 1 - tests/test_injector.py | 7 ++++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_generator.py b/tests/test_generator.py index ce33f4ff..30ad01bd 100644 --- a/tests/test_generator.py +++ b/tests/test_generator.py @@ -45,7 +45,6 @@ def test_generator_injector(): == "[苏向夜]Event1 文本: 变量测试" ) - def test_register_generator(): def name(nickname: str = "苏向夜"): return nickname diff --git a/tests/test_injector.py b/tests/test_injector.py index e6ae2e56..d6c6d3e8 100644 --- a/tests/test_injector.py +++ b/tests/test_injector.py @@ -9,16 +9,17 @@ from infini.router import Startswith def test_injector(): - def name(name: str): - return name + def name(nickname: str, c: int = 0): + return nickname, c def add(a: int, b: int = 0): return a + b injector = Injector() - injector.parameters = {"a": 12, "b": 20, "c": 0, "card_name": name} + injector.parameters = {"a": 12, "b": 20, "c": 10, "nickname": "苏向夜"} assert injector.inject(add)() == 32 assert injector.output(add) == 32 + assert injector.output(name) == ("苏向夜", 10) def test_handler_injector(): -- cgit v1.2.3-70-g09d2