def test_quadratic_trend(self):
series = TimeSeries([ (1, 32), (2, 55), (3, 40), (4, 100) ])
trend = series.trend(order=TimeSeries.QUADRATIC).round()
self.assertListEqual(trend.timestamps, [1, 2, 3, 4])
self.assertListEqual(trend.values, [38, 38, 57, 94])
def test_invalid_trend(self):
series = TimeSeries([])
with self.assertRaises(ArithmeticError):
series.trend()
def test_linear_trend(self):
series = TimeSeries([ (1, 32), (2, 55), (3, 40) ])
trend = series.trend(order=TimeSeries.LINEAR).round()
self.assertListEqual(trend.timestamps, [1, 2, 3])
self.assertListEqual(trend.values, [38, 42, 46])
