def test_error_when_division_by_zero(_func, df_vartypes): df_zero = df_vartypes.copy() df_zero.loc[1, "Marks"] = 0 transformer = RelativeFeatures( variables=["Age"], reference=["Marks"], func=_func, ) transformer.fit(df_vartypes) with pytest.raises(ValueError): transformer.transform(df_zero)
def test_error_when_null_values_in_variable(df_vartypes): df_na = df_vartypes.copy() df_na.loc[1, "Age"] = np.nan transformer = RelativeFeatures( variables=["Age", "Marks"], reference=["Age", "Marks"], func=["add", "mul"], missing_values="raise", ) with pytest.raises(ValueError): transformer.fit(df_na) transformer.fit(df_vartypes) with pytest.raises(ValueError): transformer.transform(df_na)