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')