+
    /ip                         ^ RI t ^ RIt^ RIt^ RIHt ^ RIHtHt ]P                  ]P                  .t]P                  ]P                  .t]],           tR tR t ! R R4      tR# )    N)assert_allclose)linalgsparsec                    VP                  V 4      p\        P                  ! V\        P                  4      '       d   W2P                  V 4      R ,          ,           pVP	                  V4      # )y              ?)randomnp
issubdtypecomplexfloatingastype)shapedtyperngAs   &$$ [/var/www/html/photoedit/myenv/lib/python3.14/site-packages/scipy/linalg/tests/test_batch.py
get_randomr      sK    

5A	}}UB..//

5!B&&88E?    c                    VP                  V 4      P                  V4      p\        P                  ! VP	                  RR4      4      pVP                  VP                  R7      P                  V4      V,          pWE,           V,           # )   size)r   r   r   conjswapaxesstandard_normalr   )r   r   atolr   r   Atnoises   &&&&   r   get_nearly_hermitianr      si     	

5  'A	B#	$BQWW-44U;dBE6E>r   c                     a  ] tR t^t o R^R^RRRRRRRR/R	 lt]P                  P                  R]4      R
 4       t	]P                  P                  R]4      R 4       t
]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]P                  ]P                  ]P                   ]P"                  ]P$                  ]P&                  ]P(                  ]P*                  ]P&                  ]P,                  ]P.                  ]P0                  .4      ]P                  P                  R]4      R 4       4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t]P                  P                  R]4      R 4       t ]P                  P                  R]PB                  ^3]PD                  ^3]PF                  ^3.4      ]P                  P                  R]4      R 4       4       t$]P                  P                  RRR.4      ]P                  P                  R]4      R 4       4       t%]P                  P                  R]PL                  ]PN                  ]PP                  .4      ]P                  P                  R]4      R 4       4       t)]P                  P                  R. ROO4      ]P                  P                  R]4      R 4       4       t*]P                  P                  RRNRP.RQRR..4      ]P                  P                  R]4      R  4       4       t+]P                  P                  R!RNRS.4      ]P                  P                  R"RMRT.4      ]P                  P                  R]4      R# 4       4       4       t,]P                  P                  R"RMRT.4      ]P                  P                  R]4      R$ 4       4       t-]P                  P                  R]P\                  ]P^                  .4      ]P                  P                  R]4      R% 4       4       t0]P                  P                  R&RR.4      ]P                  P                  R]4      R' 4       4       t1]P                  P                  R(RR.4      ]P                  P                  R]4      R) 4       4       t2]P                  P                  R]4      R* 4       t3]P                  P                  R+RR.4      ]P                  P                  R,]Ph                  ^3]Ph                  ^3]Pj                  ^3]Pl                  ^3.4      ]P                  P                  R]4      R- 4       4       4       t7]P                  P                  R.RR.4      ]P                  P                  R]4      R/ 4       4       t8]P                  P                  R]4      R0 4       t9]P                  P                  R]Pt                  .4      ]P                  P                  R]4      R1 4       4       t;]P                  P                  R]Px                  ^3]Pz                  ^3]P|                  ^3]P~                  ^3]P                  ^3]P                  ^3.4      ]P                  P                  R]4      R2 4       4       tB]P                  P                  R]4      R3 4       tC]P                  P                  R]4      R4 4       tD]P                  P                  R]P                  ]P                  .4      ]P                  P                  R]4      R5 4       4       tG]P                  P                  R]4      R6 4       tH]P                  P                  R]4      R7 4       tI]P                  P                  R]4      R8 4       tJ]P                  P                  R]P                  ^3]P                  ^3.4      ]P                  P                  R]M4      R9 4       4       tN]P                  P                  R:. ROO4      ]P                  P                  R]4      R; 4       4       tO]P                  P                  R:. ROO4      ]P                  P                  R]4      R< 4       4       tP]P                  P                  R=RU. RVO. RWO3.4      ]P                  P                  R:. ROO4      ]P                  P                  R]4      R> 4       4       4       tQ]P                  P                  R?RR.4      ]P                  P                  R:. ROO4      ]P                  P                  R]4      R@ 4       4       4       tR]P                  P                  R?RR.4      ]P                  P                  RA. ROO4      ]P                  P                  R]4      RB 4       4       4       tS]P                  P                  R:. ROO4      ]P                  P                  R]4      RC 4       4       tT]P                  P                  RDRR.4      ]P                  P                  R:. ROO4      ]P                  P                  R]4      RE 4       4       4       tU]P                  P                  R:. ROO4      ]P                  P                  R]4      RF 4       4       tV]P                  P                  R:. ROO4      ]P                  P                  R]4      RG 4       4       tW]P                  P                  R:. RXO4      ]P                  P                  R]4      RH 4       4       tX]P                  P                  R]4      RI 4       tYRJ tZ]P                  P                  RK]P                  ]\P                  ! RY4      33]P                  ]\P                  ! RZ4      33.4      RL 4       t_RMt`V taR# )[	TestBatchcore_dimn_outkwargsNr   	broadcastTcheck_kwargsc          
       aa Vf   / MTp\        \        P                  ! V4      P                  P	                  4       4      p	\        V\        4      '       g   V3MTpV! V/ VB p
