diff --git a/test/test_evaluate/test_helpers.py b/test/test_evaluate/test_helpers.py index 38c91267..651fe439 100644 --- a/test/test_evaluate/test_helpers.py +++ b/test/test_evaluate/test_helpers.py @@ -1,10 +1,21 @@ -from jedi.evaluate.helpers import deep_ast_copy +from jedi.evaluate import helpers from jedi.parser import representation as pr +from jedi.parser import Parser def test_deep_ast_copy(): name = pr.Name(object, [('hallo', (0, 0))], (0, 0), (0, 0)) # fast parent copy should switch parent - new_name = deep_ast_copy(name) + new_name = helpers.deep_ast_copy(name) assert new_name.names[0].parent == new_name + + +def test_statement_elements_in_statement(): + def get_stmt_els(string): + p = Parser(string) + return helpers.statement_elements_in_statement(p.module.statements[0]) + + stmt_els = get_stmt_els('foo = [(bar(f), f) for f in baz]') + # stmt_els: count all names: 6; + count all arrays: 2 = 8 + assert len(stmt_els) == 8