Add HDF5 test to firedrake-check

scripts/firedrake-check

@@ -24,6 +24,10 @@ TESTS = {
         # near nullspace
         "tests/firedrake/regression/test_nullspace.py::test_near_nullspace",
     ),
+    2: (
+        # HDF5/checkpointing
+        "tests/firedrake/output/test_io_function.py::test_io_function_base[cell_family_degree2]",
+    ),
     3: (
         "tests/firedrake/regression/test_dg_advection.py::test_dg_advection_icosahedral_sphere[nprocs=3]",
         # vertex-only mesh

setup.py

@@ -229,6 +229,7 @@ def extensions():
     "tests/firedrake/regression/test_nullspace.py",
     "tests/firedrake/regression/test_dg_advection.py",
     "tests/firedrake/regression/test_interpolate_cross_mesh.py",
+    "tests/firedrake/output/test_io_function.py",
 )
 
 

tests/firedrake/output/test_hdf5file_checkpoint.py

@@ -38,7 +38,8 @@ def f():
     return f
 
 
-def run_write_read(mesh, fs, degree, dumpfile):
+@pytest.mark.parallel([1, 2])
+def test_write_read(mesh, fs, degree, dumpfile):
 
     V = FunctionSpace(mesh, fs, degree)
 
@@ -79,15 +80,6 @@ def test_checkpoint_fails_for_non_function(dumpfile):
             h5.write(np.arange(10), "/solution")
 
 
-def test_write_read(mesh, fs, degree, dumpfile):
-    run_write_read(mesh, fs, degree, dumpfile)
-
-
-@pytest.mark.parallel(nprocs=2)
-def test_write_read_parallel(mesh, fs, degree, dumpfile):
-    run_write_read(mesh, fs, degree, dumpfile)
-
-
 def test_checkpoint_read_not_exist_ioerror(dumpfile):
     with pytest.raises(IOError):
         with HDF5File(dumpfile, file_mode="r"):

tests/firedrake/output/test_io_function.py

@@ -148,7 +148,7 @@ def _load_check_save_functions(filename, func_name, comm, method, mesh_name, var
         afile.save_function(fB)
 
 
-@pytest.mark.parallel(nprocs=2)
+@pytest.mark.parallel(2)
 @pytest.mark.parametrize('cell_family_degree', [
     ("triangle_small", "P", 1),
     ("triangle_small", "P", 6),