a
    Qez                     @   s  d dl Zd dl mZmZ d dlmZ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 d dlm  mZ G dd dZG d	d
 d
ZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZdAddZdd Zdd  Z d!d" Z!d#d$ Z"d%d& Z#d'd( Z$d)d* Z%d+d, Z&d-d. Z'd/d0 Z(d1d2 Z)d3d4 Z*d5d6 Z+d7d8 Z,d9d: Z-d;d< Z.d=d> Z/d?d@ Z0dS )B    N)arraysqrt)assert_array_almost_equalassert_equalassert_almost_equalassert_allclose)raises)	integrategammac                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
	TestChebyc                 C   s   t d}t d}tjdd8 t d}t d}t d}t d}W d    n1 s^0    Y  t|jdgd	 t|jddgd	 t|jg d
d	 t|jg dd	 t|jg dd	 t|jg dd	 d S )Nr      ignoreall               )r   r   )r   r   r   )r   r   r   r   )r   r   r   r   r   )orthchebycnperrstater   c)selfZC0ZC1ZC2ZC3ZC4ZC5 r    W/var/www/sunrise/env/lib/python3.9/site-packages/scipy/special/tests/test_orthogonal.pytest_chebyc   s    




(zTestCheby.test_chebycc                 C   s   t d}t d}t d}t d}t d}t d}t|jdgd t|jddgd t|jg dd t|jg d	d t|jg d
d t|jg dd d S )Nr   r   r   r   r   r   r   )r   r   )r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   r   )r   chebysr   r   )r   ZS0ZS1ZS2ZS3ZS4ZS5r    r    r!   test_chebys   s    





zTestCheby.test_chebysc                 C   s   t d}t d}t d}t d}t d}t d}t|jdgd t|jddgd t|jg dd t|jg d	d t|jg d
d t|jg dd d S )Nr   r   r   r   r   r   r   )r   r   r#   )r   r   r   r   )   r   ir   r   )   r   r   r   r   )r   chebytr   r   )r   ZT0ZT1ZT2ZT3ZT4ZT5r    r    r!   test_chebyt,   s    





zTestCheby.test_chebytc                 C   s   t d}t d}t d}t d}t d}t d}t|jdgd t|jddgd t|jg dd t|jg d	d t|jg d
d t|jg dd d S )Nr   r   r   r   r   r   r   )r   r   r#   )r&   r   r   r   )r'   r   r   r   )    r   ir      r   )r   chebyur   r   )r   ZU0ZU1ZU2ZU3ZU4ZU5r    r    r!   test_chebyu:   s    





zTestCheby.test_chebyuN)__name__
__module____qualname__r"   r%   r*   r/   r    r    r    r!   r      s   r   c                   @   s   e Zd Zdd ZdS )TestGegenbauerc                 C   s  dt j  d }t |dkr$d}td|}td|}td|}td|}td|}td|}t|jtdgd	 t|jtd| dgd	 t|jtd| |d  d| gd	 t|jtdt	|d dd
| |d  dgd d t|jtdt	|d ddt	|d dd| |d  gd d t|jtdt	|d ddt	|d ddt	|d dgd d d S )Nr         ?r   皙ɿr   r   r   r   r   ig      @   r+   g      @r(      g      .@)
r   randomanyr   Z
gegenbauerr   r   r   scZpoch)r   aZCa0ZCa1ZCa2ZCa3ZCa4ZCa5r    r    r!   test_gegenbauerK   sB    &&&&zTestGegenbauer.test_gegenbauerN)r0   r1   r2   r<   r    r    r    r!   r3   I   s   r3   c                   @   s   e Zd Zdd Zdd ZdS )TestHermitec                 C   s   t d}t d}t d}t d}t d}t d}t|jdgd t|jddgd t|jg dd t|jg d	d t|jg d
d t|jg dd d S )Nr   r   r   r   r   r   r   )r   r   r   )r&   r   r+   r   )r'   r   ir      r>   )r,   r   i`r   x   r   )r   hermiter   r   )r   H0H1H2H3H4H5r    r    r!   test_hermiteb   s    





zTestHermite.test_hermitec                 C   s2  t dtd dg}td}td}td}td}td}td}td|}td|td }	td|d }
td|dtd  }td|d	 }td|d	td  }t|j|jd
 t|j|	jd
 t|j|
