saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-12 17:57:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix diagnostics unit tests for new pandas

Browse source
This commit is contained in:
bl 2017-11-04 17:03:44 -07:00
parent 4ee5a971e4
commit 9ffa0d8790
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
python/fbprophet/tests/test_diagnostics.py
View file

@ -84,7 +84,9 @@ class TestDiagnostics(TestCase):
df_shf2 = diagnostics.simulated_historical_forecasts(
m, horizon='10 days', k=1, period='5 days')
self.assertAlmostEqual(
((df_shf1 - df_shf2)**2)[['y', 'yhat']].sum().sum(), 0.0)
((df_shf1['y'] - df_shf2['y']) ** 2).sum(), 0.0)
self.assertAlmostEqual(
((df_shf1['yhat'] - df_shf2['yhat']) ** 2).sum(), 0.0)
def test_cross_validation(self):
m = Prophet()
@ -111,4 +113,6 @@ class TestDiagnostics(TestCase):
df_cv2 = diagnostics.cross_validation(
m, horizon='32 days', period='10 days', initial='96 days')
self.assertAlmostEqual(
((df_cv1 - df_cv2)**2)[['y', 'yhat']].sum().sum(), 0.0)
((df_cv1['y'] - df_cv2['y']) ** 2).sum(), 0.0)
self.assertAlmostEqual(
((df_cv1['yhat'] - df_cv2['yhat']) ** 2).sum(), 0.0)