VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2026-01-17 00:25:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Retrieve all kinds of assignment targets from with test

Browse Source
This commit is contained in:
Batuhan Taskaya
2020-09-24 10:42:56 +03:00
parent 6ae0efa415
commit 606c528803
2 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_parser_tree.py
View File

@@ -229,3 +229,12 @@ def test_iter_funcdefs():
module = parse(code, version='3.8')
func_names = [f.name.value for f in module.iter_funcdefs()]
assert func_names == ['normal', 'asyn', 'dec_normal', 'dec_async']
def test_with_stmt_get_test_node_from_name():
code = "with A as X.Y, B as (Z), C as Q[0], D as Q['foo']: pass"
with_stmt = parse(code, version='3').children[0]
tests = [
with_stmt.get_test_node_from_name(name).value
for name in with_stmt.get_defined_names(include_setitem=True)
]
assert tests == ["A", "B", "C", "D"]