From c826511330fac36dca01851cfe9fc813ecc10b8a Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 20 May 2026 06:59:55 +0000
Subject: [PATCH] test(cost_calculator): assert output_cost_per_reasoning_token
 for gemini-3.1-flash-lite

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

diff --git a/tests/test_litellm/test_cost_calculator.py b/tests/test_litellm/test_cost_calculator.py
index db3e6511db2..18ab8a2a07a 100644
--- a/tests/test_litellm/test_cost_calculator.py
+++ b/tests/test_litellm/test_cost_calculator.py
@@ -2073,6 +2073,7 @@ def test_gemini_3_1_flash_lite_pricing():
         assert model_info["input_cost_per_token"] == 4.5e-07
         assert model_info["input_cost_per_audio_token"] == 9e-07
         assert model_info["output_cost_per_token"] == 2.7e-06
+        assert model_info["output_cost_per_reasoning_token"] == 2.7e-06
         assert model_info["cache_read_input_token_cost"] == 4.5e-08
         assert model_info["max_input_tokens"] == 1048576