ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: opt / bitninja-python-dojo / embedded / lib / python3.9 / test

import inspect
import types
import unittest

from test.support import import_module
from test.support import gc_collect
asyncio = import_module("asyncio")

class AwaitException(Exception):
    pass

@types.coroutine
def awaitable(*, throw=False):
    if throw:
        yield ('throw',)
    else:
        yield ('result',)

def run_until_complete(coro):
    exc = False
    while True:
        try:
            if exc:
                exc = False
                fut = coro.throw(AwaitException)
            else:
                fut = coro.send(None)
        except StopIteration as ex:
            return ex.args[0]

if fut == ('throw',):
            exc = True

def to_list(gen):
    async def iterate():
        res = []
        async for i in gen:
            res.append(i)
        return res

return run_until_complete(iterate())

class AsyncGenSyntaxTest(unittest.TestCase):

def test_async_gen_syntax_01(self):
        code = '''async def foo():
            await abc
            yield from 123
        '''

with self.assertRaisesRegex(SyntaxError, 'yield from.*inside async'):
            exec(code, {}, {})

def test_async_gen_syntax_02(self):
        code = '''async def foo():
            yield from 123
        '''

with self.assertRaisesRegex(SyntaxError, 'yield from.*inside async'):
            exec(code, {}, {})

def test_async_gen_syntax_03(self):
        code = '''async def foo():
            await abc
            yield
            return 123
        '''

with self.assertRaisesRegex(SyntaxError, 'return.*value.*async gen'):
            exec(code, {}, {})

def test_async_gen_syntax_04(self):
        code = '''async def foo():
            yield
            return 123
        '''

with self.assertRaisesRegex(SyntaxError, 'return.*value.*async gen'):
            exec(code, {}, {})

def test_async_gen_syntax_05(self):
        code = '''async def foo():
            if 0:
                yield
            return 12
        '''

with self.assertRaisesRegex(SyntaxError, 'return.*value.*async gen'):
            exec(code, {}, {})

class AsyncGenTest(unittest.TestCase):

def compare_generators(self, sync_gen, async_gen):
        def sync_iterate(g):
            res = []
            while True:
                try:
                    res.append(g.__next__())
                except StopIteration:
                    res.append('STOP')
                    break
                except Exception as ex:
                    res.append(str(type(ex)))
            return res

def async_iterate(g):
            res = []
            while True:
                an = g.__anext__()
                try:
                    while True:
                        try:
                            an.__next__()
                        except StopIteration as ex:
                            if ex.args:
                                res.append(ex.args[0])
                                break
                            else:
                                res.append('EMPTY StopIteration')
                                break
                        except StopAsyncIteration:
                            raise
                        except Exception as ex:
                            res.append(str(type(ex)))
                            break
                except StopAsyncIteration:
                    res.append('STOP')
                    break
            return res

sync_gen_result = sync_iterate(sync_gen)
        async_gen_result = async_iterate(async_gen)
        self.assertEqual(sync_gen_result, async_gen_result)
        return async_gen_result

def test_async_gen_iteration_01(self):
        async def gen():
            await awaitable()
            a = yield 123
            self.assertIs(a, None)
            await awaitable()
            yield 456
            await awaitable()
            yield 789

self.assertEqual(to_list(gen()), [123, 456, 789])

def test_async_gen_iteration_02(self):
        async def gen():
            await awaitable()
            yield 123
            await awaitable()

g = gen()
        ai = g.__aiter__()

an = ai.__anext__()
        self.assertEqual(an.__next__(), ('result',))

try:
            an.__next__()
        except StopIteration as ex:
            self.assertEqual(ex.args[0], 123)
        else:
            self.fail('StopIteration was not raised')

an = ai.__anext__()
        self.assertEqual(an.__next__(), ('result',))

try:
            an.__next__()
        except StopAsyncIteration as ex:
            self.assertFalse(ex.args)
        else:
            self.fail('StopAsyncIteration was not raised')

def test_async_gen_exception_03(self):
        async def gen():
            await awaitable()
            yield 123
            await awaitable(throw=True)
            yield 456

with self.assertRaises(AwaitException):
            to_list(gen())

def test_async_gen_exception_04(self):
        async def gen():
            await awaitable()
            yield 123
            1 / 0

g = gen()
        ai = g.__aiter__()
        an = ai.__anext__()
        self.assertEqual(an.__next__(), ('result',))

try:
            an.__next__()
        except StopIteration as ex:
            self.assertEqual(ex.args[0], 123)
        else:
            self.fail('StopIteration was not raised')

with self.assertRaises(ZeroDivisionError):
            ai.__anext__().__next__()

def test_async_gen_exception_05(self):
        async def gen():
            yield 123
            raise StopAsyncIteration

with self.assertRaisesRegex(RuntimeError,
                                    'async generator.*StopAsyncIteration'):
            to_list(gen())

def test_async_gen_exception_06(self):
        async def gen():
            yield 123
            raise StopIteration

with self.assertRaisesRegex(RuntimeError,
                                    'async generator.*StopIteration'):
            to_list(gen())

def test_async_gen_exception_07(self):
        def sync_gen():
            try:
                yield 1
                1 / 0
            finally:
                yield 2
                yield 3

yield 100

async def async_gen():
            try:
                yield 1
                1 / 0
            finally:
                yield 2
                yield 3

yield 100

self.compare_generators(sync_gen(), async_gen())

def test_async_gen_exception_08(self):
        def sync_gen():
            try:
                yield 1
            finally:
                yield 2
                1 / 0
                yield 3

yield 100

async def async_gen():
            try:
                yield 1
                await awaitable()
            finally:
                await awaitable()
                yield 2
                1 / 0
                yield 3

yield 100

self.compare_generators(sync_gen(), async_gen())

def test_async_gen_exception_09(self):
        def sync_gen():
            try:
                yield 1
                1 / 0
            finally:
                yield 2
                yield 3

yield 100

async def async_gen():
            try:
                await awaitable()
                yield 1
                1 / 0
            finally:
                yield 2
                await awaitable()
                yield 3

yield 100

self.compare_generators(sync_gen(), async_gen())

