phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_long_term_spend_accuracy_with_bursts

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-31 19:17:13 -07:00
parent 4959512170
commit a753fc9d9f
2 changed files with 9 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/otel_tests/local_test_spend_accuracy_tests.py
View file

@ -266,14 +266,14 @@ async def test_long_term_spend_accuracy_with_bursts():
abs(key_info["info"]["spend"] - expected_spend) < TOLERANCE
), f"Key spend {key_info['info']['spend']} does not match expected {expected_spend}"
# assert (
# abs(user_info["user_info"]["spend"] - expected_spend) < TOLERANCE
# ), f"User spend {user_info['user_info']['spend']} does not match expected {expected_spend}"
assert (
abs(user_info["user_info"]["spend"] - expected_spend) < TOLERANCE
), f"User spend {user_info['user_info']['spend']} does not match expected {expected_spend}"
# assert (
# abs(team_info["team_info"]["spend"] - expected_spend) < TOLERANCE
# ), f"Team spend {team_info['team_info']['spend']} does not match expected {expected_spend}"
assert (
abs(team_info["team_info"]["spend"] - expected_spend) < TOLERANCE
), f"Team spend {team_info['team_info']['spend']} does not match expected {expected_spend}"
# assert (
# abs(org_info["spend"] - expected_spend) < TOLERANCE
# ), f"Organization spend {org_info['spend']} does not match expected {expected_spend}"
assert (
abs(org_info["spend"] - expected_spend) < TOLERANCE
), f"Organization spend {org_info['spend']} does not match expected {expected_spend}"

6
tests/proxy_unit_tests/test_update_spend.py
View file

@ -27,12 +27,6 @@ class MockPrismaClient:
# Initialize transaction lists
self.spend_log_transactions = []
self.user_list_transactons = {}
self.end_user_list_transactons = {}
self.key_list_transactons = {}
self.team_list_transactons = {}
self.team_member_list_transactons = {}
self.org_list_transactons = {}
self.daily_user_spend_transactions = {}
def jsonify_object(self, obj):