Exemplo n.º 1
0
def test_invalid_transf_type():
    ft = FillNaTransformer_forward(columns=col_a)
    ft.fit(data)
    with pytest.raises(NameError):
        ft.transform(data.values)
Exemplo n.º 2
0
def test_col_no_include():
    ft = FillNaTransformer_forward(columns=col_no_include)
    with pytest.raises(NameError):
        ft.fit(data)
Exemplo n.º 3
0
def test_tranf_equal_null():
    ft = FillNaTransformer_forward()
    ft.fit(data)
    assert_frame_equal(ft.transform(data), data_out)
Exemplo n.º 4
0
def test_tranf_equal_all():
    ft = FillNaTransformer_forward(columns=col_test)
    ft.fit(data)
    assert_frame_equal(ft.transform(data), data_out)
Exemplo n.º 5
0
def test_tranf_equal_b():
    ft1 = FillNaTransformer_forward(columns=col_b)
    ft1.fit(data)
    assert_frame_equal(ft1.transform(data), data_out_b)
Exemplo n.º 6
0
def test_col_fail_type():
    with pytest.raises(NameError):
        FillNaTransformer_forward(columns=np.array(col_test))
Exemplo n.º 7
0
def test_fitting_error():
    """FeatureDropper has not been fitted, yet."""
    ft = FillNaTransformer_forward(columns=col_a)
    with pytest.raises(AttributeError):
        ft.transform(data)