diff --git a/tests/firedrake/adjoint/test_burgers_newton.py b/tests/firedrake/adjoint/test_burgers_newton.py index a8cd11d600..d9a97f00a8 100644 --- a/tests/firedrake/adjoint/test_burgers_newton.py +++ b/tests/firedrake/adjoint/test_burgers_newton.py @@ -264,6 +264,8 @@ def test_checkpointing_validity(solve_type, checkpointing, basics): tape.progress_bar = ProgressBar if checkpointing == "Revolve": tape.enable_checkpointing(Revolve(steps, steps//3)) + elif checkpointing == "SingleMemory": + tape.enable_checkpointing(SingleMemoryStorageSchedule()) elif checkpointing == "Mixed": enable_disk_checkpointing() tape.enable_checkpointing(MixedCheckpointSchedule(steps, steps//3, storage=StorageType.DISK))