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