a
    rhh                  
   @   s|  d Z ddlmZmZmZ ddlmZ ddlmZm	Z	m
Z
 ddlmZmZmZ ddlmZ ddlmZmZ ddlmZ dd	lmZmZmZ dd
lmZ ddlmZ ddlmZ ddl m!Z!m"Z"m#Z#m$Z$m%Z% ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7 ddl8m9Z9m:Z: ddl;m<Z< ddl=Z=ddl>m?Z? e
d\
Z@ZAZBZCZDZEZFZGZHZIdd ZJdd ZKdd ZLdd ZMdd  ZNd!d" ZOd#d$ ZPd%d& ZQd'd( ZRd)d* ZSd+d, ZTd-d. ZUd/d0 ZVd1d2 ZWd3d4 ZXd5d6 ZYd7d8 ZZd9d: Z[d;d< Z\d=d> Z]d?d@ Z^dAdB Z_dCdD Z`dEdF ZadGdH ZbdIdJ Zce:dKdL ZddMdN ZedOdP ZfdQdR ZgdSdT ZhdUdV ZidWdX ZjdS )YzBTests for algorithms for computing symbolic roots of polynomials.     )IRationalpi)S)SymbolWildsymbols)	conjugateimre)exp)rootsqrt)	Piecewise)acoscossin)ZZInterval)powsimp)Polycyclotomic_poly	intervalsnrootsrootof)
root_factorsroots_linearroots_quadraticroots_cubicroots_quarticroots_quinticroots_cyclotomicroots_binomialpreprocess_rootsroots)legendre_poly)PolynomialErrorUnsolvableFactorError)_nsort)raisesslow)verify_numericallyN)productza,b,c,d,e,q,t,x,y,zc                 C   s   t dd | D }t| d | t| d | ks6J t|t| dD ]6}| |d  | | ksF| |d  t| | ksF dS qFdS )Nc                 s   s   | ]}|j rd ndV  qdS )   r   Nis_real.0i r4   \/var/www/html/assistant/venv/lib/python3.9/site-packages/sympy/polys/tests/test_polyroots.py	<genexpr>(       z_check.<locals>.<genexpr>   r.   FT)sumsortedlistrangelenr	   )r%   ZnrealZixr4   r4   r5   _check$   s    $r>   c                   C   s*   t tdt d ttddgks&J d S )Nr8   r.   )r   r   xr   r4   r4   r4   r5   test_roots_linear2   s    rA   c                  C   s  t tdtd  tddgks"J t tdtd  dt  ttdddgksRJ t tdtd  d tt td d ttd d gksJ t tdtd  dt  d tdttd d  dttd d  gksJ ttdtd  dt  d t  td dt t	 dt
 t	  tt
  t  tt tt	d   t
t	d   tt
   } t t| tt	 tt
  tt
  ttt t
t  tt  tt
  dt t
 t	d   tt
   t	 tt
  tt
  ttt t
t  tt  tt
  dt t
 t	d   tt
   gksJ tttd  dt  dt  t} t | tdtd  d  t dt  tdtd  d t dt  gksJ ttd td  d t  td  d t} t | dtd d gksJ ttdtd  d t} t | }|t|ks
J td	td  d
t  d } dd | jddD dd | jddD ksXJ tdddD ]B\}}}t|td  |t  | } t | }|t|ksdJ qdd S )Nr8   r            r?   r.   i   i  c                 S   s   g | ]}| d qS r8   nr2   wr4   r4   r5   
<listcomp>P   r7   z(test_roots_quadratic.<locals>.<listcomp>T)Zradicalsc                 S   s   g | ]}| d qS rG   rH   rJ   r4   r4   r5   rL   Q   r7   F)r8   )r   r?   )r   r   r@   r   r   r   r>   	all_rootsaecdbyr)   r-   )frZ_a_bZ_cr%   r4   r4   r5   test_roots_quadratic6   s<    "0@N&TTT
"@
,

