From da1a1554c706508d790771a9bbdde446adb574b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joakim=20And=C3=A9n?= Date: Fri, 14 Jul 2023 13:33:56 -0500 Subject: [PATCH] tests: reduce size of test signals --- python/cufinufft/tests/test_basic.py | 4 ++-- python/cufinufft/tests/test_simple.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/cufinufft/tests/test_basic.py b/python/cufinufft/tests/test_basic.py index ea2d38af2..70bc91b00 100644 --- 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 index 5cc572d56..d7c64a75b 100644 --- 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]