V'       dP   V! R/ \        \        W4      4      BVB p\        W4       F$  w  r\        P                  P                  W4       K&  	  V^8X  d   V
3MT
pV'       d   \        P                  ! V!  pV^ ,          P                  R V)  p\        V^ ,          4       F  o\        V^,          4       F  oVV3R lV 4       pV! V/ VB pV^8X  d   \        P                  ! V4      3M+\        ;QJ d    . R V 4       F  NK  	  5M! R V 4       4      p\        V4       Fd  p\!        VV,          SS3,          VV,          4       \        P                  ! VV,          SS3,          4      VV,          P                  8X  d   Kd  Q h	  K  	  K  	  \        \#        X4      4       F6  pVf   VV,          P$                  MTpVV,          P$                  V8X  d   K6  Q h	  V
# )Nc              3   8   <"   T F  qSS3,          x  K  	  R # 5iN ).0arrayijs   & r   	<genexpr>'TestBatch.batch_test.<locals>.<genexpr>?   s     =fU1a4[[fs   c              3   N   "   T F  p\         P                  ! V4      x  K  	  R # 5ir)   )r   asarray)r+   refks   & r   r/   r0   B   s     >#$RZZ--#s   #%r*   )listinspect	signature
parameterskeys
isinstancetupledictzipr   testingassert_equalbroadcast_arraysr   ranger2   r   lenr   )selffunarraysr"   r#   r$   r   r%   r&   r7   res2res1out1out2resbatch_shape	arrays_ijrefk	out_dtyper-   r.   s   &&&$$$$$$           @@r   
batch_testTestBatch.batch_test    s    ~6'++C0;;@@BC
",VU";";& F%f%Ac*56A&AD!$o