rX   c                  C   s   t td t td  t t} t| ttd tdt d  dttd t d td  dttd tdt d   dttd t d td   diksJ d S )NrE   r8      r.   )r   r@   r   r%   r   eqr4   r4   r5   test_issue_7724X   s     r\   c                     s   t dtddgt  tt  tt} tddtd d  fdd| D }t|ttt  ttkstJ t	 fdd	|D sJ d S )
Nr.   rM   rC   r8      c                    s   g | ]}| t qS r4   )subsrT   r2   rV   )zr4   r5   rL   e   r7   z#test_issue_8438.<locals>.<listcomp>c                 3   s.   | ]&}  tt|ijd dddkV  qdS )r8   TZchopr   N)r^   rT   r@   rI   r1   pr`   r4   r5   r6   i   r7   z"test_issue_8438.<locals>.<genexpr>)
r   rT   r@   Zas_exprr   r   r   setr^   all)r%   postr4   rb   r5   test_issue_8438a   s    rg   c                     s   t dtd  d tt td d   } t| s6J t td dtd   d t  fddtdD }t td dtd   d t } | |ksJ t| sJ t dtd  d  } t| sJ tt dtd	  d  d	ksJ d S )
NrE      r.   r8   rY   rD   c                    s   g | ]}t  |qS r4   r   r1   rU   r4   r5   rL   p   r7   z#test_issue_8285.<locals>.<listcomp>
   )r   r@   rN   r>   r<   r=   )r%   ror4   rj   r5   test_issue_8285l   s    *"rm   c                  C   s   t td d t td d   } t| s0J t td dtd   d t } t| s^J t td t d  } t| sJ t td dtd   d t } t| sJ d S )Nr8   rE   rD   rB   r.   )r   r@   rN   r>   r%   r4   r4   r5   test_issue_8289{   s    $""ro   c                     s   t td d d td d d  td   ddt dt dtdt  dtdt  gksbJ td dtd   d   fddtdD } t   | ksJ t|  d S )Nr.   r8   rE   rk   c                    s   g | ]}t  |qS r4   ri   r1   rc   r4   r5   rL      r7   z$test_issue_14291.<locals>.<listcomp>)r   r@   rN   r   r   r<   r>   )ansr4   rp   r5   test_issue_14291   s    .,rr   c                  C   s>   t td ttt  t tdd} t| }t|dks:J d S )NrB   EXdomain)r   rT   r   r@   r%   r=   )r[   Zroots_dr4   r4   r5   test_issue_13340   s    "rv   c                     sx   t td td ddt    td ddt    tdd	t    d
 dt  t t } t fdd| D stJ d S )NrE   rB          r8   ii  ii  i
  i  c                 3   s   | ]} |d kV  qdS r   Nr4   r_   rZ   r4   r5   r6      r7   z#test_issue_14522.<locals>.<genexpr>)r   r@   r   r%   re   )roots_eqr4   rZ   r5   test_issue_14522   s    Rr{   c                  C   s@   t ttd dtd   tt  d t} | d ts<J d S )NrE   rD   r8   rB   r   )r    r   tr@   has)solr4   r4   r5   test_issue_15076   s    *r   c                  C   sb   t td dtd td   dtd   dtd td   dt  t} t| }d|v s^J d S )NrE   rh   r8   rB   @   i   r   )r   r@   r   r%   )r[   rz   r4   r4   r5   test_issue_16589   s    Jr   c               	   C   sl  t tdtd  tg dks"J t ttd dtd   dt  d tg dksXJ ttd} t ttd t t| | tj tdt d   | tj tdt d   gksJ t ttd d tdtjttd d  tjttd d  gksJ t tdtd  dtd   dt  d td tjdt	dd d  dt	dd d  ks`J td  dtd   dt  d }t
