From 838c72fb851abc98a893fc5433a25d7291590a69 Mon Sep 17 00:00:00 2001 From: Sean Taylor Date: Wed, 2 Sep 2020 17:38:20 -0700 Subject: [PATCH] Split test into fast and slow part --- python/fbprophet/tests/test_prophet.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/fbprophet/tests/test_prophet.py b/python/fbprophet/tests/test_prophet.py index 8489e65..fd15e77 100644 --- a/python/fbprophet/tests/test_prophet.py +++ b/python/fbprophet/tests/test_prophet.py @@ -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)