[experimental] Implement cached residual variant in test

Implement the timestepping test in a ways that the residuals of the last stage do not have to be recomputed. This means we also don't have to store the variables of the last step.