def test_map_return_type(self): series = TimeSeries([ (1, 2), (3, 4), (5, 6) ]) double = series.map(lambda y: y * 2) self.assertTrue(isinstance(double, TimeSeries)) self.assertListEqual([ (1, 4), (3, 8), (5, 12) ], double.points)