def test_async_gen_exception_10(self):
        async def gen():
            yield 123
        with self.assertRaisesRegex(TypeError,
                                    "non-None value .* async generator"):
            gen().__anext__().send(100)

def test_async_gen_exception_11(self):
        def sync_gen():
            yield 10
            yield 20

def sync_gen_wrapper():
            yield 1
            sg = sync_gen()
            sg.send(None)
            try:
                sg.throw(GeneratorExit())
            except GeneratorExit:
                yield 2
            yield 3

async def async_gen():
            yield 10
            yield 20

async def async_gen_wrapper():
            yield 1
            asg = async_gen()
            await asg.asend(None)
            try:
                await asg.athrow(GeneratorExit())
            except GeneratorExit:
                yield 2
            yield 3

self.compare_generators(sync_gen_wrapper(), async_gen_wrapper())

def test_async_gen_api_01(self):
        async def gen():
            yield 123

g = gen()

self.assertEqual(g.__name__, 'gen')
        g.__name__ = '123'
        self.assertEqual(g.__name__, '123')

self.assertIn('.gen', g.__qualname__)
        g.__qualname__ = '123'
        self.assertEqual(g.__qualname__, '123')

self.assertIsNone(g.ag_await)
        self.assertIsInstance(g.ag_frame, types.FrameType)
        self.assertFalse(g.ag_running)
        self.assertIsInstance(g.ag_code, types.CodeType)

self.assertTrue(inspect.isawaitable(g.aclose()))

class AsyncGenAsyncioTest(unittest.TestCase):

def setUp(self):
        self.loop = asyncio.new_event_loop()
        asyncio.set_event_loop(None)

def tearDown(self):
        self.loop.close()
        self.loop = None
        asyncio.set_event_loop_policy(None)

async def to_list(self, gen):
        res = []
        async for i in gen:
            res.append(i)
        return res

def test_async_gen_asyncio_01(self):
        async def gen():
            yield 1
            await asyncio.sleep(0.01)
            yield 2
            await asyncio.sleep(0.01)
            return
            yield 3

res = self.loop.run_until_complete(self.to_list(gen()))
        self.assertEqual(res, [1, 2])

def test_async_gen_asyncio_02(self):
        async def gen():
            yield 1
            await asyncio.sleep(0.01)
            yield 2
            1 / 0
            yield 3

with self.assertRaises(ZeroDivisionError):
            self.loop.run_until_complete(self.to_list(gen()))

def test_async_gen_asyncio_03(self):
        loop = self.loop

class Gen:
            async def __aiter__(self):
                yield 1
                await asyncio.sleep(0.01)
                yield 2

res = loop.run_until_complete(self.to_list(Gen()))
        self.assertEqual(res, [1, 2])

def test_async_gen_asyncio_anext_04(self):
        async def foo():
            yield 1
            await asyncio.sleep(0.01)
            try:
                yield 2
                yield 3
            except ZeroDivisionError:
                yield 1000
            await asyncio.sleep(0.01)
            yield 4

async def run1():
            it = foo().__aiter__()

self.assertEqual(await it.__anext__(), 1)
            self.assertEqual(await it.__anext__(), 2)
            self.assertEqual(await it.__anext__(), 3)
            self.assertEqual(await it.__anext__(), 4)
            with self.assertRaises(StopAsyncIteration):
                await it.__anext__()
            with self.assertRaises(StopAsyncIteration):
                await it.__anext__()

async def run2():
            it = foo().__aiter__()

self.assertEqual(await it.__anext__(), 1)
            self.assertEqual(await it.__anext__(), 2)
            try:
                it.__anext__().throw(ZeroDivisionError)
            except StopIteration as ex:
                self.assertEqual(ex.args[0], 1000)
            else:
                self.fail('StopIteration was not raised')
            self.assertEqual(await it.__anext__(), 4)
            with self.assertRaises(StopAsyncIteration):
                await it.__anext__()

self.loop.run_until_complete(run1())
        self.loop.run_until_complete(run2())

def test_async_gen_asyncio_anext_05(self):
        async def foo():
            v = yield 1
            v = yield v
            yield v * 100

async def run():
            it = foo().__aiter__()

try:
                it.__anext__().send(None)
            except StopIteration as ex:
                self.assertEqual(ex.args[0], 1)
            else:
                self.fail('StopIteration was not raised')

try:
                it.__anext__().send(10)
            except StopIteration as ex:
                self.assertEqual(ex.args[0], 10)
            else:
                self.fail('StopIteration was not raised')

try:
                it.__anext__().send(12)
            except StopIteration as ex:
                self.assertEqual(ex.args[0], 1200)
            else:
                self.fail('StopIteration was not raised')

with self.assertRaises(StopAsyncIteration):
                await it.__anext__()

self.loop.run_until_complete(run())

def test_async_gen_asyncio_anext_06(self):
        DONE = 0

# test synchronous generators
        def foo():
            try:
                yield
            except:
                pass
        g = foo()
        g.send(None)
        with self.assertRaises(StopIteration):
            g.send(None)

# now with asynchronous generators

async def gen():
            nonlocal DONE
            try:
                yield
            except:
                pass
            DONE = 1

async def run():
            nonlocal DONE
            g = gen()
            await g.asend(None)
            with self.assertRaises(StopAsyncIteration):
                await g.asend(None)
            DONE += 10

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 11)

def test_async_gen_asyncio_anext_tuple(self):
        async def foo():
            try:
                yield (1,)
            except ZeroDivisionError:
                yield (2,)

async def run():
            it = foo().__aiter__()

self.assertEqual(await it.__anext__(), (1,))
            with self.assertRaises(StopIteration) as cm:
                it.__anext__().throw(ZeroDivisionError)
            self.assertEqual(cm.exception.args[0], (2,))
            with self.assertRaises(StopAsyncIteration):
                await it.__anext__()

self.loop.run_until_complete(run())

def test_async_gen_asyncio_anext_stopiteration(self):
        async def foo():
            try:
                yield StopIteration(1)
            except ZeroDivisionError:
                yield StopIteration(3)

async def run():
            it = foo().__aiter__()

