L iddlZddlZddlmZddlmZmZddlmZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZdZdZdZy)N)exp)assert_ assert_equal)rootc d}d}tttdddtttdddtttdd d tttd d d tt t d d d tttd ddtttdddtttd ddtttdddtttdddg }tjgdgdddgD]\}|D] d} fd}|f} d|z}tj ||zz|tj"j%|||zz}dt'z z z} z } tj(d5t+|||td| ddz| | |d !} dddt-| tj"j%| j.||| j0| j2| j4g } | j0sJ| | j4 ddzksJ| | j2 d"ksJ| tj"j%|| j.||krJ| y#1swYxYw)#Ngh㈵>-C6?i)Fx0nnitnfevi'i idci )?绽|=g _Bcruzchengr c(d|z |zS)Nr )xr problemxscaleyscales i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/optimize/tests/test__spectral.pyfuncztest_performance..func0s  QZ ;;;r rignore)overrr)ftolfatolmaxfevsigma_0 sigma_eps line_searchDF-SANE)argsoptionsmethodr )dictF_1x0_1F_2x0_2F_6x0_6F_7x0_7F_9x0_9F_10x0_10 itertoolsproductnpsqrtlinalgnormminerrstaterreprrsuccessr r)e_ae_rtable_1r-r r#r/r r)r,r+solerr_msgrr r!s @@@r"test_performancerN s C C stt3 stu!!4 sts4 stt"5 sts2 strr3 sts4 sts4 stt"5 t115#G*(1'8'8.0A(D#  DG A <4Dq!F*BGGAJ$v-biinnT"a[6Q0QQEF6M6&= AAIVmG(+ -4$#'QeGFOVWDW0794?$A#, - -gryy~~d355RSn7U SWWchh8G;; ' ';88wv22 ;G ;277gen, 5g 5,99>>$suua.1U: CG C:7 DD - -s -&KK c$d}d}d}t|||d}t|jtjj ||}tjj ||j }t|||zky)Nc|dzdz dzS)Nrr'@r)zs r"r#ztest_complex..funcMs!tax"}r$rQrr.)tolr1)rrrHrArCrDr)r#r r(rLf0fxs r" test_complexrVLsm B D tRT) 4C CKK R !B SUU $B B$r'Mr$cd d}d}tjjdtj||zj ||}|||ztj dtj|zzz}t tjj|jdkDtj|dz}|||d|||d|| |d|| |dy) Nrc |fd}tjj}tjj||dz}t |t |dd|d}t |jt tjj||j|ky)Nc,j|z SN)dot)rAbs r"r#z=test_linear_definite..check_solvability..funccs558a< r$g@@risD)r)r(r*r-r.)r0r1) rArCsolverDrr2rrHr)r\r]r-r#xpepsrLs`` r"check_solvabilityz/test_linear_definite..check_solvabilitybs YY__Q "iinnT"X&, !s5kR    tCEE{+s23r$Zir'rrr)r) rArandomseedarangereshapediagrrCeigvalsrE)rar r\r]s r"test_linear_definiteri[s 4 AIINN4 !A#q!$A AaC"''!biil*+ ++A BII  a $ $ & *+ ! sAaF#aG$qb!V$qb!W%r$c*d}ttfD]}tjddg}tjddg|}t |||fd}t |jt|jj|jy)Nc ||z SrZr)rargs r"fztest_shape..fs 3wr$r)dtyper.)r/r1) floatcomplexrAzerosonesrrrHrrshape)rmdtrrlrLs r" test_shaperusvg+ HHaUOggqe2&1asfY7 SUU[[!''* +r$ctj|g}tjd|dz}t|ddz dz |d<|t|dddz |ddz z|dd|S)Nrr'rrArqrerrr gis r"r3r3sm ! A !QqSA qtax=1 AaD s1QR519~!"% &AabE Hr$c`tj|g}|j||dz z |SNr'rAemptyfillr r s r"r4r4s* 1#BGGAqsG Ir$ctj|g}tjd|dz}t|ddz |d<d|zt|dd|ddzdz z|dd|S)Nrr'rg?rwrxs r"r5r5sm ! A !QqSA qt9q=AaD E3qu:#2&* +AabE Hr$c`tj|g}|jd|dzz |S)Nr'rr}rs r"r6r6s) 1#BGGAadFO Ir$ct|dzdtj|g}d|dddzd|ddddzzzd|ddddzzz d|dddzzd z |ddd<d |dddzd |ddddzzz d |ddddzzzd |dddzz |dddz d|ddddzzzdz|ddd<d|dddzd|ddddzzz |ddd<|S)Nrg333333?g?r'g@rg333333#@g333333@gQ?g ףp= ?gQ @gHzG@g?gHzG@g?g?)rrArq)rr rys r"F_4rsSQ ! A1SqS6\C!ADqD'1*, ,sQqt!tWaZ/? ?#!$Q$- ORU UAccFa!f}ta1gqj004!ADqD'1*3DDtaPQPTSTPTg~U14a414a4!+,.23AaddGQqt!tWntAaddGQJ.AaddG Hr$c^t|dzdtjgd|dzz}|S)Nrr)r?rrrAarrayrs r"x0_4rs,Q -1a4( )B Ir$cd}tjd|dzdz |z }|dd|d|zz |dddf|z|dddf|zz jdzz z z S)Ng?r'rr)axis)rAresum)rr cmus r"r7r7sr A ))Aqs c !1 $B q!a1gAdFAAdFb!A F FA F NNNO OOr$c.tj|gSrZrArrr s r"r8r8 77A3<r$ct|dzdd}tj|g}d|ddddzzdz |ddd<t|ddd t|ddd zdz |ddd<||ddd|ddd<|S)Nrrcd|zdz }d|dzzd|dzzzd|zzdz dz |d kD||d kD<d|zdz|dk\||dk\<|S) Nrririxiiirr)tvs r"phizF_7..phisp EAI1a4i#ad(*T!V3d:D@!b&I!b& UQYQ'!q& r$g@r'rgqh?)rrArqr)rr rrys r"r9r9sQ ! A 1QTT7A:  !AccF1SqS6'lS!ADqD']*V3AaddG!ADqD'lAaddG Hr$cZt|dzdtjgd|dzzS)Nrr)gMbP?rr'rrs r"r:r:s(Q 88MQT* ++r$c"tj|g}tjd|}|ddzdz |ddzdz z|d<|dddz dz ||dddzzdz z|dddzdz z|dd|ddz dz ||ddzzdz z|d<|S)Nrrrr'r)rArqrerxs r"r;r;s ! A !QA Q4719qtQwqy AaD2wzk!ma!B l1n,quaxz9AaG rUAXIaK!AbE1H*Q, &AbE Hr$c.tj|gSrZrrs r"r<r<rr$c>tjd|z||z z Sr|)rAlog)rr s r"r=r=s 66!a%=1Q3 r$c.tj|gSrZrrs r"r>r>rr$)r?numpyrAr numpy.testingrrscipy.optimizerrNrVrirur3r4r5r6rrr7r8r9r:r;r<r=r>rr$r"rsr/?DD !&H +       P   , r$