%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/
Upload File :
Create Path :
Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_hermite.cpython-39.pyc

a

z[yc�H�
@s�dZddlmZddlZddlmmZddl	m
Z
ddlmZm
Z
mZmZe�dg�Ze�ddg�Ze�gd��Ze�gd	��Ze�gd
��Ze�gd��Ze�gd��Ze�gd
��Ze�gd��Ze�gd��Zeeeeeeeeeeg
Zdd�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�Z Gdd�d�Z!Gdd�d�Z"Gdd�d�Z#Gd d!�d!�Z$Gd"d#�d#�Z%Gd$d%�d%�Z&dS)&zTests for hermite module.

�)�reduceN��polyval)�assert_almost_equal�
assert_raises�assert_equal�assert_��)���r�)ri�r�)�ri���r�)r�xri`���r� )i����ri�ri ���r�@)rip���ri 
ri����r�)	i�ri���ri�4ri�r�)
ri vri��ri�ri��ricCstj|dd�S)Ng���ư>)Ztol)�herm�hermtrim��x�r�I/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_hermite.py�trimsrc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�
TestConstantscCsttjddg�dS)N���r	)rrZ
hermdomain��selfrrr�test_hermdomain!szTestConstants.test_hermdomaincCsttjdg�dS)Nr)rrZhermzerorrrr�
test_hermzero$szTestConstants.test_hermzerocCsttjdg�dS�Nr	)rrZhermonerrrr�test_hermone'szTestConstants.test_hermonecCsttjddg�dS)Nr��?)rrZhermxrrrr�
test_hermx*szTestConstants.test_hermxN)�__name__�
__module__�__qualname__r r!r#r%rrrrrsrc@sJeZdZe�ddd�Zdd�Zdd�Zdd	�Zd
d�Z	dd
�Z
dd�ZdS)�TestArithmetic�����dcCs�td�D]�}td�D]�}d|�d|��}t�t||�d�}||d7<||d7<t�dg|dgdg|dg�}tt|�t|�|d�qqdS�N��At i=�, j=r	r��err_msg)�range�np�zeros�maxr�hermaddrr�r�i�j�msg�tgt�resrrr�test_hermadd1s$zTestArithmetic.test_hermaddcCs�td�D]�}td�D]�}d|�d|��}t�t||�d�}||d7<||d8<t�dg|dgdg|dg�}tt|�t|�|d�qqdSr-)r3r4r5r6rZhermsubrrr8rrr�test_hermsub;s$zTestArithmetic.test_hermsubcCsvtt�dg�dg�tt�dg�ddg�tdd�D]<}dg|dg}dg|d|ddg}tt�|�|�q4dS)Nrr	r$r.)rrZhermmulxr3)rr9Zserr<rrr�
test_hermmulxEszTestArithmetic.test_hermmulxc
Cs�td�D]�}dg|dg}t�|j|�}td�D]x}d|�d|��}dg|dg}t�|j|�}t�||�}t�|j|�}	tt|�||dk|�t|	|||d�q2qdS)Nr.rr	r/r0r1)r3r�hermvalr�hermmulr�lenr)
rr9Zpol1Zval1r:r;Zpol2Zval2Zpol3Zval3rrr�test_hermmulMszTestArithmetic.test_hermmulc
Cs�td�D]�}td�D]z}d|�d|��}dg|dg}dg|dg}t�||�}t�||�\}}t�t�||�|�}	tt|	�t|�|d�qqdS)Nr.r/r0rr	r1)r3rr7ZhermdivrBrr)
rr9r:r;ZciZcjr<ZquoZremr=rrr�test_hermdiv[szTestArithmetic.test_hermdivcCs|td�D]n}td�D]`}d|�d|��}t�|d�}ttj|g|t�dg��}t�||�}tt	|�t	|�|d�qqdS)Nr.r/r0r	r1)
r3r4�arangerrrB�arrayZhermpowrr)rr9r:r;�cr<r=rrr�test_hermpowfszTestArithmetic.test_hermpowN)r&r'r(r4�linspacerr>r?r@rDrErIrrrrr).s