jd
 t|j|jd
 t|j|jd
 t|j|jd
 d S )N      ?r   r   r   r   r   r          @      @r   )r   poly1dr   r   hermitenormr@   r   r   )r   psubrA   rB   rC   rD   rE   rF   Zhe0Zhe1Zhe2Zhe3Zhe4Zhe5r    r    r!   test_hermitenormp   s&    





zTestHermite.test_hermitenormN)r0   r1   r2   rG   rN   r    r    r    r!   r=   a   s   r=   c                   @   s   e Zd Zdd ZdS )TestShLegendrec                 C   s  t ddg}td}td}td}td}td}td}td|}td|}	td|}
td|}td|}td|}t|j|jd t|j|	jd t|j|
jd t|j|jd t|j|jd	 t|j|jd	 d S 
Nr   r#   r   r   r   r   r   r   r>   )r   rK   r   sh_legendrelegendrer   r   )r   rM   ZPs0ZPs1ZPs2ZPs3ZPs4ZPs5Zpse0Zpse1Zpse2Zpse3Zpse4Zpse5r    r    r!   test_sh_legendre   s&    





zTestShLegendre.test_sh_legendreN)r0   r1   r2   rS   r    r    r    r!   rO      s   rO   c                   @   s   e Zd Zdd ZdS )TestShChebytc                 C   s  t ddg}td}td}td}td}td}td}td|}td|}	td|}
td|}td|}td|}t|j|jd t|j|	jd t|j|
jd t|j|jd t|j|jd	 t|j|jd	 d S rP   )r   rK   r   	sh_chebytr)   r   r   )r   rM   ZTs0ZTs1ZTs2ZTs3ZTs4ZTs5Ztse0Ztse1Ztse2Ztse3Ztse4Ztse5r    r    r!   test_sh_chebyt   s&    





zTestShChebyt.test_sh_chebytN)r0   r1   r2   rV   r    r    r    r!   rT      s   rT   c                   @   s   e Zd Zdd ZdS )TestShChebyuc                 C   s  t ddg}td}td}td}td}td}td}td|}td|}	td|}
td|}td|}td|}t|j|jd t|j|	jd t|j|
jd t|j|jd t|j|jd	 t|j|jd
 d S )Nr   r#   r   r   r   r   r   r   r>   r6   )r   rK   r   	sh_chebyur.   r   r   )r   rM   ZUs0ZUs1ZUs2ZUs3ZUs4ZUs5Zuse0Zuse1Zuse2Zuse3Zuse4Zuse5r    r    r!   test_sh_chebyu   s&    





zTestShChebyu.test_sh_chebyuN)r0   r1   r2   rY   r    r    r    r!   rW      s   rW   c                   @   s   e Zd Zdd ZdS )TestShJacobic                 C   s  dd }t ddg}dt j  }|d dt j   }td||}td||}td||}td||}td||}	td	||}
td|| |d ||d| }td|| |d ||d| }td|| |d ||d| }td|| |d ||d| }td|| |d ||d| }td	|| |d ||d	| }t|j|jd
 t|j|jd
 t|j|jd
 t|j|jd
 t|	j|jd
 t|
j|jd
 d S )Nc                 S   s(   t | d t | |  t d|  |  S )Nr   r   r
   )npr    r    r!   conv   s    z)TestShJacobi.test_sh_jacobi.<locals>.convr   r#   r   r   r   r   r   r   )r   rK   r8   r   Z	sh_jacobiZjacobir   r   )r   r]   rM   qr\   ZG0ZG1ZG2ZG3ZG4ZG5Zge0Zge1Zge2Zge3Zge4Zge5r    r    r!   test_sh_jacobi   s,    $$$$$$zTestShJacobi.test_sh_jacobiN)r0   r1   r2   r_   r    r    r    r!   rZ      s   rZ   c                   @   s   e Zd Zdd ZdS )TestCallc                 C   s   g }t dD ]&}|dd dt|d  D  qtjdd@ |D ]*}t|}t|dt|j	d|d	 qFW d    n1 s0    Y  d S )
