def test_no_error_when_null_values_in_variable():
math_combinator_mean = MathematicalCombination(
variables_to_combine=['fixed acidity', 'volatile acidity'],
math_operations=['mean'],
new_variables_names=['avg_acidity'],
missing_values="ignore"
)
ref = pd.DataFrame.from_dict({
"fixed acidity": None,
"volatile acidity": 0.7
})
math_combinator_mean.fit(ref)
math_combinator_mean.transform(ref)
def test_error_when_null_values_in_variable(df_vartypes):
df_na = df_vartypes.copy()
df_na.loc[1, "Age"] = np.nan
with pytest.raises(ValueError):
math_combinator = MathematicalCombination(
variables_to_combine=["Age", "Marks"],
math_operations=["sum", "mean"],
missing_values="raise",
)
math_combinator.fit(df_na)
with pytest.raises(ValueError):
math_combinator = MathematicalCombination(
variables_to_combine=["Age", "Marks"],
math_operations=["sum", "mean"],
missing_values="raise",
)
math_combinator.fit(df_vartypes)
math_combinator.transform(df_na)