r)c@s�eZdZe�gd��Ze�dee�Ze�deee�Zej	�	d�ddZ
ee
gd��Zdd	�Z
d
d�Zdd
�Zdd�Zdd�ZdS)�TestEvaluation)g@��?g�?�i,j->ij�
i,j,k->ijk�r+r.r
r	)rLg@g@cs�tt�gdg�jd�t�dd���fdd�tD�}td�D]<}d|��}||}t��dg|dg�}t|||d�q<td	�D]^}d
g|}t�	|��tt��dg�j
|�tt��ddg�j
|�tt��gd��j
|�q�dS)Nr	rrcsg|]}t�|��qSrr��.0rHrrr�
<listcomp>��z/TestEvaluation.test_hermval.<locals>.<listcomp>�
r/r1r+r
)r	rr)rrrA�sizer4rJ�Hlistr3rr5�shape)r�yr9r;r<r=Zdimsrrr�test_hermvalzs


zTestEvaluation.test_hermvalc
Cs�|j\}}}|j\}}}tttj||dd�|j�||}t�|||j�}t||�t�	d�}	t�|	|	|j�}t
|jdk�dS�Nr
�r
r+)rrXr�
ValueErrorr�	hermval2d�c2drr4�onesrrW�
r�x1�x2�x3Zy1Zy2Zy3r<r=�zrrr�test_hermval2d�s

zTestEvaluation.test_hermval2dc
Cs�|j\}}}|j\}}}tttj|||dd�|j�|||}t�||||j�}t||�t�	d�}	t�|	|	|	|j�}t
|jdk�dSrZ)rrXrr\r�	hermval3d�c3drr4r_rrWr`rrr�test_hermval3d�s

zTestEvaluation.test_hermval3dc
Csl|j\}}}|j\}}}t�d||�}t�|||j�}t||�t�d�}	t�|	|	|j�}t	|j
dk�dS)NrMr[)r
r+r
r+)rrXr4�einsumrZ
hermgrid2dr^rr_rrWr`rrr�test_hermgrid2d�s

