Replace "np.testing.assert_almost_equal" with "np.testing.assert_arra…

…y_almost_equal" for "Cupy".
colour-science · Jul 3, 2022 · ae18e37 · ae18e37
1 parent 04b2991
commit ae18e37
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/colour_demosaicing/bayer/demosaicing/tests/test_bilinear.py b/colour_demosaicing/bayer/demosaicing/tests/test_bilinear.py
@@ -52,7 +52,7 @@ def test_demosaicing_CFA_Bayer_bilinear(self):
                 BAYER_DIRECTORY, f"Lighthouse_Bilinear_{pattern}.exr"
             )
 
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 demosaicing_CFA_Bayer_bilinear(
                     read_image(str(CFA))[..., 0], pattern
                 ),

diff --git a/colour_demosaicing/bayer/demosaicing/tests/test_malvar2004.py b/colour_demosaicing/bayer/demosaicing/tests/test_malvar2004.py
@@ -52,7 +52,7 @@ def test_demosaicing_CFA_Bayer_Malvar2004(self):
                 BAYER_DIRECTORY, f"Lighthouse_Malvar2004_{pattern}.exr"
             )
 
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 demosaicing_CFA_Bayer_Malvar2004(
                     read_image(str(CFA))[..., 0], pattern
                 ),

diff --git a/colour_demosaicing/bayer/demosaicing/tests/test_menon2007.py b/colour_demosaicing/bayer/demosaicing/tests/test_menon2007.py
@@ -52,7 +52,7 @@ def test_demosaicing_CFA_Bayer_Menon2007(self):
                 BAYER_DIRECTORY, f"Lighthouse_Menon2007_{pattern}.exr"
             )
 
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 demosaicing_CFA_Bayer_Menon2007(
                     read_image(str(CFA))[..., 0], pattern
                 ),
@@ -63,7 +63,7 @@ def test_demosaicing_CFA_Bayer_Menon2007(self):
             RGB = os.path.join(
                 BAYER_DIRECTORY, f"Lighthouse_Menon2007_NR_{pattern}.exr"
             )
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 demosaicing_CFA_Bayer_Menon2007(
                     read_image(str(CFA))[..., 0],
                     pattern,

diff --git a/colour_demosaicing/bayer/tests/test_masks.py b/colour_demosaicing/bayer/tests/test_masks.py
@@ -44,7 +44,7 @@ def test_masks_CFA_Bayer(self):
 
         for pattern in ("RGGB", "BGGR", "GRBG", "GBRG"):
             mask = os.path.join(BAYER_DIRECTORY, f"{pattern}_Masks.exr")
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 tstack(masks_CFA_Bayer((8, 8), pattern)),
                 read_image(str(mask)),
                 decimal=7,

diff --git a/colour_demosaicing/bayer/tests/test_mosaicing.py b/colour_demosaicing/bayer/tests/test_mosaicing.py
@@ -52,7 +52,7 @@ def test_mosaicing_CFA_Bayer(self):
             CFA = os.path.join(
                 BAYER_DIRECTORY, f"Lighthouse_CFA_{pattern}.exr"
             )
-            np.testing.assert_almost_equal(
+            np.testing.assert_array_almost_equal(
                 mosaicing_CFA_Bayer(image, pattern),
                 read_image(str(CFA))[..., 0],
                 decimal=7,