diff options
Diffstat (limited to 'silx/math/fft/test/test_fft.py')
| -rw-r--r-- | silx/math/fft/test/test_fft.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/silx/math/fft/test/test_fft.py b/silx/math/fft/test/test_fft.py index 14b1243..9ef2fd2 100644 --- a/silx/math/fft/test/test_fft.py +++ b/silx/math/fft/test/test_fft.py @@ -156,6 +156,9 @@ class TestFFT(ParametricTestCase): tol = self.tol[np.dtype(input_data.dtype)] if trdim == "3D": tol *= 10 # Error is relatively high in high dimensions + # It seems that cuda has problems with C2D batched 1D + if trdim == "batched_1D" and backend == "cuda" and mode == "C2C": + tol *= 10 # Python < 3.5 does not want to mix **extra_args with existing kwargs fft_args = { @@ -177,9 +180,10 @@ class TestFFT(ParametricTestCase): res = F.fft(input_data) res_np = F_np.fft(input_data) mae = self.calc_mae(res, res_np) + all_close = np.allclose(res, res_np, atol=tol, rtol=tol), self.assertTrue( - mae < np.abs(input_data.max()) * tol, - "FFT %s:%s, MAE(%s, numpy) = %f" % (mode, trdim, backend, mae) + all_close, + "FFT %s:%s, MAE(%s, numpy) = %f (tol = %.2e)" % (mode, trdim, backend, mae, tol) ) # Inverse FFT |