From bd46174a5a09a54e5ae1077909f923f56a7cf710 Mon Sep 17 00:00:00 2001 From: Matthias Bussonnier Date: Mon, 6 Jul 2020 14:26:52 -0700 Subject: bpo-41218: Only mark async code with CO_COROUTINE. (#21357) 3.8.3 had a regression where compiling with ast.PyCF_ALLOW_TOP_LEVEL_AWAIT woudl agressively mark things are coroutine even if there were not. --- Lib/test/test_builtin.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'Lib/test/test_builtin.py') diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py index 40df7b606ae..3dcdf8d9030 100644 --- a/Lib/test/test_builtin.py +++ b/Lib/test/test_builtin.py @@ -371,6 +371,25 @@ class BuiltinTest(unittest.TestCase): rv = ns['f']() self.assertEqual(rv, tuple(expected)) + def test_compile_top_level_await_no_coro(self): + """Make sure top level non-await codes get the correct coroutine flags. + """ + modes = ('single', 'exec') + code_samples = [ + '''def f():pass\n''', + '''[x for x in l]''' + ] + for mode, code_sample in product(modes, code_samples): + source = dedent(code_sample) + co = compile(source, + '?', + mode, + flags=ast.PyCF_ALLOW_TOP_LEVEL_AWAIT) + + self.assertNotEqual(co.co_flags & CO_COROUTINE, CO_COROUTINE, + msg=f"source={source} mode={mode}") + + def test_compile_top_level_await(self): """Test whether code some top level await can be compiled. -- cgit v1.2.3