Пример #1
0
    def test_inheritance(self):
        """Test that OneHotEncodingTransformer inherits from BaseNominalTransformer and sklean's OneHotEncoder."""

        x = OneHotEncodingTransformer()

        h.assert_inheritance(x, tubular.nominal.BaseNominalTransformer)
        h.assert_inheritance(x, sklearn.preprocessing.OneHotEncoder)
Пример #2
0
    def test_inheritance(self):
        """Test that NominalToIntegerTransformer inherits from BaseNominalTransformer."""

        x = OrdinalEncoderTransformer(response_column="a")

        h.assert_inheritance(x, tubular.nominal.BaseNominalTransformer)
        h.assert_inheritance(x, tubular.mapping.BaseMappingTransformMixin)
Пример #3
0
    def test_inheritance(self):
        """Test that MappingTransformer inherits from BaseMappingTransformer and BaseMappingTransformMixin."""

        x = MappingTransformer(mappings={"a": {"a": 1}})

        h.assert_inheritance(x, tubular.mapping.BaseMappingTransformer)
        h.assert_inheritance(x, tubular.mapping.BaseMappingTransformMixin)
Пример #4
0
def test_isinstance_call():
    """Test the call to isinstance."""

    with mock.patch("tubular.testing.helpers.isinstance") as mocked_method:

        h.assert_inheritance(1, int)

        assert (
            mocked_method.call_count == 1
        ), f"Unexpected number of call to h.check_is_class -\n  Expected: 1\n  Actual: {mocked_method.call_count}"

        call_1_args = mocked_method.call_args_list[0]
        call_1_pos_args = call_1_args[0]
        call_1_kwargs = call_1_args[1]

        call_1_expected_kwargs = {}

        call_1_expected_pos_args = (1, int)

        assert (
            call_1_expected_kwargs == call_1_kwargs
        ), f"Unexpected kwargs -\n  Expected: {call_1_expected_kwargs}\n  Actual: {call_1_kwargs}"

        assert len(call_1_expected_pos_args) == len(
            call_1_pos_args
        ), f"Unexpected number of positional -\n  Expected: {len(call_1_expected_pos_args)}\n  Actual: {len(call_1_pos_args)}"

        assert (
            call_1_expected_pos_args[0] == call_1_pos_args[0]
        ), f"Unexpected number of positional arg in index 0 -\n  Expected: {call_1_expected_pos_args[0]}\n  Actual: {call_1_pos_args[0]}"

        assert (
            call_1_expected_pos_args[1] == call_1_pos_args[1]
        ), f"Unexpected number of positional arg in index 1 -\n  Expected: {call_1_expected_pos_args[1]}\n  Actual: {call_1_pos_args[1]}"
    def test_inheritance(self):
        """Test that DataFrameMethodTransformer inherits from BaseTransformer."""

        x = DataFrameMethodTransformer(new_column_name="a",
                                       pd_method_name="sum",
                                       columns=["b", "c"])

        h.assert_inheritance(x, tubular.base.BaseTransformer)
    def test_inheritance(self):
        """Test that SeriesDtMethodTransformer inherits from BaseTransformer."""

        x = SeriesDtMethodTransformer(new_column_name="a",
                                      pd_method_name="year",
                                      column="b")

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #7
0
    def test_inheritance(self):
        """Test that SeriesStrMethodTransformer inherits from BaseTransformer."""

        x = SeriesStrMethodTransformer(new_column_name="a",
                                       pd_method_name="find",
                                       columns=["b"])

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #8
0
    def test_inheritance(self):
        """Test that CrossColumnAddTransformer inherits from BaseMappingTransformer."""

        x = CrossColumnAddTransformer(mappings={"a": {
            "a": 1
        }},
                                      adjust_column="b")

        h.assert_inheritance(x, tubular.mapping.BaseMappingTransformer)
    def test_inheritance(self):
        """Test that BetweenDatesTransformer inherits from BaseTransformer."""

        x = BetweenDatesTransformer(column_lower="a",
                                    column_between="b",
                                    column_upper="c",
                                    new_column_name="d")

        h.assert_inheritance(x, BaseTransformer)
