@@ -2320,19 +2320,15 @@ def test_broadcast_rhs(self, b_shape):
2320
2320
2321
2321
assert dpnp .allclose (a_dp @ x , b_dp , rtol = 1e-5 , atol = 1e-5 )
2322
2322
2323
- @pytest .mark .parametrize ("shape" , [(0 , 0 ), (0 , 5 ), (5 , 5 )])
2324
- @pytest .mark .parametrize ("rhs_cols" , [None , 0 , 3 ])
2325
- def test_empty_shapes (self , shape , rhs_cols ):
2326
- a_dp = dpnp .empty (shape , dtype = dpnp .default_float_type (), order = "F" )
2327
- if min (shape ) > 0 :
2328
- for i in range (min (shape )):
2323
+ @pytest .mark .parametrize ("a_shape" , [(0 , 0 ), (5 , 5 )])
2324
+ @pytest .mark .parametrize ("b_shape" , [(0 ,), (0 , 0 ), (0 , 5 )])
2325
+ def test_empty_shapes (self , a_shape , b_shape ):
2326
+ a_dp = dpnp .empty (a_shape , dtype = dpnp .default_float_type (), order = "F" )
2327
+ n = a_shape [0 ]
2328
+
2329
+ if n > 0 :
2330
+ for i in range (n ):
2329
2331
a_dp [i , i ] = a_dp .dtype .type (1.0 )
2330
-
2331
- n = shape [0 ]
2332
- if rhs_cols is None :
2333
- b_shape = (n ,)
2334
- else :
2335
- b_shape = (n , rhs_cols )
2336
2332
b_dp = dpnp .empty (b_shape , dtype = dpnp .default_float_type (), order = "F" )
2337
2333
2338
2334
lu , piv = dpnp .linalg .lu_factor (a_dp , check_finite = False )
@@ -2537,7 +2533,7 @@ def test_diff_type(self, dtype_a, dtype_b, b_shape):
2537
2533
((0 , 0 , 0 ), (0 , 0 )),
2538
2534
],
2539
2535
)
2540
- def test_empty_inputs (self , a_shape , b_shape ):
2536
+ def test_empty_shapes (self , a_shape , b_shape ):
2541
2537
a = dpnp .empty (a_shape , dtype = dpnp .default_float_type (), order = "F" )
2542
2538
b = dpnp .empty (b_shape , dtype = dpnp .default_float_type (), order = "F" )
2543
2539
0 commit comments