Improve Name.is_definition

2014-11-02 17:53:04 +01:00
parent 81174741a4
commit f4d7020ebf
1 changed files with 5 additions and 2 deletions
--- a/jedi/parser/representation.py
+++ b/jedi/parser/representation.py
@@ -255,8 +255,11 @@ class Name(Leaf):

    def is_definition(self):
        stmt = self.get_definition()
-        return isinstance(stmt, (ExprStmt, Import)) \
-            and self in stmt.get_defined_names()
+        if isinstance(stmt, (Function, Class, Module)):
+            return True
+        else:
+            return isinstance(stmt, (ExprStmt, Import)) \
+                and self in stmt.get_defined_names()

    def assignment_indexes(self):
        """