[3.8] bpo-40726: handle uninitalized end_lineno on ast.increment_line… · python/cpython@b24c9d2 · GitHub
Skip to content

Commit b24c9d2

Browse files
authored
[3.8] bpo-40726: handle uninitalized end_lineno on ast.increment_lineno (GH-21745)
…no (GH-20312). (cherry picked from commit 8f4380d) Co-authored-by: Batuhan Taskaya <batuhanosmantaskaya@gmail.com> Automerge-Triggered-By: @pablogsal
1 parent f0e030c commit b24c9d2

3 files changed

Lines changed: 30 additions & 5 deletions

File tree

Lib/ast.py

Lines changed: 13 additions & 5 deletions

Lib/test/test_ast.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -718,6 +718,15 @@ def test_copy_location(self):
718718
'lineno=1, col_offset=4, end_lineno=1, end_col_offset=5), lineno=1, '
719719
'col_offset=0, end_lineno=1, end_col_offset=5))'
720720
)
721+
src = ast.Call(col_offset=1, lineno=1, end_lineno=1, end_col_offset=1)
722+
new = ast.copy_location(src, ast.Call(
723+
col_offset=None, lineno=None,
724+
end_lineno=None, end_col_offset=None
725+
))
726+
self.assertIsNone(new.end_lineno)
727+
self.assertIsNone(new.end_col_offset)
728+
self.assertEqual(new.lineno, 1)
729+
self.assertEqual(new.col_offset, 1)
721730

722731
def test_fix_missing_locations(self):
723732
src = ast.parse('write("spam")')
@@ -757,6 +766,12 @@ def test_increment_lineno(self):
757766
'lineno=4, col_offset=4, end_lineno=4, end_col_offset=5), lineno=4, '
758767
'col_offset=0, end_lineno=4, end_col_offset=5))'
759768
)
769+
src = ast.Call(
770+
func=ast.Name("test", ast.Load()), args=[], keywords=[],
771+
lineno=1, end_lineno=None
772+
)
773+
self.assertEqual(ast.increment_lineno(src).lineno, 2)
774+
self.assertIsNone(ast.increment_lineno(src).end_lineno)
760775

761776
def test_iter_fields(self):
762777
node = ast.parse('foo()', mode='eval')
Lines changed: 2 additions & 0 deletions

0 commit comments

Comments
 (0)