From c33df193bfbc09dc5c3b62db6507ad1a79e92ea6 Mon Sep 17 00:00:00 2001 From: Damien George Date: Wed, 13 Apr 2016 15:27:06 +0100 Subject: tests: Add 6 tests for async await/for/with. --- tests/basics/async_for.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 tests/basics/async_for.py (limited to 'tests/basics/async_for.py') diff --git a/tests/basics/async_for.py b/tests/basics/async_for.py new file mode 100644 index 0000000000..6b4e136d59 --- /dev/null +++ b/tests/basics/async_for.py @@ -0,0 +1,29 @@ +# test basic async for execution +# example taken from PEP0492 + +class AsyncIteratorWrapper: + def __init__(self, obj): + print('init') + self._it = iter(obj) + + async def __aiter__(self): + print('aiter') + return self + + async def __anext__(self): + print('anext') + try: + value = next(self._it) + except StopIteration: + raise StopAsyncIteration + return value + +async def coro(): + async for letter in AsyncIteratorWrapper('abc'): + print(letter) + +o = coro() +try: + o.send(None) +except StopIteration: + print('finished') -- cgit v1.2.3