Пример #10
0
def test_error():
    """Test an exception with the right info is raised if obj is not an instance of cls."""

    with pytest.raises(
            AssertionError,
            match=
            f"Incorrect inheritance - passed obj of class {(1).__class__.__name__} is not an instance of {float}",
    ):

        h.assert_inheritance(1, float)
Пример #11
0
    def test_inheritance(self):
        """Test that DateDiffLeapYearTransformer inherits from BaseTransformer."""

        x = DateDiffLeapYearTransformer(
            column_lower="dummy_1",
            column_upper="dummy_2",
            new_column_name="dummy_3",
            drop_cols=True,
        )

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #12
0
def test_inheritance():
    """Test ReturnKeyDict inherits from dict."""

    x = ReturnKeyDict()

    h.assert_inheritance(x, dict)
    def test_inheritance(self):
        """Test that DateDifferenceTransformer inherits from BaseTransformer."""

        x = DateDifferenceTransformer("column_lower", "column_upper")

        h.assert_inheritance(x, tubular.base.BaseTransformer)
    def test_inheritance(self):
        """Test that BaseMappingTransformMixin inherits from BaseTransformer."""

        x = BaseMappingTransformMixin()

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #15
0
    def test_inheritance(self):
        """Test that ToDatetimeTransformer inherits from BaseTransformer."""

        to_dt = ToDatetimeTransformer(column="a", new_column_name="b")

        h.assert_inheritance(to_dt, tubular.base.BaseTransformer)
    def test_inheritance(self):
        """Test that BaseMappingTransformer inherits from BaseTransformer."""

        x = BaseMappingTransformer(mappings={"a": {"a": 1}})

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #17
0
    def test_inheritance(self):
        """Test that NominalToIntegerTransformer inherits from BaseNominalTransformer."""

        x = GroupRareLevelsTransformer()

        h.assert_inheritance(x, tubular.nominal.BaseNominalTransformer)
    def test_inheritance(self):
        """Test that BaseNominalTransformer inherits from BaseTransformer."""

        x = BaseNominalTransformer()

        h.assert_inheritance(x, BaseTransformer)
Пример #19
0
    def test_inheritance(self):
        """Test that ModeImputer inherits from BaseImputer."""

        x = ModeImputer()

        h.assert_inheritance(x, tubular.imputers.BaseImputer)
Пример #20
0
    def test_inheritance(self):
        """Test that BaseImputer inherits from BaseTransformer."""

        x = BaseImputer()

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #21
0
    def test_inheritance(self):
        """Test that CappingTransformer inherits from BaseTransformer."""

        x = CappingTransformer(capping_values={"a": [1, 3]})

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #22
0
    def test_inheritance(self):
        """Test that NullIndicator inherits from BaseTransformer."""

        x = NullIndicator()

        h.assert_inheritance(x, tubular.base.BaseTransformer)
    def test_inheritance(self):
        """Test that ArbitraryImputer inherits from BaseTransformer."""

        x = ArbitraryImputer(impute_value=1, columns="a")

        h.assert_inheritance(x, tubular.imputers.BaseImputer)
    def test_inheritance(self):
        """Test that CutTransformer inherits from BaseTransformer."""

        x = CutTransformer(column="a", new_column_name="b")

        h.assert_inheritance(x, tubular.base.BaseTransformer)
    def test_inheritance(self):
        """Test that ScalingTransformer inherits from BaseTransformer."""

        x = ScalingTransformer(columns=["a"], scaler="standard")

        h.assert_inheritance(x, tubular.base.BaseTransformer)
Пример #26
0
    def test_inheritance(self):
        """Test that OutOfRangeNullTransformer inherits from CappingTransformer."""

        x = OutOfRangeNullTransformer(capping_values={"a": [1, 3]})

        h.assert_inheritance(x, tubular.capping.CappingTransformer)
    def test_inheritance(self):
        """Test that LogTransformer inherits from BaseTransformer."""

        x = LogTransformer(columns="a")

        h.assert_inheritance(x, tubular.base.BaseTransformer)