Add _make_nonsingular_nd_np to TestLuFactorBatched

Author: vlad-perevezentsev

dpnp/tests/test_linalg.py

@@ -2052,6 +2052,23 @@ def _apply_pivots_rows(A_dp, piv_dp):
         rows = dpnp.asarray(rows)
         return A_dp[rows]
 
+    @staticmethod
+    def _make_nonsingular_nd_np(shape, dtype, order):
+        A = generate_random_numpy_array(shape, dtype, order)
+        m, n = shape[-2], shape[-1]
+        k = min(m, n)
+        A3 = A.reshape((-1, m, n))
+        for B in A3:
+            for i in range(k):
+                off = numpy.sum(numpy.abs(B[i, :n])) - numpy.abs(B[i, i])
+                B[i, i] = A.dtype.type(off + 1.0)
+
+        A = A3.reshape(shape)
+        # A3.reshape returns an array in C order by default
+        if order != "C":
+            A = numpy.array(A, order=order)
+        return A
+
     @staticmethod
     def _split_lu(lu, m, n):
         L = dpnp.tril(lu, k=-1)
@@ -2068,7 +2085,7 @@ def _split_lu(lu, m, n):
     @pytest.mark.parametrize("order", ["C", "F"])
     @pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
     def test_lu_factor_batched(self, shape, order, dtype):
-        a_np = generate_random_numpy_array(shape, dtype, order)
+        a_np = self._make_nonsingular_nd_np(shape, dtype, order)
         a_dp = dpnp.array(a_np, order=order)
 
         lu, piv = dpnp.linalg.lu_factor(
@@ -2092,7 +2109,8 @@ def test_lu_factor_batched(self, shape, order, dtype):
     @pytest.mark.parametrize("dtype", get_float_complex_dtypes())
     @pytest.mark.parametrize("order", ["C", "F"])
     def test_overwrite(self, dtype, order):
-        a_dp = dpnp.arange(2 * 2 * 3, dtype=dtype).reshape(3, 2, 2, order=order)
+        a_np = self._make_nonsingular_nd_np((3, 2, 2), dtype, order)
+        a_dp = dpnp.array(a_np, order=order)
         a_dp_orig = a_dp.copy()
         lu, piv = dpnp.linalg.lu_factor(
             a_dp, overwrite_a=True, check_finite=False
@@ -2123,13 +2141,11 @@ def test_empty_inputs(self, shape):
         assert piv.shape == (*shape[:-2], min(m, n))
 
     def test_strided(self):
-        a = (
-            dpnp.arange(5 * 3 * 3, dtype=dpnp.default_float_type()).reshape(
-                5, 3, 3, order="F"
-            )
-            + 0.1
+        a_np = self._make_nonsingular_nd_np(
+            (5, 3, 3), dpnp.default_float_type(), "F"
         )
-        a_stride = a[::2]
+        a_dp = dpnp.array(a_np, order=order)
+        a_stride = a_dp[::2]
         lu, piv = dpnp.linalg.lu_factor(a_stride, check_finite=False)
         for i in range(a_stride.shape[0]):
             L, U = self._split_lu(lu[i], 3, 3)