Update lu_factor tests by using dpnp.allclose

Author: vlad-perevezentsev

dpnp/tests/test_linalg.py

@@ -1911,7 +1911,7 @@ def test_lu_factor(self, shape, order, dtype):
         A_cast = a_dp.astype(LU.dtype, copy=False)
         PA = self._apply_pivots_rows(A_cast, piv)
 
-        assert_allclose(LU, PA, rtol=1e-6, atol=1e-6)
+        assert dpnp.allclose(LU, PA, rtol=1e-6, atol=1e-6)
 
     @pytest.mark.parametrize("dtype", get_float_complex_dtypes())
     def test_overwrite_inplace(self, dtype):
@@ -1928,7 +1928,7 @@ def test_overwrite_inplace(self, dtype):
         PA = self._apply_pivots_rows(a_dp_orig, piv)
         LU = L @ U
 
-        assert_allclose(LU, PA, rtol=1e-6, atol=1e-6)
+        assert dpnp.allclose(LU, PA, rtol=1e-6, atol=1e-6)
 
     @pytest.mark.parametrize("dtype", get_float_complex_dtypes())
     def test_overwrite_copy(self, dtype):
@@ -1945,7 +1945,7 @@ def test_overwrite_copy(self, dtype):
         PA = self._apply_pivots_rows(a_dp_orig, piv)
         LU = L @ U
 
-        assert_allclose(LU, PA, rtol=1e-6, atol=1e-6)
+        assert dpnp.allclose(LU, PA, rtol=1e-6, atol=1e-6)
 
     def test_overwrite_copy_special(self):
         # F-contig but dtype != res_type
@@ -1972,7 +1972,7 @@ def test_overwrite_copy_special(self):
                 a_orig.astype(L.dtype, copy=False), piv
             )
             LU = L @ U
-            assert_allclose(LU, PA, rtol=1e-6, atol=1e-6)
+            assert dpnp.allclose(LU, PA, rtol=1e-6, atol=1e-6)
 
     @pytest.mark.parametrize("shape", [(0, 0), (0, 2), (2, 0)])
     def test_empty_inputs(self, shape):
@@ -2003,7 +2003,7 @@ def test_strided(self, sl):
         PA = self._apply_pivots_rows(a_dp, piv)
         LU = L @ U
 
-        assert_allclose(LU, PA, rtol=1e-6, atol=1e-6)
+        assert dpnp.allclose(LU, PA, rtol=1e-6, atol=1e-6)
 
     def test_singular_matrix(self):
         a_dp = dpnp.array([[1.0, 2.0], [2.0, 4.0]])
@@ -2070,7 +2070,7 @@ def test_lu_factor_batched(self, shape, order, dtype):
             L, U = self._split_lu(lu_3d[i], m, n)
             A_cast = a_3d[i].astype(L.dtype, copy=False)
             PA = self._apply_pivots_rows(A_cast, piv_2d[i])
-            assert_allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
+            assert dpnp.allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
 
     @pytest.mark.parametrize("dtype", get_float_complex_dtypes())
     @pytest.mark.parametrize("order", ["C", "F"])
@@ -2082,7 +2082,7 @@ def test_overwrite(self, dtype, order):
         )
 
         assert lu is not a_dp
-        assert_allclose(a_dp, a_dp_orig)
+        assert dpnp.allclose(a_dp, a_dp_orig)
 
         m = n = 2
         lu_3d = lu.reshape((-1, m, n))
@@ -2092,7 +2092,7 @@ def test_overwrite(self, dtype, order):
             L, U = self._split_lu(lu_3d[i], m, n)
             A_cast = a_3d[i].astype(L.dtype, copy=False)
             PA = self._apply_pivots_rows(A_cast, piv_2d[i])
-            assert_allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
+            assert dpnp.allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
 
     @pytest.mark.parametrize(
         "shape", [(0, 2, 2), (2, 0, 2), (2, 2, 0), (0, 0, 0)]
@@ -2119,7 +2119,7 @@ def test_strided(self):
             PA = self._apply_pivots_rows(
                 a_stride[i].astype(L.dtype, copy=False), piv[i]
             )
-            assert_allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
+            assert dpnp.allclose(L @ U, PA, rtol=1e-6, atol=1e-6)
 
     def test_singular_matrix(self):
         a = dpnp.zeros((3, 2, 2), dtype=dpnp.default_float_type())