v = await it.__anext__()
            self.assertIsInstance(v, StopIteration)
            self.assertEqual(v.value, 1)
            with self.assertRaises(StopIteration) as cm:
                it.__anext__().throw(ZeroDivisionError)
            v = cm.exception.args[0]
            self.assertIsInstance(v, StopIteration)
            self.assertEqual(v.value, 3)
            with self.assertRaises(StopAsyncIteration):
                await it.__anext__()

self.loop.run_until_complete(run())

def test_async_gen_asyncio_aclose_06(self):
        async def foo():
            try:
                yield 1
                1 / 0
            finally:
                await asyncio.sleep(0.01)
                yield 12

async def run():
            gen = foo()
            it = gen.__aiter__()
            await it.__anext__()
            await gen.aclose()

with self.assertRaisesRegex(
                RuntimeError,
                "async generator ignored GeneratorExit"):
            self.loop.run_until_complete(run())

def test_async_gen_asyncio_aclose_07(self):
        DONE = 0

async def foo():
            nonlocal DONE
            try:
                yield 1
                1 / 0
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE += 1
            DONE += 1000

async def run():
            gen = foo()
            it = gen.__aiter__()
            await it.__anext__()
            await gen.aclose()

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_aclose_08(self):
        DONE = 0

fut = asyncio.Future(loop=self.loop)

async def foo():
            nonlocal DONE
            try:
                yield 1
                await fut
                DONE += 1000
                yield 2
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE += 1
            DONE += 1000

async def run():
            gen = foo()
            it = gen.__aiter__()
            self.assertEqual(await it.__anext__(), 1)
            await gen.aclose()

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

# Silence ResourceWarnings
        fut.cancel()
        self.loop.run_until_complete(asyncio.sleep(0.01))

def test_async_gen_asyncio_gc_aclose_09(self):
        DONE = 0

async def gen():
            nonlocal DONE
            try:
                while True:
                    yield 1
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE = 1

async def run():
            g = gen()
            await g.__anext__()
            await g.__anext__()
            del g
            gc_collect()  # For PyPy or other GCs.

await asyncio.sleep(0.1)

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_aclose_10(self):
        DONE = 0

# test synchronous generators
        def foo():
            try:
                yield
            except:
                pass
        g = foo()
        g.send(None)
        g.close()

# now with asynchronous generators

async def gen():
            nonlocal DONE
            try:
                yield
            except:
                pass
            DONE = 1

async def run():
            nonlocal DONE
            g = gen()
            await g.asend(None)
            await g.aclose()
            DONE += 10

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 11)

def test_async_gen_asyncio_aclose_11(self):
        DONE = 0

# test synchronous generators
        def foo():
            try:
                yield
            except:
                pass
            yield
        g = foo()
        g.send(None)
        with self.assertRaisesRegex(RuntimeError, 'ignored GeneratorExit'):
            g.close()

# now with asynchronous generators

async def gen():
            nonlocal DONE
            try:
                yield
            except:
                pass
            yield
            DONE += 1

async def run():
            nonlocal DONE
            g = gen()
            await g.asend(None)
            with self.assertRaisesRegex(RuntimeError, 'ignored GeneratorExit'):
                await g.aclose()
            DONE += 10

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 10)

def test_async_gen_asyncio_aclose_12(self):
        DONE = 0

async def target():
            await asyncio.sleep(0.01)
            1 / 0

async def foo():
            nonlocal DONE
            task = asyncio.create_task(target())
            try:
                yield 1
            finally:
                try:
                    await task
                except ZeroDivisionError:
                    DONE = 1

async def run():
            gen = foo()
            it = gen.__aiter__()
            await it.__anext__()
            await gen.aclose()

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_asend_01(self):
        DONE = 0

# Sanity check:
        def sgen():
            v = yield 1
            yield v * 2
        sg = sgen()
        v = sg.send(None)
        self.assertEqual(v, 1)
        v = sg.send(100)
        self.assertEqual(v, 200)

async def gen():
            nonlocal DONE
            try:
                await asyncio.sleep(0.01)
                v = yield 1
                await asyncio.sleep(0.01)
                yield v * 2
                await asyncio.sleep(0.01)
                return
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE = 1

async def run():
            g = gen()

v = await g.asend(None)
            self.assertEqual(v, 1)

v = await g.asend(100)
            self.assertEqual(v, 200)

with self.assertRaises(StopAsyncIteration):
                await g.asend(None)

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_asend_02(self):
        DONE = 0

async def sleep_n_crash(delay):
            await asyncio.sleep(delay)
            1 / 0

async def gen():
            nonlocal DONE
            try:
                await asyncio.sleep(0.01)
                v = yield 1
                await sleep_n_crash(0.01)
                DONE += 1000
                yield v * 2
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE = 1

async def run():
            g = gen()

v = await g.asend(None)
            self.assertEqual(v, 1)

await g.asend(100)

with self.assertRaises(ZeroDivisionError):
            self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_asend_03(self):
        DONE = 0

async def sleep_n_crash(delay):
            fut = asyncio.ensure_future(asyncio.sleep(delay),
                                        loop=self.loop)
            self.loop.call_later(delay / 2, lambda: fut.cancel())
            return await fut

async def run():
            g = gen()

v = await g.asend(None)
            self.assertEqual(v, 1)

await g.asend(100)

with self.assertRaises(asyncio.CancelledError):
            self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_athrow_01(self):
        DONE = 0

class FooEr(Exception):
            pass

# Sanity check:
        def sgen():
            try:
                v = yield 1
            except FooEr:
                v = 1000
            yield v * 2
        sg = sgen()
        v = sg.send(None)
        self.assertEqual(v, 1)
        v = sg.throw(FooEr)
        self.assertEqual(v, 2000)
        with self.assertRaises(StopIteration):
            sg.send(None)

async def gen():
            nonlocal DONE
            try:
                await asyncio.sleep(0.01)
                try:
                    v = yield 1
                except FooEr:
                    v = 1000
                    await asyncio.sleep(0.01)
                yield v * 2
                await asyncio.sleep(0.01)
                # return
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE = 1

async def run():
            g = gen()

v = await g.asend(None)
            self.assertEqual(v, 1)

v = await g.athrow(FooEr)
            self.assertEqual(v, 2000)

