Skip to content

Commit 28db978

Browse files
authored
gh-101229: Add tests for aliases of imported names (#101230)
1 parent 7a37523 commit 28db978

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

Lib/test/test_ast.py

+18
Original file line numberDiff line numberDiff line change
@@ -408,6 +408,24 @@ def test_alias(self):
408408
self.assertEqual(alias.col_offset, 16)
409409
self.assertEqual(alias.end_col_offset, 17)
410410

411+
im = ast.parse("from bar import y as z").body[0]
412+
alias = im.names[0]
413+
self.assertEqual(alias.name, "y")
414+
self.assertEqual(alias.asname, "z")
415+
self.assertEqual(alias.lineno, 1)
416+
self.assertEqual(alias.end_lineno, 1)
417+
self.assertEqual(alias.col_offset, 16)
418+
self.assertEqual(alias.end_col_offset, 22)
419+
420+
im = ast.parse("import bar as foo").body[0]
421+
alias = im.names[0]
422+
self.assertEqual(alias.name, "bar")
423+
self.assertEqual(alias.asname, "foo")
424+
self.assertEqual(alias.lineno, 1)
425+
self.assertEqual(alias.end_lineno, 1)
426+
self.assertEqual(alias.col_offset, 7)
427+
self.assertEqual(alias.end_col_offset, 17)
428+
411429
def test_base_classes(self):
412430
self.assertTrue(issubclass(ast.For, ast.stmt))
413431
self.assertTrue(issubclass(ast.Name, ast.expr))

0 commit comments

Comments
 (0)