From d9ddc4b9010f2a5d71cd27fd26461274ca7e4848 Mon Sep 17 00:00:00 2001 From: yassin Date: Thu, 17 Sep 2026 19:50:50 +0000 Subject: [PATCH] test(proxy): assert temp budget increase stops at the exact expiry instant Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/test_litellm/proxy/auth/test_auth_checks.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py index 518dad8c48f..81bf1471a00 100644 --- a/tests/test_litellm/proxy/auth/test_auth_checks.py +++ b/tests/test_litellm/proxy/auth/test_auth_checks.py @@ -8477,12 +8477,10 @@ def test_effective_team_member_budget_applies_unexpired_increase() -> None: def test_effective_team_member_budget_ignores_expired_increase() -> None: from litellm.proxy.auth.auth_checks import _effective_team_member_budget - budget: Final = LiteLLM_BudgetTable( - max_budget=100.0, - temp_budget_increase=50.0, - temp_budget_expiry=datetime(2020, 1, 1, tzinfo=timezone.utc), - ) + expiry: Final = datetime(2020, 1, 1, tzinfo=timezone.utc) + budget: Final = LiteLLM_BudgetTable(max_budget=100.0, temp_budget_increase=50.0, temp_budget_expiry=expiry) assert _effective_team_member_budget(budget, now=datetime(2026, 1, 1, tzinfo=timezone.utc)) == 100.0 + assert _effective_team_member_budget(budget, now=expiry) == 100.0 def test_effective_team_member_budget_without_increase() -> None: