o "`< @sddlZddlTddlmZmZmZmZmZmZm Z m Z m Z m Z m Z ddZddZdd Zd d Zd d ZddZddZddZdS)N)*) SecantMuller BisectionIllinoisPegasusAndersonRidderANewtonNewtonMNewtonMDNewtonc Cs>dt_tddtddsJtttdtsJtttdtdfts-Jtddtd d s;Jd d}tt t t t fD]}t|d |d }t ||tksYJqFt t tttttfD]}t|d|d }t ||tksvJqcdd}tt|dddtsJttj|dddtjsJttj|dddtsJttj|dddtsJtt'tdtddtdddWdn1swYWdn1swYtttdddWdn1swYtttdddWddS1swYdS)NcSs d|dS)Nxrr?/usr/lib/python3/dist-packages/mpmath/tests/test_rootfinding.py  ztest_findroot..g?rgQ @cSs ||dSNrrrrrr ry@@y?cSst|SN)Zcosrrrrr@solver)?rcSs |ddSNrrrrrrrrg|=)Ztolry?r!cSs|dd|dS)Nr!iClF[rrrrrr"z 5.278e+13z -5.278e+13)realimagcSdS)Nrrrrrr'rcSsd|ddS)Nrr)prrrr+g?)mpZdpsfindrootZmpfZaeZsinZpiZmpcr rr rr absepsrrrrr isinstanceZivfpfloatcomplexpytestZraises ValueErrorZworkprec)frrrrr test_findrootsB       $r6cCstddddddks JdS)NcSs |ddS)Nr!rrrrrrr/rz test_bisection..)rr!Zbisectrrr,rrrrtest_bisection-sr8cCs.dd}t|ddd}t||tksJdS)NcSstgd|S)N)rrrr)Zpolyvalrrrrr2ztest_mnewton..gZmnewtonr)r,r-r.r5rrrr test_mnewton1r;cC.dd}t|dtd}t||tksJdS)NcSs |ddS)Nr!drrrrrr7rztest_anewton..rr)r,r r-r.r:rrr test_anewton6r<r?cCr=)NcSsd|ddS)Nr!rrrrrrr<r*ztest_muller..rr)r,rr-r.r:rrr test_muller;r<r@csDtddD]tfdddksJqtddddks JdS)Nrrcs |dSrrrirrrBrz#test_multiplicity..cSs|dSr rrrrrrCrr)rangeZ multiplicityrrrArtest_multiplicity@srDcsdd}tt|dtddgddggdd ksJtt|ddd d d D]\}}q%t||d dks5Jdd dd dd fdd}t|d}dd|Dddgks[JdS)NcWsXd|ddd|ddd|ddd|d|ddd|ddgS)Nrrr!rrrrrrr5FsXz test_multidimensional..f)rrErrFrgHz>cSs t|tSr)norminfrrrrrJs z'test_multidimensional..)verboserHr!g+=cSs| d|Sr rryrrrrOr9cSs(|d||ddd||dS)Nr!rrrKrrrrPs(cSst|d|dSr )ZsqrtrKrrrrQr"cs&||}||||||fSrr)rrLZf1xf1f2Zf3rrr5Rs ) rPcSsg|]}tt|qSr)intround).0rBrrr Vsz)test_multidimensional..rr)ZmnormZjacobianZmatrixr r+rHr,)r5rerrorrrMrtest_multidimensionalEs,  rVcCs0tddddks JtddddksJdS)NcSr%)NrrrrrrrYr'ztest_trivial..rcSs|SrrrrrrrZr'rr7rrrr test_trivialXsrW)r3ZmpmathZmpmath.calculus.optimizationrrrrrrr r r r r r6r8r;r?r@rDrVrWrrrrs4'