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