def test_predict_remove_data(): # GH6887 endog = [i + np.random.normal(scale=0.1) for i in range(100)] exog = [i for i in range(100)] model = OLS(endog, exog, weights=[1 for _ in range(100)]).fit() model.remove_data() scalar = model.get_prediction(1).predicted_mean one_d = model.get_prediction([1]).predicted_mean assert_allclose(scalar, one_d) series = model.get_prediction(pd.Series([1])).predicted_mean assert_allclose(scalar, series)