zTestEvaluation.test_hermgrid2dc
Csr|j\}}}|j\}}}t�d|||�}t�||||j�}t||�t�d�}	t�|	|	|	|j�}t	|j
dk�dS)NrNr[)r
r+r
r+r
r+)rrXr4rirZ
hermgrid3drgrr_rrWr`rrr�test_hermgrid3d�s

zTestEvaluation.test_hermgrid3dN)r&r'r(r4rGZc1drir^rg�randomrrrXrYrerhrjrkrrrrrKpsrKc@seZdZdd�Zdd�ZdS)�TestIntegralc
	Cs2tttjdgd�tttjdgd�tttjdgdddg�tttjdgdgd�tttjdgdgd�tttjdgdd�tdd	�D]8}dg|ddg}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}|dd�}t|	�D]}tj|dd
�}�qJtj||	d
�}tt	|�t	|���q"�qtd	�D]�}tdd	�D]n}	dg|dg}|dd�}t|	�D]}tj|d|gd
�}�q�tj||	tt|	��d
�}tt	|�t	|���q��q�td	�D]�}tdd	�D]r}	dg|dg}|dd�}t|	�D]}tj|d|gdd�}�qPtj||	tt|	��dd�}tt	|�t	|���q(�qtd	�D]�}tdd	�D]r}	dg|dg}|dd�}t|	�D]}tj|d|gdd�}�q�tj||	tt|	��dd�}tt	|�t	|���q��q�dS)Nrr$rr	)�lbnd)�scl��axisr
r.)�m�k)rrrsrn)rrrsro�rr)r�	TypeErrorr�hermintr\r3r�	poly2herm�	herm2polyrrA�list)
rr9rsr=ro�polr<Zhermpolrvr:rrr�test_hermint�s�




zTestIntegral.test_hermintcCs�tj�d�}t�dd�|jD��j}tj|dd�}t||�t�dd�|D��}tj|dd�}t||�t�dd�|D��}tj|d	dd
�}t||�dS)N�r+rcSsg|]}t�|��qSr�rrvrPrrrrR'rSz2TestIntegral.test_hermint_axis.<locals>.<listcomp>rrpcSsg|]}t�|��qSrr}rPrrrrR+rSr	cSsg|]}tj|dd��qS)r+)rsr}rPrrrrR/rSr+)rsrq)r4rl�vstack�Trrvr�rr^r<r=rrr�test_hermint_axis#s

zTestIntegral.test_hermint_axisN)r&r'r(r{r�rrrrrm�sSrmc@seZdZdd�Zdd�ZdS)�TestDerivativecCstttjdgd�tttjdgd�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�dS)	Nrr$rr.r	rtr
)rrro)
rrur�hermderr\r3rrrvr)rr9r<r=r:rrr�test_hermder6s zTestDerivative.test_hermdercCsltj�d�}t�dd�|jD��j}tj|dd�}t||�t�dd�|D��}tj|dd�}t||�dS)Nr|cSsg|]}t�|��qSr�rr�rPrrrrRSrSz4TestDerivative.test_hermder_axis.<locals>.<listcomp>rrpcSsg|]}t�|��qSrr�rPrrrrRWrSr	)r4rlr~rrr�rr�rrr�test_hermder_axisOs
z TestDerivative.test_hermder_axisN)r&r'r(r�r�rrrrr�4sr�c@s8eZdZej�d�ddZdd�Zdd�Zdd	�Zd
S)�
TestVanderrOr
r	cCs�t�d�}t�|d�}t|jdk�td�D].}dg|dg}t|d|ft�||��q,t�	ddgddgdd	gg�}t�|d�}t|jd
k�td�D].}dg|dg}t|d|ft�||��q�dS)Nr+r|rrr	.r
r.�)r+r
r)
r4rFr�
hermvanderrrWr3rrArG)rr�vr9�coefrrr�test_hermvander`s
zTestVander.test_hermvandercCsx|j\}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||�t�|g|gddg�}t	|j
dk�dS)Nr[r	r
)r	r.r�)rr4rlrZhermvander2dr]�dot�flatrrrW�rrarbrcrHZvanr<r=rrr�test_hermvander2dqs
zTestVander.test_hermvander2dcCs�|j\}}}tj�d�}t�|||gd��}t�||||�}t�||j�}t||�t�|g|g|ggd��}t	|j
dk�dS)N)r
r+r)r	r
r+)r	r.�)rr4rlrZhermvander3drfr�r�rrrWr�rrr�test_hermvander3d~s
zTestVander.test_hermvander3dN)	r&r'r(r4rlrr�r�r�rrrrr�\s
r�c@seZdZdd�ZdS)�TestFittingcCsdd�}dd�}tttjdgdgd�tttjdggdgd�tttjgdgd�tttjdgdgggd�tttjddgdgd�tttjdgddgd�tttjdgdgddggd	�tttjdgdgdddgd	�tttjdgdgdg�tttjdgdggd
��tttjdgdgg�t�dd�}||�}t�||d�}tt|�d�t	t�
||�|�t�||gd
��}tt|�d�t	t�
||�|�t�||d�}tt|�d�t	t�
||�|�t�||gd��}tt|�d�t	t�
||�|�t�||gd��}tt|�d�t	t�
||�|�t�|t�||g�jd�}t	|t�||g�j�t�|t�||g�jgd
��}t	|t�||g�j�t�
|�}|��}	d|ddd�<d|ddd�<tj||	d|d	�}
t	|
|�tj||	gd
�|d	�}
t	|
|�tj|t�|	|	g�jd|d	�}t	|t�||g�j�tj|t�|	|	g�jgd
�|d	�}t	|t�||g�j�gd�}t	t�||d�ddg�t	t�||ddg�ddg�t�dd�}||�}t�||d�}t	t�
||�|�t�||gd��}
t	t�
||
�|�t	||
�dS)NcSs||d|dS)Nr	r
rrrrr�f�sz#TestFitting.test_hermfit.<locals>.fcSs|d|ddS)Nrr
r	rrrrr�f2�sz$TestFitting.test_hermfit.<locals>.f2r	rrr
)�w)r
rr�r+r)rr	r
r+r.)rr	r
r+r)r
r+rr	r)r	y�?ry��r$)rr
r)rr\rZhermfitrur4rJrrCrrArGrZ
zeros_like�copy)rr�r�rrXZcoef3Zcoef4Zcoef2dr�ZywZwcoef3Zwcoef2dZcoef1Zcoef2rrr�test_hermfit�sp


