Skip to content

Commit 97ffaf0

Browse files
committed
test: Add failing generator test for inspect.getsourcelines
1 parent 481d5b5 commit 97ffaf0

2 files changed

Lines changed: 13 additions & 0 deletions

File tree

Lib/test/test_inspect/inspect_fodder2.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -369,3 +369,11 @@ class dc364:
369369
# line 369
370370
dc370 = dataclasses.make_dataclass('dc370', (('x', int), ('y', int)))
371371
dc371 = dataclasses.make_dataclass('dc370', (('x', int), ('y', int)), module=__name__)
372+
373+
# line 373
374+
from inspect import currentframe
375+
def generator_frame():
376+
loops = (
377+
currentframe() for _ in [0]
378+
)
379+
return list(loops)[0]

Lib/test/test_inspect/test_inspect.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1195,6 +1195,11 @@ def test_class_definition_same_name_diff_methods(self):
11951195
self.assertSourceEqual(mod2.cls296, 296, 298)
11961196
self.assertSourceEqual(mod2.cls310, 310, 312)
11971197

1198+
def test_generator_frame(self):
1199+
frame = mod2.generator_frame()
1200+
assert frame is not None
1201+
self.assertSourceEqual(frame, 375, 379)
1202+
11981203
class TestNoEOL(GetSourceBase):
11991204
def setUp(self):
12001205
self.tempdir = TESTFN + '_dir'

0 commit comments

Comments
 (0)