Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_hermite_e.cpython-39.pyc
a
z[yc�I �
@ s� d Z ddlmZ ddlZddlm mZ ddl m
Z
ddl m Z m
Z
mZmZ e�dg�Ze�ddg�Ze�g d��Ze�g d��Ze�g d ��Ze�g d
��Ze�g d ��Ze�g d ��Ze�g d
��Ze�g d��Zeeeeeeeeeeg
Zdd� Z G 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&dS )%z Tests for hermite_e module.
� )�reduceN��polyval)�assert_almost_equal�
assert_raises� assert_equal�assert_� )���r r )r ���r r )� r i����r r )r � r i����r r )��r �- r r r r )r i����r �i r i��r r ) r r i\���r �� r i��r r )
r i� r i���r iz r i���r r c C s t j| dd�S )Ng���ư>)Ztol)�herme� hermetrim��x� r �K/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_hermite_e.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 hermedomain��selfr r r �test_hermedomain! s z TestConstants.test_hermedomainc C s t tjdg� d S )Nr )r r Z hermezeror r r r �test_hermezero$ s z TestConstants.test_hermezeroc C s t tjdg� d S �Nr )r r Zhermeoner r r r �
test_hermeone' s zTestConstants.test_hermeonec C s t tjddg� d S )Nr r )r r Zhermexr r r r � test_hermex* s zTestConstants.test_hermexN)�__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 )�TestArithmeticr r �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 �hermeaddr r �r �i�j�msg�tgt�resr r r �
test_hermeadd1 s $z TestArithmetic.test_hermeaddc 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& )r, r- r. r/ r Zhermesubr r r1 r r r �
test_hermesub; s $z TestArithmetic.test_hermesubc C sv t t�dg�dg� t t�dg�ddg� tdd�D ]<}dg| dg }dg|d |ddg }t t�|�|� q4d S )Nr r r' )r r Z hermemulxr, )r r2 Zserr5 r r r �test_hermemulxE s z TestArithmetic.test_hermemulxc
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* )r, r �hermevalr �hermemulr �lenr )
r r2 Zpol1Zval1r3 r4 Zpol2Zval2Zpol3Zval3r r r �
test_hermemulM s z TestArithmetic.test_hermemulc
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 r0 Zhermedivr; r r )
r r2 r3 r4 ZciZcjr5 ZquoZremr6 r r r �
test_hermediv[ s z TestArithmetic.test_hermedivc 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 r; �arrayZhermepowr r )r r2 r3 r4 �cr5 r6 r r r �
test_hermepowf s z TestArithmetic.test_hermepowN) r! r" r# r- �linspacer r7 r8 r9 r= r> rB 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 @� @� @�i,j->ij�
i,j,k->ijk�r r' � r )g �?rE rF 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 ��.0rA r r r �
<listcomp>� � z0TestEvaluation.test_hermeval.<locals>.<listcomp>�
r( r* r rJ )r r r ) r r r: �sizer- rC �Helistr, r r. �shape)r �yr2 r4 r5 r6 Zdimsr r r �
test_hermevalz s
z TestEvaluation.test_hermevalc
C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t
|j dk� d S �NrJ �rJ r ) r rS r �
ValueErrorr �
hermeval2d�c2dr r- �onesr rR �
r �x1�x2�x3Zy1Zy2Zy3r5 r6 �zr r r �test_hermeval2d� s
z TestEvaluation.test_hermeval2dc
C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t
|j dk� d S rU ) r rS r rW r �
hermeval3d�c3dr r- rZ r rR r[ r r r �test_hermeval3d� s
z TestEvaluation.test_hermeval3dc
C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j
dk� d S )NrG rV )rJ r rJ r ) r rS r- �einsumr Z hermegrid2drY r rZ r rR r[ r r r �test_hermegrid2d� s
z TestEvaluation.test_hermegrid2dc
C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j
dk� d S )NrH rV )rJ r rJ r rJ r ) r rS r- rd r Z hermegrid3drb r rZ r rR r[ r r r �test_hermegrid3d� s
z TestEvaluation.test_hermegrid3dN)r! r" r# r- r@ Zc1drd rY rb �randomr r rS rT r` rc re rf r r r r rD p s rD 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��axisrJ r' )�m�k)rn ro rj )rn ro rk �rn ) r � TypeErrorr �hermeintrW r, r �
poly2herme�
herme2polyr r: �list)
r r2 ro r6 rk �polr5 Zhermepolrr r3 r r r �
test_hermeint� s�
zTestIntegral.test_hermeintc 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 rr rK r r r rM ' rN z3TestIntegral.test_hermeint_axis.<locals>.<listcomp>r rl c S s g | ]}t �|��qS r rz rK r r r rM + rN r c S s g | ]}t j|d d��qS )r )ro rz rK r r r rM / rN r )ro rm )r- rg �vstack�Tr rr r �r rY r5 r6 r r r �test_hermeint_axis# s
z TestIntegral.test_hermeint_axisN)r! r" r# rw r~ r r r r rh � s Srh 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 ri r
r' r rp rJ )rn rk )
r rq r �hermederrW r, r r rr r )r r2 r5 r6 r3 r r r �
test_hermeder6 s$ �z TestDerivative.test_hermederc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )Nrx c S s g | ]}t �|��qS r �r r� rK r r r rM T rN z5TestDerivative.test_hermeder_axis.<locals>.<listcomp>r rl c S s g | ]}t �|��qS r r� rK r r r rM X rN r )r- rg r{ r| r r� r r} r r r �test_hermeder_axisP s
z!TestDerivative.test_hermeder_axisN)r! r" r# r� r� r r r r r 4 s r c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd
S ) �
TestVanderrI rJ 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 rx ry r r .rJ r' � )r rJ ry )
r- r? r � hermevanderr rR r, r r: r@ )r r �vr2 �coefr r r �test_hermevandera s
zTestVander.test_hermevanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j
dk� d S )NrV r rJ )r r' r� ) r r- rg r Z
hermevander2drX �dot�flatr r rR �r r\ r] r^ rA Zvanr5 r6 r r r �test_hermevander2dr s
z TestVander.test_hermevander2dc C s� | j \}}}tj�d�}t�|||g d��}t�||||�}t�||j�}t||� t�|g|g|gg d��}t |j
dk� d S )N)rJ r ry )r rJ r )r r' � ) r r- rg r Z
hermevander3dra r� r� r r rR r� r r r �test_hermevander3d s
z TestVander.test_hermevander3dN) r! r" r# r- rg r r� r� r� r r r r r� ] 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 rJ r r r r r �f� s z$TestFitting.test_hermefit.<locals>.fc S s | d | d d S )Nry rJ r r r r r r �f2� s z%TestFitting.test_hermefit.<locals>.f2r r
r rJ )�w)rJ r
r� r ry )r r rJ r r' )r r rJ r ry )rJ r ry r r )r y �?r
y � �)r rJ ry )r rW r Zhermefitrq r- rC r r< r r: r@ r| Z
zeros_like�copy)r r� r� r rS Zcoef3Zcoef4Zcoef2dr� ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r �
test_hermefit� sp
" zTestFitting.test_hermefitN)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 rW r �hermecompanionr 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 )r, r r r� rR )r r2 r� r r r �test_dimensions� s z TestCompanion.test_dimensionsc C s t t�ddg�d dk� d S )Nr rJ )r r � �)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�� t�dtj
�}t|� � |� d S )Nr% �c r rJ ) r Z
hermegaussr� r- r� r| ZsqrtZdiagonalr Zeye�pi�sum)r r r� r� ZvvZvdr5 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 rJ r
)
r �hermefromrootsr r r, r- ZcosrC r� r: r r< rt )r r6 r2 �rootsrv r5 r r r �test_hermefromroots� s
*
z TestMisc.test_hermefromrootsc C sl t t�dg�g � t t�ddg�dg� tdd�D ]4}t�dd|�}t�t�|��}t t|�t|�� q2d S )Nr r
rJ r' )r r Z
hermerootsr, r- rC r� r )r r2 r5 r6 r r r �test_hermeroots
s zTestMisc.test_hermerootsc C sb g d�}t ttj|d� tt�|�|d d� � tt�|d�|d d� � tt�|d�dg� d S )N)rJ r
r r r
r r rJ r )r rW r r r )r r� r r r �test_hermetrim s
zTestMisc.test_hermetrimc C s t t�dd�ddg� d S )Nr ry )r r Z hermeliner r r r �test_hermeline s zTestMisc.test_hermelinec C s2 t d�D ]$}tt�dg| dg �t| � qd S �NrO r r )r, r r rt rQ �r r2 r r r �test_herme2poly s zTestMisc.test_herme2polyc C s2 t d�D ]$}tt�t| �dg| dg � qd S r� )r, r r rs rQ r� r r r �test_poly2herme$ s zTestMisc.test_poly2hermec C s8 t �ddd�}t �d|d �}t�|�}t||� d S )N���r' � r� rJ )r- rC Zexpr Z hermeweightr )r r r5 r6 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 Znumpyr- Znumpy.polynomial.hermite_eZ
polynomialZ hermite_er Znumpy.polynomial.polynomialr Z
numpy.testingr r r r r@ ZHe0ZHe1ZHe2ZHe3ZHe4ZHe5ZHe6ZHe7ZHe8ZHe9rQ r r r$ rD rh r r� r� r� r� r� r r r r �<module> s6 B^f)0M
Mini Shell