From: Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com>
Date: Sat, 14 May 2022 06:19:00 +0000 (-0700)
Subject: gh-92311: Add tests for frame_setlineno jumping over listcomps (GH-92741)
X-Git-Tag: v3.11.0b2~98
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=964067635602d1be5782b68a44123d59f3b92520;p=thirdparty%2FPython%2Fcpython.git

gh-92311: Add tests for frame_setlineno jumping over listcomps (GH-92741)

(cherry picked from commit 8cf2906828b4ea281ea5381bf59b9052bae99f53)

Co-authored-by: Dennis Sweeney <36520290+sweeneyde@users.noreply.github.com>
---

diff --git a/Lib/test/test_sys_settrace.py b/Lib/test/test_sys_settrace.py
index 1a67fa673dd6..9cc6bcfcab5e 100644
--- a/Lib/test/test_sys_settrace.py
+++ b/Lib/test/test_sys_settrace.py
@@ -2484,6 +2484,54 @@ output.append(4)
         next(gen())
         output.append(5)
 
+    @jump_test(2, 3, [1, 3])
+    def test_jump_forward_over_listcomp(output):
+        output.append(1)
+        x = [i for i in range(10)]
+        output.append(3)
+
+    # checking for segfaults.
+    # See https://github.com/python/cpython/issues/92311
+    @jump_test(3, 1, [])
+    def test_jump_backward_over_listcomp(output):
+        a = 1
+        x = [i for i in range(10)]
+        c = 3
+
+    @jump_test(8, 2, [2, 7, 2])
+    def test_jump_backward_over_listcomp_v2(output):
+        flag = False
+        output.append(2)
+        if flag:
+            return
+        x = [i for i in range(5)]
+        flag = 6
+        output.append(7)
+        output.append(8)
+
+    @async_jump_test(2, 3, [1, 3])
+    async def test_jump_forward_over_async_listcomp(output):
+        output.append(1)
+        x = [i async for i in asynciter(range(10))]
+        output.append(3)
+
+    @async_jump_test(3, 1, [])
+    async def test_jump_backward_over_async_listcomp(output):
+        a = 1
+        x = [i async for i in asynciter(range(10))]
+        c = 3
+
+    @async_jump_test(8, 2, [2, 7, 2])
+    async def test_jump_backward_over_async_listcomp_v2(output):
+        flag = False
+        output.append(2)
+        if flag:
+            return
+        x = [i async for i in asynciter(range(5))]
+        flag = 6
+        output.append(7)
+        output.append(8)
+
 
 class TestExtendedArgs(unittest.TestCase):