test: fix some test to work with a variety of nranks

PyLops · Oct 23, 2024 · 2e745c7 · 2e745c7
1 parent a4406f9
commit 2e745c7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/tests/test_fredholm.py b/tests/test_fredholm.py
@@ -16,7 +16,7 @@
 
 
 par1 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 5,
@@ -26,7 +26,7 @@
     "dtype": "float32",
 }  # real, saved Gt
 par2 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 5,
@@ -36,7 +36,7 @@
     "dtype": "float32",
 }  # real, unsaved Gt
 par3 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 5,
@@ -46,7 +46,7 @@
     "dtype": "complex64",
 }  # complex, saved Gt
 par4 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 5,
@@ -56,7 +56,7 @@
     "dtype": "complex64",
 }  # complex, unsaved Gt
 par5 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 1,
@@ -66,7 +66,7 @@
     "dtype": "float32",
 }  # real, saved Gt, nz=1
 par6 = {
-    "nsl": 6,
+    "nsl": 12,
     "ny": 6,
     "nx": 4,
     "nz": 1,

diff --git a/tests/test_linearop.py b/tests/test_linearop.py
@@ -351,4 +351,4 @@ def test_adj_mpilinop(par):
         VStack = pylops.VStack(ops=[(i + 1) * Sop for i in range(size)])
         FullOp = VStack @ Fop
         y_np = FullOp.H @ x_global
-        assert_allclose(y, y_np.flatten(), rtol=1e-14)
+        assert_allclose(y, y_np.flatten(), rtol=1e-12)