Comprehension parent

This commit is contained in:
Nicholas Gates
2018-09-09 22:49:06 +01:00
parent 84b07a8809
commit a68e35c895

View File

@@ -289,6 +289,20 @@ def test_parent_on_completion(Script):
assert parent.type == 'class' assert parent.type == 'class'
def test_parent_on_comprehension():
ns = jedi.names('''\
def spam():
return [i for i in range(5)]
''', all_scopes=True)
assert [name.name for name in ns] == ['spam', 'i']
assert ns[0].parent().name == ''
assert ns[0].parent().type == 'module'
assert ns[1].parent().name == 'spam'
assert ns[1].parent().type == 'function'
def test_type(Script): def test_type(Script):
for c in Script('a = [str()]; a[0].').completions(): for c in Script('a = [str()]; a[0].').completions():
if c.name == '__class__': if c.name == '__class__':