|ddd	t t|tdd
  krbt	dddtd ttdtd d d  d  dtd ttdtd d  d td   d t	dd dtd ttdtd d  d td   d t	dd gkshn J d S )Nr8   rB   )r   r   r   r.   r.   r.   r.   r?   r   T)trigmultiple)r      rh      rM   rD   )r   r   r@   r   rT   r   Halfr   r   r   r%   r   r   r   r   )rV   r[   r4   r4   r5   test_roots_cubic   s,    "6

,0(
"
4>>r   c                     sd  t ttd tg dksJ t ttd td  tg dg dg dg dfv sXJ t ttd td  tg dg d	g d
g dfv sJ t ttd t t} tjttd d  tjttd d  tjtj g}t	| t
dt	|t
dksJ tg dD ]\}\}}}}|dkrD| |d td |td   }n<|dkr|||d tdd |td   |td   }td |td   |td   |t  |  t t t}t fdd|D sJ qttt td  td  td  dtd   tdd t t  }t fdd|D s:J tdddtd dtd   dtd   td   d  t t t}	t fdd|	D sJ tttd  td  t  t t  }tdd |D sJ tddtdddtddtdddtddddf}
|
D ]<t t t}tfd dt||D s"J q"d S )!NrE   r   r   r   r   rB   )r?   r   r   r   )r   r?   r   r   )r   r   r?   r   )r   r   r   r?   )r.   r   r   r   )r   r.   r   r   )r   r   r.   r   )r   r   r   r.   r8   key))r.   r8   rB   r   )rB   i	   r.   r8   rB   rE   r   )r   rC   rB   )rC   rY   r   )rD   irC   rh      rw   c                 3   s&   | ]}  t|jd ddkV  qdS )Tra   r   Nr^   r@   rI   )r2   ZairZ   r4   r5   r6      r7   z%test_roots_quartic.<locals>.<genexpr>r.   c                 3   s    | ]}t  t|d V  qdS ry   r,   r^   r@   r1   rZ   r4   r5   r6      r7   r`   T)negative   rY   c                 3   s*   | ]"}t  t|fd ffdV  qdS )r?   r   Nr   r1   )r[   r`   r4   r5   r6      r7   c                 s   s   | ]}t |tkV  qd S N)typer   r1   r4   r4   r5   r6      r7   r?   )rT   r`   rM   c                 3   s&   | ]\}}t | | d V  qdS ry   )r,   r^   )r2   rK   s)repr4   r5   r6      r7   )r    r   r@   r   r   r   r   ZeroOner:   hash	enumerater   re   qr   rT   r^   zip)lhsrhsr3   rO   rS   rQ   rR   rq   r~   ZzansZrepsr4   )r[   r   r`   r5   test_roots_quartic   sR    

6
$
2, <0"r   c                	   C   sX   t dd tttd td ddt    dt dt   d dt  tD rTJ d S )	Nc                 s   s   | ]}t |tV  qd S r   )
isinstancer   r1   r4   r4   r5   r6      r7   z#test_issue_21287.<locals>.<genexpr>rE   r8   rB   rY   r?   r.   )anyr    r   r@   r   r4   r4   r4   r5   test_issue_21287   s    
<r   c                     s   t d d t d d d dt d d   d t d dt d   dt d   dt   d	 f} | D ]: tt }t|dks~J t fd
d|D s^J q^d S )NrY   r8   r.      n   rB   7   i	  i  c                 3   s,   | ]$}  t|d jdddkV  qdS )rk   gh㈵>ra   r   Nr   r_   rZ   r4   r5   r6     r7   z%test_roots_quintic.<locals>.<genexpr>)r@   r!   r   r=   re   )Zeqsr%   r4   rZ   r5   test_roots_quintic   s    
"*r   c                   C   s  t tdtdddgksJ t tdtdddgks8J t tdtddtddttd d  tddttd d  gksJ t tdtddt tgksJ t tdtddtjttd d  tjttd d  gksJ t td	tddtt	d	  tt
t	d	   tt	d	  tt
t	d	   tt	tdd	  tt
t	tdd	   tt	tdd	  tt
t	tdd	   tt	tdd	 tt
t	tdd	   tt	tdd	 tt
t	tdd	   gksJ t td
tddtd d ttd d  td d ttd d  tdd ttd d  tdd ttd d  gks\J t tdtddtd d td  td d td  tdd td  tdd td  gksJ t tdtdddddgksJ t tdtdddddgksJ t tdtddddtdd dtdd gks>J t tdtddddt tgksdJ t tdtddddtdd tddd  tddd dtddd  tdd tddd  gksJ t tdtdddddtdd tddgksJ d S )Nr.   TZpolysr8   r?   rB   rE   rD   r]   rh   r   )factorrY   )r"   r   r@   r   r   r   r   r   r   r   r   r   r4   r4   r4   r5   test_roots_cyclotomic  sn    2 *((&&	



