From ec1540d7905ec8f5e9383d928a6bfd89860b94a3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20van=20Noord?= <13665637+DanielNoord@users.noreply.github.com> Date: Mon, 3 Apr 2023 20:00:43 +0200 Subject: Fix constructors of ``AssignName`` (#2077) Co-authored-by: Pierre Sassoulas --- tests/test_nodes.py | 7 ++++++- tests/test_transforms.py | 18 ++++++++++++++++-- 2 files changed, 22 insertions(+), 3 deletions(-) (limited to 'tests') diff --git a/tests/test_nodes.py b/tests/test_nodes.py index b51bd926..9d5f3ba0 100644 --- a/tests/test_nodes.py +++ b/tests/test_nodes.py @@ -1018,7 +1018,12 @@ class AliasesTest(unittest.TestCase): def test_assname(node: AssignName) -> AssignName | None: if node.name == "foo": return nodes.AssignName( - "bar", node.lineno, node.col_offset, node.parent + "bar", + node.lineno, + node.col_offset, + node.parent, + end_lineno=node.end_lineno, + end_col_offset=node.end_col_offset, ) return None diff --git a/tests/test_transforms.py b/tests/test_transforms.py index 8494694d..8eb4cdea 100644 --- a/tests/test_transforms.py +++ b/tests/test_transforms.py @@ -88,7 +88,14 @@ class TestTransforms(unittest.TestCase): def test_transform_patches_locals(self) -> None: def transform_function(node: FunctionDef) -> None: assign = nodes.Assign() - name = nodes.AssignName(name="value") + name = nodes.AssignName( + name="value", + lineno=0, + col_offset=0, + parent=assign, + end_lineno=None, + end_col_offset=None, + ) assign.targets = [name] assign.value = nodes.const_factory(42) node.body.append(assign) @@ -183,7 +190,14 @@ class TestTransforms(unittest.TestCase): def test_transforms_are_called_for_builtin_modules(self) -> None: # Test that transforms are called for builtin modules. def transform_function(node: FunctionDef) -> FunctionDef: - name = nodes.AssignName(name="value") + name = nodes.AssignName( + name="value", + lineno=0, + col_offset=0, + parent=node.args, + end_lineno=None, + end_col_offset=None, + ) node.args.args = [name] return node -- cgit v1.2.1