def METHOD_NAME(self):
    time = Time()
    time.set_control_vars(
        initial_time=2, final_time=100, saveper=1, time_step=0.25)
    assert time.in_return()
    time.update(4)
    assert time.in_return()
    time.update(10)
    assert time.in_return()
    time.update(12)
    assert time.in_return()
    time.update(37)
    assert time.in_return()
    time.reset()
    time.add_return_timestamps([2, 10, 37])
    assert time.in_return()
    time.update(4)
    assert not time.in_return()
    time.update(10)
    assert time.in_return()
    time.update(12)
    assert not time.in_return()
    time.update(37)
    assert time.in_return()
    time.reset()
    time.add_return_timestamps(np.array([4, 12]))
    assert not time.in_return()
    time.update(4)
    assert time.in_return()
    time.update(10)
    assert not time.in_return()
    time.update(12)
    assert time.in_return()
    time.update(37)
    assert not time.in_return()
    time.reset()
    time.add_return_timestamps(37)
    assert not time.in_return()
    time.update(4)
    assert not time.in_return()
    time.update(10)
    assert not time.in_return()
    time.update(12)
    assert not time.in_return()
    time.update(37)
    assert time.in_return()
    time.reset()
    time.add_return_timestamps(None)
    assert time.in_return()
    time.update(4)
    assert time.in_return()
    time.update(10)
    assert time.in_return()
    time.update(12)
    assert time.in_return()
    time.update(37)
    assert time.in_return()