N
r   c               	   C   sH  t tdt tdgksJ t tdtd  tg dks>J t tdt d ttddgksdJ dtdd d } t tdtd  d t|  | t  |  | t  | | t  | | t  gksJ tt ttd	 d  td
dd}tddd}tt|td  | t}t t|td  | t}t|d t|d ksFJ t|d t|d ksdJ t	ddddD ]X\}}}}||kr|dkrqrt|t|  ||  }	t |	}
|
t
|
ksrJ qrttdtd  dtd   tdt tddtdt d   ddt ddt tddtdt d   diksDJ d S )NrY   r   rE   r   r8   rM   rk   rB   rh   a1T)Znonnegativeb1r.   )r.   r8   )r?   r.   )r8   rB   rE   rY   rw   r?   )r#   r   r@   r   r   r>   r   r   r   r-   r)   r%   rT   r   )Ar   r   Zr0r1rO   rS   r   rI   rc   rq   r4   r4   r5   test_roots_binomial7  s2    "&. ""r   c                  C   sl  t t td  t t } tt| t\}}|dks6J |tt t td  t t tks\J td td  td td   tt  t  } tt| t\}}|dt ksJ |ttd td  t t  tksJ td td  td td   t  } tt| t\}}|dt ksJ |ttd td  t  tks<J td td  tt  t  } tt| t\}}|dt ks|J |ttd t t  tksJ td td  t  } tt| t\}}|dt ksJ |ttd t  tksJ td\}}}}d|d  |d  |d  d| t |d	  |d	  |d
   d|d  |d  |d  td  |d   d|d  |d  |d  td  |d   d|d  |d  |d  td  |d   d|d  |d  |d  td  |d   d|d  |d  |d  td  d|d    d| | |d	  td	  d|d    d|d  td  d  } tt| t\}}|d| | ||d   ksJ |dtd  dtd	   dtd   dtd   dtd   dtd   dtd   dt  d  ksJ ttd  td tt  tt	tttf d!} ttd  tt tt	tt d!}t| t|fkshJ d S )"Nr8   r.   rB   E,F,J,L [r|)s2^ rh   rw       83ll8r]         ]$=,y rD   r      /@:rY   rk      D!^ rE   q 
    P   y   q    id i B i߰l   ul   /1g&l   g l   FW&k rt   )
rO   rT   r@   rS   r$   r   rQ   r   r   r   )rU   ZcoeffZpolyEFJLgr4   r4   r5   test_roots_preprocessingX  sh    &,&$$ "****.&
"<

0$r   c                  C   s  t dti ksJ t tttjdiks*J t td ttjdiksFJ t td td  td   ttd dtddtddiksJ t dt d ttdddiksJ t dt d d ttdddiksJ t dt d d ttdddiksJ t dt d d ttdddiks(J t td d ttdtjdtj dt diks\J t td d d ttdtjdtj dt diksJ t dt d d  ttdddiksJ t dt d d  ttd	ddiksJ t dt d d  ttdddiksJ t dt d d  ttd	ddiksLJ t dt d d  ttdddikszJ t dt d d  ttd	ddiksJ t tt t	 d  tt	t diksJ t tt t	 d  tt	 t diksJ t td t d t  t ttdtjdiks6J t td dtd   d ttjdtj
diksjJ t td
 dtd   dtd   td  ttjddtd dtjddtd diksJ t td d ttdd ttd d  dtdd ttd d  dtd d ttd d  dtd d ttd d  dtjdtj dtdt diksnJ dtd  dtd   dtd   dtd   dtd
   dtd   dtd   dtd   } t | tjdtd dtddtdddtd	ddtdddtdddiks J t tt	 t t tt	 t t  ttt	 t t tt	 t  dikslJ t td td  t d tddi ksJ t td td  td   tddtd dtddtddiksJ t td td  td  td   tddtd dtddtddtddiks<J t td dtd   dt  d ttd ddt ddt diksJ t td dtd   dt  d tdddt ddt dtd diksJ t td t td dtd   dt  d  ttjdtjdtd ddt ddt diks6J tjtdd }}dtd d | }d| d }|d }tdt d }|| }| | }t td td  t d tdd| | | d| | ||  | d| | ||  | diksJ td dt  d tdtd  dt  tdt d } dd dD \}	}
