mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
test_long_term_spend_accuracy_with_bursts
This commit is contained in:
parent
4959512170
commit
a753fc9d9f
2 changed files with 9 additions and 15 deletions
|
@ -266,14 +266,14 @@ async def test_long_term_spend_accuracy_with_bursts():
|
||||||
abs(key_info["info"]["spend"] - expected_spend) < TOLERANCE
|
abs(key_info["info"]["spend"] - expected_spend) < TOLERANCE
|
||||||
), f"Key spend {key_info['info']['spend']} does not match expected {expected_spend}"
|
), f"Key spend {key_info['info']['spend']} does not match expected {expected_spend}"
|
||||||
|
|
||||||
# assert (
|
assert (
|
||||||
# abs(user_info["user_info"]["spend"] - expected_spend) < TOLERANCE
|
abs(user_info["user_info"]["spend"] - expected_spend) < TOLERANCE
|
||||||
# ), f"User spend {user_info['user_info']['spend']} does not match expected {expected_spend}"
|
), f"User spend {user_info['user_info']['spend']} does not match expected {expected_spend}"
|
||||||
|
|
||||||
# assert (
|
assert (
|
||||||
# abs(team_info["team_info"]["spend"] - expected_spend) < TOLERANCE
|
abs(team_info["team_info"]["spend"] - expected_spend) < TOLERANCE
|
||||||
# ), f"Team spend {team_info['team_info']['spend']} does not match expected {expected_spend}"
|
), f"Team spend {team_info['team_info']['spend']} does not match expected {expected_spend}"
|
||||||
|
|
||||||
# assert (
|
assert (
|
||||||
# abs(org_info["spend"] - expected_spend) < TOLERANCE
|
abs(org_info["spend"] - expected_spend) < TOLERANCE
|
||||||
# ), f"Organization spend {org_info['spend']} does not match expected {expected_spend}"
|
), f"Organization spend {org_info['spend']} does not match expected {expected_spend}"
|
||||||
|
|
|
@ -27,12 +27,6 @@ class MockPrismaClient:
|
||||||
|
|
||||||
# Initialize transaction lists
|
# Initialize transaction lists
|
||||||
self.spend_log_transactions = []
|
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 = {}
|
self.daily_user_spend_transactions = {}
|
||||||
|
|
||||||
def jsonify_object(self, obj):
|
def jsonify_object(self, obj):
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue