@@ -1747,7 +1747,8 @@ def check_qr_stacked(self, a):
17471747 I_mat = np .identity (q1 .shape [- 1 ])
17481748 stack_I_mat = np .broadcast_to (I_mat ,
17491749 q1 .shape [:- 2 ] + (q1 .shape [- 1 ],)* 2 )
1750- assert_almost_equal (matmul (swapaxes (q1 , - 1 , - 2 ).conj (), q1 ), stack_I_mat )
1750+ assert_almost_equal (matmul (swapaxes (q1 , - 1 , - 2 ).conj (), q1 ),
1751+ stack_I_mat )
17511752 assert_almost_equal (np .triu (r1 [..., :, :]), r1 )
17521753
17531754 # mode == 'r'
@@ -1756,12 +1757,12 @@ def check_qr_stacked(self, a):
17561757 assert_ (isinstance (r2 , a_type ))
17571758 assert_almost_equal (r2 , r1 )
17581759
1759- @pytest .mark .parametrize ("size" , [
1760+ @pytest .mark .parametrize ("size" , [
17601761 (3 , 4 ), (4 , 3 ), (4 , 4 ),
17611762 (3 , 0 ), (0 , 3 )])
1762- @pytest .mark .parametrize ("outer_size" , [
1763+ @pytest .mark .parametrize ("outer_size" , [
17631764 (2 , 2 ), (2 ,), (2 , 3 , 4 )])
1764- @pytest .mark .parametrize ("dt" , [
1765+ @pytest .mark .parametrize ("dt" , [
17651766 np .single , np .double ,
17661767 np .csingle , np .cdouble ])
17671768 def test_stacked_inputs (self , outer_size , size , dt ):
0 commit comments