ENH: update aliasedmixin repr

This commit is contained in:
Joe Jevnik
2016-10-04 16:09:00 -04:00
parent f9ce7f060e
commit 499680553f
2 changed files with 15 additions and 2 deletions
+10 -1
View File
@@ -29,9 +29,18 @@ class WithAlias(object):
def test_repr(self):
assert_equal(
repr(self.Term().alias('ayy lmao')),
"Aliased%s(..., name='ayy lmao')" % self.Term.__base__.__name__,
"Aliased%s(Term(...), name='ayy lmao')" % (
self.Term.__base__.__name__,
),
)
def test_short_repr(self):
for name in ('a', 'b'):
assert_equal(
self.Term().alias(name).short_repr(),
name,
)
class TestFactorAlias(WithAlias, BasePipelineTestCase):
class Term(Factor):
+5 -1
View File
@@ -272,11 +272,15 @@ class AliasedMixin(SingleInputMixin):
return inputs[0]
def __repr__(self):
return '{type}(..., name={name!r})'.format(
return '{type}({inner_type}(...), name={name!r})'.format(
type=type(self).__name__,
inner_type=type(self.inputs[0]).__name__,
name=self.name,
)
def short_repr(self):
return self.name
@classmethod
def make_aliased_type(cls, other_base):
"""