pythongh-94814: Improve coverage of _PyCode_CreateLineArray

The case where there are more than (1 << 15) lines was not covered
mdboom · Jul 14, 2022 · 1c6af67 · 1c6af67
1 parent 8d7089f
commit 1c6af67
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/Lib/test/test_sys_settrace.py b/Lib/test/test_sys_settrace.py
@@ -1571,6 +1571,31 @@ def func():
 
         self.run_and_compare(func, EXPECTED_EVENTS)
 
+    def test_very_large_function(self):
+        # There is a separate code path when the number of lines > (1 << 15).
+        d = {}
+        exec("""def f():              # line 0
+            x = 0                     # line 1
+            y = 1                     # line 2
+            '''                       # line 3
+            %s                        # lines 4 through (1 << 16)
+            '''                       #
+            x += 1                    #
+            return""" % ('\n' * (1 << 16),), d)
+        f = d['f']
+
+        EXPECTED_EVENTS = [
+            (0, 'call'),
+            (1, 'line'),
+            (2, 'line'),
+            (3, 'line'),
+            (65542, 'line'),
+            (65543, 'line'),
+            (65543, 'return'),
+        ]
+
+        self.run_and_compare(f, EXPECTED_EVENTS)
+
 
 EVENT_NAMES = [
     'call',