''3 . !tg ((&1FQioojy1{1~&A;q>*=f=	9//-2aZ

3)u>#>uu>#>> uA#CF1a4L#a&988CF1a4L1SV\\AAA & + ' s3xA(-A5Iq6<<9,,, ! r   c                    \         P                  P                  R 4      pVP                  R4      P                  V4      pV P	                  \
        P                  V4       R# )   TAH4fCN         rV   )r   r   default_rngr   rO   r   	expm_condrB   r   r   r   s   &&  r   test_expm_condTestBatch.test_expm_condM   sD    ii##$:;JJ|$++E2((!,r   c                0   \         P                  P                  R 4      p\        RVRV4      pV P	                  \
        P                  V\        RR7      R7      p\         P                  ! V4      '       d   Q h\         P                  ! V4      '       g   Q hR# rR   ga2U0*3?gMbP?r   r$   NrS   )
r   r   rW   r   rO   r   issymmetricr;   allanyrB   r   r   r   rI   s   &&   r   test_issymmetricTestBatch.test_issymmetricS   i    ii##$:; udC@oof00!DdOoL66#;;vvc{{{r   c                0   \         P                  P                  R 4      p\        RVRV4      pV P	                  \
        P                  V\        RR7      R7      p\         P                  ! V4      '       d   Q h\         P                  ! V4      '       g   Q hR# r]   )
r   r   rW   r   rO   r   ishermitianr;   ra   rb   rc   s   &&   r   test_ishermitianTestBatch.test_ishermitian[   rf   r   c           	     N   \         P                  P                  R 4      pVP                  R4      P                  V4      pV P	                  \
        P                  V\        ^^R7      ^R7      p\
        P                  ! V^^4      p\         P                  P                  WE4       R# )rR   )MN)r$   r"   N)rT   rU   rV   )
r   r   rW   r   rO   r   diagsvdr;   r=   r>   rB   r   r   r   rF   rE   s   &&    r   test_diagsvdTestBatch.test_diagsvdc   su    ii##$:;JJy!((/v~~qQRST~~aA&


+r   rC   c                    \         P                  P                  R 4      p\        RW#R7      pV\        P
                  8X  d'   V^\         P                  ! ^VR7      ,          ,           pV P                  W4       R# )rR   r   r   r   NrS   )r   r   rW   r   r   sqrtmeyerO   rB   rC   r   r   r   s   &&&  r   test_matmatTestBatch.test_matmatl   sY     ii##$:;|5: &,,AbffQe,,,Ar   c                    \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V4       R# )rR   rs   NrT   rU   rV      )r   r   rW   r   rO   r   
null_spacerY   s   &&  r   test_null_spaceTestBatch.test_null_space|   s7    ii##$:;|5:))1-r   c                    \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V\        \         P                  R7      R7       R# )rR   rs   )funcr_   N   rV   rU   rU   )	r   r   rW   r   rO   r   funmr;   sinrY   s   &&  r   	test_funmTestBatch.test_funm   sB    ii##$:;|5:Qt/@Ar   c                   \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  VRR/R7      p\
        P                  ! VR4      p\         P                  P                  WE4       R# )rR   rs   tg      ?r_   Nr   )	r   r   rW   r   rO   r   fractional_matrix_powerr=   r>   ro   s   &&    r   test_fractional_matrix_power&TestBatch.test_fractional_matrix_power   se    ii##$:;|5:v==q#cS--a5


+r   c                x   \         P                  P                  R 4      p\        RWR7      pV^\         P                  ! ^4      ,          ,           pV P                  \        P                  V4      p\        P                  ! V4      p\        WE4       F$  w  rg\         P                  P                  Wg4       K&  	  R# )l	   8*k"XtlTC rs   NrS   )r   r   rW   r   rv   rO   r   logmr<   r=   r>   )rB   r   r   r   rF   rE   res1ires2is   &&      r   	test_logmTestBatch.test_logm   s}     ii##$JK|5:"&&)Ov{{A.{{1~OLEJJ##E1 ,r   c           	         \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V^\        RR7      R7       R# )rR   rs   T)return_rankr#   r$   NrS   )r   r   rW   r   rO   r   pinvr;   rY   s   &&  r   	test_pinvTestBatch.test_pinv   s@    ii##$:;|5:Qa8NOr   c                    \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V^R7       V P	                  \
        P                  V^RR/R7       R# )rR   rs   r#   separateTr   NrS   )r   r   rW   r   rO   r   matrix_balancerY   s   &&  r   test_matrix_balanceTestBatch.test_matrix_balance   sZ    ii##$:;|5:--q:--q:dBSTr   c           	     R   \         P                  P                  R 4      p\        RWR7      p\         P                  ! \        R^4       Uu. uF  p\         P                  ! W44      NK  	  up4      P                  R4      pV P                  \        P                  V^R7       R# u upi )rR   rs   r   N)rV   rV   r   rU   rV   rV   )r   r   rW   r   r2   r@   triureshaperO   r   	bandwidth)rB   r   r   r   rM   s   &&   r   test_bandwidthTestBatch.test_bandwidth   sv    ii##$:;vU4JJuR|<|!|<=EElS((!15 =s   B$	fun_n_outc                    \         P                  P                  R 4      pVw  rE\        RV^ V4      pV^\         P                  ! ^VR7      ,          ,           pV P                  WFVR7       R# )rR   rt   r   NrS   )r   r   rW   r   rv   rO   )rB   r   r   r   rC   r#   r   s   &&&    r   test_ldl_choleskyTestBatch.test_ldl_cholesky   sZ     ii##$:;
 ua="&&%(((e,r   
compute_uvFc           	         \         P                  P                  R 4      p\        RW#R7      pV'       d   ^M^pV P	                  \
        P                  WE\        VR7      R7       R# )rR   rs   )r   r   NrT   rU   r   rV   )r   r   rW   r   rO   r   svdr;   )rB   r   r   r   r   r#   s   &&&   r   test_svdTestBatch.test_svd   sI     ii##$:;|5:Q

A4:;VWr   c                    \         P                  P                  R 4      p\        RW#R7      pV P	                  W^R7       R# )rR   rs   r   Nr   r   r   rW   r   rO   rw   s   &&&  r   test_polar_qr_rqTestBatch.test_polar_qr_rq   5     ii##$:;|5:a(r   cdimc                   \         P                  P                  R 4      p\        RW#R7      p\        WVR7      p\        P
                  ! WERR7      p\        P                  ! V4      w  rxWu,          p	V\         P                  \         P                  09   d   RMRp
\        V^ ,          WR7       \        V^,          WR7       R# )	rR   rs   left)modeư>-q=r^   Nr   rU   rT   rT   )
r   r   rW   r   r   qr_multiplyqrfloat32	complex64r   )rB   r   r   r   r   crI   qrrL   r   s   &&&        r   test_qr_multiplyTestBatch.test_qr_multiply   s     ii##$:;|5:tc2  F3yy|eR\\ ::tA/A-r   uvdimc                <   \         P                  P                  R 4      pVw  rE\        RW#R7      p\        WBVR7      p\        WRVR7      p\        P
                  ! V4      w  r\        P                  ! WWx4      p\        ^4       F  pW,          W,          rVP                  ^8X  d   Wx3MW|,          W,          3w  pp\        P                  ! WVV4      p\        V^ ,          V,          V^ ,          4       \        V^,          V,          V^,          4       K  	  R# )rR   rs   N)rV   rT   rU   )
r   r   rW   r   r   r   	qr_updater@   ndimr   )rB   r   r   r   udimvdimr   uvr   r   rI   r-   qiriuiviref_is   &&&               r   test_qr_updateTestBatch.test_qr_update   s     ii##$:;
y7tc2tc2yy|qQ*qAT14 vv{aVqtFB$$RR4ECF1IuQx0CF1IuQx0 r   r   kdimc                ^   \         P                  P                  R 4      p\        RW4R7      p\        WVR7      pVP	                  ^ ^VR7      p\
        P                  ! V4      w  r\
        P                  ! WWg4      p
\        ^4       F  pW,          W,          rVP                  ^ 8X  d   TMW{,          pVP                  ^8X  d   TMWk,          p\
        P                  ! WW4      p\        V
^ ,          V,          V^ ,          4       \        V
^,          V,          V^,          4       K  	  R# rR   rs   r   N)rV   rT   rT   )r   r   rW   r   integersr   r   	qr_insertr@   r   r   )rB   r   r   r   r   r   r   rM   r   r   rI   r-   r   r   kir   r   s   &&&&             r   test_qr_insertTestBatch.test_qr_insert   s     ii##$:;y7tc2LLADL)yy|qQ*qAT14ffkqtBffkqtB$$RR4ECF1IuQx0CF1IuQx0 r   c                   \         P                  P                  R 4      p\        RW#R7      pVP	                  ^ ^VR7      p\
        P                  ! V4      w  rg\
        P                  ! WgV4      p\        ^4       F  p	Wi,          Wy,          rVP                  ^ 8X  d   TMWY,          p\
        P                  ! WV4      p\        V^ ,          V	,          V^ ,          4       \        V^,          V	,          V^,          4       K  	  R# r   )r   r   rW   r   r   r   r   	qr_deleter@   r   r   )rB   r   r   r   r   rM   r   r   rI   r-   r   r   r   r   s   &&&           r   test_qr_deleteTestBatch.test_qr_delete   s     ii##$:;y7LLADL)yy|qQ'qAT14ffkqtB$$RR0ECF1IuQx0CF1IuQx0 r   c                    \         P                  P                  R 4      p\        RW#R7      pV P	                  W^R7       R# )rR   rs   r   NrS   r   rw   s   &&&  r   test_schur_luTestBatch.test_schur_lu	  r   r   calc_qc           	         \         P                  P                  R 4      p\        RW#R7      pV'       d   ^M^pV P	                  \
        P                  WE\        VR7      R7       R# )rR   rs   )r   r   NrS   )r   r   rW   r   rO   r   
hessenbergr;   )rB   r   r   r   r   r#   s   &&&   r   test_hessenbergTestBatch.test_hessenberg  sK     ii##$:;|5:))1$fBUVr   eigvals_onlyc           	         \         P                  P                  R 4      p\        RW#R7      pV'       d   ^M^pV P	                  \
        P                  WE\        VR7      R7       R# )rR   rs   r   r   NrS   )r   r   rW   r   rO   r   
eig_bandedr;   )rB   r   r   r   r   r#   s   &&&   r   test_eig_bandedTestBatch.test_eig_banded  sR     ii##$:;|5:!q))1#> 	 	@r   c                    \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V4       R# )rR   rs   NrS   )r   r   rW   r   rO   r   eigvals_bandedrY   s   &&  r   test_eigvals_bandedTestBatch.test_eigvals_banded!  s7    ii##$:;|5:--q1r   two_in
fun_n_noutc                   \         P                  P                  R 4      pVw  rV\        RV^ V4      p\        RV^ V4      pV^\         P                  ! ^4      P                  V4      ,          ,           pV'       d   Wx3MV3p	V^8X  d"   V\        P                  8X  d   \        RR7      M/ p
V P                  WYWjR7       R# )rR   Tr   r   Nr   rU   rV   rV   r   r   rV   rV   )
r   r   rW   r   rv   r   r   eighr;   rO   )rB   r   r   r   r   rC   r#   r   Bargsr$   s   &&&&       r   	test_eighTestBatch.test_eigh'  s    
 ii##$:;
 ua= ua="&&)""5)))vaT-2aZC<L4(SU>r   compute_expmc           	         \         P                  P                  R 4      p\        RW#R7      p\        RW#R7      pV'       d   ^M^pV P	                  \
        P                  WE3V\        VR7      R7       R# )rR   rs   )r   r   Nr   r   )r   r   rW   r   rO   r   expm_frechetr;   )rB   r   r   r   r   Er#   s   &&&    r   test_expm_frechetTestBatch.test_expm_frechet5  sb     ii##$:;|5:|5:!q++aV5#> 	 	@r   c                j   \         P                  P                  R 4      p\        RWR7      p\        RWR7      pV P	                  \
        P                  W434       ^^^rvp\        ^^WV3WR7      p\        ^^WW3WR7      p\
        P                  ! W44      P                  ^^\        Wg4      38X  g   Q hR# )rR   rs   N)r   rU   rV   rU   )r   r   rV   rU   )	r   r   rW   r   rO   r   subspace_anglesr   min)rB   r   r   r   r   rl   rm   Ks   &&      r   test_subspace_anglesTestBatch.test_subspace_angles?  s    ii##$:;|5:|5:..7Qa1a|5:1a|5:%%a+11aCI5FFFFr   c                    \         P                  P                  R 4      p\        RW#R7      pV P	                  W4       R# )rR   rs   Nr   rU   rV   rT   r   rw   s   &&&  r   test_svdvalsTestBatch.test_svdvalsK  s1     ii##$:;|5:r   c                    \         P                  P                  R 4      pVw  rE\        RW#R7      p\        RW#R7      pV P	                  WFV3VR7       R# rR   rs   r   Nr   r   )rB   r   r   r   rC   r#   r   r   s   &&&     r   test_two_generic_matrix_inputs(TestBatch.test_two_generic_matrix_inputsR  sL     ii##$:;
