hR<SrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK J r J r J r JrJrJrJrJrJrJrJrJrJr SSKJr SSKJr SSKJrJrJrJ r J!r! SSK"J#r# SSK$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/ SSK0r S r2\*r3SS jr*S r4S r5S r61Skr7"SS5r8Sr9\ RtRwS5 /r<\<\9"S\8"S\ "SS/SS//\S9\ "SS/\S95\8"S\ "SS/SS//\S9\ "SS/\S95\8"S\ "SS/SS//\S9\ "/SQ/SQ/\S95\8"S\ "SS /S!S"//\S9\ "S#S/\S95\8"S$\ "SS /S!S"//\S9\ "S#S/\S95\8"S%\ "SS /S!S"//\S9\ "/S&Q/S'Q/\S95\8"S(\ Rz"S)\S9\ Rz"S*\S9S+1S,9\8"S-\ RtR}S.S.5\ RtR}S.55\8"S/\ RtR}S0S05\ RtR}S055\8"S1S0S2/S3S4//S2S0/5/ 5- r<\<\9"S5\8"S6\ "/S7Q/SQ/\S9\ "SS/\S95\8"S8\ "SS/SS/S9S://\S9\ "/S;Q\S95\8"S<\ "/S7Q/SQ/\S9\ "SS/\S95\8"S=\ "SS/SS/S9S://\S9\ "/S;Q\S95\8"S>\ "/S?Q/S@Q/\S9\ "S#S/\S95\8"SA\ "SBSC/SDSE/SFSG//\S9\ "/SHQ\S95\8"SI\ "/S?Q/S@Q/\S9\ "S#S/\S95\8"SJ\ "SBSC/SDSE/SFSG//\S9\ "/SHQ\S95\8"SK\ "/S?Q/S@Q/\S9\ "S#S/SLSM//\S95\8"SN\ "SBSC/SDSE/SFSG//\S9\ "S#S/SLSM/SLSM//\S95\8"SO\ RtR}S.SP5\ RtR}S.55\8"SQ\ RtR}S0SR5\ RtR}S055\8"SS\ RtR}SRS05\ RtR}SR55\8"ST\ RtR}SS45\ RtR}S5S+1S,9\8"SU\ RtR}S4S5\ RtR}S45S+1S,9/5- r<\<\9"SV\8"SW\ "SS/SS//\S9S5\8"SX\ "SS/SS//\S9S5\8"SY\ "SS /SZS0//\S9S5\8"S[\ "SS /SZS0//\S9S5\8"S\\ Rz"S)\S9SS+1S,9\8"S]S0S2/S2S0//S5\8"S^\ "SS/SS//5S5\8"S_\ RtR}S0S05S5/5- rgdy=)rrr8s r+get_rtolr=Us !!r->squarestrided hermitian nonsquare generalizedsize-0c6\rSrSr\"54SjrSrSrSrg) LinalgCasedcz[[U[55 XlX lX0l[ U5Ulg)z A bundle of arguments to be passed to a test case, with an identifying name, the operands a and b, and a set of tags to filter the tests N)rr&strnamer2r3 frozensettags)selfrIr2r3rKs r+__init__LinalgCase.__init__es-  4%& dO r-cPU"URURURS9 g)z> Run the function `do` on this test case, expanding arguments rKN)r2r3rK)rLdos r+checkLinalgCase.checkps 466466 *r-c"SURS3$)Nz )rIrLs r+__repr__LinalgCase.__repr__vstyyk++r-)r2r3rIrKN) __name__ __module__ __qualname____firstlineno__setrMrRrW__static_attributes__r-r+rErEds(+ $+ ,r-rEcdU[;dS5eUHnURU1-UlM U$)zR Add the given tag (a string) to each of the cases (a list of LinalgCase objects) z Invalid tag)all_tagsrK)tagcasescases r+ apply_tagrezs6 (?)M)?II%  Lr-r>r?@@@r8r double_2)rhrgrj)rirj@r?@y@@y@@y@@@?r cdouble_2)rnrmy?@)y?y?y?0x0rrrrCrP8x81x1nonarrayrA single_nsq_1)rgrhri single_nsq_2g@rl)rhrgri double_nsq_1 double_nsq_2 csingle_nsq_1)??@@@)y@y@"@y@@ csingle_nsq_2rrry@"y@y@ @)rnrmr cdouble_nsq_1 cdouble_nsq_2cdouble_nsq_1_2y?y@cdouble_nsq_2_28x11 1x55x10x44x0r@hsinglehdoublehcsingley@hcdoublehempty hnonarray matrix_b_only hmatrix_1x1c/n[GHsn[UR[R5(dM/[R "URSUR-SUR-/5nUR cSnOdUR RS:Xa UR nO=[R "UR SUR -SUR -/5n[URS-X#URS1-S9nURU5 [R "UR/S-S-5RS URR-5nUR cSnOUR RS:Xaa[R "UR /S-S-URS -5RS URRS S-5nOM[R "UR /S-S-5RS UR R-5n[URS -X#URS1-S9nURU5 GMv U$) Nrxryrv_tile3rBrPryrx_tile213)CASESr&r2r'r(rr3ndimrErIrKappendreshapeshape) new_casesrdr2r3new_cases r+_make_generalized_casesr sI$&&"**--  HHdffa$&&j!dff*5 6 66>A VV[[A A$&&!dff*a$&&j9:Adii(2A#'99 #>@" HHdffX\A% & . .v /D E 66>A VV[[A $&&A)AGGBK7876DFFLL$556 $&&A)*226DFFLL3HIAdii*4a#'99 #>@"36 r-c ## [U[R5(dUS4v gS/UR-nSUS'URS:aSUS'URS:aS US '[R "[ U56GHPn[URU5VVs/sHup4[X4-5PM nnn[ S U55n[R"XPRS 9nUR[R5RS 5 XvnXS'URUR5n[![R""Xp:H55 USSR%UVs/sHnSU-PM sn5-4v URS:aZURSS:XaG['UR(5n SU S'[R*R,R/X S9nUS4v URS:aZURSS:XaG['UR(5n SU S'[R*R,R/X S9nUS4v URS:dGMURSSS:XdGM['UR(5n SU S'SU S'[R*R,R/X S9nUS4v GMS gs snnfs snf7f)z4 Generate cartesian product of strides for all axes nopNrv)rvryrrvrrx)rvrc3<# UHn[SSU5v M g7fr$)slice).0repeats r+ $_stride_comb_iter..CsGwVuT400wsr8l>[=.stride__z%+dr)strides stride_xxx_0stride_xxx_0_xrvrvstride_xxx_0_0)r&r'r(r itertoolsproducttupleziprabsemptyr0viewuint32fill __class__ralljoinlistrlib stride_tricks as_strided) x stride_setrepeatsr2r3 new_shapeslicesxijss r+_stride_comb_iterr1sd a $ $h!&&JJrNvvz# 2vvz 2$$eJ&78,/,AB,ADASZ,A BGwGGXXiww /   + Z3 WWQ[[ !rw )chh7'C7a 7'CDDDD 66Q;1772;!+QYYAAbE%%000>Bn$ $ 66Q;1772;!+QYYAAbE%%000>B&& & 66Q;1773B<61QYYAAbEAbE%%000>B&& &;9B(Ds-BK%!K:B6K%0K ?C.K%2K%AK%c /n[Hn[UR5Hcup#[UR5HEupE[ UR S-U-S-U-X$UR S1-S9nURU5 MG Me M U$)Nrr?rP)rrr2r3rErIrKr)rrdr2a_labelr3b_labelrs r+_make_strided_casesrasI+DFF3JA/7 %dii#o&?#&E&OQR+/99 {+BD  *84 r-c:\rSrSr\r\"5\"54SjrSrg)LinalgTestCaseirc2URHJnURU-U:waMURU-(aM.URUR5 ML g![a0nSU<S3nU[ R "5- n[U5UeSnAff=f)z` Run func on each of the cases with all of the tags in require, and none of the tags in exclude zIn test case: z N) TEST_CASESrKrRrQ Exception traceback format_excAssertionError)rLrequireexcluderdemsgs r+ check_casesLinalgTestCase.check_casesus OODyy7"g-yy7" 1 477#$ 1&thd3y++--$S)q0 1sA B&+BBr_N) rYrZr[r\rrr]rr^r_r-r+rrrsJ"%%1r-rc \rSrSrSrSrSrg)LinalgSquareTestCaseic*URS1SS1S9 g)Nr>rBrCrrrrVs r+ test_sq_cases"LinalgSquareTestCase.test_sq_casess! ("/!:  rCrBrrrVs r+test_empty_sq_cases(LinalgSquareTestCase.test_empty_sq_casess! (H!5"/  2r-r_N)rYrZr[r\rrr^r_r-r+rr <2r-rc \rSrSrSrSrSrg)LinalgNonsquareTestCaseic*URS1SS1S9 g)NrArBrCrrrVs r+test_nonsq_cases(LinalgNonsquareTestCase.test_nonsq_cases! +"/!:  rCrrrVs r+test_generalized_sq_cases9LinalgGeneralizedSquareTestCase.test_generalized_sq_casess! -!:"*  -r-c&UR1SkS9 g)N>r>rBrCrrrVs r+test_generalized_empty_sq_cases?LinalgGeneralizedSquareTestCase.test_generalized_empty_sq_casess !DEr-r_N) rYrZr[r\pytestmarkslowrrr^r_r-r+rrs> [[-- [[FFr-rc\rSrSr\R R S5r\R R S5rSr g)"LinalgGeneralizedNonsquareTestCaseic*URSS1S1S9 g)NrBrArCrrrVs r+test_generalized_nonsq_cases?LinalgGeneralizedNonsquareTestCase.test_generalized_nonsq_cases! -!="*  -r-c&UR1SkS9 g)N>rArBrCrrrVs r+"test_generalized_empty_nonsq_casesELinalgGeneralizedNonsquareTestCase.test_generalized_empty_nonsq_casess !GHr-r_N) rYrZr[r\r r r rrr^r_r-r+rrs> [[-- [[IIr-rc\rSrSr\R R S5r\R R S5rSr g)HermitianGeneralizedTestCaseic*URSS1S1S9 g)NrBr@rCrrrVs r+test_generalized_herm_cases8HermitianGeneralizedTestCase.test_generalized_herm_casesrr-c*UR1SkS1S9 g)N>r@rBrCnonerrrVs r+!test_generalized_empty_herm_cases>HermitianGeneralizedTestCase.test_generalized_empty_herm_casess !G"(  +r-r_N) rYrZr[r\r r r rrr^r_r-r+rrs< [[-- [[++r-rc[U5nURS:aF[R"URUR S9n[ URS5US'U$[ URS5$)Nryr8r.r)rrr'rrr0r )r2rs r+identity_like_generalizedr!s\ Avv{ HHQWWAGG ,!''"+&# ##r-c\rSrSrSrSrg) SolveCasesicD[R"X5n[R"U5RS:Xa<[ XS5Sn[ [R"X%R5U5 O[ X5n[ X%5 [[XB55 g)Nrv).N.r) r solver'rrr r broadcast_torrr,)rLr2r3rKradotxs r+rQ SolveCases.dosq LL  88A;  q 1 l+F3E ;; ? G1LE  )#A)*r-r_NrYrZr[r\rQr^r_r-r+r#r#s +r-r#cr\rSrSr\R R S\\\ \ /5S5r Sr Sr SrSrg) TestSolveir0c[R"SS/SS//US9n[[R"X"5R U5 gNrv?r8)r'rrr r&r0rLr0rs r+ test_typesTestSolve.test_typess8 HHq#ha) 7V\\!'--u5r-c"SS[R5n[R"S5RSSS5n[R"S5R U5n[ R "X#5nURS:Xde[R"S5RSS5R U5n[ R "X#5nURS:Xde[R"S5RSS5R U5n[[[ R X#5 g) Nc\rSrSrSrg))TestSolve.test_1_d..ArraySubclassir_NrYrZr[r\r^r_r-r+ ArraySubclassr5 r-r7rtrxrxrxrz)rxrxrxrv) r'r(arangerrr r&rr ValueError)rLr7r2r3results r+test_1_dTestSolve.test_1_ds BJJ  IIaL Aq ) IIaL  m ,a#||v%%% IIaL A & + +M :a#||y((( IIaL A & + +M :j&,,5r-c "SS[R5n[R"S5RSSS5n[R"S5RSSS5R U5n[ R "X#5SS2SS2SS24n[ R "USS2SS2SS24USS2SS2SS245n[XT5 [[XQ55 [[ R[ R USS2SS2SS24U5 [[[ R X#SS2SS2SS245 [R"S5RSSS5n[[[ R X#5 [[[ R USSUSS5 [R"S5R U5n[ R "X#5SS2SS24n[ R "USS2SS2SS24USS5n[XT5 [[XQ55 [R"S5RSS5n[[[ R X#5 [[[ R USSUSS5 [[[ R USS2SS2SS24U5 g) Nc\rSrSrSrg),TestSolve.test_0_size..ArraySubclassir_Nr6r_r-r+r7rAr8r-r7rtrxrrvryr) r'r(r:rrr r&rrr&r rr;rLr7r2r3expectedr<s r+ test_0_sizeTestSolve.test_0_sizes- BJJ  IIaL Aq ) IIaL Aq ) . .} =<<%a1ai0a1Q3! na1Q3 l;6, 612 f((&,,!QqS!A#+Jj&,,Q!QY<@ IIaL Aq )j&,,5j&,,!A!A? IIaL  m ,<<%a1f-a1Q3! na!f56, 612 IIaL A &j&,,5j&,,!A!A?j&,,!QqS!A#+Br-c"SS[R5n[R"S5RSSS5n[R"S5RSSS5R U5n[ R "X#5SS2SS2SS24n[ R "X#SS2SS2SS245n[XT5 [[XQ55 [ R "X#5SS2SS2SS24n[ R "USS2SS2SS24USS2SS2SS245n[XT5 [[XQ55 g) Nc\rSrSrSrg).TestSolve.test_0_size_k..ArraySubclassi"r_Nr6r_r-r+r7rH"r8r-r7rzrvrxrryr) r'r(r:rrr r&rrr&rBs r+ test_0_size_kTestSolve.test_0_size_k s BJJ  IIaL Aq ) IIaL Aq ) . .} =<<%aAaCi0a1a19.6, 612<<%a1ack2a1Q3! na1Q3! n=6, 612r-r_N)rYrZr[r\r r  parametrizerr rrr1r=rDrIr^r_r-r+r,r,s@ [[Wvvw&HI6J66"CB3r-r,c\rSrSrSrSrg)InvCasesi3c[R"U5n[[X5[ U55 [ [ XA55 gr$)r invrr r!rr,)rLr2r3rKa_invs r+rQ InvCases.do5s4 1 F1,5a8 :#E-.r-r_Nr*r_r-r+rMrM3s/r-rMcf\rSrSr\R R S\\\ \ /5S5r Sr Sr g)TestInvi<r0c[R"SS/SS//US9n[[R"U5R U5 gr.)r'rrr rOr0r0s r+r1TestInv.test_types=s7 HHq#ha) 7VZZ]((%0r-c"SS[R5n[R"S[RS9R U5n[ R "U5n[URR[RL5 [URUR5 [[X155 [R"S[RS9R U5n[ R "U5n[URR[RL5 [URUR5 [[X155 g)Nc\rSrSrSrg)*TestInv.test_0_size..ArraySubclassiDr_Nr6r_r-r+r7rXDr8r-r7rrvrvr8rq)r'r(zerosint_rr rOrr0r%float64rrr& complex64rLr7r2ress r+rDTestInv.test_0_sizeBs BJJ  HHYbgg . 3 3M Bjjm "**,-QWWcii( 3./ HHV2<< 0 5 5m Djjm ",,./QWWcii( 3./r-r_NrYrZr[r\r r rKrr rrr1rDr^r_r-r+rSrS<s4 [[Wvvw&HI1J10r-rSc\rSrSrSrSrg) EigvalsCasesiScv[R"U5n[R"U5upV[XE5 gr$)r eigvalseigr)rLr2r3rKevevaluesevectorss r+rQEigvalsCases.doUs( ^^A "JJqMB(r-r_Nr*r_r-r+rcrcSs)r-rccf\rSrSr\R R S\\\ \ /5S5r Sr Sr g) TestEigvalsi[r0c*[R"SS/SS//US9n[[R"U5R U5 [R"SS/SS//US9n[[R"U5R [ U55 gNrvr/r8r)r'rrr rer0r;r0s r+r1TestEigvals.test_types\st HHq#ha) 7V^^A&,,e4 HHq#hQ( 6V^^A&,,.?.FGr-c"SS[R5n[R"S[RS9R U5n[ R "U5n[URR[RL5 [SUR5 [[U[R55 [R"S[RS9R U5n[ R "U5n[URR[RL5 [SUR5 [[U[R55 g)Nc\rSrSrSrg).TestEigvals.test_0_size..ArraySubclassier_Nr6r_r-r+r7rrer8r-r7rYr8rrvrqrr)r'r(rZr[rr rerr0r%r\rrr&r]r^s r+rDTestEigvals.test_0_sizecs BJJ  HHYbgg . 3 3M BnnQ "**,-VSYY' 3 +, HHV2<< 0 5 5m DnnQ ",,./T399% 3 +,r-r_Nrar_r-r+rlrl[s6 [[Wvvw&HIHJH -r-rlc\rSrSrSrSrg)EigCasesivc :[R"U5nURURpe[ [ X5[ R"U5[ R"U5SSSS24-[UR5S9 [[Xa55 g)N.rtol) r rf eigenvalues eigenvectorsrr r'rr=r0rr,)rLr2r3rKr_rzr{s r+rQ EigCases.doxstjjm$'OOS5E5E\q/ <02::k3J3PTVW<3XX%k&7&78 : #L45r-r_Nr*r_r-r+rvrvvs6r-rvcf\rSrSr\R R S\\\ \ /5S5r Sr Sr g)TestEigir0c[R"SS/SS//US9n[RRU5up4[ UR U5 [ UR U5 [R"SS/SS//US9n[RRU5up4[ UR [ U55 [ UR [ U55 grn)r'rr rfrr0r;rLr0rwvs r+r1TestEig.test_typess HHq#ha) 7yy}}QQWWe$QWWe$ HHq#hQ( 6yy}}QQWW/67QWW/67r-c:"SS[R5n[R"S[RS9R U5n[ R "U5up4[URR[RL5 [URR[RL5 [URUR5 [SUR5 [[U[R55 [R"S[RS9R U5n[ R "U5up4[URR[RL5 [URR[RL5 [URUR5 [SUR5 [[U[R55 g)Nc\rSrSrSrg)*TestEig.test_0_size..ArraySubclassir_Nr6r_r-r+r7rr8r-r7rYr8rsrqrr)r'r(rZr[rr rfrr0r%r\rrr&r]rLr7r2r_res_vs r+rDTestEig.test_0_sizes< BJJ  HHYbgg . 3 3M BZZ]    BJJ./ "**,-QWWekk*VSYY' 1bjj)* HHV2<< 0 5 5m DZZ]    BLL01 ",,./QWWekk*T399% 1bjj)*r-r_Nrar_r-r+r~r~s4 [[Wvvw&HI 8J 8+r-r~cd\rSrSrSr\R RS\\ \ \ /5S5r Sr g) SVDBaseTestsiFr0c[R"SS/SS//US9n[R"U5nURUR UR pen[URU5 [UR[U55 [URU5 [R"USURS9n[UR[U55 g)Nrvr/r8F compute_uvr@) r'rr svdUSVhrr0r9r@)rLr0rr_rrrrs r+r1SVDBaseTests.test_typess HHq#ha) 7jjm55#%%bQWWe$QWWnU34RXXu% JJqUdnn EQWWnU34r-r_N)rYrZr[r\r@r r rKrr rrr1r^r_r-r+rrs4I [[Wvvw&HI5J5r-rc\rSrSrSrSrg)SVDCasesic f[R"US5upEn[U[[R "U5[R "U5SSSS24-[R "U55[ UR5S9 [[XA55 [[Xa55 g)NF.rx) r rrr r'rr=r0rr,)rLr2r3rKurvts r+rQ SVDCases.dos~::a'b6"**Q-"**Q-T1 2M"M+-::b>;%agg. 0 #A)*#B*+r-r_Nr*r_r-r+rrs,r-rc \rSrSrSrSrSrg)TestSVDic[R"S5n[R"USURS9up#n[ UR S5 [ UR S5 [ U[R"S55 [R"S5n[R"USURS9up#n[ UR S5 [ UR S5 [ U[R"S55 g) z5Empty input should put an identity matrix in u or vh )rzrTrrzrzrqrz)rrzN)r'rr rr@rreye)rLrrrvhs r+test_empty_identityTestSVD.test_empty_identitys HHV ::aDDNNKbQWWf%RXXv&Qq " HHV ::aDDNNKbQWWf%RXXv&R#r-c[R"SS/SS//5n[R"USURS9n[R "U5n[ X#5 g)Nrvr/Fr)r'rr rr@svdvalsr)rLr s_from_svds_from_svdvalss r+ test_svdvalsTestSVD.test_svdvalssI HHq#ha) *ZZet~~N *J7r-r_N)rYrZr[r\rrr^r_r-r+rrs  $8r-rc\rSrSrSrSrg)SVDHermitianCasesic R[R"USSS9upEn[U[[R "U5[R "U5SSSS24-[R "U55[ UR5S9 Sn[[R"XG"U55[R"[R"URS5UR55 [[R"Xg"U55[R"[R"URS5UR55 [[R"U5SSSS24U5 [[XA55 [[Xa55 g)NFTr@.rxc[[UR55nUSUSsUS'US'[R"[R "XS95$)Nrraxes)rrangerr'conj transpose)matrs r+r@'SVDHermitianCases.do..hermitiansHchh(D!%b48 DHd2h772<<78 8r-r)r rrr r'rr=r0rr'rrrsortrr,)rLr2r3rKrrrr@s r+rQSVDHermitianCases.dos"::a$7b6"**Q-"**Q-T1 2M"M+-::b>;%agg. 0 9 BIIa16qwwWY{H[]^]d]d8efBIIb)B-8"//"&&QSQYQYZ\Q]J^`b`h`h:ijRWWQZTrT *A.#A)*#B*+r-r_Nr*r_r-r+rrs,r-rc\rSrSrSrSrg)TestSVDHermitianiTr_N)rYrZr[r\r@r^r_r-r+rrsIr-rc\rSrSrSrSrg) CondCasesic [U5nSU;a [[[RU5 g[R "USS9n[ [R"U5USUS- SSS9 [ [R"US 5USUS- SSS9 [ [R"US 5USUS- SSS9 [RRU5n[ [R"US 5[U5RS 5RS 5[U5RS 5RS 5-SSS9 [ [R"US 5[U5RS 5RS 5[U5RS 5RS 5-SSS9 [ [R"U[R5[U5RS 5RS 5[U5RS 5RS 5-SSS9 [ [R"U[R*5[U5RS 5RS 5[U5RS 5RS 5-SSS9 [ [R"US 5[R"[U5S -RS 5RS 5[U5S -RS 5RS 5-5SSS9 g)NrCF)rr%).rrrr4r5rxrrvrfro)rr rr condrrr'rOrsummaxminr sqrt)rLr2r3rKcrcinvs r+rQ CondCases.dos AJ t  +v{{A 6  JJqU + KKNAfI' 2R 1  KK1 qy1W:5R 1  KK2 ' QvY 6R 1 yy}}Q KK1  FJJrN  r "SY]]2%6%:%:2%> >R 1  KK2  FJJrN  r "SY]]2%6%:%:2%> >R 1  KK266 " FJJrN  r "SY]]2%6%:%:2%> >R 1  KKBFF7 # FJJrN  r "SY]]2%6%:%:2%> >R 1  KK5 ! GGSVQYOOB'++B/4y!|((,0045 6R  1r-r_Nr*r_r-r+rrs(1r-rch\rSrSrSrSr\RRSSSS9S5r S r S r g ) TestCondic[/SQ/SQ/SQ/5n[[R"U[5S5 [[R"U[*5S5 [[R"US5S5 [[R"US5S5 [[R"US 5[ R "S 55 g) N)rgrrv)rgr)rrrirzgUUUUUU?rvrr/rgUUUUU6@)rrr rr r'rrLAs r+test_basic_nonsvdTestCond.test_basic_nonsvds :{J7 8FKK3/3FKKC40%8FKK1-q1FKK2.4FKK5127783DEr-cX[R"S5[R"S5/n/SQnSS/n[R"X5H'upE[ [ R"XE5S:5 M) [R"X5HupE[ R"XE5 M g)Nr9Nrvrxrrr4&k C)r'rZonesrrrr r)rLAsp_posp_negrps r+ test_singularTestCond.test_singular%shhv 0#R%%b0DA FKK%, -1%%b0DA KK 1r-TFz/Platform/LAPACK-dependent failure, see gh-18914)runreasonc/SQn/SQn[R"S5n[RUS'UH[n[R"X45n[ [ U[R55 [ [R"U55 M] [R"S5n[RUS'UHn[R"X45n[ [R"US55 XB;a$[ USS :5 [ US S :5 Md[ [R"US5(+5 [ [R"US 5(+5 M g) N)Nrvrrxrrrr9rs)ryrxrxrvrrvrvrrrx) r'rnanr rrr&r\isnan)rLpsrrrrs r+test_nanTestCond.test_nan3s )# GGFO&&$A A!A Jq"**- . BHHQK  GGI VV' A A!A BHHQqTN #z!t $!t $BHHQqTN*+BHHQqTN*+r-c[RRS5 [RRSSSS5nSUS'SUS'SHn[R "X5n[ US[R5 [ US[R5 [[R"US55 [[R"US55 M g) Nrfrxrrqr)Nrvrxrrrrsrvr) r'randomseedrandr rrr risfinite)rLrrrs r+test_stacked_singularTestCond.test_stacked_singularNs t IINN1aA &$$,A A!A 4"&& ) 4"&& ) BKK$( ) BKK$( ) -r-r_N) rYrZr[r\rrr r xfailrrr^r_r-r+rrsCF  [[t-.,.,0 *r-rc\rSrSrSrSrg) PinvCasesi^c[R"U5n[n[U"U"X5U5USSS9 [ [ XA55 g)Nrrrr pinvr rrr,rLr2r3rKa_ginvrs r+rQ PinvCases.docs<QCA2AaXZ[#F./r-r_Nr*r_r-r+rr^s 0r-rc\rSrSrSrg)TestPinvikr_Nr6r_r-r+rrkr-rc\rSrSrSrSrg)PinvHermitianCasesioc[R"USS9n[n[U"U"X5U5USSS9 [ [ XA55 g)NTrrrrrrs r+rQPinvHermitianCases.doqs?Q$/CA2AaXZ[#F./r-r_Nr*r_r-r+rros0r-rc\rSrSrSrg)TestPinvHermitianiyr_Nr6r_r-r+rryrr-rcn[R"/SQ/SQ/SQ/5n[[RR USS9[RR USS95 [ R "[SS9 [RR USSS 9 SSS5 g!,(df  g=f) Nrvrxry)rzrvrv)rxryrvr/rcondrxz%`rtol` and `rcond` can't be both set.)match)rry)r'rrr rr raisesr;)r2s r+test_pinv_rtol_argr}s )Y 23A q$ qs# B  q#.   s = B&& B4c\rSrSrSrSrg)DetCasesic[R"U5n[R"U5nURURpv[ U5R R[[4;a[ U5R[5nO[ U5R[5n[R"U5n [U[R"U SS95 [U[ R""U5-[R"U SS95 [ R$"U5n[ R$"U5nUS:gn [[ R&"Xj5S5 [)Xz)[**5 g)Nraxisrrv)r detslogdetsign logabsdetrr0r%rr astyperrerrreducer'exp atleast_1drrr ) rLr2r3rKdr_rldadrgms r+rQ DetCases.dos JJqMnnQ#--2 1:   VV$4 4""6*B""7+B ^^B Axr;<Ar NHOOBR,HI MM!  ]]2  !VBFF14L!,RVcT"r-r_Nr*r_r-r+rrs#r-rcl\rSrSrSr\R RS\\ \ \ /5S5r Sr Srg)TestDetic|[[R"S//5S5 [[[R"S//55[5 [[R"S//5S5 [[[R"S//55[ 5 [[R "S//5S[*45 [[[R "S//5S5[5 [[[R "S//5S5[5 [[R "S//5S[*45 [[[R "S//5S5[ 5 [[[R "S//5S5[5 g)Nyrrv)rr rr%r rrr rVs r+ test_zeroTestDet.test_zeros&VZZ#(#.T&**seW-.7VZZ$)3/T&**tfX./9V^^cUG,sSDk:T&..3%1!45v>T&..3%1!45v>V^^dVH-sd|<T&..4&2156@T&..4&2156?r-r0cL[R"SS/SS//US9n[[RR U5R U5 [RR U5up4[UR [U55 [UR U5 gr.)r'rrr rr0rr9)rLr0rphrs r+r1TestDet.test_typesss HHq#ha) 7RYY]]1%++U3 !!!$QWWnU34RXXu%r-c[R"S[RS9n[R"U5n[ US5 [ URR[RL5 [R"U5n[ US5 [ USRR[RL5 [ USRR[RL5 [R"S[RS9n[R"U5n[ US5 [ URR[RL5 [R"U5n[ US5 [ USRR[RL5 [ USRR[RL5 g)Nrqr8rgrrrv) r'rZr]r rrrr0r%rfloat32r\)rLr2r_s r+rDTestDet.test_0_sizes6 HHV2<< 0jjmS" ",,./nnQS&!A !!R\\12A !!RZZ/0 HHV2:: .jjmS" "**,-nnQS&!A !!RZZ/0A !!RZZ/0r-r_N)rYrZr[r\rr r rKrr rrr1rDr^r_r-r+rrs< @ [[Wvvw&HI&J&1r-rc\rSrSrSrSrg) LstsqCasesic [R"U5nURupV[R"US5upxn [R "XSS9uppUS:Xa[ U S:HR55 XV::a![U[X55 [X5 O [X5 [XRU55 X:XaXV:a[R"[[R"X5U- 55S-RSS9n[R"U5n[R"U5RS:Xa'SUl[U RUR5 O.[R"/5R![#U 55n[X5 [ [R$"U R&[R(55 [ [+X55 [ [+X55 g) NFrrrrxrrvr)r'rrr rlstsqrrrrr__array_wrap__rrrrrr% issubdtyper0floatingr,)rLr2r3rKarrrnrrrr residualsranksv expect_residss r+rQ LstsqCases.dosijjmyy::a'b!'a"!=d 6 Q!VLLN # 6 3q9 -  !  !B 1 1! 45 9 3rvva|a/01Q6  JJ}5Mzz!}!!Q&&* #Y__m.A.ABHHRL--d1g6MI5 ioor{{;<#A)*#I12r-r_Nr*r_r-r+rrs3r-rcl\rSrSrSr\R R/SQ/SQ5S5rSr Sr g) TestLstsqicr[R"/SQ/SQ/SQ/SQ/5Rn[R"/SQ5n[R"XSS9up4pV[ US:H5 [R"X5up4pV[ US :H5 [R"XSS9up4pV[ US :H5 g) N)rrgrrgrhr)rrhrrrgr)rgrrgrrrj)rrrrhrir)rvrrrrrrrrzry)r'rTr r!r)rLr2r3rr'r(rs r+ test_rcondTestLstsq.test_rconds HH22224 567Q HH' ( & Q <d  & Q 2d  & Q >d r-)rr&n_rhs))rzrxrx)rrzrv)rrzrx)rzrrv)rzrrx)rzrxrrrrcF[R"X-5RX5n[R"X45n[R "XESS9upgpUS:Xa[ US:HR55 [URX#45 [URX:aU4OS5 X:a:US:a4U[R"XF5- n [XzU -RSS95 [U[X55 [U R[X545 g)Nrrrrrr)r'r:rrr r!rrrrrrrr) rLrr&r2r2r3rr'r(rr s r+test_empty_a_bTestLstsq.test_empty_a_bs IIae  $ $Q * GGQJ  & Q >d 6 Q!VLLN #QWWqj)Y__15xdD 5UQYBFF1L A  E;;B;+? @T3q9%QWWs1yl+r-c`[R"/SQ5n[R"/SQ5n[R"U[R"[ U55/5R n[ [S5 [R"X2SS9 SSS5 g!,(df  g=f)N)rrvrxry)rg?g?g@gffffff @zIncompatible dimensionsr) r'rvstackrlenr/r!rr r!)rLryrs r+test_incompatible_dims TestLstsq.test_incompatible_dimssj HH\ " HH- . IIq"''#a&/* + - - .G H LLT *I H Hs B B-r_N) rYrZr[r\r0r r rKr5r;r^r_r-r+r-r-s7  [[03 , ,+r-r-dtz?bBhHiIqQefdgFDGOc\rSrSr\R "S5r\/SQr\/SQr\/SQr \\\\ /r \ "SS/SS//5r \R"\///S-5r\\R "S 5\R "S 5\R "S 5/rS rS rSrSrSrSrSr\R4R7\SS9S5rSrg)TestMatrixPowerirz)ryrrvrx)rxryrrv)rvrxryrrvrrxrgGcURRU5n[[USS-S-S-S-5UR5 [[USS-S-S-S-5UR5 [[USS-S-S-S-5UR 5 [[USS-S-S-S-5UR 5 g)NrxrFi rrvry)rshft_1r rrrshft_0rshft_2rshft_3)rLr=rshfts r+test_large_power TestMatrixPower.test_large_power&s ##B' 3 5 9 :DLL J 3 5 9 :DLL J 3 5 9 :DLL J 3 5 9 :DLL Jr-cSnURHGnU"URU55 U[:wdM&U"URRU55 MI g)Nc[US5n[U[U55 [URUR5 g)Nr)rrr!r0)Mmzs r+tz.TestMatrixPower.test_power_is_zero..tz2s0a#B 6q9 : 177 +r- rshft_allr objectstackedrLr=rOrs r+test_power_is_zero"TestMatrixPower.test_power_is_zero1sE , >>C szz"~ V|4<<&&r*+"r-cSnURHGnU"URU55 U[:wdM&U"URRU55 MI g)Ncr[US5n[X5 [URUR5 g)Nrv)rrr0)rrNs r+rO-TestMatrixPower.test_power_is_one..tz=s(c1%B  ! 399 -r-rQrUs r+test_power_is_one!TestMatrixPower.test_power_is_one<sE . >>C szz"~ V|4<<&&r*+"r-cSnURHGnU"URU55 U[:wdM&U"URRU55 MI g)Nc[US5nUR[:wa[O[n[ X"X55 [ URUR5 g)Nrx)rr0rSr rr)rrNmmuls r+rO-TestMatrixPower.test_power_is_two..tzHs@c1%B YY&06cD T#^ , 399 -r-rQrUs r+test_power_is_two!TestMatrixPower.test_power_is_twoGsE . >>C szz"~ V|4<<&&r*+"r-cSnURH+nXR;dMU"URU55 M- g)Nc[US5nUR[:wa[O[n[ U"X5[ U55 gNr)rr0rSr rrr!)rinvmatr_s r+rO3TestMatrixPower.test_power_is_minus_one..tzTs9!#r*F YY&06cD V!#>C%3::b>""r-cURRU5n[[[US5 [[[US/5 g)Ng?rv)rEr r TypeErrorrrLr=rs r+test_exceptions_bad_power)TestMatrixPower.test_exceptions_bad_power^s4ll!!"%isC8isQC8r-c[[[[R"S/U5S5 [[[[R"S/S//U5S5 [[[[R "SU5S5 g)Nrvrx)rzryrx)r rrr'rr)rLr=s r+test_exceptions_non_square*TestMatrixPower.test_exceptions_non_squarecsWk<1#r1BAFk<A3*b1I1Mk<B1GKr-fp errors don't work in wasmrcXR;agURRU5n[[[ US5 gre)rhnoninvr r rrrms r+test_exceptions_not_invertible.TestMatrixPower.test_exceptions_not_invertiblehs1   kk  $k<b9r-r_N)rYrZr[r\r'rrErDrFrGrRrrvblockrTrSr0rhrIrVr[rarirnrqr r skipifrrwr^r_r-r+r?r?sffQiGl#Gl#Gl#G'7G4I QFQF# $Fhh' }q()Grxx}bhhsmRXXc]CG J , , , #9 L  [[(FG:H:r-r?c\rSrSrSrSrg)TestEigvalshCasesipc"[R"US5n[R"U5upVURSS9 [ XE[ UR 5S9 [R"US5n[ Xu[ UR 5S9 g)NLrrrxr)r eigvalshrfrrr=r0)rLr2r3rKrgrhriev2s r+rQTestEigvalshCases.dorsh__Q $"JJqM " (288*<=ooa%8BHH+=>r-r_Nr*r_r-r+r|r|ps ?r-r|cr\rSrSr\R R S\\\ \ /5S5r Sr Sr SrSrg) TestEigvalshi~r0c[R"SS/SS//US9n[RRU5n[ UR [ U55 gr.)r'rr rrr0r9)rLr0rrs r+r1TestEigvalsh.test_typessF HHq#ha) 7 II  q !QWWnU34r-cN[R"SS/SS//[RS9n[[[R R USS9 [[[R R US5 [[[R R US5 gNrvr/r8lrongUPLOlowerupper)r'rrr r;r rrLrs r+ test_invalidTestEigvalsh.test_invalidsk HHq#ha) <j"))"4"4agFj"))"4"4aAj"))"4"4aAr-c[R"SS/SS//[RS9n[R"SS/SS//[RS9n[R"SS/[RS9n[[R5n[RR U5n[ XSUS9 [RR USS9n[ XSUS9 [RR USS9n[ XSUS9 [RR US S9n[ XSUS9 [RR US S9n[ XSUS9 g Nrrvr8rrxr~rlrr)r'rr r=r rr)rLKloKuptgtryrs r+ test_UPLOTestEigvalsh.test_UPLOshhAA'ryy9hhAA'ryy9hhAwbii0 " II  s #T* II  s  -T* II  s  -T* II  s  -T* II  s  -T*r-c"SS[R5n[R"S[RS9R U5n[ R "U5n[URR[RL5 [SUR5 [[U[R55 [R"S[RS9R U5n[ R "U5n[URR[RL5 [SUR5 [[U[R55 g)Nc\rSrSrSrg)/TestEigvalsh.test_0_size..ArraySubclassir_Nr6r_r-r+r7rr8r-r7rYr8rsrqrr)r'r(rZr[rr rrr0r%r\rrr&r]rr^s r+rDTestEigvalsh.test_0_sizes BJJ  HHYbgg . 3 3M Booa  "**,-VSYY' 3 +, HHV2<< 0 5 5m Dooa  "**,-T399% 3 +,r-r_NrYrZr[r\r r rKrr rrr1rrrDr^r_r-r+rr~s? [[Wvvw&HI5J5 B +,-r-rc\rSrSrSrSrg) TestEighCasesic[R"U5nURURpe[R"U5upxUR SS9 [ XW5 [[X5[R"U5SSSS24[R"U5-[UR5S9 [R"US5up[ X5 [[X5[R"U 5SSSS24[R"U 5-[UR5[U5S9 g)Nrr.rxr)ryerr_msg)r eighrzr{rfrrrr r'rr=r0repr) rLr2r3rKr_rgevcrhrirevc2s r+rQTestEighCases.doskk!n//3#3#3C"JJqM " B(q 2sD!|4rzz#F%bhh/ 1KK3' C)q 3T1 5 48HH%bhh/a Br-r_Nr*r_r-r+rrsBr-rcr\rSrSr\R R S\\\ \ /5S5r Sr Sr SrSrg) TestEighir0c[R"SS/SS//US9n[RRU5up4[ UR [ U55 [ UR U5 gr.)r'rr rrr0r9rs r+r1TestEigh.test_typessT HHq#ha) 7yy~~a QWWnU34QWWe$r-cN[R"SS/SS//[RS9n[[[R R USS9 [[[R R US5 [[[R R US5 gr)r'rrr r;r rrs r+rTestEigh.test_invalidse HHq#ha) <j"))..!'Bj"))..!W=j"))..!W=r-c[R"SS/SS//[RS9n[R"SS/SS//[RS9n[R"SS/[RS9n[[R5n[RR U5upV[ XSUS9 [RR USS9upV[ XSUS9 [RR USS9upV[ XSUS9 [RR US S9upV[ XSUS9 [RR US S9upV[ XSUS9 gr)r'rr r=r rr)rLrrrryrrs r+rTestEigh.test_UPLOshhAA'ryy9hhAA'ryy9hhAwbii0 "yy~~c"T*yy~~c~,T*yy~~c~,T*yy~~c~,T*yy~~c~,T*r-c:"SS[R5n[R"S[RS9R U5n[ R "U5up4[URR[RL5 [URR[RL5 [URUR5 [SUR5 [[U[R55 [R"S[RS9R U5n[ R "U5up4[URR[RL5 [URR[RL5 [URUR5 [SUR5 [[U[R55 g)Nc\rSrSrSrg)+TestEigh.test_0_size..ArraySubclassir_Nr6r_r-r+r7rr8r-r7rYr8rsrqrr)r'r(rZr[rr rrr0r%r\rrr&r]rrs r+rDTestEigh.test_0_sizes< BJJ  HHYbgg . 3 3M B[[^    BJJ./ "**,-QWWekk*VSYY' 1bjj)* HHV2<< 0 5 5m D[[^    BLL01 "**,-QWWekk*T399% 1bjj)*r-r_Nrr_r-r+rrs> [[Wvvw&HI%J% > +,+r-rc,\rSrSrSrSr\S5rSrg) _TestNormBaseiNc.[URR[R5(a+[ URUR R5 g[[URR[R55 gr$) issubclassr0r%r'inexactrrealrr$)rr_s r+ check_dtype_TestNormBase.check_dtype sN aggllBJJ / / AFFLL 1 Jsyy~~r{{; ?Er-c~[R"/SQ5n[RSn[RSnX#-nUGHnURU5n[ U[R *5nUR Xg5 [US5 [5nUR[S5 [ US5nUR Xg5 [US5 SSS5 [ US5nUR Xg5 [US5 [ US 5nUR Xg5 [US 5 [ US5nUR Xg5 [XwRRS 5URRS 5-5 [ US 5nUR Xg5 [XwRRS 5URRS 5-5 [ U[R 5nUR Xg5 [US5 GM g!,(df  GNX=f)Nr AllIntegerAllFloatrdivide by zero encounteredrrrxrvrhr/rzg?rg) r'r typecodesr rr rrr"filterRuntimeWarningr0r% rLr2 exact_types inexact_types all_types each_typeatansups r+test_vector_return_type(_TestNormGeneral.test_vector_return_types HHY ll<0  Z0 / "I)$Bb266'"B   R $ C ("$ >+GH"b\  (#B, % b!B   R $ A &b!B   R $ C (b!B   R $ HHMM#$6 i8P$P Qb!B   R $ HHMM#$6 i8P$P Qb"&&!B   R $ C (?#%$s AH-- H< c^/SQn/SQn/SQnU4SjnXU4H nU"U5 M [UTRS9[UTRS9[UTRS94H nU"U5 M g)N)rvrxryrz)rrrr)rrxrrzcF>[RR[U5STRS9 [RR[U[ 5STRS9 [RR[U[ *5STRS9 [RR[US5STRS9 [RR[US5STRS9 [RR[US 5STRS9 [RR[US 5S TRS9 [RR[US 5S TRS9 g)Ng6ҭ@r/rjrgrv$@rgQ?rxrgsc?rrz)r'testingrrrr )rrLs r+_test+_TestNormGeneral.test_vector.._testHsD JJ * *47I3788 + = JJ * *43<3788 + = JJ * *4C4=#3788 + = JJ * *41:t3788 + = JJ * *42; 3788 + = JJ * *41:y3788 + = JJ * *42;9M3788 + = JJ * *41:q3788 + =r-r8)rr=)rLr2r3rrrs` r+ test_vector_TestNormGeneral.test_vectorCsl    =$A !H)5$''+B)+A !H+r-c [/SQ/SQ/URS9nSSSSSS[R[R*4Hn[ UR S5Vs/sHn[ USS2U4US 9PM nn[[ XSS 9U5 [ UR S5Vs/sHn[ XSS24US 9PM nn[[ XSS 9U5 M [R"SS URS9RSSS 5nURnSS SSS[R[R*S4GH n[R"[ U*U5S5HnUupU S:aX- n U S:aX- n X:Xa[[[ XbUS 9 M6[ XbUS 9n XyU -- n X:aA[ UR U 5Vs/sHn[ USSRXIr-r_N) rYrZr[r\rrrrrr^r_r-r+rrsF ')R<#5J(Qr-rcJ\rSrSr\R rSrSrSrSr Sr Sr g) _TestNorm2Dic `[[UR//URS95S5 g)Nr8r)rrrr=rVs r+test_matrix_empty_TestNorm2D.test_matrix_emptys$T$**bT*9:C@r-cUR/SQ/SQ/5n[RSnSnX#-nUGHnURU5n[ U[R *5nUR Xg5 [US5 [5nUR[S5 [ US5nUR Xg5 [US5 SSS5 [ US 5nUR Xg5 [US5 [ US 5nUR Xg5 [US 5 [ US 5nUR Xg5 [US5 [ U[R 5nUR Xg5 [US5 [ US 5nUR Xg5 [US5 [ US5nUR Xg5 [RRUSSS9 GM g!,(df  GN,=f)NrrYrfdFDrhrrrgrvrxgLXz?rrrgU&,t=@rr) rr'rr rr rrr"rrrrs r+test_matrix_return_type#_TestNorm2D.test_matrix_return_types JJ 9- .ll<0  / "I)$Bb266'"B   R $ C ("$ >+GH"b\  (#B, % b!B   R $ C (b!B   R $ $4 5b"B   R $ C (b"&&!B   R $ C (b%B   R $ C (b%B   R $ JJ * *2/A1 * MK#%$s AG// G> cdURSS/SS//URS9n[[U5S5 [[US5S5 [[US5S 5 [[U[5S 5 [[U[*5S 5 [[US5S 5 [[US 5S 5 [[US5S5 [[US5S5 [ [ [US5 [ [ [US5 [ [ [US5 g)Nrvryrrr8gT"@rrrg(@rjrrlrxgW?"@rgїʄ?nofrorr)rr=rrr r r;rs r+test_matrix_2x2_TestNorm2D.test_matrix_2x2s JJAA'twwJ 7DGY/DENI6DEND1DCL$/DSDM3/DAJ-DBK-DAJ(:;DBK)<=j$73j$2.j$1-r-cSUR/SQ/SQ/SQ/URS9-n[[U5S5 [[US5S5 [[US5S 5 [[U[5S 5 [[U[*5S 5 [[US 5S 5 [[US5S5 [[US5S5 [[US5S5 g)Ng?r)rrr)ryrxrvr8grj=Q0?rrgB+pc?g?g333333?rvrgrg?rxg&.d?rgܸՠ?)rr=rrr rs r+test_matrix_3x3_TestNorm2D.test_matrix_3x3s  JJ 9i8J H IDG%9:DEN,@ADEN,>?DCL#.DSDM3/DAJ,DBK-DAJ(;<DBK)<=r-cUR/SQ/SQ/URS9n[R"SSURS9R SSS5n[ [ [US S 5 [ [ [US S 5 [ [ [SS/S S5 [ [ [SS/S S5 [ [ [SS/S S5 S HEn[ [ [XS5 [ [ [XS5 [ [ [X#S5 MG [ [[USS5 [ [[USS5 [ [ [USS5 g)Nrrr8rvrrxryrzrrrtestrryrs)rvrxrxryrrvrx) rr=r'r:rr r;rr)rLrrrs r+ test_bad_args_TestNorm2D.test_bad_args s  JJ 9-TWWJ = IIa477 + 3 3Aq! < j$5!4j$5!4j$At<j$At<j$A=E *dAd ; *dAf = *dAf = iq$2iq$7j$4;r-r_N) rYrZr[r\r'rrrrrrr^r_r-r+rrs) HHEA1Nf. >"  t$ /   s * E?? F c [S[[R"S5SS95 [S[[R"S5SS95 [S[[R "S5SS95 [R"S5nSUS'[S [USS95 S US'[S[USS S 95 [S [USS S 95 g)NrzTrrvrrrr6ryg:0yE>g|"BE>)r@r7g;䡈E>)rrr'rrrZ)rLr8s r+test_symmetric_rank"TestMatrixRank.test_symmetric_ranksQ BFF1I>?Q BGGFOtDEQ BHHV$4EF FF1I& Q A67& Q A7CDQ A7CDr-r_N)rYrZr[r\r:r=r^r_r-r+r4r4ds 02 Er-r4c@[RRS5n[S5HpnUR SS9nUSS2S4USS2S4-USS2S4'[ [ U5S5 USS2S 4USS2S 4-USS2S 4'[ [ U5S 5 Mr g) Ni3rF)(rsizervrxrr rzrryrt)r'r RandomStaternormalrr)rngiXs r+test_reduced_rankrHs ))   )C 3Z JJHJ %AqD'AadG#!Q$[^Q'AqD'AadG#!Q$[^Q'r-c \rSrSr\R rSr\RRSS//SQ5S5r Sr Sr S r \RRS /S Q5\RRS /S Q5\RRS\R\R\R \R"/5S555rSrg)TestQRic[U5nURnURupE[XE5n[R "USS9nUR URp[URU:H5 [U RU:H5 [[X55 [[X55 [URXD4:H5 [U RXE4:H5 [[X5U5 [[URR5U5[R"U55 [[R "U 5U 5 [R "USS9up[U RU:H5 [U RU:H5 [[X55 [[X55 [U RXF4:H5 [U RXe4:H5 [[X5U5 [[U RR5U 5[R"U55 [[R "U 5U 5 [R "USS9n [U RU:H5 [[X55 [X5 g)Ncompletemodereducedr )r%r0rrr qrQRrr&rrr/rr'rtriu) rLr2a_typea_dtyperr&rr_rQrRq1r1r2s r+check_qrTestQR.check_qrsa''ww Iii +uucee17"#7"# 1%& 1%&A6!"A6!"CIq)C A.q :BGGAJ*19-G#$G#$ 2&' 2&'QF"#QF"#CK+C R0"&&)<BGGBK,YYqs #G#$ 2&'B#r-rr&)ryrr rqc[X5n[R"X45nURU5 [RR USS9upV[ UR[R5 [ UR[R5 [ URX!45 [ URU45 g)NrawrM) rr'rrYr rPrr0r r)rLrr&rr2htaus r+ test_qr_emptyTestQR.test_qr_emptys I HHaV  aae,QWWbii(SYY *QWWqf%SYY%r-cURSS/SS/SS//[RS9n[R"USS 9up#[ UR [R:H5 [ UR [R:H5 [ URS :H5 [ URS :H5 [R"URSS 9up#[ UR [R:H5 [ UR [R:H5 [ URS :H5 [ URS :H5 g) Nrvrxryrzrrr8r]rMr rr) rr'r r rPrr0rr/)rLr2r^r_s r+ test_mode_rawTestQR.test_mode_raws JJAAA/ryyJ A15)299$% RYY&'6!" T!"133U+299$% RYY&'6!" T!"r-cURSS/SS//5nURSS/SS/SS//5nSHbnURU5nURU5nURU5 URU5 URUR5 Md SHnnSSURU5--nSSURU5--nURU5 URU5 URUR5 Mp g) Nrvrxryrzrrfd?)rr rYr/)rLr2r3r=m1m2s r+test_mode_all_but_economic!TestQR.test_mode_all_but_economics JJAA' ( JJAAA/ 0B"B"B MM"  MM"  MM"$$  BR!((2,&&BR!((2,&&B MM"  MM"  MM"$$  r-c [U5nURnURSSupE[XE5n[R "USS9upx[ URU:H5 [ URU:H5 [ [Xr55 [ [X55 [ URSSXD4:H5 [ URSSXE4:H5 [[Xx5U5 [R"URS5n [R"U URSSURS4S--5n [[[USS5R5U5U 5 [[R"USSS2SS245U5 [R "USS9up[ U RU:H5 [ U RU:H5 [ [X55 [ [X55 [ U RSSXF4:H5 [ U RSSXe4:H5 [[X5U5 [R"U RS5n [R"U U RSSU RS4S--5n [[[U SS5R5U 5U 5 [[R"U SSS2SS245U 5 [R "USS9n [ U RU:H5 [ [X55 [X5 g) NrrLrMrrx.rOr )r%r0rrr rPrr&rr r'r r'rrrS)rLr2rTrUrr&rqr I_mat stack_I_matrVrWrXs r+check_qr_stackedTestQR.check_qr_stackedsa''wwrs| Iyy,7"#7"# 1%& 1%& &' &'F1L!, AGGBK(ooe  ~'99; F8Ar2#6#;#;#=qA;OBGGAc1aiL11519-G#$G#$ 2&' 2&' !'( !'(F2NA. BHHRL)ooe" "!(;;= F8BB#7#<#<#>C' )BGGBsAqyM2B7YYqs #G#$ 2&'B#r-rB))ryrz)rzryrr[r  outer_size)r9r)rxryrzr=c[RRS5nURX-S9R U5nURX-S9R U5nUR U5 UR USU--5 g)N{rArg)r'r default_rngrDr rp)rLrrrBr=rErrs r+test_stacked_inputsTestQR.test_stacked_inputs*sxii##C( JJJ-J . 5 5b 9 JJJ-J . 5 5b 9 a  a#'k*r-r_N)rYrZr[r\r'rrYr r rKr`rcrjrprr rrrvr^r_r-r+rJrJs HHE%$N [[c3Z* &  &#, "+$Z [[V& [[\,!" [[T 299 BJJ$ !+!"+r-rJc@\rSrSr\R R S/SQ5\R R S\R\R\R\R45\R R SSS/5S555r S r S rS rg ) TestCholeskyi;r)rr9ryry)2r{)ryrrr0rFTc h[RRS5 [RR"U6n[R"U[R 5(a#US[RR"U6--n[ [[U555nSUSS&[R"URU5R5U5n[R"XBS9n[RRXCS9nU(a5[R"URU5R5U5nO3[R"XfRU5R55nSURS-[R "U5R"-n[%XtXS US US U3S 9 [R&"USS S 9n [)[R*"[R,"U 555 [)[R*"U S:55 g)Nrvrg)rrrr8rir  )atolrr)axis1axis2)r'rrrandnr#complexfloatingrrr9r rrrr choleskyrfinfoepsrdiagonalrrisreal) rLrr0rr2trr3rr s r+test_basic_property TestCholesky.test_basic_property=s q IIOOU # == 2 2 3 3B%000A s5z" #"# IIakk!n))+Q / JJq & II  q  .  !++a.--/3A ![[^0023AQWWQZ"((5/"5"5547!E7"QCr!1MN KK2 .ryy|$%qAvr-c"SS[R5n[R"S[RS9R U5n[ R "U5n[URUR5 [URR[RL5 [[U[R55 [R"S[RS9R U5n[ R "U5n[URUR5 [URR[RL5 [[U[R55 g)Nc\rSrSrSrg)/TestCholesky.test_0_size..ArraySubclassibr_Nr6r_r-r+r7rbr8r-r7rYr8)rvrr)r'r(rZr[rr rrrrr0r%r\r&r]r^s r+rDTestCholesky.test_0_sizeas BJJ  HHYbgg . 3 3M Booa QWWcii( "**,- 3 +, HHYbll 3 8 8 Gooa QWWcii( ",,./ 3 +,r-c8[R"SS/SS//5n[[R"U5[R"USS95 [[R"USS9[R"U5R R 55 g)Ny?yy@y@Fr}T)r'rrr rr/r)rLr2s r+test_upper_lower_arg!TestCholesky.test_upper_lower_argqsn HHvv&(89 :V__Q'%)HI OOAT * OOA  % % ' r-r_N)rYrZr[r\r r rKr'rr\r]r"rrDrr^r_r-r+ryry;s [[@ [["**bjj",, F [[%    8-  r-ryc|\rSrSr\R "S5r\R "S5r\R"/SQ/SQ/SQ/5r \ "\RR\\5\ 5 \ "\S5 \RR\SS2\R4\5 SSS5 Srg!,(df  Srg=f) TestOuteri}ryr3r)rrxrzz$Input arrays must be one-dimensionalNr_)rYrZr[r\r'r:arr1arr2rrCrr outerr!r;newaxisr^r_r-r+rr}s 99Q>& !Y . KK5 6 /I ^ $1 /01   # HHR\\ " # $*c!f4#  $sBE ,(E8AF8 &E54E58 F5-F54F58 G2%G--G2zCannot start subprocessc*SS/n[R"S5nUHsnURSSUS9n[R"[ R SU/5 URSSUS9n[R"[ R SU/5 Mu g) NzPyQt5.QtWidgetsIPythonz import sys {before} try: import {bad_lib} except ImportError: sys.exit(0) {after} x = np.ones(2, dtype=np.float32) sys.exit(0 if np.allclose(x.dot(x), 2.0) else 1) zimport numpy as np)beforeafterbad_libz-c)rrr)textwrapdedentr subprocess check_callr executable)bad_libstemplatercodes r+test_sdot_bug_8577rs"9-H   H&:"'.0s~~tT:;%9"'.0s~~tT:;r-cV\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rg) TestMultiDoti c [RRS5n[RRS5n[RRS5n[[XU/5UR U5R U55 [[XU/5[R"U[R"X#555 gNrrxrxrr'rrrr)rLrrCs r+(test_basic_function_with_three_arguments5TestMultiDot.test_basic_function_with_three_argumentss II  V $ II  V $ II  V $IqQi0!%%(,,q/BIqQi0"&&BFF1L2IJr-c[RRS5n[RRS5n[[X/5UR U55 [[X/5[R"X55 grr)rLrrs r+&test_basic_function_with_two_arguments3TestMultiDot.test_basic_function_with_two_argumentss\ II  V $ II  V $Iqf-quuQx8Iqf-rvva|[RRS5n[RRS5n[RRS5n[RRS5n[[XX4/5RS5 g)Nrxrrr9rr'rrrr)rLA1drrrs r+test_vector_as_first_argument*TestMultiDot.test_vector_as_first_argument*smiiq! II  V $ II  V $ II  V $ Y~.44d;r-c>[RRS5n[RRS5n[RRS5n[RRS5n[[XX4/5RS5 g)Nrrrx)rr)rLrrrD1ds r+test_vector_as_last_argument)TestMultiDot.test_vector_as_last_argument4sm II  V $ II  V $ II  V $iiq! Ya~.44d;r-c>[RRS5n[RRS5n[RRS5n[RRS5n[[XX4/5RS5 g)Nrxrrr_r)rLrrrrs r+&test_vector_as_first_and_last_argument3TestMultiDot.test_vector_as_first_and_last_argument>sniiq! II  V $ II  V $iiq! Y/066;r-c [RRS5n[RRS5n[RRS5n[R"S5n[XU/US9nXELde[ XAR U5R U55 [ U[R "U[R "X#555 g)Nrrr)r'rrZrrr)rLrrrr)rets r+test_three_arguments_and_out)TestMultiDot.test_three_arguments_and_outHs II  V $ II  V $ II  V $hhvq s+zzCqa1C266!.>v.F-GHr-r_N)rYrZr[r\rrrrrrrrrrrr^r_r-r+rr s<K=M<<< :/ 9=<Ir-rc:\rSrSr\R R S\R"S5S4\R"S5S4/5S5r \R R SS S /5S 5r \R R S S S/5S5r Sr Sr g) TestTensorinvizarr, indrzrrtrxrxryryrxrvc~[[5 [R"XS9 SSS5 g!,(df  g=f)Nind)r rr tensorinv)rLr%rs r+test_non_square_handling&TestTensorinv.test_non_square_handlings& ; '   S *( ' 's. <z shape, ind)rzrrtryrx)rtryrvc[R"S5nXl[R"X2S9nURUSURSU-nURn[ Xe5 g)Nrr2r)r'rrr rr)rLrrr2ainvrCactuals r+test_tensorinv_shape"TestTensorinv.test_tensorinv_shapesU FF2J!-7734=1774C=0V&r-rrrc[R"S5nSUl[[5 [ R "X!S9 SSS5 g!,(df  g=f)Nrrr)r'rrr r;r r)rLrr2s r+test_tensorinv_ind_limit&TestTensorinv.test_tensorinv_ind_limits: FF2J : &   q *' & &s A  Ac[R"S5nSUl[R"USS9n[R "S5n[ [R"X#S5[RRX55 g)Nrrrvr) r'rrr rrr tensordot tensorsolve)rLr2rr3s r+test_tensorinv_result#TestTensorinv.test_tensorinv_resultsY FF2Jq) GGBK Ta0"))2G2G2MNr-r_N)rYrZr[r\r r rKr'rrrrr#r^r_r-r+r r s [[Z  "  Q*  +  + [[\,  '  ' [[U 2%  + + Or-r c\rSrSr\R R S\R"S5S4\R"S5S4/5S5r \R R S/S Q5S 5r S r g) TestTensorsolveiza, axesr Nr rrxc[[5 [R"URSS5n[ R "XUS9 SSS5 g!,(df  g=fNrxr)r rr'rrr r")rLr2rr3s r+r(TestTensorsolve.test_non_square_handlings@ ; ' $A   q$ /( ' 's 9A A r))rxryr)ryrzrzry)rryryrc &[RR"U6n[R"URSS5n[R R X#5n[[R"X$[UR5S9U5 gr)) r'rrrrr r"rr!r9)rLrr2r3rs r+test_tensorsolve_result'TestTensorsolve.test_tensorsolve_results_ IIOOU # GGAGGBQK  II ! !! ' QAGG =qAr-r_) rYrZr[r\r r rKr'rrr,r^r_r-r+r&r&sv [[Y  %  V$)  0  0  [[W/BBr-r&c[R"SS/SS//SS9n[[S5 [R "U5 SSS5 g!,(df  g=f)Nrvrrxrfloat16r8zunsupported in linalg)r'rr!rlr r)r%s r+test_unsupported_commontyper0sC ((QGaV$I 6C Y(? @ A @ @s A  Az,Bad memory reports lead to OOM in ci testingcSn[R"SU/[RS9n[R"SS/[RS9nSUS'[R"X!5n[ USS5 g)Nrvr8)rr)r'rZrrrr)r&r2r3rs r+test_blas64_dotr3s[ A !Qrzz*A Abjj)AAeH q A51r-z*Numpy not compiled with 64-bit BLAS/LAPACKc [Rn[RRRnSnSnUn[R "SS/US9n[R "S/US9n[R "S/US9nU"X#XTXvSS5n[ USS5 [ USU5 [ USU5 [UR55n [S U s=:=(a S :5 gs 5 g) Nlrvr8rrinforr&r2l) r'r\r rdgeqrfrZrritemr) r0lapack_routinerr&ldar2workr_resultslworks r+!test_blas64_geqrf_lwork_smoketestr=s JJEYY**11NAA C !Qu%A 88QCu %D ((A3e $CQ13b!rus    !&PP>     #*.95 7   ,,, t 8xr2hR)8b"XV,.xr2hR)8b"XV,.zr2hR)8lL1@Byw'&&)9:'Jgw'w79yw'&&)9:'Jgw'w79{w'&&)9:'J02JKSZ[]uxxf-xxF+J uyy~~a#yy~~a "uyy~~a#yy~~a "zAA1v;   F;~lL1@b"XV,.~r2hR2r(36B\02~lL1@b"XV,.~r2hR2r(36B\02/1LMU\^gw'w79g&'(:Wg22n222 Fn F I I +> +$ +%'F +H3 H3V/#%D/0h0.)')H)-,-66#%D6!+h!+H 5 5,#%D,8h 8,,)+G,((,+1$&E+1\@*y@*F 0$'/2 0 y 0*,H0 *  /##%D#*'1h'1T3%'>3<.+ .+b5HI5H 5HIJS:S:KS:l ?)+G ?3-3-lB%'CB.8+8+v = =[Q}[Q|u<-u<?OJ   N Jsd d dd