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)
