|
@@ -323,8 +323,9 @@ class TestProphet(TestCase):
|
|
|
holidays2 = pd.concat((holidays, holidays2))
|
|
|
m = Prophet(holidays=holidays2)
|
|
|
feats, priors, names = m.make_holiday_features(df['ds'])
|
|
|
- self.assertEqual(priors, [8., 8., 5., 5.])
|
|
|
- self.assertEqual(set(names), {'xmas', 'seans-bday'})
|
|
|
+ pn = zip(priors, [s.split('_delim_')[0] for s in feats.columns])
|
|
|
+ for t in pn:
|
|
|
+ self.assertIn(t, [(8., 'seans-bday'), (5., 'xmas')])
|
|
|
holidays2 = pd.DataFrame({
|
|
|
'ds': pd.to_datetime(['2012-06-06', '2013-06-06']),
|
|
|
'holiday': ['seans-bday'] * 2,
|
|
@@ -335,7 +336,7 @@ class TestProphet(TestCase):
|
|
|
feats, priors, names = Prophet(
|
|
|
holidays=holidays2, holidays_prior_scale=4
|
|
|
).make_holiday_features(df['ds'])
|
|
|
- self.assertEqual(priors, [4., 4., 5., 5.])
|
|
|
+ self.assertEqual(set(priors), {4., 5.})
|
|
|
# Check incompatible priors
|
|
|
holidays = pd.DataFrame({
|
|
|
'ds': pd.to_datetime(['2016-12-25', '2016-12-27']),
|