From c353161456145161f57d8eb09aa8135f3f2641b1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 3 Feb 2024 20:59:07 -0800 Subject: [PATCH] (fix) test_parallel limiter fix --- litellm/tests/test_parallel_request_limiter.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_parallel_request_limiter.py b/litellm/tests/test_parallel_request_limiter.py index 27d81356ff4..34dc0e3b57b 100644 --- a/litellm/tests/test_parallel_request_limiter.py +++ b/litellm/tests/test_parallel_request_limiter.py @@ -371,14 +371,13 @@ async def test_normal_router_tpm_limit(): print(f"response: {response}") try: - await parallel_request_handler.async_pre_call_hook( - user_api_key_dict=user_api_key_dict, - cache=local_cache, - data={}, - call_type="", + assert ( + parallel_request_handler.user_api_key_cache.get_cache( + key=request_count_api_key + )["current_tpm"] + > 0 ) - pytest.fail(f"Expected call to fail") except Exception as e: assert e.status_code == 429