K i< zddlZddlddlmZmZmZmZmZmZm Z m Z m Z m Z m Z dZdZdZdZdZd Zd Zd Zy) N)*) SecantMuller BisectionIllinoisPegasusAndersonRidderANewtonNewtonMNewtonMDNewtonc dt_tdtdj dsJtt tdj t sJtt tdtdfj t sJtdtdj d sJd }tttttfD]*}t|d | }t||tkr*Jtttt t"t$t&fD]*}t|d | }t||tkr*Jd}t)t|ddtsJt)t*j|ddt*jsJt)t,j|ddt.sJt)t,j|ddt0sJt3j4t65t9d5tdtddddddddt3j4t65tdddddt3j4t65tdddddy#1swYpxYw#1swYtxYw#1swYQxYw#1swYyxYw)Ncd|zdz S)Nxs c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/mpmath/tests/test_rootfinding.pyztest_findroot.. ac!eg?rgQ @c||zdzSNrrs rrztest_findroot.. rry@@y?ct|SN)cosrs rrztest_findroot..s #a&r@solver)?r"c|dz dzSNrrs rrztest_findroot..s1q51*rrg|=)tolr%y?r(c|dzd|zz dzS)Nr(iClF[rrs rrztest_findroot.."sq!tgai/.@rz 5.278e+13z -5.278e+13)realimagcy)Nrrs rrztest_findroot..'rrcd|z dzdz S)Nrr)ps rrztest_findroot..+sAEB;?rg?)mpdpsfindrootmpfaesinpimpcr rr rr absepsrrrr r isinstanceivfpfloatcomplexpytestraises ValueErrorworkprec)fr$rs r test_findrootrGsI BF OSV , / / 55 5 CQ # #B '' ' C#a&#d), - 0 0 44 4 OSY / 2 22 66 6A67FG< Q6 *1Q4y369h Q 01Q4y3 A hq!/ 55 5 bkk!RUk3RVV << < bkk!QEk2E :: : bkk!Tuk5w ?? ? z "? a[ ? @k = ? ?? z ""q!" z "1*C011 ? ??? ""11s< J:&J.?J:(KK.J7 3J::KKKc,tddddk(sJy)Nc|dzdz S)Nr(rrrs rrz test_bisection../sad1fr)rr(bisectr#rr5rrrtest_bisectionrL-s $U( ;q @@ @rcZd}t|dd}t||tksJy)Nctgd|S)N)rrrr)polyvalrs rrztest_mnewton..2s ')A&rgmnewtonr#)r5r;r<rFrs r test_mnewtonrR1s)&AD+A qt9s??rcbd}t|dt}t||tksJy)Nc|dz dzS)Nr(drrs rrztest_anewton..7s1q53,rr%r#)r5r r;r<rQs r test_anewtonrV6s)ABw'A qt9s??rcbd}t|dt}t||tksJy)Ncd|zdzdzS)Nr(rrrs rrztest_muller..<s1q51*q.rr%r#)r5rr;r<rQs r test_mullerrY;s) ABv&A qt9s??rcttddD]tfddk(rJtdddk(sJy)Nrrc|dz zSrr)ris rrz#test_multiplicity..Bsq1uqjrc |dzSr'rrs rrz#test_multiplicity..Cs !Q$rr)range multiplicity)r\s@rtest_multiplicityr`@sC 1a[:0!4999:  *a // /rc`d}tt|dtddgddggz ddksJtt|ddd D]\}}t |d d ksJd ddfd}t |d}|Dcgc]}tt|c}ddgk(sJycc}w)Ncd|ddzzd|ddzzz dz |ddzd|dzz |ddzzd|dzzdz gS)Nrrr(rrrs rrFz test_multidimensional..fFsb!A$' !AaD!G)#A%qtQwq1v~adAg'=a!f'DQ'FGGr)rrcrrdrgHz>c"t|tSr )norminfrs rrz'test_multidimensional..Js DCLr)verbosergr(g+=c| d|zzSr'rrys rrz'test_multidimensional..Osqb1Q3hrc<|dz||dzdz zzd|zz |dzz S)Nr(rrrks rrz'test_multidimensional..Ps-q!taAl*QqS0q1u=rc*t|dz|dzzSr')sqrtrks rrz'test_multidimensional..Qsd1a4!Q$;'rcF||}|||z |||z fSr r)rrlf1xf1f2f3s rrFz test_multidimensional..fRs-Ah1a31a3//r) rurr) mnormjacobianmatrixrr3rgr5intround)rFrerrorr\rrrsrts @@@rtest_multidimensionalr|EsH !V$v!uaVn'==a @5 HH HRFA"8: 5   1q>E !! ! B =B 'B0 HA#$ %aCaM %!Q // / %sB+cLtdddk(sJtdddk(sJy)Ncy)Nrrrs rrztest_trivial..Yr/rrc|Sr rrs rrztest_trivial..ZsarrrKrrr test_trivialrXs, K #q (( ( K #q (( (r)rBmpmathmpmath.calculus.optimizationrrrrrr r r r r rrGrLrRrVrYr`r|rrrrrsM BBBB%1NA   0 0&)r