NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes test_cli_eval_success_path test.

Browse Source 
PiperOrigin-RevId: 758981214
This commit is contained in:
Ankur Sharma 2025-05-14 22:20:13 -07:00 committed by Copybara-Service
parent ada24d7171
commit 3686608f8d
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/unittests/cli/utils/test_cli_tools_click.py
View File

@ -244,6 +244,9 @@ def test_cli_eval_success_path(
# stub cli_eval module # stub cli_eval module
stub = types.ModuleType("google.adk.cli.cli_eval") stub = types.ModuleType("google.adk.cli.cli_eval")
eval_sets_manager_stub = types.ModuleType(
"google.adk.evaluation.local_eval_sets_manager"
)
class _EvalMetric: class _EvalMetric:
@ -252,10 +255,20 @@ def test_cli_eval_success_path(
class _EvalCaseResult: class _EvalCaseResult:
def __init__(self, eval_set_file: str, final_eval_status: str) -> None: def __init__(self, eval_set_id: str, final_eval_status: str) -> None:
self.eval_set_file = eval_set_file self.eval_set_id = eval_set_id
self.final_eval_status = final_eval_status self.final_eval_status = final_eval_status
class EvalCase:
def __init__(self, eval_id: str):
self.eval_id = eval_id
class EvalSet:
def __init__(self, eval_cases: list[EvalCase]):
self.eval_cases = eval_cases
# minimal enum-like namespace # minimal enum-like namespace
_EvalStatus = types.SimpleNamespace(PASSED="PASSED", FAILED="FAILED") _EvalStatus = types.SimpleNamespace(PASSED="PASSED", FAILED="FAILED")
@ -269,6 +282,9 @@ def test_cli_eval_success_path(
stub.get_root_agent = lambda _p: object() stub.get_root_agent = lambda _p: object()
stub.try_get_reset_func = lambda _p: None stub.try_get_reset_func = lambda _p: None
stub.parse_and_get_evals_to_run = lambda _paths: {"set1.json": ["e1", "e2"]} stub.parse_and_get_evals_to_run = lambda _paths: {"set1.json": ["e1", "e2"]}
eval_sets_manager_stub.load_eval_set_from_file = lambda x, y: EvalSet(
[EvalCase("e1"), EvalCase("e2")]
)
# Create an async generator function for run_evals # Create an async generator function for run_evals
async def mock_run_evals(*_a, **_k): async def mock_run_evals(*_a, **_k):
@ -290,6 +306,9 @@ def test_cli_eval_success_path(
# inject stub # inject stub
sys.modules["google.adk.cli.cli_eval"] = stub sys.modules["google.adk.cli.cli_eval"] = stub
sys.modules["google.adk.evaluation.local_eval_sets_manager"] = (
eval_sets_manager_stub
)
# create dummy agent directory # create dummy agent directory
agent_dir = tmp_path / "agent5" agent_dir = tmp_path / "agent5"