Skip to content

Commit 343157b

Browse files
committed
removed unneeded test variable
1 parent 06824b9 commit 343157b

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

tests/asyncio/test_retry_async.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,9 +138,8 @@ async def test_retry_target_bad_sleep_generator():
138138
async def test_retry_streaming_target_bad_sleep_generator():
139139
from google.api_core.retry_streaming_async import retry_target_stream
140140

141-
gen = retry_target_stream(None, lambda x: True, [], None)
142141
with pytest.raises(ValueError, match="Sleep generator"):
143-
await gen.__anext__()
142+
await retry_target_stream(None, None, [], None).__anext__()
144143

145144

146145
class TestAsyncRetry:

tests/unit/test_retry.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,10 @@ def test_retry_target_bad_sleep_generator():
166166
def test_retry_streaming_target_bad_sleep_generator():
167167
from google.api_core.retry_streaming import retry_target_stream
168168

169-
gen = retry_target_stream(None, lambda x: True, [], None)
170169
with pytest.raises(
171170
ValueError, match="Sleep generator stopped yielding sleep values"
172171
):
173-
next(gen)
172+
next(retry_target_stream(None, None, [], None))
174173

175174

176175
class TestRetry(object):

0 commit comments

Comments
 (0)