with self.assertRaises(StopAsyncIteration):
                await g.asend(None)

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_athrow_02(self):
        DONE = 0

class FooEr(Exception):
            pass

async def gen():
            nonlocal DONE
            try:
                await asyncio.sleep(0.01)
                try:
                    v = yield 1
                except FooEr:
                    await sleep_n_crash(0.01)
                yield v * 2
                await asyncio.sleep(0.01)
                # return
            finally:
                await asyncio.sleep(0.01)
                await asyncio.sleep(0.01)
                DONE = 1

async def run():
            g = gen()

v = await g.asend(None)
            self.assertEqual(v, 1)

try:
                await g.athrow(FooEr)
            except asyncio.CancelledError:
                self.assertEqual(DONE, 1)
                raise
            else:
                self.fail('CancelledError was not raised')

with self.assertRaises(asyncio.CancelledError):
            self.loop.run_until_complete(run())
        self.assertEqual(DONE, 1)

def test_async_gen_asyncio_athrow_03(self):
        DONE = 0

# test synchronous generators
        def foo():
            try:
                yield
            except:
                pass
        g = foo()
        g.send(None)
        with self.assertRaises(StopIteration):
            g.throw(ValueError)

# now with asynchronous generators

async def gen():
            nonlocal DONE
            try:
                yield
            except:
                pass
            DONE = 1

async def run():
            nonlocal DONE
            g = gen()
            await g.asend(None)
            with self.assertRaises(StopAsyncIteration):
                await g.athrow(ValueError)
            DONE += 10

self.loop.run_until_complete(run())
        self.assertEqual(DONE, 11)

def test_async_gen_asyncio_athrow_tuple(self):
        async def gen():
            try:
                yield 1
            except ZeroDivisionError:
                yield (2,)

async def run():
            g = gen()
            v = await g.asend(None)
            self.assertEqual(v, 1)
            v = await g.athrow(ZeroDivisionError)
            self.assertEqual(v, (2,))
            with self.assertRaises(StopAsyncIteration):
                await g.asend(None)

self.loop.run_until_complete(run())

def test_async_gen_asyncio_athrow_stopiteration(self):
        async def gen():
            try:
                yield 1
            except ZeroDivisionError:
                yield StopIteration(2)

async def run():
            g = gen()
            v = await g.asend(None)
            self.assertEqual(v, 1)
            v = await g.athrow(ZeroDivisionError)
            self.assertIsInstance(v, StopIteration)
            self.assertEqual(v.value, 2)
            with self.assertRaises(StopAsyncIteration):
                await g.asend(None)

self.loop.run_until_complete(run())

def test_async_gen_asyncio_shutdown_01(self):
        finalized = 0

async def waiter(timeout):
            nonlocal finalized
            try:
                await asyncio.sleep(timeout)
                yield 1
            finally:
                await asyncio.sleep(0)
                finalized += 1

async def wait():
            async for _ in waiter(1):
                pass

t1 = self.loop.create_task(wait())
        t2 = self.loop.create_task(wait())

self.loop.run_until_complete(asyncio.sleep(0.1))

# Silence warnings
        t1.cancel()
        t2.cancel()

with self.assertRaises(asyncio.CancelledError):
            self.loop.run_until_complete(t1)
        with self.assertRaises(asyncio.CancelledError):
            self.loop.run_until_complete(t2)

self.loop.run_until_complete(self.loop.shutdown_asyncgens())

self.assertEqual(finalized, 2)

def test_async_gen_asyncio_shutdown_02(self):
        messages = []

def exception_handler(loop, context):
            messages.append(context)

async def async_iterate():
            yield 1
            yield 2

it = async_iterate()
        async def main():
            loop = asyncio.get_running_loop()
            loop.set_exception_handler(exception_handler)

async for i in it:
                break

asyncio.run(main())

self.assertEqual(messages, [])

def test_async_gen_asyncio_shutdown_exception_01(self):
        messages = []

def exception_handler(loop, context):
            messages.append(context)

async def async_iterate():
            try:
                yield 1
                yield 2
            finally:
                1/0

it = async_iterate()
        async def main():
            loop = asyncio.get_running_loop()
            loop.set_exception_handler(exception_handler)

async for i in it:
                break

asyncio.run(main())

message, = messages
        self.assertEqual(message['asyncgen'], it)
        self.assertIsInstance(message['exception'], ZeroDivisionError)
        self.assertIn('an error occurred during closing of asynchronous generator',
                      message['message'])

def test_async_gen_asyncio_shutdown_exception_02(self):
        messages = []

def exception_handler(loop, context):
            messages.append(context)

async def async_iterate():
            try:
                yield 1
                yield 2
            finally:
                1/0

async def main():
            loop = asyncio.get_running_loop()
            loop.set_exception_handler(exception_handler)

async for i in async_iterate():
                break

asyncio.run(main())

message, = messages
        self.assertIsInstance(message['exception'], ZeroDivisionError)
        self.assertIn('unhandled exception during asyncio.run() shutdown',
                      message['message'])

def test_async_gen_expression_01(self):
        async def arange(n):
            for i in range(n):
                await asyncio.sleep(0.01)
                yield i

def make_arange(n):
            # This syntax is legal starting with Python 3.7
            return (i * 2 async for i in arange(n))

async def run():
            return [i async for i in make_arange(10)]

res = self.loop.run_until_complete(run())
        self.assertEqual(res, [i * 2 for i in range(10)])

def test_async_gen_expression_02(self):
        async def wrap(n):
            await asyncio.sleep(0.01)
            return n

def make_arange(n):
            # This syntax is legal starting with Python 3.7
            return (i * 2 for i in range(n) if await wrap(i))

async def run():
            return [i async for i in make_arange(10)]

res = self.loop.run_until_complete(run())
        self.assertEqual(res, [i * 2 for i in range(1, 10)])

def test_asyncgen_nonstarted_hooks_are_cancellable(self):
        # See https://bugs.python.org/issue38013
        messages = []

def exception_handler(loop, context):
            messages.append(context)

async def async_iterate():
            yield 1
            yield 2

async def main():
            loop = asyncio.get_running_loop()
            loop.set_exception_handler(exception_handler)

async for i in async_iterate():
                break

asyncio.run(main())

