def test_invalid_transf_type():
ft = FillNaTransformer_forward(columns=col_a)
ft.fit(data)
with pytest.raises(NameError):
ft.transform(data.values)
def test_col_no_include():
ft = FillNaTransformer_forward(columns=col_no_include)
with pytest.raises(NameError):
ft.fit(data)
def test_tranf_equal_null():
ft = FillNaTransformer_forward()
ft.fit(data)
assert_frame_equal(ft.transform(data), data_out)
def test_tranf_equal_all():
ft = FillNaTransformer_forward(columns=col_test)
ft.fit(data)
assert_frame_equal(ft.transform(data), data_out)
def test_tranf_equal_b():
ft1 = FillNaTransformer_forward(columns=col_b)
ft1.fit(data)
assert_frame_equal(ft1.transform(data), data_out_b)
def test_col_fail_type():
with pytest.raises(NameError):
FillNaTransformer_forward(columns=np.array(col_test))
def test_fitting_error():
"""FeatureDropper has not been fitted, yet."""
ft = FillNaTransformer_forward(columns=col_a)
with pytest.raises(AttributeError):
ft.transform(data)