scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test: use _local_model_cost_map fixture in gemini 3.7 flash pricing tests

Browse source
This commit is contained in:
mateo-berri 2026-08-13 10:13:40 -07:00
parent d3d259b211
commit d9020795dd
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py
View file

@ -2913,10 +2913,7 @@ GEMINI_37_FLASH_LAUNCH_PRICING = [
@pytest.mark.parametrize("model,input_cost,output_cost,cache_read_cost", GEMINI_37_FLASH_LAUNCH_PRICING)
def test_gemini_37_flash_launch_pricing(model, input_cost, output_cost, cache_read_cost):
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
def test_gemini_37_flash_launch_pricing(model, input_cost, output_cost, cache_read_cost, _local_model_cost_map):
model_cost_map = litellm.model_cost[model]
assert model_cost_map["input_cost_per_token"] == input_cost
assert model_cost_map["output_cost_per_token"] == output_cost
@ -2928,10 +2925,7 @@ def test_gemini_37_flash_launch_pricing(model, input_cost, output_cost, cache_re
assert model_cost_map["max_input_tokens"] == 1048576
def test_generic_cost_per_token_gemini_37_flash():
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
def test_generic_cost_per_token_gemini_37_flash(_local_model_cost_map):
usage = Usage(
prompt_tokens=1000,
completion_tokens=500,