td}t | t|	|
tddt |    d|	|
tddt |    d|	|
tddt |    d|	|
tddt |    diksJ td td  td  t d } dd d D \}}}t | t| |d|   t||d|   |   d| |d|   t||d|   |   d| |d|   t||d|   |   d| |d|   t||d|   |   dik	sJ td dt td   ddt  dtd   t  t dtd   } t | ttjdtjtjt  tjtddt  td  dtd     dtjtjt  tjtddt  td  dtd     dik
sJ t tt	 t td  dtd   dt  d tddi k
sJ t tt	 t td  dtd   dt  d tddi ksJ t td d td!d"tjdtj diksBJ t td d td#d"tdt dikslJ t td td  ttjdtj diksJ t td td  td$d% d&tjdiksJ t td d td!dd'tj tjgksJ t td d td#dd'tt gksJ td(dd)\}}td || d  dt ||   d }t |td*d"d||  diksxJ t td tdd+tjtjtjgksJ t d,tdd+g ksJ td
 td  td  td  td  t d } t | t ttd  ttd  dt tttdd  tttdd  dt tttdd  tttdd  dtttd  ttd  dttttdd  tttdd  dttttdd  tttdd  diksJ td d | d   }t |t ttd  ttd  dt tttdd  tttdd  dt tttdd  tttdd  dtttd  ttd  dttttdd  tttdd  dttttdd  tttdd  dt dtdiksJ t td d-t  d }d.d |D d/ }d0d
td1  }|dt|d d d2t|d  ksNJ tddtd  td  d3dtd  td   td d t  d td4d5}t |dtd dddtd  dtd d diksJ td6td  d7td td   d8td   d0td td   d9td   d:td td   d;td   d<td td   d=td t  d=t  d> td4d5}t |td d dddtd  ddtd dd?dtd  dd	dtd  diksJ ttd dtd   d
td td   dtd t  d@t  dA dAtd  td4d5}t |dtd d ddtd d ddtd d diksxJ t tttd d d td4d5td tddtj tdt d    dtd tddtj tdt d    dtd tdd diksJ d S )BNr.   r   r8   rB   rE   r?   rY   rk   rC   rD   rh   i i  i  i  i     r]   i  r   r   F)ZcubicsrM   T!      c                 S   s   g | ]}t | qS r4   r   r_   r4   r4   r5   rL     s   ztest_roots0.<locals>.<listcomp>))r   r   )r.   r   c                 S   s   g | ]}t | qS r4   r   r_   r4   r4   r5   rL     r7   ))r.   rE   )r.   rh   )rY   rh   Zfilterr   c                 S   s   | j S r   )Zis_positive)rV   r4   r4   r5   <lambda>  r7   ztest_roots0.<locals>.<lambda>)	predicate)r   r   za, b)realRr   i  (   c                 S   s   g | ]}|j r|qS r4   r/   )r2   rxr4   r4   r5   rL   *  r7   r   l   i2  r   r   rs   rt   )            }   -            r      r   )r%   r@   r   r   expandr   r   r   rO   rS   ZNegativeOner   rQ   rR   r   r^   r`   rT   r   r   r   r   r   r   )rU   Zr1_2Zr1_3Zx0x1Zx2Zx3Zx4Zx5Zr13_20Zr1_20s2Zr1_4Zr1_8Zr5_8arbrrc   r   rV   Z	real_rootcrr[   r4   r4   r5   test_roots0  sX   D$((*48......,.44,&
:


(L,
,$
J(
0(

 <
 ****D::@@.*.
,(($(0**((	**((

*P<@

0 D
B ,,r   c                     s  t d\} }}}d | | |  | |  |  }d | |||  |    |  | | || |   | | ||   }tt| ddgksJ tt| ddgksJ t d\}}}}	}
}}|| ||  ||  |	| |
  |
 |	 | }|| | | |
||  |
  |	|	 ||   }tt|| | g dksTJ d dd   d  tt  }tdd	  fd
d|D D rJ dS )z6Just test that calculating these roots does not hang. z	a,b,c,d,xr8   r.   zzz,yy,xx,zy,zx,yx,kr   rB   rh   c                 s   s   | ]
}|V  qd S r   r4   r1   r4   r4   r5   r6   T  r7   z"test_roots_slow.<locals>.<genexpr>c                    s    g | ]}  |jd dqS )Tra   )r^   rI   )r2   rirU   r@   r4   r5   rL   T  r7   z#test_roots_slow.<locals>.<listcomp>N)r   r;   r%   valueskeysr   )rO   rS   rQ   rR   f1f2zzyyxxZzyZzxZyxke1e2r   r4   r   r5   test_roots_slow@  s    $H00$r   c                  C   s   t td t d tdd} t td t d tdd}t| |D ]\}}t|| dk s>J q>td dtd	 td
   ddtd  td   ddtd d   }t |dd} d}t| |D ]\}}t|| dk sJ qd S )Nr8   r.   Tr   g      ?g-q=rE   g      @g       @rB   g     S@g      8@g      b@g      "@)g؋~)g)cg3!	@g
 @绽|=)r%   r@   r   absr   )R1R2r   r2rU   r4   r4   r5   test_roots_inexactW  s    2r   c                  C   s  t d\} }}}d| d  |d  |d  d| t | d  |d  |d   d| d	  |d
  |d	  td
  |d   d| d  |d  |d  td  |d   d| d  |d  |d  td  |d   d| d  |d  |d  td  |d	   d| d
  |d	  |d
  td	  d|d    d|  | |d  td  d|d
    d|d  td  d  }t|ti ksJ t| tdd}g d}td}||  | ||d
   }t|t|ksJ t||D ]8\}	}
|	|}|d urt|| |
 dk sJ qd S )Nr   r   rh   rw   r   r]   r   r   rD   r8   r   r   rY   rB   rk   r   rE   r   r   r   r   r   r   Tr   )g?|YcgOXLzGX@g'I^g@g|۰n@gdZw@g5`e@gS$Ȫ@gJm@rK   r   )	r   r@   r%   evalfr   r=   r   matchr   )r   r   r   r   rU   r   r   rK   rc   r   r   r   r4   r4   r5   test_roots_preprocessedi  s6    "****.&

