g< 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 R/var/www/openai/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..s 1q51*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..'s2rrcd|z dzdz S)Nrr)ps rrztest_findroot..+sAEB;?rg?)mpdpsfindrootmpfaesinpimpcr rr rr absepsrrrr r isinstanceivfpfloatcomplexpytestraises ValueErrorworkprec)fr$rs r test_findrootrFs  BF OSV , / / 55 5 CQ # #B '' ' C#a&#d), - 0 0 44 4 OSY / 2 22 66 6A67FG< Q6 *1Q4y3=69h Q 01Q4y3 A hq!/ 55 5 bkk!RUk3RVV << < bkk!QEk2E :: : bkk!Tuk5w ?? ? z " a[ @k = ? # z "q! # z "*C0 # "[ # " # " # "s< J:&J.?J:(KK.J7 3J::KKKc,tddddk(sJy)Nc|dzdz S)Nr(rrrs rrz test_bisection../s ad1fr)rr(bisectr#rr4rrrtest_bisectionrK-s $U( ;q @@ @rcZd}t|dd}t||tksJy)Nctgd|S)N)rrrr)polyvalrs rrztest_mnewton..2s ')A&rgmnewtonr#)r4r:r;rErs r test_mnewtonrQ1s)&AD+A qt9s??rcbd}t|dt}t||tksJy)Nc|dz dzS)Nr(drrs rrztest_anewton..7s 1q53,rr%r#)r4r r:r;rPs r test_anewtonrU6s)ABw'A qt9s??rcbd}t|dt}t||tksJy)Ncd|zdzdzS)Nr(rrrs rrztest_muller..<s1q51*q.rr%r#)r4rr:r;rPs r test_mullerrX;s) ABv&A qt9s??rcttddD]tfddk(rJtdddk(sJy)Nrrc|dz zSrr)ris rrz#test_multiplicity..Bs q1uqjrc |dzSr'rrs rrz#test_multiplicity..Cs!Q$rr)range multiplicity)r[s@rtest_multiplicityr_@s> 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 rrEz test_multidimensional..fFsb!A$' !AaD!G)#A%qtQwq1v~adAg'=a!f'DQ'FGGr)rrbrrcrgHz>c"t|tSr )norminfrs rrz'test_multidimensional..Js DCLr)verboserfr(g+=c| d|zzSr'rrys rrz'test_multidimensional..Osqb1Q3hrc<|dz||dzdz zzd|zz |dzz S)Nr(rrrjs rrz'test_multidimensional..Ps+q!taAl*QqS0q1u=rc*t|dz|dzzSr')sqrtrjs rrz'test_multidimensional..Qsd1a4!Q$;'rcF||}|||z |||z fSr r)rrkf1xf1f2f3s rrEz test_multidimensional..fRs-Ah1a31a3//r) rtrr) mnormjacobianmatrixrr2rfr4intround)rErerrorr[rqrrrss @@@rtest_multidimensionalr{EsH !V$v!uaVn'==a @5 HH HRFA"8:5 : 1q>E !! ! B =B 'B0 HA#$ %1aCaM1 %!Q // / %sB+cLtdddk(sJtdddk(sJy)Ncy)Nrrrs rrztest_trivial..Ysarrc|Sr rrs rrztest_trivial..ZsarrrJrrr test_trivialrXs, K #q (( ( K #q (( (r)rAmpmathmpmath.calculus.optimizationrrrrrr r r r r rrFrKrQrUrXr_r{rrrrrsM BBBB%1NA   0 0&)r