From 4df3bfac12417e5589b9e944f4e7b3927b3d652e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 19:59:18 -0800 Subject: [PATCH] (test) assert error raised is ExceededBudget --- litellm/tests/test_key_generate_dynamodb.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py index 96da4e9df2..559cc9ff52 100644 --- a/litellm/tests/test_key_generate_dynamodb.py +++ b/litellm/tests/test_key_generate_dynamodb.py @@ -190,8 +190,9 @@ def test_call_with_key_under_budget(): asyncio.run(test()) except Exception as e: - pass - print("Got Exception", e) + error_detail = e.detail + assert "Authentication Error, ExceededBudget:" in error_detail + print(vars(e)) # def test_call_with_key_over_budget():