Get checkpointing working for simulations which solve multiple equations per timestep.
That was surprisingly easy, once I realised how.

Add a checkpoint_stokes test. We do 11 timesteps, but only give 5 checkpoint slots, so it has
to replay several times during the adjoint run in order to acquire the solutions about
which it wants to linearise.