Skip to content

Commit 1fe57e0

Browse files
committed
removed unneeded functions
1 parent c312262 commit 1fe57e0

File tree

2 files changed

+2
-16
lines changed

2 files changed

+2
-16
lines changed

tests/asyncio/test_retry_async.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -138,15 +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-
async def target_fn():
142-
async def inner_gen():
143-
raise RuntimeError("initiate retry")
144-
yield None
145-
146-
return inner_gen()
147-
148141
with pytest.raises(ValueError, match="Sleep generator"):
149-
gen = retry_target_stream(target_fn, lambda x: True, [], None)
142+
gen = retry_target_stream(None, lambda x: True, [], None)
150143
await gen.__anext__()
151144

152145

tests/unit/test_retry.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -166,17 +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-
def target_fn():
170-
def inner_gen():
171-
raise RuntimeError("initiate retry")
172-
yield None
173-
174-
return inner_gen()
175-
176169
with pytest.raises(
177170
ValueError, match="Sleep generator stopped yielding sleep values"
178171
):
179-
gen = retry_target_stream(target_fn, lambda x: True, [], None)
172+
gen = retry_target_stream(None, lambda x: True, [], None)
180173
next(gen)
181174

182175

0 commit comments

Comments
 (0)