Add test for MCMC with 0 changepoints

This commit is contained in:
Maxim Grishin 2019-10-26 00:01:28 +03:00 committed by Ben Letham
parent a6646670b9
commit ca9a49d328

View file

@ -55,6 +55,10 @@ class TestProphet(TestCase):
forecaster.fit(train)
forecaster.predict(future)
forecaster = Prophet(n_changepoints=0, mcmc_samples=100)
forecaster.fit(train)
forecaster.predict(future)
def test_fit_changepoint_not_in_history(self):
train = DATA[(DATA['ds'] < '2013-01-01') | (DATA['ds'] > '2014-01-01')]
future = pd.DataFrame({'ds': DATA['ds']})