catalyst/tests/pipeline/test_alias.py

from nose.tools import nottest
import numpy as np

from catalyst.testing.predicates import assert_equal
from catalyst.pipeline import Classifier, Factor, Filter
from catalyst.utils.numpy_utils import float64_dtype, int64_dtype

from .base import BasePipelineTestCase


@nottest
class BaseAliasTestCase(BasePipelineTestCase):

    def test_alias(self):
        f = self.Term()
        alias = f.alias('ayy lmao')

        f_values = np.random.RandomState(5).randn(5, 5)

        self.check_terms(
            terms={
                'f_alias': alias,
            },
            expected={
                'f_alias': f_values,
            },
            initial_workspace={f: f_values},
            mask=self.build_mask(np.ones((5, 5))),
        )

    def test_repr(self):
        assert_equal(
            repr(self.Term().alias('ayy lmao')),
            "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(BaseAliasTestCase):
    class Term(Factor):
        dtype = float64_dtype
        inputs = ()
        window_length = 0


class TestFilterAlias(BaseAliasTestCase):
    class Term(Filter):
        inputs = ()
        window_length = 0


class TestClassifierAlias(BaseAliasTestCase):
    class Term(Classifier):
        dtype = int64_dtype
        inputs = ()
        window_length = 0
        missing_value = -1