scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix: update test

Browse source
This commit is contained in:
Rhys 2026-02-23 10:29:56 +07:00
parent f0a8f2b425
commit 51b229156e
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/test_gpt_image_cost_calculator.py
View file

@ -269,9 +269,9 @@ class TestGPTImage15OutputImageTokens:
f"Image tokens may not be included in cost calculation."
)
def test_azure_gpt_image_15_output_image_tokens_cost(self):
def test_azure_gpt_image_15_output_tokens_cost(self):
"""
Test that output image tokens are correctly included in cost calculation for Azure.
Test that output tokens are correctly calculated in cost calculation for Azure.
"""
# Simulate gpt-image-1.5 response with output_tokens_details
usage = Usage(
@ -316,7 +316,7 @@ class TestGPTImage15OutputImageTokens:
assert abs(cost - expected_cost) < 1e-6, (
f"Expected {expected_cost}, got {cost}. "
f"Image tokens may not be included in cost calculation."
f"Output tokens may be calculated in a wrong rate for Azure."
)