mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-12 13:41:29 +08:00
ENH: update aliasedmixin repr
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user