"zTestFitting.test_hermfitN)r&r'r(r�rrrrr��sr�c@s$eZdZdd�Zdd�Zdd�ZdS)�
TestCompanioncCs"tttjg�tttjdg�dSr")rr\r�
hermcompanionrrrr�test_raises�szTestCompanion.test_raisescCs<tdd�D],}dg|dg}tt�|�j||fk�q
dS)Nr	r.r)r3rrr�rW)rr9r�rrr�test_dimensions�szTestCompanion.test_dimensionscCstt�ddg�ddk�dS)Nr	r
)rrgп)rrr�rrrr�test_linear_root�szTestCompanion.test_linear_rootN)r&r'r(r�r�r�rrrrr��sr�c@seZdZdd�ZdS)�	TestGausscCs�t�d�\}}t�|d�}t�|j||�}dt�|���}|dd�df||}t|t�	d��t�tj
�}t|��|�dS)Nr,�cr	)rZ	hermgaussr�r4r�rZsqrtZdiagonalrZeye�pi�sum)rrr�r�ZvvZvdr<rrr�test_100�szTestGauss.test_100N)r&r'r(r�rrrrr��sr�c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�TestMiscc	Cs�t�g�}tt|�dg�tdd�D]z}t�t�tjdd|d�ddd��}t�|�}t�	||�}d}t
t|�|dk�tt�|�dd�t||�q$dS)Nr	r.rr
r)
r�
hermfromrootsrrr3r4ZcosrJr�rArrCrx)rr=r9�rootsrzr<rrr�test_hermfromroots�s
*
zTestMisc.test_hermfromrootscCsltt�dg�g�tt�ddg�dg�tdd�D]4}t�dd|�}t�t�|��}tt|�t|��q2dS)Nr	g�r
r.r)rrZ	hermrootsr3r4rJr�r)rr9r<r=rrr�test_hermroots	szTestMisc.test_hermrootscCsbgd�}tttj|d�tt�|�|dd��tt�|d�|dd��tt�|d�dg�dS)N)r
rr	rrr	r*r
r)rr\rrr)rr�rrr�
test_hermtrims
zTestMisc.test_hermtrimcCstt�dd�ddg�dS)Nr+rr
)rrZhermlinerrrr�
test_hermlineszTestMisc.test_hermlinecCs2td�D]$}tt�dg|dg�t|�qdS�NrTrr	)r3rrrxrV�rr9rrr�test_herm2polyszTestMisc.test_herm2polycCs2td�D]$}tt�t|�dg|dg�qdSr�)r3rrrwrVr�rrr�test_poly2herm#szTestMisc.test_poly2hermcCs6t�ddd�}t�|d�}t�|�}t||�dS)N���r.�r
)r4rJZexprZ
hermweightr)rrr<r=rrr�test_weight's
zTestMisc.test_weightN)
r&r'r(r�r�r�r�r�r�r�rrrrr��sr�)'�__doc__�	functoolsrZnumpyr4Znumpy.polynomial.hermiteZ
polynomialZhermiterZnumpy.polynomial.polynomialrZ
numpy.testingrrrrrGZH0ZH1ZH2ZH3ZH4ZH5ZH6ZH7ZH8ZH9rVrrr)rKrmr�r�r�r�r�r�rrrr�<module>s6B^f(0M

Zerion Mini Shell 1.0