Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_chebyshev.cpython-39.pyc
a
z[yc*P �
@ sh d Z ddlmZ ddlZddlm mZ ddl m
Z
ddl m Z m
Z
mZmZ dd� ZdgZddgZg d �Zg d
�Zg d �Zg d �Zg d
�Zg d�Zg d�Zg d�Zeeeeeeeeeeg
ZG dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd� d�Z G dd � 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'dS ))z Tests for chebyshev module.
� )�reduceN��polyval)�assert_almost_equal�
assert_raises� assert_equal�assert_c C s t j| dd�S )Ng���ư>)Ztol)�cheb�chebtrim��x� r
�K/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_chebyshev.py�trim s r � )���r � )r ���r � )r r i����r � )r � r i��r � )r r � r i���r � )r i����r �8 r i����r �@ ) r r i��r � r i ���r � )
r � r i����r i� r i����r � c @ s e Zd Zdd� Zdd� ZdS )� TestPrivatec C sd t d�D ]V}t�dgdg| tj�}t�dg| dg dg| tj�}t�|�}t||� qd S )Nr r r � �?)�range�np�array�doubler Z_cseries_to_zseriesr ��self�i�inp�tgt�resr
r
r �test__cseries_to_zseries! s
$
z$TestPrivate.test__cseries_to_zseriesc C sd t d�D ]V}t�dg| dg dg| tj�}t�dgdg| tj�}t�|�}t||� qd S )Nr r! r r )r" r# r$ r% r Z_zseries_to_cseriesr r& r
r
r �test__zseries_to_cseries( s
$
z$TestPrivate.test__zseries_to_cseriesN)�__name__�
__module__� __qualname__r, r- r
r
r
r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestConstantsc C s t tjddg� d S )Nr r )r r Z
chebdomain�r' r
r
r �test_chebdomain2 s z TestConstants.test_chebdomainc C s t tjdg� d S )Nr )r r Zchebzeror2 r
r
r �
test_chebzero5 s zTestConstants.test_chebzeroc C s t tjdg� d S �Nr )r r Zcheboner2 r
r
r � test_chebone8 s zTestConstants.test_chebonec C s t tjddg� d S )Nr r )r r Zchebxr2 r
r
r �
test_chebx; s zTestConstants.test_chebxN)r. r/ r0 r3 r4 r6 r7 r
r
r
r r1 0 s r1 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestArithmeticc C s� t d�D ]�}t d�D ]�}d|� d|� �}t�t||�d �}|| d7 < || d7 < t�dg| dg dg| dg �}tt|�t|�|d� qqd S �Nr �At i=�, j=r r ��err_msg)r" r# �zeros�maxr �chebaddr r �r' r( �j�msgr* r+ r
r
r � test_chebaddA s $zTestArithmetic.test_chebaddc C s� t d�D ]�}t d�D ]�}d|� d|� �}t�t||�d �}|| d7 < || d8 < t�dg| dg dg| dg �}tt|�t|�|d� qqd S r9 )r" r# r> r? r Zchebsubr r rA r
r
r � test_chebsubK s $zTestArithmetic.test_chebsubc C st t t�dg�dg� t t�dg�ddg� tdd�D ]:}dg| dg }dg|d g d� }t t�|�|� q4d S )Nr r r )r! r r! )r r Zchebmulxr" )r' r( Zserr* r
r
r �
test_chebmulxU s z TestArithmetic.test_chebmulxc C s� t d�D ]�}t d�D ]�}d|� d|� �}t�|| d �}||| d7 < |t|| � d7 < t�dg| dg dg| dg �}tt|�t|�|d� qqd S )Nr r: r; r r! r r< )r" r# r> �absr �chebmulr r rA r
r
r � test_chebmul] s $zTestArithmetic.test_chebmulc
C s� t d�D ]�}t d�D ]z}d|� d|� �}dg| dg }dg| dg }t�||�}t�||�\}}t�t�||�|�} tt| �t|�|d� qqd S )Nr r: r; r r r< )r" r r@ ZchebdivrH r r )
r' r( rB rC ZciZcjr* ZquoZremr+ r
r
r � test_chebdivg s zTestArithmetic.test_chebdivc C s| t d�D ]n}t d�D ]`}d|� d|� �}t�|d �}ttj|g| t�dg��}t�||�}tt |�t |�|d� qqd S )Nr r: r; r r< )
r" r# �aranger r rH r$ Zchebpowr r )r' r( rB rC �cr* r+ r
r
r � test_chebpowr s zTestArithmetic.test_chebpowN) r. r/ r0 rD rE rF rI rJ rM r
r
r
r r8 ? s
r8 c @ s� e Zd Ze�g d��Ze�dee�Ze�deee�Zej � d�d d Z
e e
g d��Z dd � Z
d
d � Zd d
� Zdd� Zdd� ZdS )�TestEvaluation)g @� @� �?�i,j->ij�
i,j,k->ijk�� r r r )� �?rO g @c s� t t�g dg�jd� t�dd�� � fdd�tD �}td�D ]<}d|� �}|| }t�� dg| dg �}t|||d� q<td �D ]^}d
g| }t� |�� t t�� dg�j
|� t t�� ddg�j
|� t t�� g d ��j
|� q�d S ) Nr r r c s g | ]}t � |��qS r
r ��.0rL r r
r �
<listcomp>� � z/TestEvaluation.test_chebval.<locals>.<listcomp>�
r: r< rT r )r r r ) r r �chebval�sizer# �linspace�Tlistr" r r> �shape)r' �yr( rC r* r+ Zdimsr
r r � test_chebval� s
zTestEvaluation.test_chebvalc
C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t
|j dk� d S �Nr �r rT ) r r` r �
ValueErrorr � chebval2d�c2dr r# �onesr r_ �
r' �x1�x2�x3Zy1Zy2Zy3r* r+ �zr
r
r �test_chebval2d� s
z TestEvaluation.test_chebval2dc
C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t
|j dk� d S rb ) r r` r rd r � chebval3d�c3dr r# rg r r_ rh r
r
r �test_chebval3d� s
z TestEvaluation.test_chebval3dc
C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j
dk� d S )NrQ rc )r rT r rT ) r r` r# �einsumr Z
chebgrid2drf r rg r r_ rh r
r
r �test_chebgrid2d� s
z TestEvaluation.test_chebgrid2dc
C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j
dk� d S )NrR rc )r rT r rT r rT ) r r` r# rq r Z
chebgrid3dro r rg r r_ rh r
r
r �test_chebgrid3d� s
z TestEvaluation.test_chebgrid3dN)r. r/ r0 r# r$ Zc1drq rf ro �randomr r r` ra rm rp rr rs r
r
r
r rN | s rN c @ s e Zd Zdd� Zdd� ZdS )� TestIntegralc
C s2 t ttjdgd� t ttjdgd� t ttjdgdddg� t ttjdgdgd� t ttjdgdgd� t ttjdgdd� tdd �D ]8}dg|d dg }tjdg||d
�}t|ddg� q�td �D ]n}|d }dg| dg }|gdg| d| g }t�|�}tj|d|gd
�}t�|�}tt |�t |�� q�td �D ]N}|d }dg| dg }t�|�}tj|d|gdd �}tt�
d|�|� �q@td �D ]r}|d }dg| dg }|gdg| d| g }t�|�}tj|d|gdd �}t�|�}tt |�t |�� �q�td �D ]r}tdd �D ]`} dg| dg }|d d � }t| �D ]}tj|dd
�}�qJtj|| d
�}tt |�t |�� �q"�qtd �D ]�}tdd �D ]n} dg| dg }|d d � }t| �D ]}tj|d|gd
�}�q�tj|| t t| ��d
�}tt |�t |�� �q��q�td �D ]�}tdd �D ]r} dg| dg }|d d � }t| �D ]}tj|d|gdd �}�qPtj|| t t| ��dd �}tt |�t |�� �q(�qtd �D ]�}tdd �D ]r} dg| dg }|d d � }t| �D ]}tj|d|gdd �}�q�tj|| t t| ��dd �}tt |�t |�� �q��q�d S )Nr r! r r )�lbnd)�scl��axisr r )�m�k)rz r{ rv )rz r{ rw �rz ) r � TypeErrorr �chebintrd r" r � poly2cheb� cheb2polyr r[ �list)
r' r( r{ r+ rw Zpolr* Zchebpolr~ rB r
r
r � test_chebint� s�
zTestIntegral.test_chebintc C s� t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� t �dd� |D ��}tj|d dd
�}t||� d S ) N�rT r c S s g | ]}t �|��qS r
�r r~ rV r
r
r rX 3 rY z2TestIntegral.test_chebint_axis.<locals>.<listcomp>r rx c S s g | ]}t �|��qS r
r� rV r
r
r rX 7 rY r c S s g | ]}t j|d d��qS )rT )r{ r� rV r
r
r rX ; rY rT )r{ ry )r# rt �vstack�Tr r~ r �r' rf r* r+ r
r
r �test_chebint_axis/ s
z TestIntegral.test_chebint_axisN)r. r/ r0 r� r� r
r
r
r ru � s Sru c @ s e Zd Zdd� Zdd� ZdS )�TestDerivativec C s t ttjdgd� t ttjdgd� td�D ]4}dg| dg }tj|dd�}tt|�t|�� q,td�D ]N}tdd�D ]>}dg| dg }tjtj||d�|d�}t t|�t|�� qxqjtd�D ]R}tdd�D ]B}dg| dg }tjtj||dd�|dd�}t t|�t|�� q�q�d S ) Nr r! r r r r| r )rz rw )
r r} r �chebderrd r" r r r~ r )r' r( r* r+ rB r
r
r � test_chebderB s zTestDerivative.test_chebderc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )Nr� c S s g | ]}t �|��qS r
�r r� rV r
r
r rX _ rY z4TestDerivative.test_chebder_axis.<locals>.<listcomp>r rx c S s g | ]}t �|��qS r
r� rV r
r
r rX c rY r )r# rt r� r� r r� r r� r
r
r �test_chebder_axis[ s
z TestDerivative.test_chebder_axisN)r. r/ r0 r� r� r
r
r
r r� @ s r� c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd
S ) �
TestVanderrS r r c C s� t �d�}t�|d�}t|jdk� td�D ].}dg| dg }t|d|f t�||�� q,t � ddgddgdd gg�}t�|d�}t|jd
k� td�D ].}dg| dg }t|d|f t�||�� q�d S ) NrT r� r r r .r r � )rT r r )
r# rK r �
chebvanderr r_ r" r r[ r$ )r' r �vr( �coefr
r
r �test_chebvanderl s
zTestVander.test_chebvanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j
dk� d S )Nrc r r )r r r� ) r r# rt r Z chebvander2dre �dot�flatr r r_ �r' ri rj rk rL Zvanr* r+ r
r
r �test_chebvander2d} s
z TestVander.test_chebvander2dc C s� | j \}}}tj�d�}t�|||g d��}t�||||�}t�||j�}t||� t�|g|g|gg d��}t |j
dk� d S )N)r rT r )r r rT )r r � ) r r# rt r Z chebvander3drn r� r� r r r_ r� r
r
r �test_chebvander3d� s
z TestVander.test_chebvander3dN) r. r/ r0 r# rt r r� r� r� r
r
r
r r� h s
r� c @ s e Zd Zdd� ZdS )� TestFittingc C s dd� }dd� }t ttjdgdgd� t ttjdggdgd� t ttjg dgd� t ttjdgdgggd� t ttjddgdgd� t ttjdgddgd� t ttjdgdgddggd � t ttjdgdgdddgd � t ttjdgdgdg� t ttjdgdgg d
�� t ttjdgdgg � t�dd�}||�}t�||d �}tt|�d � t t�
||�|� t�||g d
��}tt|�d � t t�
||�|� t�||d �}tt|�d� t t�
||�|� t�||g d��}tt|�d� t t�
||�|� t�||g d��}tt|�d� t t�
||�|� t�|t� ||g�j d �}t |t� ||g�j � t�|t� ||g�j g d
��}t |t� ||g�j � t�
|�}|�� } d|dd d�< d|dd d�< tj|| d |d �}
t |
|� tj|| g d
�|d �}
t |
|� tj|t� | | g�j d |d �} t | t� ||g�j � tj|t� | | g�j g d
�|d �} t | t� ||g�j � g d�}t t�||d�ddg� t t�||ddg�ddg� t�dd�}||�}t�||d �} t t�
|| �|� t�||g d��}
t t�
||
�|� t | |
� d S )Nc S s | | d | d S �Nr r r
r r
r
r �f� s z#TestFitting.test_chebfit.<locals>.fc S s | d | d d S )Nr r r r
r r
r
r �f2� s z$TestFitting.test_chebfit.<locals>.f2r r r r )�w)r r r� rT r )r r r rT r )r r r rT r )r rT r r r )r y �?r y � �)r r r )r rd r Zchebfitr} r# r] r �lenr r[ r$ r� Z
zeros_like�copy)r' r� r� r r` Zcoef3Zcoef4Zcoef2dr� ZywZwcoef3Zwcoef2dZcoef1Zcoef2r
r
r � test_chebfit� sp
" zTestFitting.test_chebfitN)r. r/ r0 r� r
r
r
r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestInterpolatec C s ||d |d S r� r
)r' r r
r
r r� � s zTestInterpolate.fc C s( t ttj| jd� t ttj| jd� d S )Nr g $@)r rd r �chebinterpolater� r} r2 r
r
r � test_raises� s zTestInterpolate.test_raisesc C s2 t dd�D ]"}tt�| j|�j|d fk� q
d S )Nr r )r" r r r� r� r_ )r' �degr
r
r �test_dimensions� s z TestInterpolate.test_dimensionsc C sj dd� }t �ddd�}tdd�D ]D}td|d �D ]0}t�|||f�}tt�||�|||�dd� q2q d S ) Nc S s | | S )Nr
)r �pr
r
r �powx� s z0TestInterpolate.test_approximation.<locals>.powxr r rZ r � )�decimal)r# r] r" r r� r r[ )r' r� r r� r� rL r
r
r �test_approximation� s z"TestInterpolate.test_approximationN)r. r/ r0 r� r� r� r� r
r
r
r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
TestCompanionc C s" t ttjg � t ttjdg� d S r5 )r rd r �
chebcompanionr2 r
r
r r� s zTestCompanion.test_raisesc C s< t dd�D ],}dg| dg }tt�|�j||fk� q
d S )Nr r r )r" r r r� r_ )r' r( r� r
r
r r� s z TestCompanion.test_dimensionsc C s t t�ddg�d dk� d S )Nr r )r r � �)r r r� r2 r
r
r �test_linear_root s z TestCompanion.test_linear_rootN)r. r/ r0 r� r� r� r
r
r
r r� � s r� c @ s e Zd Zdd� ZdS )� TestGaussc C s~ t �d�\}}t �|d�}t�|j| |�}dt�|�� � }|d d �d f | | }t|t� d�� tj
}t|� � |� d S )N�d �c r ) r Z chebgaussr� r# r� r� �sqrtZdiagonalr Zeye�pi�sum)r' r r� r� ZvvZvdr* r
r
r �test_100 s zTestGauss.test_100N)r. r/ r0 r� r
r
r
r r�
s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestMiscc C s� t �g �}tt|�dg� tdd�D ]f}t�t�tj dd| d �dd d� �}dg| dg }t �|�d|d }tt|�t|�� q$d S )Nr r r r ) r �
chebfromrootsr r r" r# Zcosr] r� )r' r+ r( �rootsr* r
r
r �test_chebfromroots" s
*zTestMisc.test_chebfromrootsc C sl t t�dg�g � t t�ddg�dg� tdd�D ]4}t�dd|�}t�t�|��}t t|�t|�� q2d S )Nr r r� r r )r r Z chebrootsr" r# r] r� r )r' r( r* r+ r
r
r �test_chebroots+ s zTestMisc.test_chebrootsc C sb g d�}t ttj|d� tt�|�|d d� � tt�|d�|d d� � tt�|d�dg� d S )N)r r r r r r r r r )r rd r r
r )r' r� r
r
r �
test_chebtrim3 s
zTestMisc.test_chebtrimc C s t t�dd�ddg� d S )NrT r )r r Zchebliner2 r
r
r �
test_chebline> s zTestMisc.test_cheblinec C s2 t d�D ]$}tt�dg| dg �t| � qd S �NrZ r r )r" r r r� r^ �r' r( r
r
r �test_cheb2polyA s zTestMisc.test_cheb2polyc C s2 t d�D ]$}tt�t| �dg| dg � qd S r� )r" r r r r^ r� r
r
r �test_poly2chebE s zTestMisc.test_poly2chebc C sN t �ddd�dd� }dt �d| �t �d| � }t�|�}t||� d S )Nr r � rU )r# r] r� r Z
chebweightr )r' r r* r+ r
r
r � test_weightI s
zTestMisc.test_weightc C s~ t ttjd� t ttjd� dg}tt�d�|� ddg}tt�d�|� g d�}tt�d�|� g d �}tt�d
�|� d S ) NrP r r g�;f���g�;f���?r )g�LX�z��r g�LX�z��?rT )g( 1�k��gŜ��}ؿgŜ��}�?g( 1�k��?r )r rd r Zchebpts1r �r' r* r
r
r �
test_chebpts1O s zTestMisc.test_chebpts1c C s� t ttjd� t ttjd� ddg}tt�d�|� g d�}tt�d�|� g d�}tt�d�|� g d �}tt�d
�|� d S ) NrP r r r )r r r rT )r r� r! r r )g �g�Kf���r g�Kf���?rU r )r rd r Zchebpts2r r� r
r
r �
test_chebpts2^ s zTestMisc.test_chebpts2N) r. r/ r0 r� r� r� r� r� r� r� r� r� r
r
r
r r� s r� )(�__doc__� functoolsr Znumpyr# Znumpy.polynomial.chebyshevZ
polynomialZ chebyshevr Znumpy.polynomial.polynomialr Z
numpy.testingr r r r r ZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9r^ r r1 r8 rN ru r� r� r� r� r� r� r� r
r
r
r �<module> s: =^f(0M
Mini Shell