def test_transform_value_error(self):
# arrange
year = 2004
input_df = pd.DataFrame(columns=["year"])
with pytest.raises(ValueError):
transform.transform_enade_year(input_df, year)
def test_transform_2005(self, mocker: MockerFixture):
# arrange
year = 2005
def side_effect(df: pd.DataFrame) -> pd.DataFrame:
return df.append({"year": f"{year}"}, ignore_index=True)
mocker.patch(
"src.transformation.transform_2005.Transform2005.transform",
side_effect=side_effect)
input_df = pd.DataFrame(columns=["year"])
expected_df = pd.DataFrame({"year": [f"{year}"]})
output = transform.transform_enade_year(input_df, year)
assert output.equals(expected_df)