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

update test

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-09 22:19:32 -07:00
parent 37f9ff1a9c
commit c12ceb69ae
1 changed files with 1 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_budget_manager.py
View file

@ -33,8 +33,6 @@ def test_user_budget_enough():
print(f"response: {response}") print(f"response: {response}")
test_user_budget_enough()
## Scenario 2: User budget not enough to make call ## Scenario 2: User budget not enough to make call
def test_user_budget_not_enough(): def test_user_budget_not_enough():
user = "12345" user = "12345"
@ -56,16 +54,10 @@ def test_user_budget_not_enough():
print(f"response: {response}") print(f"response: {response}")
test_user_budget_not_enough()
## Scenario 3: Saving budget to disk ## Scenario 3: Saving budget to disk
def test_budget_save_to_disk(): def test_budget_save_to_disk():
budget_manager.save_data() budget_manager.save_data()
test_budget_save_to_disk()
## Scenario 4: Loading budget from disk ## Scenario 4: Loading budget from disk
def test_budget_load_from_disk(): def test_budget_load_from_disk():
budget_manager_2 = BudgetManager(type="local") budget_manager_2 = BudgetManager(type="local")
test_budget_load_from_disk()