def test_forecast_on_empty_series(self):
with self.assertRaises(ArithmeticError):
series = TimeSeries([])
forecast = series.forecast(7)
def test_ets_forecast_with_frequency(self):
series = TimeSeries([ (1, 100), (2, 200), (3, 100), (4, 200), (5, 100) ])
forecast = series.forecast(3, method=TimeSeries.ETS, frequency=4)
self.assertTrue(isinstance(forecast, TimeSeries))
self.assertListEqual(forecast.timestamps, [6, 7, 8])
def test_invalid_forecast_method(self):
series = TimeSeries([ (1, 100), (2, 200), (3, 100), (4, 200), (5, 100) ])
with self.assertRaises(ValueError):
forecast = series.forecast(7, method='huh')
