improved test decorators

tests/asyncio/test_retry_async.py

@@ -766,10 +766,9 @@ async def __anext__(self):
             await retryable.__anext__()
 
     @pytest.mark.parametrize("yield_method", ["__anext__", "asend"])
-    @mock.patch("random.uniform", autospec=True, side_effect=lambda m, n: n)
     @mock.patch("asyncio.sleep", autospec=True)
     @pytest.mark.asyncio
-    async def test_yield_stream_after_deadline(self, sleep, uniform, yield_method):
+    async def test_yield_stream_after_deadline(self, sleep, yield_method):
         """
         By default, if the deadline is hit between yields, the generator will continue.
 
@@ -828,8 +827,9 @@ async def test_yield_stream_after_deadline(self, sleep, uniform, yield_method):
                 await check_yield()
             await no_check_yield()
 
+    @mock.patch("asyncio.sleep", autospec=True)
     @pytest.mark.asyncio
-    async def test_generator_error_list(self):
+    async def test_generator_error_list(self, sleep):
         """
         generator should keep history of errors seen
         """

tests/unit/test_retry.py

@@ -773,9 +773,8 @@ def test___call___with_is_stream(self, sleep):
         assert unpacked == [0, 1, 2, 3, 4, 5, 0, 1, 2, 3]
 
     @pytest.mark.parametrize("yield_method", ["__next__", "send"])
-    @mock.patch("random.uniform", autospec=True, side_effect=lambda m, n: n)
     @mock.patch("asyncio.sleep", autospec=True)
-    def test_yield_stream_after_deadline(self, sleep, uniform, yield_method):
+    def test_yield_stream_after_deadline(self, sleep, yield_method):
         """
         By default, if the deadline is hit between yields, the generator will continue.
 
@@ -833,7 +832,8 @@ def test_yield_stream_after_deadline(self, sleep, uniform, yield_method):
                 check_yield()
             no_check_yield()
 
-    def test_generator_error_list(self):
+    @mock.patch("asyncio.sleep", autospec=True)
+    def test_generator_error_list(self, sleep):
         """
         generator should keep history of errors seen
         """