Improve api class tests.

2016-12-04 20:04:54 +01:00
parent 439e394535
commit 6f4cd7e6d3
7 changed files with 75 additions and 40 deletions
--- a/jedi/evaluate/init.py
+++ b/jedi/evaluate/init.py
@@ -488,6 +488,8 @@ class Evaluator(object):
            # a name it's something you can "goto" again.
            return [TreeNameDefinition(context, name)]
        elif isinstance(par, (tree.Param, tree.Function, tree.Class)) and par.name is name:
+            if par.type in ('funcdef', 'classdef', 'module'):
+                return [context.name]
            return [TreeNameDefinition(context, name)]
        elif isinstance(stmt, tree.Import):
            module_names = imports.ImportWrapper(context, name).follow(is_goto=True)