docker-python/tests/test_prophet.py at tmp-gpu-test · Kaggle/docker-python · GitHub

18 lines (13 loc) · 542 Bytes

import unittest

import numpy as np

import pandas as pd

from prophet import Prophet

class TestProphet(unittest.TestCase):

def test_fit(self):

train = pd.DataFrame({

'ds': np.array(['2012-05-18', '2012-05-19', '2012-05-20', '2012-05-21', '2012-05-22']),

'y': np.array([38.23, 21.25, 20.18, 20.01, 19.02])

})

forecaster = Prophet(mcmc_samples=1)

forecaster.fit(train, adapt_engaged=False)

self.assertEqual(len(train) + 1, len(forecaster.make_future_dataframe(periods=1)))

Expand file tree