diff options
| author | Daniël van Noord <13665637+DanielNoord@users.noreply.github.com> | 2023-04-03 20:00:43 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-03 20:00:43 +0200 |
| commit | ec1540d7905ec8f5e9383d928a6bfd89860b94a3 (patch) | |
| tree | 3bf5c3aa61f6c502cfb24f2eacfdb409f69ee757 /tests | |
| parent | 8a588e386b6f31c1223c119f40ca40346f90dd8e (diff) | |
| download | astroid-git-ec1540d7905ec8f5e9383d928a6bfd89860b94a3.tar.gz | |
Fix constructors of ``AssignName`` (#2077)
Co-authored-by: Pierre Sassoulas <pierre.sassoulas@gmail.com>
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_nodes.py | 7 | ||||
| -rw-r--r-- | tests/test_transforms.py | 18 |
2 files changed, 22 insertions, 3 deletions
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 |