saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-29 20:14:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Split test into fast and slow part

Browse source
This commit is contained in:
Sean Taylor 2020-09-02 17:38:20 -07:00
parent 96848bcc92
commit 838c72fb85
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/fbprophet/tests/test_prophet.py
View file

@ -102,6 +102,12 @@ class TestProphet(TestCase):
forecaster.fit(train)
forecaster.predict(future)
@skipUnless("--test-slow" in sys.argv, "Skipped due to the lack of '--test-slow' argument")
def test_fit_predict_no_changepoints_mcmc(self):
N = DATA.shape[0]
train = DATA.head(N // 2)
future = DATA.tail(N // 2)
forecaster = Prophet(n_changepoints=0, mcmc_samples=100)
forecaster.fit(train)
forecaster.predict(future)