|5:|5:V51r   c                   \         P                  P                  R 4      p^^rC\        RWR7      pVRRV1RV13,          VRRV1VR13,          VRVR1RV13,          VRVR1VR13,          3w  rgr\        P
                  ! WSV4      p
\        P
                  ! WgW34      p\        W4       F$  w  r\         P                  P                  W4       K&  	  \        ^4       Fr  p\        ^4       F`  p\        P
                  ! W^V3,          W44      p\        V
V4       F.  w  pp\         P                  P                  WV3,          V4       K0  	  Kb  	  Kt  	  R# )rR   rs   .N)r   rU   
   r  )
r   r   rW   r   r   cossinr<   r=   r>   r@   )rB   r   r   pr   Xx11x12x21x22rI   rL   res_ir   r.   rM   ref_jkref_ijks   &&                r   test_cossinTestBatch.test_cossin`  s(   ii##$:;!1~U<RaR!naRaRnQR!naQRn>#mmA!$mmSs01MLEJJ##E1 * qA1XqAw5&)#v&6NE7JJ++EQ$KA '7  r   c                    \         P                  P                  R 4      p\        RWR7      p\        RWR7      p\        RWR7      pV P	                  \
        P                  W4V34       R# )rR   rs   Nr   )r   r   rW   r   rO   r   solve_sylvester)rB   r   r   r   r   Cs   &&    r   test_sylvesterTestBatch.test_sylvesterr  sS    ii##$:;|5:|5:|5:..q	:r   c           	        \         P                  P                  R 4      p\        RW#R7      p\        RW#R7      p\	        RV^ VR7      p\	        RV^ VR7      pV^\         P
                  ! ^4      ,          ,           pV^\         P
                  ! ^4      ,          ,           pV^\         P
                  ! ^4      ,          ,           pV^\         P
                  ! ^4      ,          ,           p\         P
                  ! ^4      p\         P                  ! R4      p	V P                  WWVV34       V P                  WWVWx34       V P                  WWVWxV	34       V! WEWg4      p
V! WEWgV	R7      p\         P                  P                  W4       R# )rR   rs   r   r   r   )sNr   )rT   rT   )
r   r   rW   r   r   rv   zerosrO   r=   r   )rB   rC   r   r   abr   r   er&  rI   rL   s   &&&         r   test_areTestBatch.test_arez  s$    ii##$:;|5:|5: UL UL"&&)O"&&)O"&&)O"&&)OFF1IHHVq\*q_-qQ/0!o!"


