def test_error_if_input_df_contains_na_in_transform(df_vartypes, df_na): # test case 4: when dataset contains na, transform method with pytest.raises(ValueError): transformer = EqualWidthDiscretiser() transformer.fit(df_vartypes) transformer.transform(df_na[["Name", "City", "Age", "Marks", "dob"]])
def test_error_if_input_df_contains_na_in_fit(df_na): # test case 3: when dataset contains na, fit method with pytest.raises(ValueError): transformer = EqualWidthDiscretiser() transformer.fit(df_na)