Ejemplo n.º 1
0
    def test_target_encoder_clean_column_other_name(self):
        column_name = "test_column"
        expected = "test_column_enc"

        encoder = TargetEncoder()
        actual = encoder._clean_column_name(column_name)

        assert actual == expected
Ejemplo n.º 2
0
    def test_target_encoder_clean_column_name_processed_column(self):
        column_name = "test_column_processed"
        expected = "test_column_enc"

        encoder = TargetEncoder()
        actual = encoder._clean_column_name(column_name)

        assert actual == expected