13486bea951688cffe91312ae11caee7a8d1f265,tests/test_functions.py,,test_get_features_target_multi_columns,#Any#,205

Before Change

def test_get_features_target_multi_columns(dataframe):
    dataframe = dataframe.clean_names()
    X, y = dataframe.get_features_targets(target_columns=["a", "bell_chart"])
    assert X.shape == (9, 3)
    assert y.shape == (9, 2)

After Change

@pytest.mark.hyp
@given(dfstrategy())
def test_get_features_target_multi_columns(df):
    X, y = df.clean_names().get_features_targets(
        target_columns=["a", "bell_chart"]
    )
    assert X.shape[1] == 3

Instances