Update tests/test_legacy_readers.py

Co-authored-by: Jørgen Schartum Dokken <dokken@simula.no>

Author: finsberg

tests/test_legacy_readers.py

@@ -153,10 +153,8 @@ def test_read_legacy_function_from_checkpoint(backend):
     f = ufl.conditional(ufl.gt(x[0], 0.5), x[1], 2 * x[0])
     L = ufl.inner(f, v) * ufl.dx
 
-    try:
-        from dolfinx.fem.petsc import LinearProblem
-    except ImportError:
-        pytest.skip("dolfinx.fem.petsc.LinearProblem not available")
+    if not dolfinx.has_petsc4py:
+        pytest.skip("dolfinx not configured with PETSc4py")
 
     uh = dolfinx.fem.Function(V)
     if "petsc_options_prefix" in inspect.signature(LinearProblem.__init__).parameters.keys():