@@ -293,7 +293,9 @@ def test_empty(self, shape, p):
293
293
expected = numpy .linalg .cond (a , p = p )
294
294
assert_dtype_allclose (result , expected )
295
295
296
- @pytest .mark .parametrize ("dtype" , get_all_dtypes (no_bool = True ))
296
+ @pytest .mark .parametrize (
297
+ "dtype" , get_all_dtypes (no_none = True , no_bool = True )
298
+ )
297
299
@pytest .mark .parametrize (
298
300
"shape" , [(4 , 4 ), (2 , 4 , 3 , 3 )], ids = ["(4, 4)" , "(2, 4, 3, 3)" ]
299
301
)
@@ -2408,10 +2410,8 @@ def test_qr(self, dtype, shape, mode):
2408
2410
if mode in ("complete" , "reduced" ):
2409
2411
result = dpnp .linalg .qr (ia , mode )
2410
2412
dpnp_q , dpnp_r = result .Q , result .R
2411
- assert_almost_equal (
2412
- dpnp .matmul (dpnp_q , dpnp_r ),
2413
- a ,
2414
- decimal = 5 ,
2413
+ assert dpnp .allclose (
2414
+ dpnp .matmul (dpnp_q , dpnp_r ), ia , atol = 1e-05
2415
2415
)
2416
2416
else : # mode=="raw"
2417
2417
dpnp_q , dpnp_r = dpnp .linalg .qr (ia , mode )
@@ -2424,15 +2424,13 @@ def test_qr(self, dtype, shape, mode):
2424
2424
@pytest .mark .parametrize (
2425
2425
"shape" ,
2426
2426
[(32 , 32 ), (8 , 16 , 16 )],
2427
- ids = [
2428
- "(32, 32)" ,
2429
- "(8, 16, 16)" ,
2430
- ],
2427
+ ids = ["(32, 32)" , "(8, 16, 16)" ],
2431
2428
)
2432
2429
@pytest .mark .parametrize ("mode" , ["r" , "raw" , "complete" , "reduced" ])
2433
2430
def test_qr_large (self , dtype , shape , mode ):
2434
2431
a = generate_random_numpy_array (shape , dtype , seed_value = 81 )
2435
2432
ia = dpnp .array (a )
2433
+
2436
2434
if mode == "r" :
2437
2435
np_r = numpy .linalg .qr (a , mode )
2438
2436
dpnp_r = dpnp .linalg .qr (ia , mode )
@@ -2443,11 +2441,7 @@ def test_qr_large(self, dtype, shape, mode):
2443
2441
if mode in ("complete" , "reduced" ):
2444
2442
result = dpnp .linalg .qr (ia , mode )
2445
2443
dpnp_q , dpnp_r = result .Q , result .R
2446
- assert_almost_equal (
2447
- dpnp .matmul (dpnp_q , dpnp_r ),
2448
- a ,
2449
- decimal = 5 ,
2450
- )
2444
+ assert dpnp .allclose (dpnp .matmul (dpnp_q , dpnp_r ), ia , atol = 1e-5 )
2451
2445
else : # mode=="raw"
2452
2446
dpnp_q , dpnp_r = dpnp .linalg .qr (ia , mode )
2453
2447
assert_allclose (dpnp_q , np_q , atol = 1e-4 )
@@ -3169,9 +3163,7 @@ def test_test_tensorinv_errors(self):
3169
3163
class TestTensorsolve :
3170
3164
@pytest .mark .parametrize ("dtype" , get_all_dtypes ())
3171
3165
@pytest .mark .parametrize (
3172
- "axes" ,
3173
- [None , (1 ,), (2 ,)],
3174
- ids = ["None" , "(1,)" , "(2,)" ],
3166
+ "axes" , [None , (1 ,), (2 ,)], ids = ["None" , "(1,)" , "(2,)" ]
3175
3167
)
3176
3168
def test_tensorsolve_axes (self , dtype , axes ):
3177
3169
a = numpy .eye (12 ).reshape (12 , 3 , 4 ).astype (dtype )
0 commit comments