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

Fix unittest for Py3.6 ordered dicts

Browse source
This commit is contained in:
Ben Letham 2018-06-11 15:18:40 -07:00
parent 5cd35f7d6d
commit a5869505c0
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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']),