def test_evaluate_score_ndim(X, y):
"""Test fit() works for both y.ndim == 1 and y.ndim == 2. Two test cases
are listed in @pytest.mark.parametrize()
"""
sr = SimpleRegressor(random_state=0)
print(f"Data ndim: X: {X.shape}, y: {y.shape}")
sr.fit(X, y)
def test_regression_boston():
boston = load_boston()
data = data_df_from_bunch(boston)
er = SimpleRegressor()
er.fit(data, target_col='target')
# test nupmy array
er = SimpleRegressor()
er.fit(boston.data, boston.target)