From df0d8ff15ea2f65e9f1ba90209239f241cf16d91 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 18 Aug 2026 22:38:13 -0700
Subject: [PATCH] test: assert the vertex batch output rate is a real discount

---
 tests/test_litellm/batches/test_batch_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_litellm/batches/test_batch_utils.py b/tests/test_litellm/batches/test_batch_utils.py
index 49ff8de1ab9..50f1380a4f8 100644
--- a/tests/test_litellm/batches/test_batch_utils.py
+++ b/tests/test_litellm/batches/test_batch_utils.py
@@ -895,6 +895,7 @@ async def test_handle_completed_vertex_batch_computes_cost_usage_and_models(monk
     batch_output = pricing["output_cost_per_token_batches"]
 
     assert batch_input < pricing["input_cost_per_token"]
+    assert batch_output < pricing["output_cost_per_token"]
     assert cost > 0
     assert cost == pytest.approx(30 * batch_input + 15 * batch_output)
     assert (usage.prompt_tokens, usage.completion_tokens, usage.total_tokens) == (30, 15, 45)