1
0
forked from VimPlug/jedi

Improve api class tests.

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

View File

@@ -520,13 +520,16 @@ class BaseNode(Base):
def end_pos(self):
return self.children[-1].end_pos
def get_code(self, normalized=False, include_prefix=True):
def _get_code_for_children(self, children, normalized, include_prefix):
# TODO implement normalized (depending on context).
if include_prefix:
return "".join(c.get_code(normalized) for c in self.children)
return "".join(c.get_code(normalized) for c in children)
else:
first = self.children[0].get_code(include_prefix=False)
return first + "".join(c.get_code(normalized) for c in self.children[1:])
first = children[0].get_code(include_prefix=False)
return first + "".join(c.get_code(normalized) for c in children[1:])
def get_code(self, normalized=False, include_prefix=True):
return self._get_code_for_children(self.children, normalized, include_prefix)
@Python3Method
def name_for_position(self, position):
@@ -1621,6 +1624,12 @@ class Param(BaseNode):
default = '' if self.default is None else '=%s' % self.default.get_code()
return '<%s: %s>' % (type(self).__name__, str(self._tfpdef()) + default)
def get_code(self, normalized=False, include_prefix=True):
children = self.children
if children[-1] == ',':
children = children[:-1]
return self._get_code_for_children(children, normalized, include_prefix)
class CompFor(BaseNode):
type = 'comp_for'