a Z?Z@ZAZBZCZDZEZFZGddZHddZIddZJddZKdd ZLd!d"ZMd#d$ZNd%d&ZOd'd(ZPd)d*ZQd+d,ZRd-d.ZSd/d0ZTd1d2ZUd3d4ZVd5d6ZWd7d8ZXd9d:ZYd;d<ZZd=d>Z[d?d@Z\dAdBZ]dCdDZ^dEdFZ_e8dGdHZ`dIdJZadKdLZbdMdNZcdOdPZddQdRZedS)SzBTests for algorithms for computing symbolic roots of polynomials. )IRationalpi)S)SymbolWildsymbols) conjugateimre)exp)rootsqrt) Piecewise)acoscossin)ZZInterval)powsimp)Polycyclotomic_poly intervalsnrootsrootof) root_factors roots_linearroots_quadratic roots_cubic roots_quarticroots_cyclotomicroots_binomialpreprocess_rootsroots) legendre_poly)PolynomialError)_nsort)raisesslow)verify_numericallyN)productza,b,c,d,e,q,t,x,y,zcCstdd|D}tt|d|t|d|ks:Jt|t|dD]6}||d||ksJ||dt||ksJdSqJdS)NcSsg|]}|jrdndqS)ris_real.0ir2p/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/polys/tests/test_polyroots.py 'z_check..r,FT)sumlistsortedrangelenr )r$Znrealixr2r2r3_check#s(r=cCs*ttdtdttddgks&JdS)Nr6r,)rrxrr2r2r2r3test_roots_linear1sr@cCsttdtdtddgks"JttdtddtttdddgksRJttdtddtt tddttddgksJttdtddtdtdttdddttddgksJttdtddtdttddtt dt t tt tt t tt dt t dtt }tt|tt tt tt ttt t t tt t t dtt t dtt t tt tt ttt t t tt t t dtt t dtt gksJtt tddtdt t}t|tdt dd t dt tdt ddt dt gksJttdt d dtt ddt}t|dt ddgksJttdtddt}t|}|t|ks Jtd tdd td }d d |jddDdd |jddDksXJtdddD]B\}}}t|td|t|}t|}|t|ksdJqddS)Nr6rr>r,iicSsg|]}|dqSr6nr0wr2r2r3r4Or5z(test_roots_quadratic..T)ZradicalscSsg|]}|dqSrFrGrIr2r2r3r4Pr5F)r6)rr>)rrr?rrrr= all_rootsaecdbyr'r+)frZ_a_bZ_cr$r2r2r3test_roots_quadratic5s<"0@N&T TT "@ ,  rVc Csttdttdtt}t|ttdtdtddttd tdtddttdtdtd dttd tdtd diksJdS)NrDr6r,)rr?rr$reqr2r2r3test_issue_7724Ws rZcstdtddgttttt}tddtddfdd|D}t|tttttkstJt fdd |DsJdS) Nr,rKrBr6csg|]}|tqSr2)subsrRr0rT)zr2r3r4dr5z#test_issue_8438..c3s.|]&}tt|ijddddkVqdS)r6TZchoprN)r\rRr?rHr/pr^r2r3 hr5z"test_issue_8438..) rrRr?Zas_exprrrrsetr\all)r$postr2r`r3test_issue_8438`srfcstdtddtttdd}t|s6JttddtddtfddtdD}ttddtddt}||ksJt|sJtdtdd}t|sJttdtd dd ksJdS) NrDr,r6rWrCcsg|]}t|qSr2rr/rSr2r3r4or5z#test_issue_8285.. )rr?rLr=r:r;)r$ror2rir3test_issue_8285ks* "   rlcCsttddttdd}t|s0Jttddtddt}t|s^Jttdtd}t|sJttddtddt}t|sJdS)Nr6rDrCrAr,)rr?rLr=r$r2r2r3test_issue_8289zs$ "  "rncsttdddtdddtdddtdtdtdtdtdtgksbJtddtddfddtdD}t|ksJt|dS)Nr,r6rDrjcsg|]}t|qSr2rhr/rar2r3r4r5z$test_issue_14291..)rr?rLrrr:r=)ansr2ror3test_issue_14291s.,rqcCs>ttdtttttdd}t|}t|dks:JdS)NrAEXdomain)rrRr r?r$r;)rYZroots_dr2r2r3test_issue_13340s"rucsxttdtdddttdddttdd td d ttt}tfd d |DstJdS)NrDrA r6iiiii ic3s|]}|dkVqdSrNr2r]rXr2r3rbr5z#test_issue_14522..)rr?rr$rd)roots_eqr2rXr3test_issue_14522sRrzcCs@tttddtdttdt}|dtsrT)trigmultiple)r rgrKrC)rrr?r rRrHalfrrrr$rrrr)rTrYr2r2r3test_roots_cubics,"6  ,0( "  4>>rc sjtttdtgdksJtttdtdtgdgdgdgdfvsXJtttdtdtgdgd gd gd fvsJtttdtt}tjttdd tjttdd tjtj g}t |t d t |t d ksJt gdD]\}\}}}}|d krD| |d td|td }n<|dkr|||d t dd|td|td}td|td|td |t|ttt}t fdd|DsJqttttdtdtdd td t ddtt}t fdd|Ds:Jtdddtdd tddtd tddttt} t fdd| DsJtttdtdttt}t dd|DsJtt ddt dddtt ddt dd dtt ddd df} | D]<ttt}t fd!dt||Ds(Jq(dS)"NrDrrrrrA)r>rrr)rr>rr)rrr>r)rrrr>)r,rrr)rr,rr)rrr,r)rrrr,r6key))r,r6rAr)rAi r,r6rArDr)rrBrA)rBrWr)rCirBrgrvc3s&|]}t|jdddkVqdS)Tr_rN)r\r?rH)r0ZairXr2r3rbr5z%test_roots_quartic..r,c3s |]}tt|dVqdSrxr*r\r?r/rXr2r3rbr5r^T)negative rWcs(g|] }tt|fdffdqS)r>rrr/)rYr^r2r3r4r5z&test_roots_quartic..css|]}t|tkVqdSN)typerr/r2r2r3rbr5r>)rRr^rKcs$g|]\}}t||dqS)r)r*r\)r0rJs)repr2r3r4r5)r rr?rrrrZeroOner9hash enumeraterrdqrrRdictr\zip) lhsrhsr1rMrQrOrPrpr|ZzansZrepsr2)rYrr^r3test_roots_quarticsR  6 $ 2, < 0"rc CsXtddtttdtdddtdtdtddttDrTJdS) Ncss|]}t|tVqdSr) isinstancerr/r2r2r3rbr5z#test_issue_21287..rDr6rArWr>r,)anyr rr?rr2r2r2r3test_issue_21287s Jttdtdddd t tgksdJttd tdddd t dd  t dd d t dd ddt dd dt dd t dd dgksJttdtdddd dt ddt ddgksJdS)Nr,TZpolysr6r>rArDrCr[rg )ZfactorrW) r!rr?rrrrrrrrr r2r2r2r3test_roots_cyclotomicsn2 *((&&       N rc CsHttdttdgksJttdtdtgdks>JttdtdttddgksdJdtddd}ttdtddt| |t| |t||t||tgksJttttd dtd d d }td d d }tt|td|t}tt|td|t}t|dt|dksFJt|dt|dksdJt ddddD]X\}}}}||kr|dkrqrt|t|||} t| } | t | ksrJqrt tdtddt dtdt tddt dtdddt ddt tddt dtddiksDJdS)NrWrrDrr6rKrjrArga1T)Z nonnegativeb1r,)r,r6)r>r,)r6rArDrWrvr>)r"rr?rrr=rrrr+r'r$rRr) ArrZr0r1rMrQrrHrarpr2r2r3test_roots_binomial,s2"&.   ""rcCsltttdtt}tt|t\}}|dks6J|ttttdtttks\Jtdtdtdtdttt}tt|t\}}|dtksJ|ttdtdtttksJtdtdtdtdt}tt|t\}}|dtksJ|ttdtdttkstd4d5}t|t d ddddt dddt ddd?dt ddd dt ddiksJttddtdd t dtdd t dtd@tdAdAt dtd4d5}t|dt ddddt ddddt dddiksxJtttt dddtd4d5t d tddtj t dtddt d tddtj t dtddt d tdddiksJdS)BNr,rr6rArDr>rWrjrBrCrgi iii ir[ir~rF)ZcubicsrKT!cSsg|] }t|qSr2rr]r2r2r3r4sztest_roots0..))rr)r,rcSsg|] }t|qSr2rr]r2r2r3r4r5))r,rD)r,rg)rWrgZfilterrcSs|jSr)Z is_positive)rTr2r2r3r5ztest_roots0..) predicate)rrza, b)realRri(cSsg|]}|jr|qSr2r-)r0rxr2r2r3r4r5rli2rrrrrs)}-rr)r$r?rrexpandrrrrMrQZ NegativeOnerrOrPrr\r^rRrrrrr r)rSZr1_2Zr1_3Zx0x1Zx2Zx3Zx4Zx5Zr13_20Zr1_20s2Zr1_4Zr1_8Zr5_8arbrrarrTZ real_rootZcrrYr2r2r3 test_roots0sXD$((*48......,.44,&  :   ( L, ,$ J( 0(     <   **** D:: @@.*. ,(($(0**(( **(( *P<@  0 D B ,,rcstd\}}}}d||||||}d|||||||||||||||}tt|ddgksJtt|ddgksJtd\}}}} } } } || || || | | | | | | } || | | | || | | | || }tt| || gdksTJddddtt}tdd fd d |DDrJd S) z6Just test that calculating these roots does not hang. z a,b,c,d,xr6r,zzz,yy,xx,zy,zx,yx,krrArgcss|] }|VqdSrr2r/r2r2r3rbIr5z"test_roots_slow..cs g|]}|jddqS)Tr_)r\rH)r0rirSr?r2r3r4Ir5z#test_roots_slow..N)rr8r$valueskeysr)rMrQrOrPf1f2zzyyxxZzyZzxZyxke1e2rr2rr3test_roots_slow5s$H00$rcCsttdtdtdd}ttdtdtdd}t||D]\}}t||dks>Jq>tddtd td d d tdtdd dtdd}t|dd}d}t||D]\}}t||dksJqdS)Nr6r,Trg?g-q=rDg@g@rAgS@g8@gb@g"@)g؋~)g)c g3! @g @绽|=)r$r?rabsr)R1R2rr2rSr2r2r3test_roots_inexactLs2 rc Cstd\}}}}d|d|d|dd|t|d|d|dd|d |d |d td |d d |d |d|d td|dd|d|d|dtd|dd|d|d |dtd |d d|d |d |d td d |dd|||dtdd|d d|dtdd}t|tiksJt|tdd}gd}td}|||||d }t|t|ksJt||D]8\} } | |} | durt| || dksJqdS)Nrrrgrvrr[rrrCr6rrrWrArjrrDrrrrrrTr)g?|YcgOXLzGX@g'I^g@g |۰n@gdZw@g5`e@gS$Ȫ@gJm@rJr) rr?r$evalfrr;rmatchr) rrrrrSrrrJrarrrr2r2r3test_roots_preprocessed^s6"****.&  rc Csddtdtddtddtd}t|d d \}}t|}t|d d }d d |D}dd |D}||}dd |D}t|ddd}t|ddd}||fD]D}t||D]4\}}|jr||vsJqt|t|f|vsJqqdS)Nipiir6i rA1rDT)rdrcSsg|]\\}}}t||qSr2rr0rMrQ_r2r2r3r4r5z$test_roots_mixed..cSs8g|]0\\}}}tt|t|tt|t|qSr2)rr r rr2r2r3r4scSsg|] }|qSr2)rr]r2r2r3r4r5cSs|Srsort_keyr?r2r2r3rr5z"test_roots_mixed..rcSs|Srrrr2r2r3rr5) r?rrr$r9rr.r r ) rS_reZ_imZ_nrootsZ_srootsZ _intervalsZ_rootsr1rTr2r2r3test_roots_mixed{s"0  rc CsttdttdtgksJtttttttgks.rArGcSsg|] }t|qSr2strr]r2r2r3r4r5z test_nroots1..)@z-0.999z-0.996z-0.991z-0.983z-0.973z-0.961z-0.946z-0.930z-0.911z-0.889z-0.866z-0.841z-0.813z-0.784z-0.753z-0.720z-0.685z-0.649z-0.611z-0.572z-0.531z-0.489z-0.446z-0.402z-0.357z-0.311z-0.265z-0.217z-0.170z-0.121z-0.0730z-0.0243z0.0243z0.0730z0.121z0.170z0.217z0.265z0.311z0.357z0.402z0.446z0.489z0.531z0.572z0.611z0.649z0.685z0.720z0.753z0.784z0.813z0.841z0.866z0.889z0.911z0.930z0.946z0.961z0.973z0.983z0.991z0.996z0.999)r%r?r(mpmathmpZ NoConvergencer)rHr$r2ror3 test_nroots1s  rcCsjttddtdt}|jdd}dd|Dgdks@J|jdd}dd|Dgd ksfJdS) NrWrAr,rGcSsg|] }t|qSr2rr]r2r2r3r4r5z test_nroots2..)z-0.332z-0.839 - 0.944*Iz-0.839 + 0.944*Iz1.01 - 0.937*Iz1.01 + 0.937*IcSsg|] }t|qSr2rr]r2r2r3r4r5)z-0.33199z-0.83907 - 0.94385*Iz-0.83907 + 0.94385*Iz1.0051 - 0.93726*Iz1.0051 + 0.93726*I)rr?r)rar$r2r2r3 test_nroots2s    rcCs>ttttdtdtttttdddks:JdS)NrAr6T)Z composite)r;r$rrRrr?r2r2r2r3test_roots_compositesrcs.ttdttdttfdddS)NrAr,cstSrrmr2rXr2r3rr5z"test_issue_19113..)rr?r(r&r2r2rXr3test_issue_19113srcCs>tddddtdddtdgd d ddgks:JdS) Nr,rBrrDr6rgrrTr)r$rr2r2r2r3test_issue_17454srcCsNttdtdgksJttddtdtdd gksJJdS)Nl l rArDr6)rr?Z real_rootsrr2r2r2r3test_issue_20913sr )f__doc__Zsympy.core.numbersrrrZsympy.core.singletonrZsympy.core.symbolrrrZ$sympy.functions.elementary.complexesr r r Z&sympy.functions.elementary.exponentialr Z(sympy.functions.elementary.miscellaneousr rZ$sympy.functions.elementary.piecewiserZ(sympy.functions.elementary.trigonometricrrrZsympy.polys.domains.integerringrZsympy.sets.setsrZsympy.simplify.powsimprZ sympy.polysrrrrrZsympy.polys.polyrootsrrrrr r!r"r#r$Zsympy.polys.orthopolysr%Zsympy.polys.polyerrorsr&Zsympy.polys.polyutilsr'Zsympy.testing.pytestr(r)Zsympy.core.randomr*r itertoolsr+rMrQrOrPrNrr{r?rRr^r=r@rVrZrfrlrnrqrurzr}rrrrrrrrrrrrrrrrrrr r2r2r2r3sj      ,     "    :0!<-