""3,r   c                    \         P                  P                  R 4      p\        RWR7      p\        P
                  ! V4      w  rEV P                  \        P                  WE3^R7       R# r  )r   r   rW   r   r   schurrO   rsf2csf)rB   r   r   r   TZs   &&    r   test_rsf2csTestBatch.test_rsf2cs  sJ    ii##$:;|5:||Aa8r   c                    \         P                  P                  R 4      p\        RWR7      p^
VR&   V P	                  \
        P                  V4       R# )rR   rs   N)rT   rV   rU   r|   ).r   NNN)r   r   rW   r   rO   r   cholesky_banded)rB   r   r   abs   &&  r   test_cholesky_bandedTestBatch.test_cholesky_banded  s@    ii##$:;E;:..3r   c                   \         P                  P                  R 4      p\        RWR7      p\        RWR7      p\        RWR7      p\         P                  ! VR4      p\         P                  ! VR	4      p\         P                  ! VR
4      pV P                  \        P                  WgV3RRR7      p	\        P                  ! W4V4      p
\        W4       R# )rR   rs   F)r&   r%   N)r   rU   r   rU   )r   r   rU   r|   )r   r   rU   r   )r   rU   r   rU   )r   rU   rU   r|   )r   rU   rU   r   )	r   r   rW   r   broadcast_torO   r   
block_diagr   )rB   r   r   r(  r)  r   a2b2c2rL   rI   s   &&         r   test_block_diagTestBatch.test_block_diag  s    ii##$:;|5:|5:|5: __Q-__Q-__Q-oof//""+0E  C a(!r   c                    \         P                  P                  R 4      pVw  rE\        RW#R7      p\        RW#R7      pV P	                  WFV3^VRR7       R# )rR   rs   F)r"   r#   r%   N)rU   rV   rT   )rU   rV   rV   r   )rB   r   r   r   rC   r#   dr*  s   &&&     r   test_eigh_tridiagonalTestBatch.test_eigh_tridiagonal  sP    
 ii##$:;
