Mark slower tests as such.
Fix use of OptimizerState.set_skip_steps().