GH-96092: restore frame height of traceback.walk_stack to that of python 3.10

graingert · graingert · commit 2376fe76c774 · 2022-11-02T16:49:27.000Z
diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
@@ -2243,6 +2243,7 @@ def deeper():
             return list(traceback.walk_stack(None))
         s1 = list(traceback.walk_stack(None))
         s2 = deeper()
+        self.assertIs(s2[0][0].f_code, deeper.__code__)
         self.assertEqual(len(s2) - len(s1), 1)
         self.assertEqual(s2[1:], s1)
 
diff --git a/Lib/traceback.py b/Lib/traceback.py
@@ -330,7 +330,7 @@ def walk_stack(f):
     current stack is used. Usually used with StackSummary.extract.
     """
     if f is None:
-        f = sys._getframe().f_back.f_back.f_back.f_back
+        f = sys._getframe().f_back.f_back
     while f is not None:
         yield f, f.f_lineno
         f = f.f_back
