Improve api class tests.

This commit is contained in:
Dave Halter committed 2016-12-04 20:04:54 +01:00
1 parent 439e394535
commit 6f4cd7e6d3
7 files changed
+75 -40

No files matched your search

+2
View File
@@ -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)