diff --git a/tests/test_client.py b/tests/test_client.py index 1342b3b..1ab1e16 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -81,8 +81,8 @@ def test_rate_limit(self, base_url, sync_maker, base_maker): assert rate_limit._tokens == 1 model.get(2) - model.get(3) - model.get(4) + for i in range(4): + model.get(i) assert time.time() - now >= 2 @@ -99,7 +99,7 @@ async def test_async_rate_limit(self, base_url, async_maker, base_maker): assert rate_limit._tokens == 1 await model.get(2) # type: ignore - await model.get(3) # type: ignore - await model.get(4) # type: ignore + for i in range(4): + await model.get(i) # type: ignore assert time.time() - now >= 2