self.assertEqual([], messages)

def test_async_gen_await_same_anext_coro_twice(self):
        async def async_iterate():
            yield 1
            yield 2

async def run():
            it = async_iterate()
            nxt = it.__anext__()
            await nxt
            with self.assertRaisesRegex(
                    RuntimeError,
                    r"cannot reuse already awaited __anext__\(\)/asend\(\)"
            ):
                await nxt

await it.aclose()  # prevent unfinished iterator warning

self.loop.run_until_complete(run())

def test_async_gen_await_same_aclose_coro_twice(self):
        async def async_iterate():
            yield 1
            yield 2

async def run():
            it = async_iterate()
            nxt = it.aclose()
            await nxt
            with self.assertRaisesRegex(
                    RuntimeError,
                    r"cannot reuse already awaited aclose\(\)/athrow\(\)"
            ):
                await nxt

self.loop.run_until_complete(run())

def test_async_gen_aclose_twice_with_different_coros(self):
        # Regression test for https://bugs.python.org/issue39606
        async def async_iterate():
            yield 1
            yield 2

async def run():
            it = async_iterate()
            await it.aclose()
            await it.aclose()

self.loop.run_until_complete(run())

def test_async_gen_aclose_after_exhaustion(self):
        # Regression test for https://bugs.python.org/issue39606
        async def async_iterate():
            yield 1
            yield 2

async def run():
            it = async_iterate()
            async for _ in it:
                pass
            await it.aclose()

self.loop.run_until_complete(run())

def test_async_gen_aclose_compatible_with_get_stack(self):
        async def async_generator():
            yield object()

async def run():
            ag = async_generator()
            asyncio.create_task(ag.aclose())
            tasks = asyncio.all_tasks()
            for task in tasks:
                # No AttributeError raised
                task.get_stack()

self.loop.run_until_complete(run())

if __name__ == "__main__":
    unittest.main()

