From fdcf7183172cff01e10321c83ea460fa8f006415 Mon Sep 17 00:00:00 2001 From: Tzerjen Wei Date: Sun, 9 Aug 2015 14:11:42 +0800 Subject: [PATCH] move set comprehension tests to comletion/comprehensions.py --- test/completion/comprehensions.py | 7 +++++++ test/test_evaluate/test_iterable.py | 31 ----------------------------- 2 files changed, 7 insertions(+), 31 deletions(-) delete mode 100644 test/test_evaluate/test_iterable.py diff --git a/test/completion/comprehensions.py b/test/completion/comprehensions.py index 98a15bcb..a8aaa0e9 100644 --- a/test/completion/comprehensions.py +++ b/test/completion/comprehensions.py @@ -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: diff --git a/test/test_evaluate/test_iterable.py b/test/test_evaluate/test_iterable.py deleted file mode 100644 index a8b29f6a..00000000 --- a/test/test_evaluate/test_iterable.py +++ /dev/null @@ -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