encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

chore: Fix evaluation test cases to only use pytest features

Browse Source 
PiperOrigin-RevId: 820700378
This commit is contained in:
Ankur Sharma
2025-10-17 08:25:17 -07:00
committed by Copybara-Service
parent 9dce06f9b0
commit cf3403231d
10 changed files with 176 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/evaluation/test_evaluation_generator.py
+8 -10
View File
@@ -14,8 +14,6 @@
from __future__ import annotations
from unittest import mock
from google.adk.evaluation.app_details import AgentDetails
from google.adk.evaluation.app_details import AppDetails
from google.adk.evaluation.evaluation_generator import EvaluationGenerator
@@ -206,17 +204,17 @@ class TestConvertEventsToEvalInvocation:
class TestGetAppDetailsByInvocationId:
"""Test cases for EvaluationGenerator._get_app_details_by_invocation_id method."""
def test_get_app_details_by_invocation_id_empty(self):
def test_get_app_details_by_invocation_id_empty(self, mocker):
"""Tests with an empty list of events."""
mock_request_intercepter = mock.MagicMock(spec=_RequestIntercepterPlugin)
mock_request_intercepter = mocker.MagicMock(spec=_RequestIntercepterPlugin)
app_details = EvaluationGenerator._get_app_details_by_invocation_id(
[], mock_request_intercepter
)
assert app_details == {}
def test_get_app_details_by_invocation_id_no_model_requests(self):
def test_get_app_details_by_invocation_id_no_model_requests(self, mocker):
"""Tests when request_intercepter returns no model requests."""
mock_request_intercepter = mock.MagicMock(spec=_RequestIntercepterPlugin)
mock_request_intercepter = mocker.MagicMock(spec=_RequestIntercepterPlugin)
mock_request_intercepter.get_model_request.return_value = None
events = [
_build_event("user", [types.Part(text="Hello")], "inv1"),
@@ -230,9 +228,9 @@ class TestGetAppDetailsByInvocationId:
events[1]
)
def test_get_app_details_single_invocation_single_agent(self):
def test_get_app_details_single_invocation_single_agent(self, mocker):
"""Tests a single invocation with one agent."""
mock_request_intercepter = mock.MagicMock(spec=_RequestIntercepterPlugin)
mock_request_intercepter = mocker.MagicMock(spec=_RequestIntercepterPlugin)
mock_llm_request = LlmRequest(model="test")
mock_llm_request.config.system_instruction = "instruction1"
mock_llm_request.config.tools = [types.Tool()]
@@ -262,9 +260,9 @@ class TestGetAppDetailsByInvocationId:
events[1]
)
def test_get_app_details_multiple_invocations_multiple_agents(self):
def test_get_app_details_multiple_invocations_multiple_agents(self, mocker):
"""Tests multiple invocations with multiple agents."""
mock_request_intercepter = mock.MagicMock(spec=_RequestIntercepterPlugin)
mock_request_intercepter = mocker.MagicMock(spec=_RequestIntercepterPlugin)
def get_model_request_side_effect(event):
mock_llm_request = LlmRequest(model="test")