Cover case b.ndim==0

vlad-perevezentsev · vlad-perevezentsev · commit c1cd0f6dd13d · 2024-11-27T06:23:29.000-08:00
diff --git a/dpnp/tests/test_linalg.py b/dpnp/tests/test_linalg.py
@@ -2830,6 +2830,10 @@ def test_solve_errors(self):
             inp.linalg.LinAlgError, inp.linalg.solve, a_dp_ndim_1, b_dp
         )
 
+        # b.ndim == 0
+        b_dp_ndim_0 = inp.array(2)
+        assert_raises(ValueError, inp.linalg.solve, a_dp, b_dp_ndim_0)
+
 
 class TestSlogdet:
     @pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))

Original file line number	Diff line number	Diff line change
`@@ -2830,6 +2830,10 @@ def test_solve_errors(self):`
`2830`	`2830`	`inp.linalg.LinAlgError, inp.linalg.solve, a_dp_ndim_1, b_dp`
`2831`	`2831`	`)`
`2832`	`2832`
	`2833`	`+ # b.ndim == 0`
	`2834`	`+ b_dp_ndim_0 = inp.array(2)`
	`2835`	`+ assert_raises(ValueError, inp.linalg.solve, a_dp, b_dp_ndim_0)`
	`2836`	`+`
`2833`	`2837`
`2834`	`2838`	`class TestSlogdet:`
`2835`	`2839`	`@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))`