r   c                   C   s   t td dt  d ddddiks(J t td dt  d ddddiksPJ t td dtd   td  dt  d ddddiksJ ttd	d
  d S )Nr8   r.   FstrictTrD   rY   rB   c                   S   s0   t td dtd   td  dt  d ddS )NrD   r8   rY   rB   Tr   )r%   r@   r4   r4   r4   r5   r     r7   z#test_roots_strict.<locals>.<lambda>)r%   r@   r*   r(   r4   r4   r4   r5   test_roots_strict  s    ((<r   c            	      C   s   ddt   dt d   dt d   dt d   } t| d	d
\}}t| }t| d	d}dd |D }dd |D }|| }dd |D }t|dd d}t|dd d}||fD ]D}t||D ]4\}}|jr||v sJ qt|t|f|v sJ qqd S )Nipi  i  r8   i
  rB   1   rE   T)re   r   c                 S   s   g | ]\\}}}t ||qS r4   r   r2   rO   rS   _r4   r4   r5   rL     r7   z$test_roots_mixed.<locals>.<listcomp>c                 S   s8   g | ]0\\}}}t t|t|t t|t| qS r4   )r   r   r
   r  r4   r4   r5   rL     s   c                 S   s   g | ]}|  qS r4   )r   r_   r4   r4   r5   rL     r7   c                 S   s   |   S r   sort_keyr@   r4   r4   r5   r     r7   z"test_roots_mixed.<locals>.<lambda>r   c                 S   s   |   S r   r  r  r4   r4   r5   r     r7   )	r@   r   r   r%   r:   r   r0   r   r
   )	rU   _reZ_imZ_nrootsZ_srootsZ
_intervalsZ_rootsr3   rV   r4   r4   r5   test_roots_mixed  s"    0r  c                
   C   s  t tdttdtgksJ t ttttttgks<J t td d ttd td gksbJ t td t tttt ttt gksJ t td d d td td td td tt tt tt tt gksJ t ttd d tddttd tttd tttd d tgks(J t dtd  dtd   dtd   td  td	dtttd dtd   dtd   d gksJ d S )
