Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_laguerre.cpython-39.pyc
a
z[ycgD � @ st d Z ddlmZ ddlZddlm mZ ddl m
Z
ddl m Z m
Z
mZmZ e�dg�d Ze�ddg�d Ze�g d��d Ze�g d
��d Ze�g d ��d
Ze�g d��d Ze�g d��d ZeeeeeeegZdd� ZG dd� d�ZG dd� d�ZG 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 laguerre module.
� )�reduceN��polyval)�assert_almost_equal�
assert_raises� assert_equal�assert_� ���)� ���r r )� i��� r
r
)� i�����H i�r r )�x i����iX i8���� r
r )�� i �i i����i� i���r r c C s t j| dd�S )Ng���ư>)Ztol)�lag�lagtrim��x� r �J/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_laguerre.py�trim s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestConstantsc C s t tjddg� d S )Nr r )r r Z lagdomain��selfr r r �test_lagdomain s z TestConstants.test_lagdomainc C s t tjdg� d S )Nr )r r Zlagzeror r r r � test_lagzero! s zTestConstants.test_lagzeroc C s t tjdg� d S �Nr )r r Zlagoner r r r � test_lagone$ s zTestConstants.test_lagonec C s t tjddg� d S )Nr r
)r r Zlagxr r r r � test_lagx' s zTestConstants.test_lagxN)�__name__�
__module__� __qualname__r r r! r" r r r r r s r c @ sJ e Zd Ze�ddd�Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )�TestArithmetic���� �d c 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 �N� �At i=�, j=r r ��err_msg)�range�np�zeros�maxr �lagaddr r �r �i�j�msg�tgt�resr r r � test_lagadd. s $zTestArithmetic.test_lagaddc 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 r* )r0 r1 r2 r3 r Zlagsubr r r5 r r r � test_lagsub8 s $zTestArithmetic.test_lagsubc C s� t t�dg�dg� t t�dg�ddg� tdd�D ]L}dg| dg }dg|d | d| d |d g }tt�|�|� q4d S )Nr r r
r+ r )r r Zlagmulxr0 r )r r6 Zserr9 r r r � test_lagmulxB s (zTestArithmetic.test_lagmulxc
C s� t d�D ]�}dg| dg }t�| j|�}t d�D ]x}d|� d|� �}dg| dg }t�| j|�}t�||�}t�| j|�} tt|�|| d k|� t| || |d� q2qd S )Nr+ r r r, r- r. )r0 r �lagvalr �lagmulr �lenr )
r r6 Zpol1Zval1r7 r8 Zpol2Zval2Zpol3Zval3r r r � test_lagmulJ s zTestArithmetic.test_lagmulc
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. )r0 r r4 Zlagdivr? r r )
r r6 r7 r8 ZciZcjr9 ZquoZremr: r r r � test_lagdivX s zTestArithmetic.test_lagdivc 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. )
r0 r1 �aranger r r? �arrayZlagpowr r )r r6 r7 r8 �cr9 r: r r r � test_lagpowc s zTestArithmetic.test_lagpowN) r# r$ r% r1 �linspacer r; r< r= rA rB rF r r r r r& + s
r& 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 "@g ,�g @�i,j->ij�
i,j,k->ijk�r( r+ r r )� �?g @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 ��.0rE r r r �
<listcomp>} � z.TestEvaluation.test_lagval.<locals>.<listcomp>� r, r. r( r )r r r ) r r r> �sizer1 rG �Llistr0 r r2 �shape)r �yr6 r8 r9 r: Zdimsr r r � test_lagvalw s
zTestEvaluation.test_lagvalc
C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t
|j dk� d S �Nr �r r( ) r rU r �
ValueErrorr �lagval2d�c2dr r1 �onesr rT �
r �x1�x2�x3Zy1Zy2Zy3r9 r: �zr r r �
test_lagval2d� s
z TestEvaluation.test_lagval2dc
C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t
|j dk� d S rW ) r rU r rY r �lagval3d�c3dr r1 r\ r rT r] r r r �
test_lagval3d� s
z TestEvaluation.test_lagval3dc
C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j
dk� d S )NrI rX )r r( r r( ) r rU r1 �einsumr Z laggrid2dr[ r r\ r rT r] r r r �test_laggrid2d� s
z TestEvaluation.test_laggrid2dc
C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j
dk� d S )NrJ rX )r r( r r( r r( ) r rU r1 rf r Z laggrid3drd r r\ r rT r] r r r �test_laggrid3d� s
z TestEvaluation.test_laggrid3dN)r# r$ r% r1 rD Zc1drf r[ rd �randomr r rU rV rb re rg rh r r r r rH m s rH 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 )�lbnd)�scl��axisr r+ )�m�k)rp rq rl )rp rq rm �rp ) r � TypeErrorr �lagintrY r0 r �poly2lag�lag2polyr r> �list)
r r6 rq r: rm �polr9 Zlagpolrt r7 r r r � test_lagint� s�
zTestIntegral.test_lagintc 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�r( � c S s g | ]}t �|��qS r �r rt rM r r r rO $ rP z1TestIntegral.test_lagint_axis.<locals>.<listcomp>r rn c S s g | ]}t �|��qS r r| rM r r r rO ( rP r c S s g | ]}t j|d d��qS )r( )rq r| rM r r r rO , rP r( )rq ro )r1 ri �vstack�Tr rt r �r r[ r9 r: r r r �test_lagint_axis s
z TestIntegral.test_lagint_axisN)r# r$ r% ry r� r r r r rj � s Srj 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 rk r
r+ r rr r )rp rm )
r rs r �lagderrY r0 r r rt r )r r6 r9 r: r7 r r r � test_lagder3 s zTestDerivative.test_lagderc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )Nrz c S s g | ]}t �|��qS r �r r� rM r r r rO P rP z3TestDerivative.test_lagder_axis.<locals>.<listcomp>r rn c S s g | ]}t �|��qS r r� rM r r r rO T rP r )r1 ri r} r~ r r� r r r r r �test_lagder_axisL s
z TestDerivative.test_lagder_axisN)r# r$ r% r� r� r r r r r� 1 s r� c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd
S ) �
TestVanderrK 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 ) Nr( rz r{ r r .r r+ r
)r( r r{ )
r1 rC r � lagvanderr rT r0 r r> rD )r r �vr6 �coefr r r �test_lagvander] s
zTestVander.test_lagvanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j
dk� d S )NrX r r )r r+ r
) r r1 ri r Z lagvander2drZ �dot�flatr r rT �r r^ r_ r` rE Zvanr9 r: r r r �test_lagvander2dn s
zTestVander.test_lagvander2dc C s� | j \}}}tj�d�}t�|||g d��}t�||||�}t�||j�}t||� t�|g|g|gg d��}t |j
dk� d S )N)r r( r{ )r r r( )r r+ r ) r r1 ri r Z lagvander3drc r� r� r r rT r� r r r �test_lagvander3d{ s
zTestVander.test_lagvander3dN) r# r$ r% r1 ri r r� r� r� r r r r r� Y s
r� c @ s e Zd Zdd� ZdS )� TestFittingc C sf 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�|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� d S )Nc S s | | d | d S )Nr r r r r r r �f� s z"TestFitting.test_lagfit.<locals>.fr r
r r )�w)r r
r
r( r{ )r r r r( r+ )r r r r( r{ )r y �?r
y � �)r rY r Zlagfitrs r1 rG r r@ r r> rD r~ Z
zeros_like�copy) r r� r rU Zcoef3Zcoef4Zcoef2dr� ZywZwcoef3Zwcoef2dr r r � test_lagfit� sZ
"zTestFitting.test_lagfitN)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 r )r rY r � lagcompanionr r r r � test_raises� s zTestCompanion.test_raisesc C s< t dd�D ],}dg| dg }tt�|�j||fk� q
d S )Nr r+ r )r0 r r r� rT )r r6 r� r r r �test_dimensions� s z TestCompanion.test_dimensionsc C s t t�ddg�d dk� d S )Nr r )r r g �?)r r r� r r r r �test_linear_root� s z TestCompanion.test_linear_rootN)r# r$ r% 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�� d}t|�
� |� d S )Nr) �c r rL ) r Zlaggaussr� r1 r� r~ ZsqrtZdiagonalr Zeye�sum)r r r� r� ZvvZvdr9 r r r �test_100� s zTestGauss.test_100N)r# r$ r% r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TestMiscc C s� t �g �}tt|�dg� tdd�D ]z}t�t�tj dd| d �dd d� �}t �|�}t � ||�}d}t
t |�|d k� tt � |�d d� t||� q$d S )Nr r+ r r r
)
r � lagfromrootsr r r0 r1 ZcosrG Zpir> r r@ rv )r r: r6 �rootsrx r9 r r r �test_lagfromroots� s
*
zTestMisc.test_lagfromrootsc 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 Zlagrootsr0 r1 rG r� r )r r6 r9 r: r r r �
test_lagroots� s zTestMisc.test_lagrootsc 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 rY r r r )r r� r r r � test_lagtrim� s
zTestMisc.test_lagtrimc C s t t�dd�ddg� d S )Nr( r{ rQ r )r r Zlagliner r r r � test_lagline
s zTestMisc.test_laglinec C s2 t d�D ]$}tt�dg| dg �t| � qd S �NrQ r r )r0 r r rv rS �r r6 r r r �
test_lag2poly
s zTestMisc.test_lag2polyc C s2 t d�D ]$}tt�t| �dg| dg � qd S r� )r0 r r ru rS r� r r r �
test_poly2lag s zTestMisc.test_poly2lagc C s2 t �ddd�}t �| �}t�|�}t||� d S )Nr �
� )r1 rG Zexpr Z lagweightr )r r r9 r: r r r � test_weight s
zTestMisc.test_weightN)
r# r$ r% r� r� r� r� r� r� r� r r r r r� � s r� )$�__doc__� functoolsr Znumpyr1 Znumpy.polynomial.laguerreZ
polynomialZlaguerrer Znumpy.polynomial.polynomialr Z
numpy.testingr r r r rD ZL0ZL1ZL2ZL3ZL4ZL5ZL6rS r r r& rH rj r� r� r� r� r� r� r r r r �<module> s0 B^f(0>
Mini Shell