phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_users.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-19 19:48:44 -07:00
parent 9140453d0c
commit 97e7113b87
1 changed files with 6 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/test_users.py
View file

@ -135,17 +135,12 @@ async def test_users_budgets_reset():
reset_at_init_value = user_info["user_info"]["budget_reset_at"]
i = 0
reset_at_new_value = None
while i < 3:
user_info = await retry_request(
get_user_info, session=session, get_user=get_user, call_user=key
)
reset_at_new_value = user_info["user_info"]["budget_reset_at"]
try:
assert reset_at_init_value != reset_at_new_value
break
except:
i + 1
await asyncio.sleep(70)
await asyncio.sleep(70)
user_info = await retry_request(
get_user_info, session=session, get_user=get_user, call_user=key
)
reset_at_new_value = user_info["user_info"]["budget_reset_at"]
assert reset_at_init_value != reset_at_new_value