y7y7VauNr   bdimc                   \         P                  P                  R 4      p\        RW#R7      p\        WVR7      p\        P
                  ! WE4      p\        V4      ^8X  d3   VR\         P                  3,          pVR\         P                  3,          p\        WF,          V,
          ^ RR7       \        V\         P                  P                  WE4      RR7       R# rR   rs   .>r^   gTqs*>Nr   )	r   r   rW   r   r   solverA   newaxisr   )rB   rF  r   r   r   r)  xs   &&&    r   
test_solveTestBatch.test_solve  s     ii##$:;|5:tc2LLt9>#rzz/"A#rzz/"A	140299??10t<r   c                   \         P                  P                  R 4      p\        RW#R7      p\        WVR7      p\        P
                  ! V4      p\        P                  ! We4      p\        V4      ^8X  d3   VR\         P                  3,          pVR\         P                  3,          p\        WG,          V,
          ^ RR7       \        V\         P                  P                  WE4      RR7       R# rH  )r   r   rW   r   r   	lu_factorlu_solverA   rK  r   rJ  )rB   rF  r   r   r   r)  
lu_and_pivrL  s   &&&     r   test_lu_solveTestBatch.test_lu_solve  s     ii##$:;|5:tc2%%a(
OOJ*t9>#rzz/"A#rzz/"A	140299??10t<r   l_and_uc                .   \         P                  P                  R 4      pVw  rV\        RW4R7      p\        W#VR7      p\        P
                  ! WV3Wx4      p	\        ^4       F  p
\        ^4       F  p\        V4      ^8:  d   TM	WV3,          p\         P                  ! V4      ^ 8X  d   TMW[,          p\         P                  ! V4      ^ 8X  d   TMWk,          p\        P
                  ! W3WzV3,          V4      p\        WV3,          V4       K  	  K  	  R# )rR   rs   Nr   rU   rU   rT   )
r   r   rW   r   r   solve_bandedr@   rA   r   r   )rB   rU  rF  r   r   lr   r7  r)  rL  r-   r.   bijljujxijs   &&&&            r   test_solve_bandedTestBatch.test_solve_banded  s     ii##$:;E;tc2.qA1Xt9>aqAw''!*/Qqt''!*/Qqt))2(B!tHcBQ$-  r   
separate_rc                   \         P                  P                  R 4      p\        RW4R7      p\        RW4R7      pV'       d   WV3MTp\        W#VR7      p\        P
                  ! Wx4      p	\        ^4       F  p
\        ^4       Fo  p\        V4      ^8:  d   TM	WV3,          pV'       d   WZV3,          WjV3,          3M	WZV3,          p\        P
                  ! W4      p\        WV3,          V4       Kq  	  K  	  R# )rR   rs   Nr   rU   rT   )	r   r   rW   r   r   solve_toeplitzr@   rA   r   )rB   r`  rF  r   r   r   r   c_or_crr)  rL  r-   r.   rZ  
c_or_cr_ijr]  s   &&&&           r   test_solve_toeplitzTestBatch.test_solve_toeplitz  s     ii##$:;y7y7&1&Atc2!!'-qA1Xt9>aqAw3=a1gqAw/1T7
++J<Q$-	  r   xdimc                   \         P                  P                  R 4      p\        RW4R7      p\        RW4R7      pV'       d   WV3MTp\        W#VR7      p\        P
                  ! Wx4      p	V'       d   \        P                  ! WV4      V,          p
