def test_get_zero_changepoints(self): m = Prophet(n_changepoints=0) N = DATA.shape[0] history = DATA.head(N // 2).copy() history = m.setup_dataframe(history, initialize_scales=True) m.history = history m.set_changepoints() cp = m.get_changepoint_indexes() self.assertEqual(cp.shape[0], 1) self.assertEqual(cp[0], 0) mat = m.get_changepoint_matrix() self.assertEqual(mat.shape[0], N // 2) self.assertEqual(mat.shape[1], 1)
def test_get_changepoints(self): m = Prophet() N = DATA.shape[0] history = DATA.head(N // 2).copy() history = m.setup_dataframe(history, initialize_scales=True) m.history = history m.set_changepoints() cp = m.get_changepoint_indexes() self.assertEqual(cp.shape[0], m.n_changepoints) self.assertEqual(len(cp.shape), 1) self.assertTrue(cp.min() > 0) self.assertTrue(cp.max() < N) mat = m.get_changepoint_matrix() self.assertEqual(mat.shape[0], N // 2) self.assertEqual(mat.shape[1], m.n_changepoints)