move set comprehension tests to comletion/comprehensions.py

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

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