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

move set comprehension tests to comletion/comprehensions.py

Browse Source
This commit is contained in:
Tzerjen Wei
2015-08-09 14:11:42 +08:00
parent 4cc6cb3ac4
commit fdcf718317
2 changed files with 7 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/completion/comprehensions.py
View File

@@ -50,6 +50,13 @@ left
##? str()
{a-1:b for a,b in {1:'a', 3:1.0}.items()}[0]
# with a set literal
#? int()
[a for a in {1, 2, 3}][0]
##? int()
{a for a in range(10)}[0]
# list comprehensions should also work in combination with functions
def listen(arg):
for x in arg:

31
test/test_evaluate/test_iterable.py
View File

@@ -1,31 +0,0 @@
from textwrap import dedent
from jedi import names
def get_scope_and_evaluator(source):
d = names(dedent(source))[0]
return d.parent()._definition, d._evaluator
def find_types(s):
scope, evaluator = get_scope_and_evaluator(s)
return evaluator.find_types(scope, s[0])
def test_comprehensions():
"""
test list/set/generator/dict comprehension syntax
"""
s = "a = [i for i in range(10)]"
assert len(find_types(s)) == 1
s = "a = [i for i in range(10)]"
assert len(find_types(s)) == 1
s = "a = {i:i for i in range(10)}"
assert len(find_types(s)) == 1
s = "a = {i for i in range(10)}"
assert len(find_types(s)) == 1