Write tests for steps implementations about yaib HOT 1 CLOSED

def test_step(example_df):
    rec = Recipe(example_df, ['y'], ['x1', 'x2'], ['id'])
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='min', suffix='min'))
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='max', suffix='max'))
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='mean', suffix='mean'))
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='median', suffix='median'))
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='count', suffix='count'))
    rec.add_step(StepHistorical(sel=all_of(['x1', 'x2']), fun='var', suffix='var'))
    df = rec.bake()
    assert df['x1_min'].iloc[-1] == df['x1'].loc[df['id'] == 2].min()
    assert df['x2_max'].iloc[-1] == df['x2'].loc[df['id'] == 2].max()
    assert df['x2_mean'].iloc[-1] == df['x2'].loc[df['id'] == 2].mean()
    assert df['x1_median'].iloc[-1] == df['x1'].loc[df['id'] == 2].median()
    assert df['x1_count'].iloc[-1] == df['x1'].loc[df['id'] == 2].count()
    assert df['x2_var'].iloc[-1] == df['x2'].loc[df['id'] == 2].var()