def setUp(self):
self.mock = make_mock(do_plot=False)
self.y = self.mock['y']
self.num = len(self.y)
self.dates = date_range(2015, 1, 2029, 12)[0:self.num]
self.assertEquals(len(self.y), len(self.dates))
seas_func = lambda date: date.month-1
self.seasonality_matrix, self.compression_dict = \
get_seasonality_matrix(self.dates, seasonality_function=seas_func)
def get_mock_with_dates():
mock = make_mock()
num = len(mock['x'])
mock['dates'] = date_range(2015, 1, 2099, 12)[:num]
return mock
def setUp(self):
mock = make_mock()
self.y = mock['y']
num = len(self.y)
self.dates = date_range(2015, 1, 2029, 12)[0: num]
