Handle not all values pulled case

aio_explicit_task_cancels
Tyler Goodlet 2022-04-12 15:24:16 -04:00
parent 333fad8819
commit fa354ffe2b
1 changed files with 3 additions and 1 deletions

View File

@ -280,7 +280,9 @@ async def stream_from_aio(
# we get double the pulled values in the
# ``.subscribe()`` fan out case.
doubled = list(itertools.chain(*zip(expect, expect)))
assert pulled == doubled
if pulled != doubled[:len(pulled)]:
print(f'uhhh pulled is {pulled}')
assert pulled == doubled[:len(pulled)]
else:
assert pulled == expect