diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py index d6b9ee1a95a..1e9012e7e5e 100644 --- a/Lib/test/test_builtin.py +++ b/Lib/test/test_builtin.py @@ -390,7 +390,14 @@ class BuiltinTest(unittest.TestCase): '''async for i in arange(1): a = 1''', '''async with asyncio.Lock() as l: - a = 1''' + a = 1''', + '''a = [x async for x in arange(2)][1]''', + '''a = 1 in {x async for x in arange(2)}''', + '''a = {x:1 async for x in arange(1)}[0]''', + '''a = [x async for x in arange(2) async for x in arange(2)][1]''', + '''a = [x async for x in (x async for x in arange(5))][1]''', + '''a, = [1 for x in {x async for x in arange(1)}]''', + '''a = [await asyncio.sleep(0, x) async for x in arange(2)][1]''' ] policy = maybe_get_event_loop_policy() try: diff --git a/Misc/NEWS.d/next/Core and Builtins/2020-03-12-22-13-50.bpo-39562.E2u273.rst b/Misc/NEWS.d/next/Core and Builtins/2020-03-12-22-13-50.bpo-39562.E2u273.rst new file mode 100644 index 00000000000..dbe83c6aa19 --- /dev/null +++ b/Misc/NEWS.d/next/Core and Builtins/2020-03-12-22-13-50.bpo-39562.E2u273.rst @@ -0,0 +1,2 @@ +Allow executing asynchronous comprehensions on the top level when the +``PyCF_ALLOW_TOP_LEVEL_AWAIT`` flag is given. Patch by Batuhan Taskaya. diff --git a/Python/compile.c b/Python/compile.c index a8ab873f319..fa4313a32ca 100644 --- a/Python/compile.c +++ b/Python/compile.c @@ -4520,11 +4520,14 @@ compiler_comprehension(struct compiler *c, expr_ty e, int type, PyCodeObject *co = NULL; comprehension_ty outermost; PyObject *qualname = NULL; - int is_async_function = c->u->u_ste->ste_coroutine; int is_async_generator = 0; - outermost = (comprehension_ty) asdl_seq_GET(generators, 0); + if (IS_TOP_LEVEL_AWAIT(c)) { + c->u->u_ste->ste_coroutine = 1; + } + int is_async_function = c->u->u_ste->ste_coroutine; + outermost = (comprehension_ty) asdl_seq_GET(generators, 0); if (!compiler_enter_scope(c, name, COMPILER_SCOPE_COMPREHENSION, (void *)e, e->lineno)) {