TST: Only test for KSP solvers if PETSc is available

pmgbergen · Aug 29, 2024 · d2d416a · d2d416a
1 parent 675ac8a
commit d2d416a
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/tests/unit/test_wasserstein.py b/tests/unit/test_wasserstein.py
@@ -5,6 +5,13 @@
 
 import darsia
 
+try:
+    import petsc4py
+
+    HAVE_PETSC = True
+except ImportError:
+    HAVE_PETSC = False
+
 # ! ---- 2d version ----
 
 # Coarse src image
@@ -148,13 +155,16 @@
 }
 
 
-solvers = [
-    lu_options,
-    amg_options,
-    ksp_direct_options,
-    ksp_krylov_options,
-    ksp_block_krylov_options,
-]
+solvers = (
+    [lu_options, amg_options]
+    + [
+        ksp_direct_options,
+        ksp_krylov_options,
+        ksp_block_krylov_options,
+    ]
+    if HAVE_PETSC
+    else []
+)
 
 
 # General options