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)