tests: reduce size of test signals

flatironinstitute · Jul 17, 2023 · da1a155 · da1a155
1 parent 9c896a1
commit da1a155
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/cufinufft/tests/test_basic.py b/python/cufinufft/tests/test_basic.py
@@ -9,10 +9,10 @@
 
 import utils
 
-# NOTE: Tests below fail for shapes of size (16, 16) and tolerances 1e-4.
+# NOTE: Tests below fail for tolerance 1e-4 (error executing plan).
 
 DTYPES = [np.float32, np.float64]
-SHAPES = [(64,), (64, 64), (64, 64, 64)]
+SHAPES = [(16,), (16, 16), (16, 16, 16)]
 MS = [256, 1024, 4096]
 TOLS = [1e-2, 1e-3]
 OUTPUT_ARGS = [False, True]

diff --git a/python/cufinufft/tests/test_simple.py b/python/cufinufft/tests/test_simple.py
@@ -10,7 +10,7 @@
 import utils
 
 DTYPES = [np.float32, np.float64]
-SHAPES = [(64,), (64, 64), (64, 64, 64)]
+SHAPES = [(16,), (16, 16), (16, 16, 16)]
 N_TRANS = [(), (1,), (2,)]
 MS = [256, 1024, 4096]
 TOLS = [1e-2, 1e-3]