Nr   c                 S   s   g | ]}|  qS r    )strip).0xr    r    r!   
<listcomp>       z&TestCall.test_call.<locals>.<listcomp>aW  
                orth.jacobi(%(n)d,0.3,0.9)
                orth.sh_jacobi(%(n)d,0.3,0.9)
                orth.genlaguerre(%(n)d,0.3)
                orth.laguerre(%(n)d)
                orth.hermite(%(n)d)
                orth.hermitenorm(%(n)d)
                orth.gegenbauer(%(n)d,0.3)
                orth.chebyt(%(n)d)
                orth.chebyu(%(n)d)
                orth.chebyc(%(n)d)
                orth.chebys(%(n)d)
                orth.sh_chebyt(%(n)d)
                orth.sh_chebyu(%(n)d)
                orth.legendre(%(n)d)
                orth.sh_legendre(%(n)d)
                )r[   r   r   g)\(?)err_msg)
rangeextenddictsplitr   r   evalr   rK   Zcoef)r   Zpolyr[   pstrr\   r    r    r!   	test_call   s    

zTestCall.test_callN)r0   r1   r2   rm   r    r    r    r!   r`      s   r`   c                   @   s   e Zd Zdd ZdS )TestGenlaguerrec                 C   sx   t tjdddddd t tjdddddd t tjddddtdd	g t tjddddtdd
g d S )Nr   F)Zmonicr   rI   T       r#   r   r   )r   r   Zgenlaguerrer   rK   )r   r    r    r!   test_regression  s     zTestGenlaguerre.test_regressionN)r0   r1   r2   rp   r    r    r    r!   rn     s   rn   V瞯<vIh%,=c                    s   | |d\}}	}
t |}||d d t jf |}t ||	 |j}dt |  }|d d t jf | | }t|t ||| t|		 |
|| dd  t
 fdd||}t  ||	}d|d k rdn
