def test_get_average(self): time_series = TimeSeries([ (datetime.date(2001, 12, 31), 1), (datetime.date(2002, 12, 31), 2), (datetime.date(2003, 12, 31), 3), (datetime.date(2004, 12, 31), 4), (datetime.date(2005, 12, 31), 6), ]) averaged = time_series.get_average() self.assertEqual(averaged.get(), [ (datetime.date(2001, 12, 31), 1), (datetime.date(2002, 12, 31), 1.5), (datetime.date(2003, 12, 31), 2.5), (datetime.date(2004, 12, 31), 3.5), (datetime.date(2005, 12, 31), 5), ])