o *ήcog @spdZddlmZmZmZddlmZddlmZm Z m Z ddl m Z m Z mZddlmZddlmZmZddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1ddl2m3Z3m4Z4ddl5m6Z6ddl7m8Z8m9Z9ddl:m;Z;ddlZ>e d\ Z?Z@ZAZBZCZDZEZFZGZHddZIddZJddZKddZLdd ZMd!d"ZNd#d$ZOd%d&ZPd'd(ZQd)d*ZRd+d,ZSd-d.ZTd/d0ZUd1d2ZVd3d4ZWd5d6ZXd7d8ZYd9d:ZZd;d<Z[d=d>Z\d?d@Z]dAdBZ^dCdDZ_dEdFZ`dGdHZae9dIdJZbdKdLZcdMdNZddOdPZedQdRZfdSdTZgdUdVZhdS)WzBTests 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)PolynomialErrorUnsolvableFactorError)_nsort)raisesslow)verify_numericallyN)productza,b,c,d,e,q,t,x,y,zcCstdd|D}tt|d|t|d|ksJt|t|dD]}||d||ks@||dt||ks@dSq%dS)NcSsg|] }|jr dndqS)ris_real.0ir3G/tmp/pip-target-vg8gfxp4/lib/python/sympy/polys/tests/test_polyroots.py (sz_check..r-FT)sumlistsortedrangelenr )r$nrealixr3r3r4_check$s(r>cCs*ttdtdttddgksJdS)Nr6r-)rrxrr3r3r3r4test_roots_linear2s*rAcCsttdtdtddgksJttdtddtttdddgks)Jttdtddtt tddttddgksIJttdtddtdtdttdddttddgkspJttdtddtdttddtt 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 gks Jtt tddtdt t}t|tdt dd t dt tdt ddt dt gksFJttdt d dtt ddt}t|dt ddgkskJttdtddt}t|}|t|ksJtd tdd td }d d |jddDdd |jddDksJtdddD]!\}}}t|td|t|}t|}|t|ksJqdS)Nr6rr?r-iicSg|]}|dqSr6nr1wr3r3r4r5Pz(test_roots_quadratic..T)radicalscSrGrHrIrKr3r3r4r5QrMF)r6)rr?)rrr@rrrr> all_rootsaecdbyr(r,)fr_a_b_cr$r3r3r4test_roots_quadratic6s>"0@N&T TT "@ ,  r\c Csttdttdtt}t|ttdtdtddttd tdtddttdtdtd dttd tdtd diksSJdS)NrEr6r-)rr@rr$reqr3r3r4test_issue_7724Xs r`cstdtddgttttt}tddtddfdd|D}t|tttttks:Jt fdd |DsHJdS) Nr-rOrCr6csg|]}|tqSr3)subsrVr1rX)zr3r4r5esz#test_issue_8438..c3s0|]}tt|ijddddkVqdS)r6TchoprN)rbrVr@rJr0prdr3r4 is.z"test_issue_8438..) rrVr@as_exprrrrsetrball)r$postr3rgr4test_issue_8438as rncstdtddtttdd}t|sJttddtddtfddtdD}ttddtddt}||ksLJt|sRJtdtdd}t|sdJttdtd dd ksvJdS) NrEr-r6r]rDcg|]}t|qSr3rr0rWr3r4r5prMz#test_issue_8285.. )rr@rPr>r:r;)r$ror3rrr4test_issue_8285ls* "   (rucCsttddttdd}t|sJttddtddt}t|s/Jttdtd}t|sAJttddtddt}t|sXJdS)Nr6rErDrBr-)rr@rPr>r$r3r3r4test_issue_8289{s$ "  "rwcsttdddtdddtdddtdtdtdtdtdtgks1JtddtddfddtdD}t|ksRJt|dS)Nr-r6rErscrpr3rqr0rhr3r4r5rMz$test_issue_14291..)rr@rPrrr:r>)ansr3rxr4test_issue_14291s*0 rzcCs>ttdtttttdd}t|}t|dksJdS)NrBEXdomain)rrVr r@r$r;)r_roots_dr3r3r4test_issue_13340s"rcsxttdtdddttdddttdd td d ttt}tfd d |Ds:JdS)NrErB r6iiiii ic3s|] }|dkVqdSrNr3rcr^r3r4riz#test_issue_14522..)rr@rr$rl)roots_eqr3r^r4test_issue_14522sRrcCs@tttddtdttdt}|dtsJdS)NrErDr6rBr)r rtr@has)solr3r3r4test_issue_15076s*rcCsbttddtdtddtddtdtddtt}t|}d|vs/JdS)NrEror6rB@ir)rr@rr$)r_rr3r3r4test_issue_16589sJrc CslttdtdtgdksJtttddtddtdtgdks,Jttd}tttdtt||tj tdtd|tj tdtdgks[JtttddtdtjttddtjttddgksJttdtddtddtdtdtjdt ddddt dddksJtd dtddtd}t |ddd tt|tdd kr1t dddtd t t d td d dddtd t t d td d  dtddt dddtd t t d td d  dtddt ddgks4JJdS)Nr6rB)rrrr-r-r-r-r?rT)trigmultiple)r rorOrD)rrr@r rVrHalfrrrr$rrrr)rXr_r3r3r4test_roots_cubics,"6  ,0("  4>>rc sbtttdtgdksJtttdtdtgdgdgdgdfvs,Jtttdtdtgdgd gd gd fvsIJtttdtt}tjttdd tjttdd tjtj g}t |t d t |t d ks}Jt gdD]e\}\}}}}|d kr| |d td|td }n|dkr|||d t dd|td|td}td|td|td |t|ttt}t fdd|DsJqttttdtdtdd td t ddtt}t fdd|DsJtdddtdd tddtd tddttt} t fdd| DsMJtttdtdttt}t dd|DsnJtt ddt dddtt ddt dd dtt ddd df} | D]ttt}t fd!dt||DsJqdS)"NrErrrrrB)r?rrr)rr?rr)rrr?r)rrrr?)r-rrr)rr-rr)rrr-r)rrrr-r6key))r-r6rBr)rBi r-r6rBrEr)rrCrB)rCr]r)rDirCrorc3s(|]}t|jdddkVqdS)TrerN)rbr@rJ)r1air^r3r4ris&z%test_roots_quartic..r-c3s"|] }tt|dVqdSrr+rbr@r0r^r3r4ris rdT)negative r]cs(g|]}tt|fdffdqS)r?rrr0)r_rdr3r4r5s(z&test_roots_quartic..css|] }t|tkVqdSN)typerr0r3r3r4rirr?)rVrdrOcs$g|]\}}t||dqS)r)r+rb)r1rLs)repr3r4r5s$)r rr@rrrrZeroOner9hash enumeraterrlqrrVdictrbzip) lhsrhsr2rQrUrSrTryrzansrepsr3)r_rrdr4test_roots_quarticsT  6$2,< 0"&rc CsXtddtttdtdddtdtdtddttDr*JdS) Ncss|]}t|tVqdSr) isinstancerr0r3r3r4risz#test_issue_21287..rEr6rBr]r?r-)anyr rr@rr3r3r3r4test_issue_21287s <rc CsttdtdddgksJttdtdddgksJttdtddtddttddtddttddgksAJttdtddt tgksQJttdtddtjttddtjttddgksrJttd tddtt d  tt t d tt d  tt t d tt tdd  tt t tdd tt tdd  tt t tdd tt tdd tt t tdd tt tdd tt t tdd gksJttd tddtd dttddtd dttddtddttddtddttddgks-Jttd tddtd dtdtd dtdtddtdtddtdgksaJttdtdddd dgksrJttdtdddd dgksJttdtdddd t dd dt ddgksJttdtdddd t tgksJttd tdddd t dd  t dd d t dd ddt dd dt dd t dd dgksJttdtdddd dt ddt ddgksJdS)Nr-Tpolysr6r?rBrErDraro )factorr]) r!rr@rrrrrrrrr r3r3r3r4test_roots_cyclotomicsn2 *((&&        N rc Cs:ttdttdgksJttdtdtgdksJttdtdttddgks2Jdtddd}ttdtddt| |t| |t||t||tgksbJttttd dtd d d }td d d }tt|td|t}tt|td|t}t|dt|dksJt|dt|dksJt ddddD]'\}}}}||kr|dkrqt|t|||} t| } | t | ksJqt tdtddt dtdt tddt dtdddt ddt tddt dtddiksJdS)Nr]rrErr6rOrsrBroa1T) nonnegativeb1r-)r-r6)r?r-)r6rBrEr]rr?)r"rr@rrr>rrrr,r(r$rVr) Arrr0r1rQrUrrJrhryr3r3r4test_roots_binomial-s2"&.   ""rcCs`tttdtt}tt|t\}}|dksJ|ttttdtttks.Jtdtdtdtdttt}tt|t\}}|dtksUJ|ttdtdtttkshJtdtdtdtdt}tt|t\}}|dtksJ|ttdtdttksJtdtdttt}tt|t\}}|dtksJ|ttdtttksJtdtdt}tt|t\}}|dtksJ|ttdttksJtd\}}}}d|d|d|dd|t|d |d |d d |d |d|d td|d d|d|d|dtd|dd|d|d|dtd|dd|d|d|dtd|d d|d|d |dtd d|dd|||d td d|dd|dtdd}tt|t\}}|d||||dksJ|dtddtd dtd dtddtddtddtddtd ksJttd tdtttt tttfd!}ttd tttt ttd!}t|t|fks.JdS)"Nr6r-rBE,F,J,L[r|)s2^ror83ll8ra]$=,yrDr/@:r]rsD!^rEq Pyqidi Bi߰lul/1g&l glFW&kr|) rQrVr@rUr#rrSrr r)rWcoeffpolyEFJLgr3r3r4test_roots_preprocessingNsh &,&$""****.& "<   0$rcCs tdtiks JttttjdiksJttdttjdiks#Jttdtdtdttd dtddtddiksEJtdtdttdddiksWJtdtddttdddikskJtdtddttdddiksJtdtddttdddiksJttddttdtjdtj dt diksJttdddttdtjdtj dt diksJtdtddttdddiksJtdtddttd ddiksJtdtddttdddiks Jtdtddttd ddiks!Jtdtddttdddiks8Jtdtddttd ddiksOJtttt dtt tdikseJtttt dtt tdiks|Jttdt dttttdtjdiksJttddtddttjdtj diksJttd dtddtdtdttjddt ddtjddt ddiksJttd dtt ddtt dddt ddtt dddt d dtt dddt d dtt dddtjdtj dtdt diks2Jd tdd tddtddtddtd dtddtd dtd}t|tjdtd dtddtdddtd ddtdddtdddiksJttt t ttt t t ttt t t tt t diksJttdtdtdtddiksJttdtdtdtddtd dtddtddiksJttdtdtdtdtddtd dtddtddtddiksJttddtddtd ttd ddtddtdiks>Jttddtddtd tdddtddtdtd dikseJttdttddtddtd ttjdtjdtd ddtddtdiksJtjtdd}}dt dd|}d|d}|d}t dtd}||}| |}ttdtdtdtdd| ||d| ||||d| ||||diksJtddtdtdtddttdtd}dddD\} } t d} t|t| | t dd t| d| | t dd t| d| | t dd t| d| | t dd t| diksdJtdtdtdtd}ddd D\} } }t|t| | d|t|| d||d| | d|t|| d||d| | d|t|| d||d| | d|t|| d||diksJtddttdddtdtdttdtd}t|ttjdtjtjttjt ddttdd tddtjtjttjt ddttdd tddiksEJttt t tddtddtd tddikseJttt t tddtddtd tddiksJttddtd!d"tjdtj diksJttddtd#d"tdt diksJttdtdttjdtj diksJttdtdtd$d%d&tjdiksJttddtd!dd'tj tjgksJttddtd#dd'tt gks Jtd(dd)\}}td||ddt||d}t|td*d"d||diks7Jttdtdd+tjtjtjgksKJtd,tdd+gksWJtd tdtdtdtdtd}t|t ttdttddt tttddtttdddt tttddtttdddtttdttddttttddtttdddttttddtttdddiksJtdd|d}t|t ttdttddt tttddtttdddt tttddtttdddtttdttddttttddtttdddttttddtttdddt dtdikspJttdd-td}d.d|Dd/}d0d t d1}|dt|ddd2t|dksJtddt dtdd3dt dtdt d dtdtd4d5}t|dt ddddt ddt d ddiksJtd6tdd7t dtdd8tdd0t dtdd9tdd:t dtdd;tddtd4d5}t|t d ddddt dddt ddd?dt ddd dt ddiksiJttddtdd t dtdd t dtd@tdAdAt dtd4d5}t|dt ddddt ddddt dddiksJtttt dddtd4d5t d tddtj t dtddt d tddtj t dtddt d tdddik sJdS)BNr-rr6rBrEr?r]rsrCrDroi iii irairrF)cubicsrOT!cSg|]}t|qSr3rrcr3r3r4r5s ztest_roots0..))rr)r-rcSrr3rrcr3r3r4r5))r-rE)r-ro)r]roZfilterrcSs|jSr) is_positive)rXr3r3r4sztest_roots0..) predicate)rrza, b)realRri(cSsg|]}|jr|qSr3r.)r1rxr3r3r4r5 rMrli2rrr{r|)}-rr)r$r@rrexpandrrrrQrU NegativeOnerrSrTrrbrdrVrrrrr r)rWr1_2r1_3x0x1x2x3x4x5r13_20r1_20s2r1_4r1_8r5_8arbrrhrrX real_rootcrr_r3r3r4 test_roots0sXD$(((26,,....,.44,&  :   ( L, ,$ J( 0(     <  **** D:: @@.*. ,(($(0**((  **((   *P<@  0  D B ,,rcstd\}}}}d||||||}d|||||||||||||||}tt|ddgksNJtt|ddgks]Jtd\}}}} } } } || || || | | | | | | } || | | | || | | | || }tt| || gdksJddddtt}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,krrBrocss|]}|VqdSrr3r0r3r3r4riJsz"test_roots_slow..cs g|] }|jddqS)Tre)rbrJ)r1rirWr@r3r4r5Js z#test_roots_slow..N)rr8r$valueskeysr)rQrUrSrTf1f2zzyyxxzyzxyxke1e2rr3r r4test_roots_slow6s$H00"*rcCsttdtdtdd}ttdtdtdd}t||D]\}}t||dks-Jqtddtd td d d tdtdd dtdd}t|dd}d}t||D]\}}t||dksmJq_dS)Nr6r-Trg?g-q=rEg@g@rBgS@g8@gb@g"@)g؋~)g)c g3! @g @绽|=)r$r@rabsr)R1R2rr2rWr3r3r4test_roots_inexactMs2 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]\} } | |} | durt| || dksJqdS)NrrrorrrarrrDr6rrr]rBrsrrErrrrrrTr)g?|YcgOXLzGX@g'I^g@g |۰n@gdZw@g5`e@gS$Ȫ@gJm@rLr) rr@r$evalfrr;rmatchr) rrrrrWrrrLrhrrrr3r3r4test_roots_preprocessed_s8"****.&  &r cCsttddtdddddiksJttddtdddddiks(JttddtdtddtdddddiksFJttd d dS) Nr6r-FstrictTrDr]rBcSs0ttddtdtddtdddS)NrDr6r]rBTr!)r$r@r3r3r3r4rs0z#test_roots_strict..)r$r@r)r'r3r3r3r4test_roots_strict|s((<r#c Csddtdtddtddtd}t|d d \}}t|}t|d d }d d |D}dd |D}||}dd |D}t|ddd}t|ddd}||fD]"}t||D]\}}|jrl||vskJq^t|t|f|vsxJq^qWdS)Nipiir6i rB1rET)rlrcSsg|] \\}}}t||qSr3rr1rQrU_r3r3r4r5sz$test_roots_mixed..cSs8g|]\\}}}tt|t|tt|t|qSr3)rr r r%r3r3r4r5s(cSsg|]}|qSr3)rrcr3r3r4r5rcS|Srsort_keyr@r3r3r4rz"test_roots_mixed..rcSr'rr(r*r3r3r4rr+) r@rrr$r9rr/r r ) rW_re_im_nroots_sroots _intervals_rootsr2rXr3r3r4test_roots_mixeds&0  r2c CsttdttdtgksJtttttttgksJttddttdtdgks1JttdttttttttgksHJttdddtdtdtdtdttttttttgksnJtttddtddttdtttdtttddtgksJtdtddtddtdtdtd dtttddtddtddgksJdS) Nr-r6rErrrorrDQ)rrr@rVrrr3r3r3r4test_root_factorss&.2*4( r4csRd}t|tddttjjfddjdd}dd |Dgd ks'JdS) NrTrcsjdddS)NrBr])rJmaxsteps)rr3rxr3r4rsztest_nroots1..rBrIcSg|]}t|qSr3strrcr3r3r4r5rz 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)mpmathmp NoConvergencer)rJr$r3rxr4 test_nroots1s   r<cCsjttddtdt}|jdd}dd|Dgdks J|jdd}dd|Dgd ks3JdS) Nr]rBr-rIcSr6r3r7rcr3r3r4r5rz test_nroots2..)z-0.332z-0.839 - 0.944*Iz-0.839 + 0.944*Iz1.01 - 0.937*Iz1.01 + 0.937*IcSr6r3r7rcr3r3r4r5r)z-0.33199z-0.83907 - 0.94385*Iz-0.83907 + 0.94385*Iz1.0051 - 0.93726*Iz1.0051 + 0.93726*I)rr@r)rhr$r3r3r4 test_nroots2s     r=cCs>ttttdtdtttttdddksJdS)NrBr6T) composite)r;r$rrVrr@r3r3r3r4test_roots_composite>r?cs.ttdttdttfdddS)NrBr-cstSrrvr3r^r3r4rr+z"test_issue_19113..)rr@r)r&r3r3r^r4test_issue_19113srAcCs>tddddtdddtdgd d ddgksJdS) Nr-rCrrEr6rorrTr)r$rr3r3r3r4test_issue_17454r@rBcCsNttdtdgksJttddtdtdd gks%JdS)Nl l rBrEr6)rr@ real_rootsrr3r3r3r4test_issue_20913s2rDc Cstdd}dt|tdd|}ttttdtddt|d| dtdtdd|dtdtddiksCJdS)Nr-rBr?r]r6)rrQr$rr@rr)rRrXr3r3r4test_issue_22768s  rE)i__doc__sympy.core.numbersrrrsympy.core.singletonrsympy.core.symbolrrr$sympy.functions.elementary.complexesr r r &sympy.functions.elementary.exponentialr (sympy.functions.elementary.miscellaneousr r$sympy.functions.elementary.piecewiser(sympy.functions.elementary.trigonometricrrrsympy.polys.domains.integerringrsympy.sets.setsrsympy.simplify.powsimpr sympy.polysrrrrrsympy.polys.polyrootsrrrrr r!r"r#r$sympy.polys.orthopolysr%sympy.polys.polyerrorsr&r'sympy.polys.polyutilsr(sympy.testing.pytestr)r*sympy.core.randomr+r9 itertoolsr,rQrUrSrTrRrrr@rVrdr>rAr\r`rnrurwrzrrrrrrrrrrrrrr r#r2r4r<r=r?rArBrDrEr3r3r3r4sp      ,    "    :0!<-