def test_invalid_moving_average(self):
series = TimeSeries([])
with self.assertRaises(ArithmeticError):
series.moving_average(3)
series = TimeSeries([ (1, 1), (2, 2) ])
with self.assertRaises(ArithmeticError):
series.moving_average(3)
def test_simple_moving_average(self):
points = [1, 2, 3, 4, 5, 6]
series = TimeSeries(zip(points, points))
ma = series.moving_average(3).round()
self.assertListEqual(ma.points, [ (3, 2), (4, 3), (5, 4), (6, 5) ])
ma = series.moving_average(5).round()
self.assertListEqual(ma.points, [ (5, 3), (6, 4) ])
