Use assert_allclose

cubed-dev · Jan 20, 2025 · fd7fdf3 · fd7fdf3
1 parent 37fee5b
commit fd7fdf3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cubed/tests/test_fft.py b/cubed/tests/test_fft.py
@@ -1,5 +1,5 @@
 import pytest
-from numpy.testing import assert_array_equal
+from numpy.testing import assert_allclose
 
 import cubed
 import cubed.array_api as xp
@@ -17,7 +17,7 @@ def test_fft(funcname, n):
     a = cubed.from_array(an, chunks=(1, 10))
     b = cb_fft(a, n=n)
 
-    assert_array_equal(b.compute(), bn)
+    assert_allclose(b.compute(), bn)
 
 
 def test_fft_chunked_axis_fails():