[+] ..
[-] test_largefile.py [edit]
[-] test_set.py [edit]
[-] test_sunau.py [edit]
[-] test_webbrowser.py [edit]
[-] test_idle.py [edit]
[-] keycert.passwd.pem [edit]
[-] bisect_cmd.py [edit]
[-] talos-2019-0758.pem [edit]
[-] audiotest.au [edit]
[-] test_selectors.py [edit]
[-] test_pydoc.py [edit]
[-] test_uu.py [edit]
[-] ssltests.py [edit]
[-] test_heapq.py [edit]
[-] test_parser.py [edit]
[-] test_bufio.py [edit]
[-] test_ucn.py [edit]
[-] sample_doctest_no_doctests.py [edit]
[-] Sine-1000Hz-300ms.aif [edit]
[-] test_pkg.py [edit]
[+] support
[-] profilee.py [edit]
[-] test_codecencodings_hk.py [edit]
[-] pydocfodder.py [edit]
[-] test_graphlib.py [edit]
[-] test_raise.py [edit]
[-] test_genericalias.py [edit]
[-] curses_tests.py [edit]
[-] _typed_dict_helper.py [edit]
[-] test_doctest2.txt [edit]
[-] test_zipfile.py [edit]
[-] test_imghdr.py [edit]
[-] test_type_comments.py [edit]
[-] bad_getattr.py [edit]
[-] test_asyncore.py [edit]
[-] test_script_helper.py [edit]
[+] sndhdrdata
[-] test_ossaudiodev.py [edit]
[-] formatfloat_testcases.txt [edit]
[-] test_grammar.py [edit]
[-] test_unicodedata.py [edit]
[-] test_re.py [edit]
[-] test_pkgutil.py [edit]
[-] test_nis.py [edit]
[-] test__xxsubinterpreters.py [edit]
[-] nullbytecert.pem [edit]
[-] pydoc_mod.py [edit]
[-] test_winconsoleio.py [edit]
[-] test_abstract_numbers.py [edit]
[-] test_glob.py [edit]
[-] test_runpy.py [edit]
[+] data
[-] test__opcode.py [edit]
[+] cjkencodings
[-] test_descr.py [edit]
[-] test_codecmaps_hk.py [edit]
[-] test_future3.py [edit]
[-] test_regrtest.py [edit]
[-] test_c_locale_coercion.py [edit]
[-] test_doctest4.txt [edit]
[-] test_dis.py [edit]
[-] test_xml_etree.py [edit]
[-] ann_module5.py [edit]
[-] test_sundry.py [edit]
[-] test_pstats.py [edit]
[-] test_socketserver.py [edit]
[-] test_codeop.py [edit]
[-] test_genericpath.py [edit]
[-] test_complex.py [edit]
[-] test_grp.py [edit]
[-] test_extcall.py [edit]
[-] test_sys_setprofile.py [edit]
[-] test_threading.py [edit]
[-] test_subclassinit.py [edit]
[-] gdb_sample.py [edit]
[-] test_wsgiref.py [edit]
[-] test_fstring.py [edit]
[-] allsans.pem [edit]
[-] inspect_fodder2.py [edit]
[-] dataclass_module_2_str.py [edit]
[-] test_unary.py [edit]
[-] test_doctest.py [edit]
[-] badsyntax_future8.py [edit]
[-] test_decorators.py [edit]
[-] _test_multiprocessing.py [edit]
[-] ieee754.txt [edit]
[-] badsyntax_future4.py [edit]
[-] test_property.py [edit]
[-] test_symbol.py [edit]
[-] test_rlcompleter.py [edit]
[-] test_sort.py [edit]
[-] test_codecencodings_cn.py [edit]
[-] test_traceback.py [edit]
[-] test_xmlrpc.py [edit]
[+] test_peg_generator
[-] test_posixpath.py [edit]
[-] test_future5.py [edit]
[-] test_textwrap.py [edit]
[-] mapping_tests.py [edit]
[-] test_gzip.py [edit]
[-] floating_points.txt [edit]
[-] test_httpservers.py [edit]
[-] test_array.py [edit]
[-] test_structmembers.py [edit]
[-] test_pwd.py [edit]
[-] test_calendar.py [edit]
[-] test_dynamicclassattribute.py [edit]
[+] test_tools
[-] dataclass_module_2.py [edit]
[-] test_codecencodings_kr.py [edit]
[-] mp_fork_bomb.py [edit]
[-] ssl_key.pem [edit]
[-] test_bz2.py [edit]
[-] test_codeccallbacks.py [edit]
[-] test_deque.py [edit]
[-] test_peepholer.py [edit]
[-] test_multiprocessing_forkserver.py [edit]
[-] test_unittest.py [edit]
[-] test_asynchat.py [edit]
[-] test_struct.py [edit]
[-] test_osx_env.py [edit]
[-] test_slice.py [edit]
[-] test_finalization.py [edit]
[-] test_pipes.py [edit]
[-] test_multiprocessing_main_handling.py [edit]
[-] test_unicode_identifiers.py [edit]
[-] test_userlist.py [edit]
[-] datetimetester.py [edit]
[-] test_fork1.py [edit]
[-] test_shelve.py [edit]
[-] test_fcntl.py [edit]
[-] test_errno.py [edit]
[-] test_tempfile.py [edit]
[-] make_ssl_certs.py [edit]
[-] randv2_64.pck [edit]
[-] string_tests.py [edit]
[-] future_test1.py [edit]
[-] test_pathlib.py [edit]
[-] nullcert.pem [edit]
[-] test_wait4.py [edit]
[-] test_cmd.py [edit]
[-] test_memoryview.py [edit]
[+] audiodata
[-] test_modulefinder.py [edit]
[-] test_optparse.py [edit]
[+] libregrtest
[-] test_bigaddrspace.py [edit]
[-] test_module.py [edit]
[-] test_structseq.py [edit]
[-] test_unicode.py [edit]
[-] test_positional_only_arg.py [edit]
[-] test_wave.py [edit]
[-] test_popen.py [edit]
[-] test_multiprocessing_fork.py [edit]
[-] test_context.py [edit]
[+] test_email
[-] test_typechecks.py [edit]
[-] mp_preload.py [edit]
[-] win_console_handler.py [edit]
[-] test_capi.py [edit]
[-] multibytecodec_support.py [edit]
[-] test_lib2to3.py [edit]
[-] test_copyreg.py [edit]
[-] test_operator.py [edit]
[-] test_urllib.py [edit]
[-] test_resource.py [edit]
[-] keycert.pem [edit]
[-] test_ioctl.py [edit]
[-] test_getopt.py [edit]
[+] eintrdata
[-] badsyntax_future3.py [edit]
[-] test_buffer.py [edit]
[-] test_math.py [edit]
[-] test_thread.py [edit]
[-] keycert3.pem [edit]
[-] test_configparser.py [edit]
[-] test_float.py [edit]
[-] test_pprint.py [edit]
[-] test_long.py [edit]
[-] test_netrc.py [edit]
[-] test_concurrent_futures.py [edit]
[-] mime.types [edit]
[-] memory_watchdog.py [edit]
[-] test_compare.py [edit]
[-] test_future.py [edit]
[-] ann_module.py [edit]
[-] signalinterproctester.py [edit]
[-] badsyntax_future9.py [edit]
[-] test_baseexception.py [edit]
[-] test_timeout.py [edit]
[-] lock_tests.py [edit]
[-] test_urllib2net.py [edit]
[-] math_testcases.txt [edit]
[-] test_listcomps.py [edit]
[-] test_filecmp.py [edit]
[-] test_unpack_ex.py [edit]
[-] test_sys.py [edit]
[-] test_bytes.py [edit]
[-] selfsigned_pythontestdotnet.pem [edit]
[-] keycert2.pem [edit]
[-] test_file_eintr.py [edit]
[-] exception_hierarchy.txt [edit]
[-] test_decimal.py [edit]
[-] test_spwd.py [edit]
[-] test_aifc.py [edit]
[-] reperf.py [edit]
[-] test_userstring.py [edit]
[-] test_ordered_dict.py [edit]
[-] ann_module3.py [edit]
[-] test_threading_local.py [edit]
[-] test_pdb.py [edit]
[-] test_smtpnet.py [edit]
[-] recursion.tar [edit]
[-] test_zipimport_support.py [edit]
[+] test_warnings
[-] test_richcmp.py [edit]
[-] test_mailcap.py [edit]
[-] test_linecache.py [edit]
[-] test_weakref.py [edit]
[-] test_iter.py [edit]
[-] test_copy.py [edit]
[-] test_scope.py [edit]
[-] __main__.py [edit]
[-] test_crypt.py [edit]
[-] test_clinic.py [edit]
[-] tokenize_tests-utf8-coding-cookie-and-utf8-bom-sig.txt [edit]
[-] test_gdb.py [edit]
[-] test_audioop.py [edit]
[-] empty.vbs [edit]
[-] test_with.py [edit]
[-] inspect_fodder.py [edit]
[-] test_iterlen.py [edit]
[-] seq_tests.py [edit]
[-] test_colorsys.py [edit]
[-] test_openpty.py [edit]
[-] test_opcodes.py [edit]
[-] test_sysconfig.py [edit]
[-] test_quopri.py [edit]
[-] test_smtpd.py [edit]
[-] idnsans.pem [edit]
[-] test_site.py [edit]
[-] test_strptime.py [edit]
[-] test_code.py [edit]
[-] mock_socket.py [edit]
[-] test_genexps.py [edit]
[-] badsyntax_future10.py [edit]
[-] tokenize_tests-no-coding-cookie-and-utf8-bom-sig-only.txt [edit]
[-] test_epoll.py [edit]
[-] test_itertools.py [edit]
[-] test_cprofile.py [edit]
[-] test_uuid.py [edit]
[-] test_lltrace.py [edit]
[-] test_zipfile64.py [edit]
[-] test_tcl.py [edit]
[-] test_cgitb.py [edit]
[-] test_datetime.py [edit]
[-] test_doctest3.txt [edit]
[-] test_pulldom.py [edit]
[+] ziptestdata
[-] future_test2.py [edit]
[-] test_winreg.py [edit]
[-] test_atexit.py [edit]
[-] test_ttk_textonly.py [edit]
[-] test_bigmem.py [edit]
[-] ffdh3072.pem [edit]
[-] test_charmapcodec.py [edit]
[-] test_utf8_mode.py [edit]
[-] test_zipimport.py [edit]
[-] test_html.py [edit]
[-] test_robotparser.py [edit]
[-] test_types.py [edit]
[-] test_codecencodings_iso2022.py [edit]
[-] test_bisect.py [edit]
[-] test_class.py [edit]
[-] test_dictviews.py [edit]
[-] test_mailbox.py [edit]
[-] test_httplib.py [edit]
[-] test_bool.py [edit]
[-] test_tarfile.py [edit]
[-] test_io.py [edit]
[-] test_enum.py [edit]
[-] test_fileio.py [edit]
[-] badsyntax_3131.py [edit]
[-] test_asdl_parser.py [edit]
[-] keycertecc.pem [edit]
[-] test_code_module.py [edit]
[-] test_descrtut.py [edit]
[-] autotest.py [edit]
[-] test_poplib.py [edit]
[-] test_compileall.py [edit]
[-] test_frame.py [edit]
[-] list_tests.py [edit]
[+] test_zoneinfo
[-] coding20731.py [edit]
[-] test_imp.py [edit]
[+] test_asyncio
[-] test_pyexpat.py [edit]
[-] test_pickle.py [edit]
[-] randv3.pck [edit]
[-] bad_coding2.py [edit]
[-] test__osx_support.py [edit]
[-] test_setcomps.py [edit]
[-] fork_wait.py [edit]
[-] test_posix.py [edit]
[-] test_enumerate.py [edit]
[-] test_eof.py [edit]
[+] __pycache__
[-] tokenize_tests-utf8-coding-cookie-and-no-utf8-bom-sig.txt [edit]
[-] pythoninfo.py [edit]
[-] pickletester.py [edit]
[-] time_hashlib.py [edit]
[-] test_shlex.py [edit]
[-] test_dict_version.py [edit]
[-] test_statistics.py [edit]
[-] test_ntpath.py [edit]
[-] test_mmap.py [edit]
[-] test_difflib.py [edit]
[-] tokenize_tests-latin1-coding-cookie-and-utf8-bom-sig.txt [edit]
[-] test_curses.py [edit]
[-] test_codecencodings_jp.py [edit]
[-] test_tix.py [edit]
[-] test_timeit.py [edit]
[-] test_memoryio.py [edit]
[-] test_ftplib.py [edit]
[-] test_gettext.py [edit]
[-] clinic.test [edit]
[-] test_distutils.py [edit]
[-] test_support.py [edit]
[-] test_augassign.py [edit]
[-] test_imaplib.py [edit]
[-] test_sndhdr.py [edit]
[-] mailcap.txt [edit]
[-] test_cmath.py [edit]
[-] test_urllib_response.py [edit]
[-] ssl_key.passwd.pem [edit]
[-] test_tokenize.py [edit]
[-] test_format.py [edit]
[-] test_dynamic.py [edit]
[-] test_global.py [edit]
[-] test_os.py [edit]
[-] badcert.pem [edit]
[-] test_secrets.py [edit]
[-] test_pyclbr.py [edit]
[-] test_picklebuffer.py [edit]
[-] badkey.pem [edit]
[-] test_dbm.py [edit]
[-] test_nntplib.py [edit]
[-] test_flufl.py [edit]
[-] test_eintr.py [edit]
[-] test_multiprocessing_spawn.py [edit]
[-] test_xmlrpc_net.py [edit]
[-] tf_inherit_check.py [edit]
[-] nokia.pem [edit]
[-] test_audit.py [edit]
[-] final_b.py [edit]
[-] tokenize_tests.txt [edit]
[-] revocation.crl [edit]
[+] subprocessdata
[-] test_codecmaps_tw.py [edit]
[-] test_file.py [edit]
[-] cmath_testcases.txt [edit]
[-] test_signal.py [edit]
[-] test_codecmaps_jp.py [edit]
[-] test_int.py [edit]
[-] test_sqlite.py [edit]
[-] test_http_cookies.py [edit]
[-] sgml_input.html [edit]
[-] badsyntax_future5.py [edit]
[-] test_unicode_file_functions.py [edit]
[-] re_tests.py [edit]
[-] mod_generics_cache.py [edit]
[-] test_repl.py [edit]
[-] test_faulthandler.py [edit]
[-] test_telnetlib.py [edit]
[-] test_exceptions.py [edit]
[-] test_builtin.py [edit]
[-] test_frozen.py [edit]
[-] sortperf.py [edit]
[-] test_named_expressions.py [edit]
[-] test_codecs.py [edit]
[-] test_zipapp.py [edit]
[-] test_dictcomps.py [edit]
[-] test___future__.py [edit]
[+] dtracedata
[-] dataclass_module_1.py [edit]
[-] test_binascii.py [edit]
[-] test___all__.py [edit]
[-] testtar.tar [edit]
[+] test_json
[-] secp384r1.pem [edit]
[-] test_locale.py [edit]
[-] cfgparser.2 [edit]
[-] test_trace.py [edit]
[-] test_binop.py [edit]
[-] test_threadsignals.py [edit]
[-] test_fractions.py [edit]
[-] test_univnewlines.py [edit]
[-] test_argparse.py [edit]
[-] zipdir.zip [edit]
[-] test_marshal.py [edit]
[-] test_strtod.py [edit]
[-] test_hmac.py [edit]
[-] test_random.py [edit]
[-] test_urllib2.py [edit]
[-] test_syntax.py [edit]
[-] test_compile.py [edit]
[-] xmltests.py [edit]
[-] test_tracemalloc.py [edit]
[-] final_a.py [edit]
[-] test_functools.py [edit]
[+] capath
[-] pyclbr_input.py [edit]
[-] test_yield_from.py [edit]
[-] badsyntax_future6.py [edit]
[-] test_xml_dom_minicompat.py [edit]
[-] test_bdb.py [edit]
[-] test_ast.py [edit]
[-] test_ensurepip.py [edit]
[-] test_threadedtempfile.py [edit]
[-] test_keyword.py [edit]
[-] test_strftime.py [edit]
[-] test_logging.py [edit]
[+] decimaltestdata
[-] test_dataclasses.py [edit]
[-] test_lzma.py [edit]
[-] test_startfile.py [edit]
[-] test_venv.py [edit]
[-] test_zlib.py [edit]
[-] test_contains.py [edit]
[-] test_docxmlrpc.py [edit]
[-] test_stringprep.py [edit]
[-] test_defaultdict.py [edit]
[-] audiotests.py [edit]
[-] test_ctypes.py [edit]
[-] imp_dummy.py [edit]
[-] relimport.py [edit]
[-] test_codecencodings_tw.py [edit]
[-] test_genericclass.py [edit]
[-] cfgparser.3 [edit]
[-] dis_module.py [edit]
[-] test_cgi.py [edit]
[-] pycakey.pem [edit]
[-] test_xxtestfuzz.py [edit]
[-] test_winsound.py [edit]
[-] test_unicode_file.py [edit]
[-] test_msilib.py [edit]
[-] __init__.py [edit]
[-] pycacert.pem [edit]
[-] test_dbm_ndbm.py [edit]
[+] tracedmodules
[-] test_cmd_line.py [edit]
[-] good_getattr.py [edit]
[+] imghdrdata
[-] test_kqueue.py [edit]
[-] test_call.py [edit]
[-] bad_getattr3.py [edit]
[-] test_abc.py [edit]
[-] test_xdrlib.py [edit]
[-] test_generator_stop.py [edit]
[-] test_hash.py [edit]
[-] test_urlparse.py [edit]
[-] test_sax.py [edit]
[-] test_dict.py [edit]
[-] test_reprlib.py [edit]
[-] pstats.pck [edit]
[-] test_list.py [edit]
[-] ann_module6.py [edit]
[-] test_utf8source.py [edit]
[-] test_symtable.py [edit]
[-] test_ttk_guionly.py [edit]
[-] test_cmd_line_script.py [edit]
[-] doctest_aliases.py [edit]
[-] test_http_cookiejar.py [edit]
[-] test_turtle.py [edit]
[-] keycert4.pem [edit]
[-] ann_module2.py [edit]
[-] test_py_compile.py [edit]
[-] test_super.py [edit]
[+] test_import
[-] test_binhex.py [edit]
[-] test_coroutines.py [edit]
[-] test_contextlib_async.py [edit]
[+] encoded_modules
[-] test_ipaddress.py [edit]
[-] test_platform.py [edit]
[-] test_longexp.py [edit]
[-] test_codecmaps_cn.py [edit]
[-] test_unpack.py [edit]
[-] test_doctest.txt [edit]
[-] test_socket.py [edit]
[-] test_difflib_expect.html [edit]
[-] test_check_c_globals.py [edit]
[-] test_sys_settrace.py [edit]
[-] test_urllib2_localnet.py [edit]
[+] xmltestdata
[-] test_dtrace.py [edit]
[-] sample_doctest.py [edit]
[-] sample_doctest_no_docstrings.py [edit]
[-] test_csv.py [edit]
[-] testcodec.py [edit]
[-] badsyntax_future7.py [edit]
[-] test_htmlparser.py [edit]
[-] bad_coding.py [edit]
[-] zip_cp437_header.zip [edit]
[-] test_tabnanny.py [edit]
[-] double_const.py [edit]
[-] test_urllibnet.py [edit]
[-] test_asyncgen.py [edit]
[-] test_time.py [edit]
[-] test_pow.py [edit]
[-] test_tk.py [edit]
[-] test_profile.py [edit]
[-] test_smtplib.py [edit]
[-] test_stat.py [edit]
[-] test_generators.py [edit]
[-] test_isinstance.py [edit]
[-] test_index.py [edit]
[-] test_int_literal.py [edit]
[-] test_string.py [edit]
[-] test_keywordonlyarg.py [edit]
[-] test_fileinput.py [edit]
[-] test_weakset.py [edit]
[-] ssl_servers.py [edit]
[-] test_sched.py [edit]
[-] test_queue.py [edit]
[-] test_unparse.py [edit]
[-] test_typing.py [edit]
[-] test_poll.py [edit]
[-] test_mimetypes.py [edit]
[-] test_xml_etree_c.py [edit]
[-] test__locale.py [edit]
[-] cfgparser.1 [edit]
[-] test_select.py [edit]
[-] test_ssl.py [edit]
[-] test_print.py [edit]
[-] test_minidom.py [edit]
[-] test_contextlib.py [edit]
[-] test_exception_variations.py [edit]
[-] bad_getattr2.py [edit]
[-] test_shutil.py [edit]
[-] test_devpoll.py [edit]
[-] test_doctest2.py [edit]
[-] test_source_encoding.py [edit]
[-] audit-tests.py [edit]
[-] test_getpass.py [edit]
[-] test_fnmatch.py [edit]
[-] test_plistlib.py [edit]
[-] test_crashers.py [edit]
[-] test_future4.py [edit]
[-] test_getargs2.py [edit]
[-] test_gc.py [edit]
[-] badsyntax_pep3120.py [edit]
[-] test_userdict.py [edit]
[-] dataclass_module_1_str.py [edit]
[-] test_inspect.py [edit]
[-] test_codecmaps_kr.py [edit]
[-] test_tuple.py [edit]
[-] test_readline.py [edit]
[-] dataclass_textanno.py [edit]
[-] test_dbm_dumb.py [edit]
[+] test_importlib
[-] test_hashlib.py [edit]
[-] test_syslog.py [edit]
[-] test_pty.py [edit]
[-] test_pickletools.py [edit]
[-] regrtest.py [edit]
[-] test_wait3.py [edit]
[-] test_collections.py [edit]
[-] test_peg_parser.py [edit]
[-] test_string_literals.py [edit]
[-] test_multibytecodec.py [edit]
[-] test_range.py [edit]
[-] test_exception_hierarchy.py [edit]
[-] randv2_32.pck [edit]
[-] nosan.pem [edit]
[-] test_metaclass.py [edit]
[-] test_numeric_tower.py [edit]
[-] test_subprocess.py [edit]
[-] test_embed.py [edit]
[-] ssl_cert.pem [edit]
[-] test_dbm_gnu.py [edit]
[-] test_base64.py [edit]
[-] test_funcattrs.py [edit]