diff --git a/tests/pipeline/test_alias.py b/tests/pipeline/test_alias.py index 3044fcdb..ecd6045a 100644 --- a/tests/pipeline/test_alias.py +++ b/tests/pipeline/test_alias.py @@ -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): diff --git a/zipline/pipeline/mixins.py b/zipline/pipeline/mixins.py index cd6048ad..b5663dcb 100644 --- a/zipline/pipeline/mixins.py +++ b/zipline/pipeline/mixins.py @@ -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): """