|d d }t|d	 ||d
 d S )NTr   c                 S   s   | d d| d   |  d S )Nr   r   r    rc   r    r    r!   f$  s    zverify_gauss_quad.<locals>.fc                    s    | |  S Nr    rs   rt   weight_funcr    r!   <lambda>&  re   z#verify_gauss_quad.<locals>.<lambda>gư>
   r   rtol)r   ZarangeZnewaxisdotTr   Zdiagonalr   Zeyesumr	   quadZvdot)Z	root_funcZ	eval_funcrw   r;   bNr{   atolrc   wmur[   vvvZvdZresIZresGr    rv   r!   verify_gauss_quad  s    
r   c               	   C   s  dd } dd }dd }t }|| dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || d
d|d
d|d
dd	d
ddd || d
d|d
d|d
dd	d
ddd || d
d|d
d|d
dd	d
ddd || d
d|d
d|d
dd	d
d dd || d!d!|d!d!|d!d!d	d
ddd || d!d!|d!d!|d!d!d	d
ddd || d!d!|d!d!|d!d!d	d
dd"d || d!d#|d!d#|d!d#d	d
ddd || d!d#|d!d#|d!d#d	d
ddd || d!d#|d!d#|d!d#d	d
dd"d || d$d%|d$d%|d$d%d	d
ddd || d$d%|d$d%|d$d%d	d
ddd || d$d%|d$d%|d$d%d	d
dd"d || d&d'|d&d'|d&d'd	d
d || d&d'|d&d'|d&d'd	d
ddd || d&d'|d&d'|d&d'd	d
ddd td(d)d)\}}td(\}}t||d*d* t||d*d* td(d+d+\}}td(d,\}}	t||d*d* t||	d*d* tddd-d.\}
}tddd-d/\}}}t|
|d*d* t||d*d* t|dd-d0d\}}t|||d1 tt	tjd2dd tt	tjd3dd tt	tjd-d4d tt	tjd-dd4 tt	tjd-d4d4 d S )5Nc                    s    fddS )Nc                    s   t |  |S ru   )r:   roots_jacobir[   r   r;   r   r    r!   rx   -  re   z/test_roots_jacobi.<locals>.rf.<locals>.<lambda>r    r   r    r   r!   rf,  s    ztest_roots_jacobi.<locals>.rfc                    s    fddS )Nc                    s   t |  |S ru   )r:   Zeval_jacobir[   rc   r   r    r!   rx   /  re   z/test_roots_jacobi.<locals>.ef.<locals>.<lambda>r    r   r    r   r!   ef.  s    ztest_roots_jacobi.<locals>.efc                    s    fddS )Nc                    s   d|    d|    S )Nr   r    rs   r   r    r!   rx   1  re   z/test_roots_jacobi.<locals>.wf.<locals>.<lambda>r    r   r    r   r!   wf0  s    ztest_roots_jacobi.<locals>.wf                  rH   r      -q=r   d   dy=r4   7QE=-=r   vIh%L=?r   vIh%<=g7QU==
ףp=2@L;@g^~z=̌G@r5   g     @   g     @绽|=g      @g     @@g     @@      @皙9Q@r-           +=rJ   g      @r   FTr#   rz   r   ffffff
@r   )
r   r:   r   roots_legendrer   roots_gegenbauerr	   r   assert_raises
ValueError)r   r   r   vgqZxjZwjZxlZwlZxcZwcrc   r   yr   mmuImuI_errr    r    r!   test_roots_jacobi+  s    $$((((($(($ ((((((         $  r   c               	   C   s  dd } dd }dd }t }|| dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
d || dd|dd|ddd	d
d || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
ddd || dd|dd|ddd	d
dd d || dd|dd|ddd	d
ddd tdd!dd"\}}tdd!dd#\}}}t||d$d$ t||d$d$ t|d!dd%d\}	}
t||	|
d& tttjd%dd tttjd'dd tttjd!dd tttjd!dd( tttjd!d)d( d S )*Nc                    s    fddS )Nc                    s   t |  |S ru   )r:   roots_sh_jacobir   r   r    r!   rx     re   z2test_roots_sh_jacobi.<locals>.rf.<locals>.<lambda>r    r   r    r   r!   r     s    z test_roots_sh_jacobi.<locals>.rfc                    s    fddS )Nc                    s   t |  |S ru   )r:   Zeval_sh_jacobir   r   r    r!   rx     re   z2test_roots_sh_jacobi.<locals>.ef.<locals>.<lambda>r    r   r    r   r!   r     s    z test_roots_sh_jacobi.<locals>.efc                    s    fddS )Nc                    s   d|     | d   S )NrH   r    rs   r   r    r!   rx     re   z2test_roots_sh_jacobi.<locals>.wf.<locals>.<lambda>r    r   r    r   r!   r     s    z test_roots_sh_jacobi.<locals>.wfr   g      ?r   rH   r   r   r   r   r   r   r4   r   r   r   r   r   r   r   r   r   皙?r   r   r   g(#=r   r   FTr   r   rz   r   r#   r   )r   r:   r   r   r	   r   r   r   )r   r   r   r   rc   r   r   r   r   r   r   r    r    r!   test_roots_sh_jacobi  sX    $$(($(($(($$(((((r   c            
   	   C   s<  t j} t j}tdj}t| ||tj tjd t| ||tj tjddd t| ||tj tjddd t dd\}}t dd\}}}t	||d	d	 t	||d	d	 t
|tj tj\}}	t	|||	d
 t dd\}}t dd\}}}t	||d	d	 t	||d	d	 t	t||d	d	 ttt jd ttt jd d S )Nr   r   r   r   r   r   FTr   rz      r   r   )r:   Zroots_hermiteZeval_hermiter   r@   rw   r   r   infr   r	   r   r~   r   r   
rootfevalfweightfrc   r   r   r   r   r   r   r    r    r!   test_roots_hermite  s&    r   c                     s   dd  d fdd	} | ddd	 | d
dd	 | ddd	 | ddd	 | ddd	 | ddd	 | ddd	 | ddd	 | ddd	 | ddd	 d S )Nc                 S   s   t | |jf}t jd t d|d   |dd d f< | dkrtd| |dd d f  |dd d f< td| D ]X}td| | ||d d d f  t|d | ||d d d f   ||d d f< qr|S )N      пr   r   r   r   rI   rH   )r   zerossizepiexpr   rg   )r[   nodesHkr    r    r!   hermite_recursion  s    ((Vz1test_roots_hermite_asy.<locals>.hermite_recursionrq   r   c                    sX   t | \}} | d |}t|dd d f t| || tt|ttj|| d S )Nr   r#   )r   Z_roots_hermite_asyr   r   r   r~   r   r   )r   r{   r   rc   r   r   r   r    r!   test  s     z$test_roots_hermite_asy.<locals>.test   r   r      i,  i-  i  i  i  i  i  i  )rq   r   r    )r   r    r   r!   test_roots_hermite_asy  s    
r   c            
   	   C   s   t j} t j}tdj}t| ||tj tjd t| ||tj tjddd t| ||tj tjddd t dd\}}t dd\}}}t	||d	d	 t	||d	d	 t
