From cb024047ac189b26fcf4a1f47a9f588c40eebef9 Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 11 Nov 2026 03:32:19 +0000 Subject: [PATCH] test(spend): expect token billing for Vertex image edits with usage (#51109) Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../integration/spend/test_vertex_image_regional_uplift.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/integration/spend/test_vertex_image_regional_uplift.py b/tests/integration/spend/test_vertex_image_regional_uplift.py index dfd52d28f1f..8cb116bd347 100644 --- a/tests/integration/spend/test_vertex_image_regional_uplift.py +++ b/tests/integration/spend/test_vertex_image_regional_uplift.py @@ -242,7 +242,7 @@ def test_image_generation_bills_the_deployment_location( [ pytest.param("global", False, _GLOBAL_IMAGE_COST, id="global-edit"), pytest.param(_REGION, False, _REGIONAL_IMAGE_COST, id="regional-edit"), - pytest.param(_REGION, True, _REGIONAL_IMAGE_COST, id="regional-edit-usage-is-per-image"), + pytest.param(_REGION, True, _REGIONAL_TOKEN_COST, id="regional-edit-usage-bills-tokens"), ], ) def test_image_edit_bills_the_deployment_location( @@ -489,8 +489,8 @@ def _burst_calls(global_model: str, regional_model: str) -> tuple[_Call, ...]: kinds: Final = ( ("/v1/images/generations", global_model, _GLOBAL_TOKEN_COST), ("/v1/images/generations", regional_model, _REGIONAL_TOKEN_COST), - ("/v1/images/edits", global_model, _GLOBAL_IMAGE_COST), - ("/v1/images/edits", regional_model, _REGIONAL_IMAGE_COST), + ("/v1/images/edits", global_model, _GLOBAL_TOKEN_COST), + ("/v1/images/edits", regional_model, _REGIONAL_TOKEN_COST), ) return tuple( _Call(call_id=uuid.uuid4().hex, route=route, model=model, expected=expected)