From 39d5af5c85661a54f2087f1746bb4c4d9d0f1be4 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 20:01:04 -0800 Subject: [PATCH] (test) assert correct exceptions --- litellm/tests/test_key_generate_dynamodb.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py index 559cc9ff52..5363bcd09a 100644 --- a/litellm/tests/test_key_generate_dynamodb.py +++ b/litellm/tests/test_key_generate_dynamodb.py @@ -97,7 +97,8 @@ def test_call_with_invalid_key(): asyncio.run(test()) except Exception as e: print("Got Exception", e) - print(str(e)) + print(e.detail) + assert "Authentication Error" in e.detail pass