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

fix: Raise NotFoundError in list_eval_sets function when app_name doesn't exist

Browse Source 
PiperOrigin-RevId: 784216832
This commit is contained in:
Ankur Sharma
2025-07-17 09:53:26 -07:00
committed by Copybara-Service
parent 377b5a9b78
commit b17d8b6e36
3 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/evaluation/test_gcs_eval_sets_manager.py
+13
View File
@@ -17,6 +17,7 @@ from google.adk.evaluation.eval_case import EvalCase
from google.adk.evaluation.eval_set import EvalSet
from google.adk.evaluation.gcs_eval_sets_manager import _EVAL_SET_FILE_EXTENSION
from google.adk.evaluation.gcs_eval_sets_manager import GcsEvalSetsManager
from google.cloud import exceptions as cloud_exceptions
import pytest
from .mock_gcs_utils import MockBlob
@@ -121,6 +122,18 @@ class TestGcsEvalSetsManager:
assert eval_sets == ["eval_set_1", "eval_set_2"]
def test_gcs_eval_sets_manager_list_eval_sets_fails(
self, gcs_eval_sets_manager, mocker
):
mocker.patch.object(
gcs_eval_sets_manager.bucket,
"list_blobs",
side_effect=cloud_exceptions.NotFound("not found"),
)
with pytest.raises(NotFoundError):
gcs_eval_sets_manager.list_eval_sets("test_app")
def test_gcs_eval_sets_manager_add_eval_case_success(
self, gcs_eval_sets_manager, mocker
):