summaryrefslogtreecommitdiff
path: root/tests/unittest_nodes.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unittest_nodes.py')
-rw-r--r--tests/unittest_nodes.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/unittest_nodes.py b/tests/unittest_nodes.py
index d138ee17..0ba6b3e5 100644
--- a/tests/unittest_nodes.py
+++ b/tests/unittest_nodes.py
@@ -1181,6 +1181,19 @@ def test_type_comments_posonly_arguments():
assert actual_arg.as_string() == expected_arg
+@pytest.mark.skipif(not HAS_TYPED_AST, reason="requires typed_ast")
+def test_correct_function_type_comment_parent():
+ data = """
+ def f(a):
+ # type: (A) -> A
+ pass
+ """
+ astroid = builder.parse(data)
+ f = astroid.body[0]
+ assert f.type_comment_args[0].parent is f
+ assert f.type_comment_returns.parent is f
+
+
def test_is_generator_for_yield_assignments():
node = astroid.extract_node(
"""