Nr.   r8   rE   r   r   rh   r   rD   Q)r   r   r@   rT   r   r   r4   r4   r4   r5   test_root_factors  s    &.2*
4(r	  c                     sR   d} t | tdd ttjj fdd  jdd}dd	 |D g d
ksNJ d S )Nr   Tr   c                      s    j dddS )NrB   rY   )rI   Zmaxsteps)r   r4   rp   r4   r5   r     r7   ztest_nroots1.<locals>.<lambda>rB   rH   c                 S   s   g | ]}t |qS r4   strr_   r4   r4   r5   rL     r7   z test_nroots1.<locals>.<listcomp>)@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.0243z0.0243z0.0730z0.121z0.170z0.217z0.265z0.311z0.357z0.402z0.446z0.489z0.531z0.572z0.611z0.649z0.685z0.720z0.753z0.784z0.813z0.841z0.866z0.889z0.911z0.930z0.946z0.961z0.973z0.983z0.991z0.996z0.999)r&   r@   r*   mpmathmpZNoConvergencer   )rI   r%   r4   rp   r5   test_nroots1  s    r  c                  C   sj   t td dt  d t} | jdd}dd |D g dks@J | jdd}dd |D g d	ksfJ d S )
NrY   rB   r.   rH   c                 S   s   g | ]}t |qS r4   r
  r_   r4   r4   r5   rL     r7   z test_nroots2.<locals>.<listcomp>)z-0.332z-0.839 - 0.944*Iz-0.839 + 0.944*Iz1.01 - 0.937*Iz1.01 + 0.937*Ic                 S   s   g | ]}t |qS r4   r
  r_   r4   r4   r5   rL     r7   )z-0.33199z-0.83907 - 0.94385*Iz-0.83907 + 0.94385*Iz1.0051 - 0.93726*Iz1.0051 + 0.93726*I)r   r@   r   )rc   r%   r4   r4   r5   test_nroots2  s    r  c                   C   s>   t tttd td tt  t t tdddks:J d S )NrB   r8   T)Z	composite)r=   r%   r   rT   r   r@   r4   r4   r4   r5   test_roots_composite  s    r  c                      s.   t td t t d  tt fdd d S )NrB   r.   c                      s   t  S r   rn   r4   rZ   r4   r5   r     r7   z"test_issue_19113.<locals>.<lambda>)r   r@   r*   r'   r4   r4   rZ   r5   test_issue_19113  s    r  c                   C   s>   t ddddt  d  d dt  dgd	d
ddgks:J d S )Nr.   rC   r   rE   r8   rh   r   r   Tr   )r%   r   r4   r4   r4   r5   test_issue_17454  s    r  c                   C   sN   t td t dgksJ t td d t dtdd   gksJJ d S )Nl          l       rB   rE   r8   )r   r@   Z
real_rootsr   r4   r4   r4   r5   test_issue_20913  s    r  c               	   C   s   t dd} dt |  td d|    }ttttd  td d  t|d| dtdt   d d|dtdt   d diksJ d S )Nr.   rB   r?   rY   r8   )r   rO   r%   r   r@   r   r   )rP   rV   r4   r4   r5   test_issue_22768  s    
 r  )k__doc__Zsympy.core.numbersr   r   r   Zsympy.core.singletonr   Zsympy.core.symbolr   r   r   Z$sympy.functions.elementary.complexesr	   r
   r   Z&sympy.functions.elementary.exponentialr   Z(sympy.functions.elementary.miscellaneousr   r   Z$sympy.functions.elementary.piecewiser   Z(sympy.functions.elementary.trigonometricr   r   r   Zsympy.polys.domains.integerringr   Zsympy.sets.setsr   Zsympy.simplify.powsimpr   Zsympy.polysr   r   r   r   r   Zsympy.polys.polyrootsr   r   r   r   r    r!   r"   r#   r$   r%   Zsympy.polys.orthopolysr&   Zsympy.polys.polyerrorsr'   r(   Zsympy.polys.polyutilsr)   Zsympy.testing.pytestr*   r+   Zsympy.core.randomr,   r  	itertoolsr-   rO   rS   rQ   rR   rP   r   r|   r@   rT   r`   r>   rA   rX   r\   rg   rm   ro   rr   rv   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r	  r  r  r  r  r  r  r  r4   r4   r4   r5   <module>   sp   0"		:
0!< -