M\        P                  ! V4      V,          p
V\         P                  \         P                  09   d   RMRp\        WVR7       R# )rR   rs   r   r   r^   Nrb  )
r   r   rW   r   r   matmul_toeplitztoeplitzr   r   r   )rB   r`  rh  r   r   r   r   rd  rL  rI   rL   r   s   &&&&        r   test_matmul_toeplitzTestBatch.test_matmul_toeplitz  s     ii##$:;y7y7&1&Atc2$$W0//!'!+C//!$q(CR\\ ::tt,r   c                ,   \         P                  P                  R 4      p\        RV^ VR7      pV^\         P                  ! ^4      ,          ,           p\
        P                  ! V4      p\        WVR7      p\
        P                  ! WV4      p\        V4      ^8X  d3   VR\         P                  3,          pVR\         P                  3,          p\        WG,          V,
          ^ RR7       \        V\         P
                  P                  WF4      RR7       R# )	rR   r%  rs   .r   r^   rI  Nr   )r   r   rW   r   rv   r   
cho_factorr   	cho_solverA   rK  r   rJ  )rB   rF  r   r   r   c_and_lowerr)  rL  s   &&&     r   test_cho_solveTestBatch.test_cho_solve  s     ii##$:; UL"&&)O''*tc2[,t9>#rzz/"A#rzz/"A	140299??10t<r   lowerc                   \         P                  P                  R 4      p\        RW4R7      pV'       d   ^ MRp^
VRRV3&   \        P
                  ! WQR7      p\        W#VR7      p\        P                  ! Wq3V4      p	\        ^4       Ff  p
\        ^4       FT  p\        V4      ^8:  d   TM	WV3,          p\        P                  ! WzV3,          V3V4      p\        WV3,          V4       KV  	  Kh  	  R# )rR   rs   r5  rt  NrW  r   )
r   r   rW   r   r   r6  cho_solve_bandedr@   rA   r   )rB   rt  rF  r   r   r   row_diagcbr)  rL  r-   r.   rZ  r]  s   &&&&          r   test_cho_solve_bandedTestBatch.test_cho_solve_banded  s     ii##$:;|5:12!Q.##A3tc2##RK3qA1Xt9>aqAw--rQ$x.?EQ$-  r   c                   \         P                  P                  R 4      p\        RW#R7      p^
VR&   \        WVR7      p\        P
                  ! WE4      p\        ^4       Fd  p\        ^4       FR  p\        V4      ^8:  d   TM	WWV3,          p	\        P
                  ! WGV3,          V	4      p
\        WgV3,          V
4       KT  	  Kf  	  R# )rR   rs   NrW  )r5  r5  r   )	r   r   rW   r   r   solveh_bandedr@   rA   r   )rB   rF  r   r   r   r)  rL  r-   r.   rZ  r]  s   &&&        r   test_solveh_bandedTestBatch.test_solveh_banded+  s     ii##$:;|5:(tc2  &qA1Xt9>aqAw**1T7C8Q$-  r   c                D   \         P                  P                  R 4      p\        R	W#R7      p\         P                  ! V4      p\        WVR7      p\
        P                  ! WERR7      p\        V4      ^8X  d3   VR\         P                  3,          pVR\         P                  3,          pV\         P                  \         P                  39   d   RMRp\        WF,          V,
          ^ VR7       \        V\         P
                  P                  WE4      ^V,          R7       R# )
rR   rs   Trv  .g|=g-C6*?r^   Nr   )r   r   rW   r   trilr   solve_triangularrA   rK  
complex128float64r   rJ  )rB   rF  r   r   r   r)  rL  r   s   &&&     r   test_solve_triangularTestBatch.test_solve_triangular9  s     ii##$:;|5:GGAJtc2##A5t9>#rzz/"A#rzz/"A"--!<<u$	140299??10qv>r   c                   \         P                  P                  R 4      p\        RW#R7      p\        WVR7      p\        P
                  ! WE4      pV^ ,          p\        V4      ^8X  d3   VR\         P                  3,          pVR\         P                  3,          p\        WG,          V,
          ^ RR7       \        V4      ^8X  g   Q hR# )rR   rs   .rI  r^   Nr
  )	r   r   rW   r   r   lstsqrA   rK  r   )rB   rF  r   r   r   r)  rI   rL  s   &&&     r   
test_lstsqTestBatch.test_lstsqH  s     ii##$:;|5:tc2ll1 Ft9>#rzz/"A#rzz/"A	1403x1}}r   c           	         \         P                  P                  R 4      p\        RWR7      pV P	                  \
        P                  V\        ^RR7      R7       R# )rR   rs   i sketch_sizer   r_   Nr{   )r   r   rW   r   rO   r   clarkson_woodruff_transformr;   rY   s   &&  r    test_clarkson_woodruff_transform*TestBatch.test_clarkson_woodruff_transformV  sI    ii##$:;|5:::A#v> 	 	@r   c                `   \         P                  P                  R 4      p\        R\         P                  VR7      p\
        P                  ! V4      pRp\        P                  ! \        VR7      ;_uu_ 4        \        P                  ! V^VR7       RRR4       R#   + '       g   i     R# ; i)rR   rs   z1Batch support for sparse arrays is not available.matchr  Nr{   )r   r   rW   r   r  r   	coo_arraypytestraisesNotImplementedErrorr   r  )rB   r   r   messages   &   r   'test_clarkson_woodruff_transform_sparse1TestBatch.test_clarkson_woodruff_transform_sparse]  sq    ii##$:;|2::3?QE]].g>>..qaSI ?>>>s   8BB-	zf, argsc                    R p\         P                  ! \        VR7      ;_uu_ 4        V! V!   RRR4       R#   + '       g   i     R# ; i)z#does not support zero-size batches.r  N)r  r  
