From c2c1f1f906cdeb40576880d4b6a4f8fcbc016eb8 Mon Sep 17 00:00:00 2001 From: Pablo Galindo Date: Mon, 6 Jul 2020 23:30:14 +0100 Subject: [PATCH] bpo-41218: Improve the test cases for test_compile_top_level_await_no_coro (GH-21363) --- Lib/test/test_builtin.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py index 3dcdf8d9030..3bc1a3e246f 100644 --- a/Lib/test/test_builtin.py +++ b/Lib/test/test_builtin.py @@ -372,12 +372,14 @@ class BuiltinTest(unittest.TestCase): 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. - """ + """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]''' + '''[x for x in l]''', + '''{x for x in l}''', + '''(x for x in l)''', + '''{x:x for x in l}''', ] for mode, code_sample in product(modes, code_samples): source = dedent(code_sample)