Make statement_elements_in_statement work with ListComprehensions, Lambdas and 'except foo as' expressions

2014-09-02 14:52:04 +02:00
parent 38c71fce3f
commit ea5b98905e
3 changed files with 21 additions and 2 deletions
--- a/jedi/api/classes.py
+++ b/jedi/api/classes.py
@@ -319,7 +319,7 @@ class BaseDefinition(object):
                    call_path = list(stmt_el.generate_call_path())
                    for i, element in enumerate(call_path):
                        if element is name_part:
-                            return call_path[:i+1]
+                            return call_path[:i + 1]

        if not isinstance(self._definition, pr.NamePart):
            raise TypeError('Definition is not a NamePart.')