ValueError)rB   fr   r  s   &&& r   test_zero_size_batchTestBatch.test_zero_size_batche  s0    
 8]]:W55tH 6555s	   8A		r*   rT   )r  )rT   rV   )r   rU   rT   rV   )rU   )rV   rT   r   )rV   rU   r   )rV   rU   rT   rV   )r   r   )r   r   r   )r   r   r   )r  )rV   rU   )r   rU   rV   rU   )r   rV   )rU   r   rT   rT   )b__name__
__module____qualname____firstlineno__rO   r  markparametrizefloatingrZ   rd   ri   rp   r   invru   signmsinmcosmtanhmsinhmcoshmr   pinvhorthrx   r~   r   r   r   r   r   r   choleskyldlro  r   r   polarr   rqr   r   r   r   r   r.  rP  r   r   r   r   r   eigvalsheigvalsr   r  r  svdvalsr  orthogonal_procrustes
khatri_raosolve_continuous_lyapunovsolve_discrete_lyapunovqzordqzr  r  r"  solve_continuous_aresolve_discrete_arer+  r2  r8  r@  eigh_tridiagonaleigvalsh_tridiagonalreal_floatingrD  rM  rS  r^  rf  rl  rr  rz  r~  r  r  r  r  rk  r   oneseigr  __static_attributes____classdictcell__)__classdict__s   @r   r!   r!      s[    +! +1 +T +QU +!+04+Z [[Wh/- 0-
 [[Wh/ 0 [[Wh/ 0 [[Wh/, 0, [[UVZZv||%+[[&++v||%+\\6<<%+[[&,,%M N [[Wh/	  0	N
	  [[Wh/. 0.
 [[Wh/B 0B
 [[Wh/, 0, [[Wh/	2 0	2 [[Wh/P 0P
 [[Wh/U 0U [[Wh/6 06 [[[FOOQ+?&**a,2,=,=q+A+C D[[Wh/- 0D- [[\E4=9[[Wh/X 0 :X [[UV\\699fii$HI[[Wh/) 0 J)
 [[V%AB[[Wh/	. 0 C	. [[Wd|i5K&LM[[Wh/1 0 N1 [[VdI%67[[Vb$Z0[[Wh/1 0 1 81 [[Vb$Z0[[Wh/1 0 11 [[UV\\63C3C$DE[[Wh/) 0 F)
 [[Xt}5[[Wh/W 0 6W [[^eT];[[Wh/@ 0 <@ [[Wh/2 02
 [[Xt}5[[\V[[!,<v{{A>N-3__a,@6>>STBU,W X[[Wh/? 0X 6? [[^eT];[[Wh/@ 0 <@ [[Wh/	G 0	G [[UV^^$45[[Wh/  0 6 
 [[[F,H,H!+L,2,=,=q+A,2,L,La+P,2,J,JA+N,2IIq>,2LL!+<+> ? [[Wh/2 0?2 [[Wh/B 0B" [[Wh/; 0; [[UV%@%@%+%>%>%@ A[[Wh/- 0A-( [[Wh/9 09 [[Wh/4 04 [[Wh/" 0"$ [[[F,C,CQ+G,2,G,G+K+M N[[Wm4O 5NO [[V%AB[[Wh/	= 0 C	= [[V%AB[[Wh/
= 0 C
= [[Y)Z1H(IJ[[V%AB[[Wh/. 0 C K. [[\E4=9[[V%AB[[Wh/. 0 C :. [[\E4=9[[V%AB[[Wh/- 0 C :- [[V%AB[[Wh/= 0 C= [[Wudm4[[V%AB[[Wh/. 0 C 5. [[V%AB[[Wh/
. 0 C
. [[V%AB[[Wh/? 0 C? [[V%AB[[Wh/
 0 C
 [[Wh/@ 0@J [[Y	2776?,-	bggl+-.) 	r   r!   )r5   r  numpyr   numpy.testingr   scipyr   r   r   r  r  r   r  complex_floatingr  r   r   r!   r*   r   r   <module>r     s\       )   RZZ(LL"--0 ++O	 O	r   