VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2025-12-07 21:34:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make is_definition work on setitem modifications, see #66

Browse Source
This commit is contained in:
Dave Halter
2019-08-25 23:55:15 +02:00
parent 3bb46563d4
commit e8653a49ff
2 changed files with 48 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
test/test_parser_tree.py
View File

@@ -180,3 +180,42 @@ def top_function_three():
r = get_raise_stmts(code, 2) # Lists inside try-catch
assert len(list(r)) == 2
@pytest.mark.parametrize(
'code, name_index, is_definition', [
('x = 3', 0, True),
('x.y = 3', 0, False),
('x.y = 3', 1, True),
('x.y = u.v = z', 0, False),
('x.y = u.v = z', 1, True),
('x.y = u.v = z', 2, False),
('x.y = u.v, w = z', 3, True),
('x.y = u.v, w = z', 4, True),
('x.y = u.v, w = z', 5, False),
('x, y = z', 0, True),
('x, y = z', 1, True),
('x, y = z', 2, False),
('x, y = z', 2, False),
('x[0], y = z', 2, False),
('x[0] = z', 0, True),
('x[0], y = z', 0, True),
('x: int = z', 0, True),
('x: int = z', 1, False),
('x: int = z', 2, False),
('x: int', 0, True),
('x: int', 1, False),
]
)
def test_is_definition(code, name_index, is_definition):
module = parse(code, version='3.8')
name = module.get_first_leaf()
while True:
if name.type == 'name':
if name_index == 0:
break
name_index -= 1
name = name.get_next_leaf()
assert name.is_definition() == is_definition