test_cwt_batch can have small differences on some arches (resolves #640…

…) (#641)
PyWavelets · Sep 12, 2022 · fbc677d · fbc677d
1 parent 3dd186c
commit fbc677d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pywt/tests/test_cwt_wavelets.py b/pywt/tests/test_cwt_wavelets.py
@@ -420,7 +420,8 @@ def test_cwt_batch(axis, method):
     assert_equal(cfs.shape[1 + axis], sst.shape[axis])
 
     # batch result on stacked input is the same as stacked 1d result
-    assert_equal(cfs, np.stack((cfs1,) * n_batch, axis=batch_axis + 1))
+    assert_almost_equal(cfs, np.stack((cfs1,) * n_batch, axis=batch_axis + 1),
+                        decimal=12)
 
 
 def test_cwt_small_scales():