o GZhh @s|dZddlmZmZmZddlmZddlmZm 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/m0Z0ddl1m2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9m:Z:ddl;mm?Z?e d\ Z@ZAZBZCZDZEZFZGZHZIddZJddZKddZLddZMdd ZNd!d"ZOd#d$ZPd%d&ZQd'd(ZRd)d*ZSd+d,ZTd-d.ZUd/d0ZVd1d2ZWd3d4ZXd5d6ZYd7d8ZZd9d:Z[d;d<Z\d=d>Z]d?d@Z^dAdBZ_dCdDZ`dEdFZadGdHZbdIdJZce:dKdLZddMdNZedOdPZfdQdRZgdSdTZhdUdVZidWdXZjdS)YzBTests 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_quartic roots_quinticroots_cyclotomicroots_binomialpreprocess_rootsroots) legendre_poly)PolynomialErrorUnsolvableFactorError)_nsort)raisesslow)verify_numericallyN)productza,b,c,d,e,q,t,x,y,zcCstdd|D}t|d|t|d|ksJt|t|dD]}||d||ks>||dt||ks>dSq#dS)Ncss|] }|jr dndVqdS)rNis_real.0ir4O/var/www/auris/lib/python3.10/site-packages/sympy/polys/tests/test_polyroots.py (sz_check..r.FT)sumsortedlistrangelenr )r%ZnrealZixr4r4r5_check$s$r=cCs*ttdtdttddgksJdS)Nr7r.)rrxrr4r4r4r5test_roots_linear2s*r@cCsttdtdtddgksJttdtddtttdddgks)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)Nr7rr>r.iicSg|]}|dqSr7nr2wr4r4r5 Pz(test_roots_quadratic..T)ZradicalscSrFrGrHrJr4r4r5rLQrMF)r7)rr>)rrr?rrrr= all_rootsaecdbyr)r-)frZ_a_bZ_cr%r4r4r5test_roots_quadratic6s>"0@N&T TT "@ ,  rYc Csttdttdtt}t|ttdtdtddttd tdtddttdtdtd dttd tdtd diksSJdS)NrDr7r.)rr?rr%reqr4r4r5test_issue_7724Xs r]cstdtddgttttt}tddtddfdd|D}t|tttttks:Jt fdd |DsHJdS) Nr.rNrBr7csg|]}|tqSr4)subsrUr2rW)zr4r5rLesz#test_issue_8438..c3s0|]}tt|ijddddkVqdS)r7TZchoprN)r_rUr?rIr1prar4r5r6is.z"test_issue_8438..) rrUr?Zas_exprrrrsetr_all)r%postr4rcr5test_issue_8438as rhcstdtddtttdd}t|sJttddtddtfddtdD}ttddtddt}||ksLJt|sRJtdtdd}t|sdJttdtd dd ksvJdS) NrDr.r7rZrCcg|]}t|qSr4rr1rVr4r5rLprMz#test_issue_8285.. )rr?rOr=r;r<)r%ror4rlr5test_issue_8285ls* "   (rocCsttddttdd}t|sJttddtddt}t|s/Jttdtd}t|sAJttddtddt}t|sXJdS)Nr7rDrCrAr.)rr?rOr=r%r4r4r5test_issue_8289{s$ "  "rqcsttdddtdddtdddtdtdtdtdtdtgks1JtddtddfddtdD}t|ksRJt|dS)Nr.r7rDrmcrjr4rkr1rdr4r5rLrMz$test_issue_14291..)rr?rOrrr;r=)ansr4rrr5test_issue_14291s*0 rtcCs>ttdtttttdd}t|}t|dksJdS)NrAEXdomain)rrUr r?r%r<)r\Zroots_dr4r4r5test_issue_13340s"rxcsxttdtdddttdddttdd td d ttt}tfd d |Ds:JdS)NrDrA r7iiiii ic3s|] }|dkVqdSrNr4r`r[r4r5r6z#test_issue_14522..)rr?rr%rf)roots_eqr4r[r5test_issue_14522sRr~cCs@tttddtdttdt}|dtsJdS)NrDrCr7rAr)r rtr?has)solr4r4r5test_issue_15076s*rcCsbttddtdtddtddtdtddtt}t|}d|vs/JdS)NrDrir7rA@ir)rr?rr%)r\r}r4r4r5test_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)Nr7rA)rrrr.r.r.r.r>rT)trigmultiple)r rirNrC)rrr?r rUrHalfrrrr%rrrr)rWr\r4r4r5test_roots_cubics,"6  ,0("  4>>rc s\tttdtgdksJtttdtdtgdgdgdgdfvs,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|DsnJt ddt dddt ddt dd dt ddddf} | D]ttt}t fd dt||DsJqdS)!NrDrrrrrA)r>rrr)rr>rr)rrr>r)rrrr>)r.rrr)rr.rr)rrr.r)rrrr.r7key))r.r7rAr)rAi r.r7rArDr)rrBrA)rBrZr)rCirBriryc3s(|]}t|jdddkVqdS)TrbrNr_r?rI)r2Zair[r4r5r6&z%test_roots_quartic..r.c3s"|] }tt|dVqdSr{r,r_r?r1r[r4r5r6s raT)negative rZc3s,|]}tt|fdffdVqdS)r>rNrr1)r\rar4r5r6s*css|] }t|tkVqdSN)typerr1r4r4r5r6r|r>)rUrarNc3s(|]\}}t||dVqdSr{)r,r_)r2rKs)repr4r5r6r)r rr?rrrrZeroOner9hash enumeraterrfqrrUr_zip) lhsrhsr3rPrTrRrSrsrZzansZrepsr4)r\rrar5test_roots_quarticsT  6$2,< 0"&rc CsXtddtttdtdddtdtdtddttDr*JdS) Ncss|]}t|tVqdSr) isinstancerr1r4r4r5r6sz#test_issue_21287..rDr7rArZr>r.)anyr rr?rr4r4r4r5test_issue_21287s <rcstddtddddtdddtddtddtddtd f}|D]tt}t|dks?Jtfd d |DsLJq/dS) NrZr7r. nrA7i ic3s.|]}t|djdddkVqdS)rmgh㈵>rbrNrr`r[r4r5r6s,z%test_roots_quintic..)r?r!rr<rf)Zeqsr%r4r[r5test_roots_quintics "* 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.TZpolysr7r>rArDrCr^rir)factorrZ) r"rr?rrrrrrrrr r4r4r4r5test_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)NrZrrDrr7rNrmrAria1T)Z nonnegativeb1r.)r.r7)r>r.)r7rArDrZryr>)r#rr?rrr=rrrr-r)r%rUr) ArrZr0r1rPrTrrIrdrsr4r4r5test_roots_binomial7s2"&.   ""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)"Nr7r.rAE,F,J,L[r|)s2^riry83ll8r^]$=,yrCr/@:rZrmD!^rDq Pyqidi Bi߰lul/1g&l glFW&krv) rPrUr?rTr$rrRrr r)rVZcoeffZpolyEFJLgr4r4r5test_roots_preprocessingXsh &,&$""****.& "<   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.rr7rArDr>rZrmrBrCrii iii ir^irrF)ZcubicsrNT!cSg|]}t|qSr4rr`r4r4r5rLs ztest_roots0..))rr)r.rcSrr4rr`r4r4r5rL))r.rD)r.ri)rZriZfilterrcSs|jSr)Z is_positive)rWr4r4r5sztest_roots0..) predicate)rrza, b)realRri(cSsg|]}|jr|qSr4r/)r2rxr4r4r5rL*rMrli2rrrurv)}-rr)r%r?rrexpandrrrrPrTZ NegativeOnerrRrSrr_rarUrrrrr r)rVZr1_2Zr1_3Zx0x1Zx2Zx3Zx4Zx5Zr13_20Zr1_20s2Zr1_4Zr1_8Zr5_8arbrrdrrWZ real_rootcrr\r4r4r5 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,xr7r.zzz,yy,xx,zy,zx,yx,krrAricss|]}|VqdSrr4r1r4r4r5r6Tsz"test_roots_slow..cs g|] }|jddqS)Trb)r_rI)r2rirVr?r4r5rLTs z#test_roots_slow..N)rr:r%valueskeysr)rPrTrRrSf1f2zzyyxxZzyZzxZyxke1e2rr4rr5test_roots_slow@s$H00"*rcCsttdtdtdd}ttdtdtdd}t||D]\}}t||dks-Jqtddtd td d d tdtdd dtdd}t|dd}d}t||D]\}}t||dksmJq_dS)Nr7r.Trg?g-q=rDg@g@rAgS@g8@gb@g"@)g؋~)g)c g3! @g @绽|=)r%r?rabsr)R1R2rr2rVr4r4r5test_roots_inexactWs2 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)Nrrriryrr^rrrCr7rrrZrArmrrDrrrrrrTr)g?|YcgOXLzGX@g'I^g@g |۰n@gdZw@g5`e@gS$Ȫ@gJm@rKr) rr?r%evalfrr<rmatchr) rrrrrVrrrKrdrrrr4r4r5test_roots_preprocessedis8"****.&  &rcCsttddtdddddiksJttddtdddddiks(JttddtdtddtdddddiksFJttd d dS) Nr7r.FstrictTrCrZrAcSs0ttddtdtddtdddS)NrCr7rZrATr)r%r?r4r4r4r5rs0z#test_roots_strict..)r%r?r*r(r4r4r4r5test_roots_stricts((<rc 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)Nipiir7i rA1rDT)rfrcSsg|] \\}}}t||qSr4rr2rPrT_r4r4r5rLsz$test_roots_mixed..cSs8g|]\\}}}tt|t|tt|t|qSr4)rr r rr4r4r5rLs(cSsg|]}|qSr4)rr`r4r4r5rLrcS|Srsort_keyr?r4r4r5rz"test_roots_mixed..rcSr rr r r4r4r5rr ) r?rrr%r9rr0r r ) rV_reZ_imZ_nrootsZ_srootsZ _intervalsZ_rootsr3rWr4r4r5test_roots_mixeds&0  rc CsttdttdtgksJtttttttgksJttddttdtdgks1JttdttttttttgksHJttdddtdtdtdtdttttttttgksnJtttddtddttdtttdtttddtgksJtdtddtddtdtdtd dtttddtddtddgksJdS) Nr.r7rDrrrirrCQ)rrr?rUrrr4r4r4r5test_root_factorss&.2*4( rcsRd}t|tddttjjfddjdd}dd |Dgd ks'JdS) NrTrcsjdddS)NrArZ)rIZmaxsteps)rr4rrr4r5rsztest_nroots1..rArHcSg|]}t|qSr4strr`r4r4r5rLrz 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)rIr%r4rrr5 test_nroots1s   rcCsjttddtdt}|jdd}dd|Dgdks J|jdd}dd|Dgd ks3JdS) NrZrAr.rHcSrr4rr`r4r4r5rLrz test_nroots2..)z-0.332z-0.839 - 0.944*Iz-0.839 + 0.944*Iz1.01 - 0.937*Iz1.01 + 0.937*IcSrr4rr`r4r4r5rLr)z-0.33199z-0.83907 - 0.94385*Iz-0.83907 + 0.94385*Iz1.0051 - 0.93726*Iz1.0051 + 0.93726*I)rr?r)rdr%r4r4r5 test_nroots2s     rcCs>ttttdtdtttttdddksJdS)NrAr7T)Z composite)r<r%rrUrr?r4r4r4r5test_roots_composite>rcs.ttdttdttfdddS)NrAr.cstSrrpr4r[r4r5rr z"test_issue_19113..)rr?r*r'r4r4r[r5test_issue_19113srcCs>tddddtdddtdgd d ddgksJdS) Nr.rBrrDr7rirrTr)r%rr4r4r4r5test_issue_17454rrcCsNttdtdgksJttddtdtdd gks%JdS)Nl l rArDr7)rr?Z real_rootsrr4r4r4r5test_issue_20913s2rc Cstdd}dt|tdd|}ttttdtddt|d| dtdtdd|dtdtddiksCJdS)Nr.rAr>rZr7)rrPr%rr?rr)rQrWr4r4r5test_issue_22768s  r)k__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$r%Zsympy.polys.orthopolysr&Zsympy.polys.polyerrorsr'r(Zsympy.polys.polyutilsr)Zsympy.testing.pytestr*r+Zsympy.core.randomr,r itertoolsr-rPrTrRrSrQrrr?rUrar=r@rYr]rhrorqrtrxr~rrrrrrrrrrrrrrrrrrrrrrrr4r4r4r5sr      0    "    : 0!<-