Update test_client.py

CrocoFactory · Sep 28, 2024 · 499b28f · 499b28f
1 parent 162050d
commit 499b28f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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