removed unneeded test variable

googleapis · Aug 15, 2023 · 343157b · 343157b
1 parent 06824b9
commit 343157b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/asyncio/test_retry_async.py b/tests/asyncio/test_retry_async.py
@@ -138,9 +138,8 @@ async def test_retry_target_bad_sleep_generator():
 async def test_retry_streaming_target_bad_sleep_generator():
     from google.api_core.retry_streaming_async import retry_target_stream
 
-    gen = retry_target_stream(None, lambda x: True, [], None)
     with pytest.raises(ValueError, match="Sleep generator"):
-        await gen.__anext__()
+        await retry_target_stream(None, None, [], None).__anext__()
 
 
 class TestAsyncRetry:

diff --git a/tests/unit/test_retry.py b/tests/unit/test_retry.py
@@ -166,11 +166,10 @@ def test_retry_target_bad_sleep_generator():
 def test_retry_streaming_target_bad_sleep_generator():
     from google.api_core.retry_streaming import retry_target_stream
 
-    gen = retry_target_stream(None, lambda x: True, [], None)
     with pytest.raises(
         ValueError, match="Sleep generator stopped yielding sleep values"
     ):
-        next(gen)
+        next(retry_target_stream(None, None, [], None))
 
 
 class TestRetry(object):