def METHOD_NAME(self):
    self.assertEqual(
        [0.1, 0.3, 0.6, 1, 1, 1, 1, 1, 1, 1, 1, 1],
        list(self.tm1._tm1_rest.wait_time_generator(10)))
    self.assertEqual(sum(self.tm1._tm1_rest.wait_time_generator(10)), 10)