|tj tj\}}	t	|||	d
 ttt jd ttt jd d S )Nr   r   r   r   r   r   FTr   rz   r   r   )r:   Zroots_hermitenormZeval_hermitenormr   rL   rw   r   r   r   r   r	   r   r   r   r   r    r    r!   test_roots_hermitenorm  s    r   c               	   C   s  dd } dd }dd }t }|| d|d|ddd	d
 || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
 || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
 || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
 || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
dd || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
dd || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
dd || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
dd || d|d|ddd	ddd || d|d|ddd	ddd || d|d|ddd	d
dd || d|d|ddd	ddd || d|d|ddd	ddd || dtj|ddd	d
 || dtj|ddd	d || dtj|ddd	ddd td
dd\}}td
dd\}}}t||dd t||dd t|dd d\}	}
t||	|
d! tttjdd tttjd"d tttjd#d$ d S )%Nc                    s    fddS )Nc                    s   t |  |S ru   )r:   r   r   r;   r    r!   rx     re   z6test_roots_gegenbauer.<locals>.rootf.<locals>.<lambda>r    r   r    r   r!   r     s    z$test_roots_gegenbauer.<locals>.rootfc                    s    fddS )Nc                    s   t |  |S ru   )r:   Zeval_gegenbauerr   r   r    r!   rx     re   z6test_roots_gegenbauer.<locals>.evalf.<locals>.<lambda>r    r   r    r   r!   r     s    z$test_roots_gegenbauer.<locals>.evalfc                    s    fddS )Nc                    s   d| d   d  S )Nr   r   r4   r    rs   r   r    r!   rx     re   z8test_roots_gegenbauer.<locals>.weightf.<locals>.<lambda>r    r   r    r   r!   r     s    z&test_roots_gegenbauer.<locals>.weightfr   r   rH   r   r   r   r   r   r   皙?r   r   ry   2      g     Pe@g-A=   g     @r   r   r   FTr   r#   rz   r   r   r   )	r   r:   eval_chebytr   r   r	   r   r   r   r   r   r   r   rc   r   r   r   r   r   r   r    r    r!   test_roots_gegenbauer  sV    """"""""""""""""""""""" r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddddd tdd\}}tdd	\}}}t||d
d
 t||d
d
 t	| dd\}}t|||d t
ttjd t
ttjd d S )Nr   r   rH   r   r   r   r   FTr   r#   r   rz   r   r   )r   r)   rw   r   r:   roots_chebytr   r   r	   r   r   r   r   rc   r   r   r   r   r   r   r    r    r!   test_roots_chebytO  s    r   c                  C   sP   t d\} }| d d | dd   }}t||d d d   t| d d d S )N   ry   r6   r#   r   )r:   r   r   )rc   r   posnegr    r    r!   test_chebyt_symmetry`  s    r   c                  C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddd tdd\}}tdd\}}}t||dd t||dd t	| d	d
\}}t|||d t
ttjd t
ttjd d S )Nr   r   rH   r   r   FTr   r#   r   rz   r   r   )r   r.   rw   r   r:   Zroots_chebyuZeval_chebyur   r	   r   r   r   r   r    r    r!   test_roots_chebyuf  s    r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddddd tdd\}}tdd	\}}}t||d
d
 t||d
d
 t	| dd\}}t|||d t
ttjd t
ttjd d S )Nr   ro   rI   r   r   r   r   FTr   r   r   rz   r   r   )r   r   rw   r   r:   Zroots_chebycZeval_chebycr   r	   r   r   r   r   r    r    r!   test_roots_chebycw  s    r   c                  C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddd tdd\}}tdd\}}}t||dd t||dd t	| d	d
\}}t|||d t
ttjd t
ttjd d S )Nr   ro   rI   r   r   FTr   r   r   rz   r   r   )r   r$   rw   r   r:   Zroots_chebysZeval_chebysr   r	   r   r   r   r   r    r    r!   test_roots_chebys  s    r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddddd tdd\}}tdd	\}}}t||d
d
 t||d
d
 t	| dd\}}t|||d t
ttjd t
ttjd d S Nr   r   rH   r   r   r   r   FTr   r   r   rz   r   )r   rU   rw   r   r:   Zroots_sh_chebytZeval_sh_chebytr   r	   r   r   r   r   r    r    r!   test_roots_sh_chebyt  s    r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddd ttjtj| ddddd tdd\}}tdd	\}}}t||d
d
 t||d
d
 t	| dd\}}t|||d t
ttjd t
ttjd d S r   )r   rX   rw   r   r:   Zroots_sh_chebyuZeval_sh_chebyur   r	   r   r   r   r   r    r    r!   test_roots_sh_chebyu  s    r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddddd ttjtj| ddddd tdd	\}}tdd
\}}}t||dd t||dd t	| dd\}}t|||d t
ttjd t
ttjd d S )Nr   r   rH   r   r   r   r   r   FTr   r#   r   rz   r   r   )r   rR   rw   r   r:   r   Zeval_legendrer   r	   r   r   r   r   r    r    r!   test_roots_legendre  s     r   c               	   C   s   t dj} ttjtj| ddd ttjtj| ddddd ttjtj| ddddd tdd	\}}tdd
\}}}t||dd t||dd t	| dd\}}t|||d t
ttjd t
ttjd d S )Nr   r   rH   r   r   r   r   r   FTr   r   r   rz   r   )r   rQ   rw   r   r:   Zroots_sh_legendreZeval_sh_legendrer   r	   r   r   r   r   r    r    r!   test_roots_sh_legendre  s     r   c               	   C   s   t dj} ttjtj| dtjd ttjtj| dtjddd ttjtj| dtjddd tdd\}}tdd	\}}}t	||d
d
 t	||d
d
 t
| dtj\}}t	|||d tttjd tttjd d S )Nr   r   r   r   r   r   r   FTr   r   rz   r   )r   Zlaguerrerw   r   r:   Zroots_laguerreZeval_laguerrer   r   r   r	   r   r   r   r   r    r    r!   test_roots_laguerre  s     r   c               
   C   s  dd } dd }dd }t }|| d|d|ddtjd	 || d|d|ddtjd
dd || d|d|ddtjddd || d|d|ddtjd	 || d|d|ddtjd
dd || d|d|ddtjddd || d|d|ddtjd	 || d|d|ddtjd
dd || d|d|ddtjddd || d|d|ddtjd	 || d|d|ddtjd
dd || d|d|ddtjddd || d|d|ddtjd	 || d|d|ddtjd
dd || d|d|ddtjdddd td	dd\}}td	dd\}}}t||dd t||dd t|ddtj\}	}
t||	|
d tt	tjdd tt	tjdd tt	tjdd  d S )!Nc                    s    fddS )Nc                    s   t |  |S ru   )r:   roots_genlaguerrer   r   r    r!   rx     re   z7test_roots_genlaguerre.<locals>.rootf.<locals>.<lambda>r    r   r    r   r!   r     s    z%test_roots_genlaguerre.<locals>.rootfc                    s    fddS )Nc                    s   t |  |S ru   )r:   Zeval_genlaguerrer   r   r    r!   rx     re   z7test_roots_genlaguerre.<locals>.evalf.<locals>.<lambda>r    r   r    r   r!   r     s    z%test_roots_genlaguerre.<locals>.evalfc                    s    fddS )Nc                    s   |   t |   S ru   )r   r   rs   r   r    r!   rx     re   z9test_roots_genlaguerre.<locals>.weightf.<locals>.<lambda>r    r   r    r   r!   r     s    z'test_roots_genlaguerre.<locals>.weightfr   r   r   r   r   r   r   r   r   g+F=r   gN<=ry   r   r   r   )r{   r   r   FTrI   rz   r   r   r   g)
r   r   r   r:   r   r   r	   r   r   r   r   r    r    r!   test_roots_genlaguerre  s8     $$ $$ $$ $$ $&r   c                   C   s   t dd d S )NA   r   )r:   r)   r    r    r    r!   test_gh_6721   s    r   )rq   rr   )1numpyr   r   r   Znumpy.testingr   r   r   r   Zpytestr   r   Zscipyr	   Zscipy.specialZspecialr:   r   Zscipy.special._orthogonalZ_orthogonalr   r   r3   r=   rO   rT   rW   rZ   r`   rn   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r    r    r!   <module>   sH   <'	 
`7@*