o o hh @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.0ir4t/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/polys/tests/test_polyroots.py (sz_check..r.FT)sumsortedlistrangelenr )r%nrealixr4r4r5_check$s$r?cCs*ttdtdttddgksJdS)Nr7r.)rrxrr4r4r4r5test_roots_linear2s*rBcCsttdtdtddgksJttdtddtttdddgks)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)radicalscSrHrIrJrLr4r4r5rNQrOF)r7)rr@)rrrArrrr? all_rootsaecdbyr)r-)fr_a_b_cr%r4r4r5test_roots_quadratic6s>"0@N&T TT "@ ,  r^c Csttdttdtt}t|ttdtdtddttd tdtddttdtdtd dttd tdtd diksSJdS)NrFr7r.)rrArr%reqr4r4r5test_issue_7724Xs rbcstdtddgttttt}tddtddfdd|D}t|tttttks:Jt fdd |DsHJdS) Nr.rQrDr7csg|]}|tqSr4)subsrXr2rZ)zr4r5rNesz#test_issue_8438..c3s0|]}tt|ijddddkVqdS)r7TchoprN)rdrXrArKr1prfr4r5r6is.z"test_issue_8438..) rrXrAas_exprrrrsetrdall)r%postr4rir5test_issue_8438as rocstdtddtttdd}t|sJttddtddtfddtdD}ttddtddt}||ksLJt|sRJtdtdd}t|sdJttdtd dd ksvJdS) NrFr.r7r_rEcg|]}t|qSr4rr1rYr4r5rNprOz#test_issue_8285.. )rrArRr?r;r<)r%ror4rsr5test_issue_8285ls* "   (rvcCsttddttdd}t|sJttddtddt}t|s/Jttdtd}t|sAJttddtddt}t|sXJdS)Nr7rFrErCr.)rrArRr?r%r4r4r5test_issue_8289{s$ "  "rxcsttdddtdddtdddtdtdtdtdtdtgks1JtddtddfddtdD}t|ksRJt|dS)Nr.r7rFrtcrqr4rrr1rjr4r5rNrOz$test_issue_14291..)rrArRrrr;r?)ansr4ryr5test_issue_14291s*0 r{cCs>ttdtttttdd}t|}t|dksJdS)NrCEXdomain)rrXr rAr%r<)raroots_dr4r4r5test_issue_13340s"rcsxttdtdddttdddttdd td d ttt}tfd d |Ds:JdS)NrFrC r7iiiii ic3s|] }|dkVqdSrNr4rer`r4r5r6z#test_issue_14522..)rrArr%rm)roots_eqr4r`r5test_issue_14522sRrcCs@tttddtdttdt}|dtsJdS)NrFrEr7rCr)r rtrAhas)solr4r4r5test_issue_15076s*rcCsbttddtdtddtddtdtddtt}t|}d|vs/JdS)NrFrpr7rC@ir)rrArr%)rarr4r4r5test_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)Nr7rC)rrrr.r.r.r.r@rT)trigmultiple)r rprQrE)rrrAr rXrHalfrrrr%rrrr)rZrar4r4r5test_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)!NrFrrrrrC)r@rrr)rr@rr)rrr@r)rrrr@)r.rrr)rr.rr)rrr.r)rrrr.r7key))r.r7rCr)rCi r.r7rCrFr)rrDrC)rDr_r)rEirDrprc3s(|]}t|jdddkVqdS)TrgrNrdrArK)r2air`r4r5r6&z%test_roots_quartic..r.c3s"|] }tt|dVqdSrr,rdrAr1r`r4r5r6s rfT)negative r_c3s,|]}tt|fdffdVqdS)r@rNrr1)rarfr4r5r6s*css|] }t|tkVqdSN)typerr1r4r4r5r6rr@)rXrfrQc3s(|]\}}t||dVqdSr)r,rd)r2rMs)repr4r5r6r)r rrArrrrZeroOner9hash enumeraterrmqrrXrdzip) lhsrhsr3rSrWrUrVrzrzansrepsr4)rarrfr5test_roots_quarticsT  6$2,< 0"&rc CsXtddtttdtdddtdtdtddttDr*JdS) Ncss|]}t|tVqdSr) isinstancerr1r4r4r5r6sz#test_issue_21287..rFr7rCr_r@r.)anyr rrArr4r4r4r5test_issue_21287s <rcstddtddddtdddtddtddtddtd f}|D]tt}t|dks?Jtfd d |DsLJq/dS) Nr_r7r. nrC7i ic3s.|]}t|djdddkVqdS)rtgh㈵>rgrNrrer`r4r5r6s,z%test_roots_quintic..)rAr!rr<rm)eqsr%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.Tpolysr7r@rCrFrErcrpr)factorr_) r"rrArrrrrrrrr 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)Nr_rrFrr7rQrtrCrpa1T) nonnegativeb1r.)r.r7)r@r.)r7rCrFr_rr@)r#rrArrr?rrrr-r)r%rXr) Arrr0r1rSrWrrKrjrzr4r4r5test_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.rCE,F,J,L[r|)s2^rpr83ll8rc]$=,yrEr/@:r_rtD!^rFq Pyqidi Bi߰lul/1g&l glFW&kr}) rSrXrArWr$rrUrr r)rYcoeffpolyEFJLgr4r4r5test_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.rr7rCrFr@r_rtrDrErpi iii ircirrF)cubicsrQT!cSg|]}t|qSr4rrer4r4r5rNs ztest_roots0..))rr)r.rcSrr4rrer4r4r5rN))r.rF)r.rp)r_rpZfilterrcSs|jSr) is_positive)rZr4r4r5sztest_roots0..) predicate)rrza, b)realRri(cSsg|]}|jr|qSr4r/)r2rxr4r4r5rN*rOrli2rrr|r})}-rr)r%rArrexpandrrrrSrW NegativeOnerrUrVrrdrfrXrrrrr r)rYr1_2r1_3x0x1x2x3x4x5r13_20r1_20s2r1_4r1_8r5_8arbrrjrrZ real_rootcrrar4r4r5 test_roots0sXD$(((26,,....,.44,&  :   ( L, ,$ J( 0(     <  **** D:: @@.*. ,(($(0**((  **((   *P<@  0  D B ,,r cstd\}}}}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,krrCrpcss|]}|VqdSrr4r1r4r4r5r6Tsz"test_roots_slow..cs g|] }|jddqS)Trg)rdrK)r2rirYrAr4r5rNTs z#test_roots_slow..N)rr:r%valueskeysr)rSrWrUrVf1f2zzyyxxzyzxyxke1e2rr4rr5test_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=rFg@g@rCgS@g8@gb@g"@)g؋~)g)c g3! @g @绽|=)r%rArabsr)R1R2rr2rYr4r4r5test_roots_inexactWs2 r#c 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)NrrrprrrcrrrEr7rrr_rCrtrrFrrrrrrTr)g?|YcgOXLzGX@g'I^g@g |۰n@gdZw@g5`e@gS$Ȫ@gJm@rMr) rrAr%evalfrr<rmatchr) rrrrrYr r!rMrjrr"r%r4r4r5test_roots_preprocessedis8"****.&  &r&cCsttddtdddddiksJttddtdddddiks(JttddtdtddtdddddiksFJttd d dS) Nr7r.FstrictTrEr_rCcSs0ttddtdtddtdddS)NrEr7r_rCTr')r%rAr4r4r4r5rs0z#test_roots_strict..)r%rAr*r(r4r4r4r5test_roots_stricts((<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)Nipiir7i rC1rFT)rmrcSsg|] \\}}}t||qSr4rr2rSrW_r4r4r5rNsz$test_roots_mixed..cSs8g|]\\}}}tt|t|tt|t|qSr4)rr r r+r4r4r5rNs(cSsg|]}|qSr4)r$rer4r4r5rNrcS|Srsort_keyrAr4r4r5rz"test_roots_mixed..rcSr-rr.r0r4r4r5rr1) rArrr%r9rr0r r ) rY_re_im_nroots_sroots _intervals_rootsr3rZr4r4r5test_roots_mixeds&0  r8c CsttdttdtgksJtttttttgksJttddttdtdgks1JttdttttttttgksHJttdddtdtdtdtdttttttttgksnJtttddtddttdtttdtttddtgksJtdtddtddtdtdtd dtttddtddtddgksJdS) Nr.r7rFrrrprrEQ)rrrArXrrr4r4r4r5test_root_factorss&.2*4( r:csRd}t|tddttjjfddjdd}dd |Dgd ks'JdS) NrTrcsjdddS)NrCr_)rKmaxsteps)rr4ryr4r5rsztest_nroots1..rCrJcSg|]}t|qSr4strrer4r4r5rNrz 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&rAr*mpmathmp NoConvergencer)rKr%r4ryr5 test_nroots1s   rBcCsjttddtdt}|jdd}dd|Dgdks J|jdd}dd|Dgd ks3JdS) Nr_rCr.rJcSr<r4r=rer4r4r5rNrz test_nroots2..)z-0.332z-0.839 - 0.944*Iz-0.839 + 0.944*Iz1.01 - 0.937*Iz1.01 + 0.937*IcSr<r4r=rer4r4r5rNr)z-0.33199z-0.83907 - 0.94385*Iz-0.83907 + 0.94385*Iz1.0051 - 0.93726*Iz1.0051 + 0.93726*I)rrAr)rjr%r4r4r5 test_nroots2s     rCcCs>ttttdtdtttttdddksJdS)NrCr7T) composite)r<r%rrXrrAr4r4r4r5test_roots_composite>rEcs.ttdttdttfdddS)NrCr.cstSrrwr4r`r4r5rr1z"test_issue_19113..)rrAr*r'r4r4r`r5test_issue_19113srGcCs>tddddtdddtdgd d ddgksJdS) Nr.rDrrFr7rprrTr)r%rr4r4r4r5test_issue_17454rFrHcCsNttdtdgksJttddtdtdd gks%JdS)Nl l rCrFr7)rrA real_rootsrr4r4r4r5test_issue_20913s2rJc Cstdd}dt|tdd|}ttttdtddt|d| dtdtdd|dtdtddiksCJdS)Nr.rCr@r_r7)rrSr%rrArr)rTrZr4r4r5test_issue_22768s  rK)k__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$r%sympy.polys.orthopolysr&sympy.polys.polyerrorsr'r(sympy.polys.polyutilsr)sympy.testing.pytestr*r+sympy.core.randomr,r? itertoolsr-rSrWrUrVrTrrrArXrfr?rBr^rbrorvrxr{rrrrrrrrrrrr rr#r&r)r8r:rBrCrErGrHrJrKr4r4r4r5sr      0    "    : 0!<-