HEX
Server: Apache
System: Linux pdx1-shared-a1-38 6.6.104-grsec-jammy+ #3 SMP Tue Sep 16 00:28:11 UTC 2025 x86_64
User: mmickelson (3396398)
PHP: 8.1.31
Disabled: NONE
Upload Files
File: //lib/python3/dist-packages/numpy/polynomial/tests/__pycache__/test_hermite_e.cpython-310.pyc
o

6��a�I�
@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#Gdd �d �Z$Gd!d"�d"�Z%Gd#d$�d$�Z&dS)%zTests for hermite_e module.

�)�reduceN��polyval)�assert_almost_equal�
assert_raises�assert_equal�assert_�)���rr	)r���rr	)�ri����rr	)r�ri����rr	)��r�-rrrr	)ri����r�iri��rr	)	rri\���r��ri��rr	)
ri�ri���rizri���rr	cCstj|dd�S)Ng���ư>)�tol)�herme�	hermetrim��x�r�G/usr/lib/python3/dist-packages/numpy/polynomial/tests/test_hermite_e.py�trimsrc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�
TestConstantscC�ttjddg�dS)Nr
r	)rr�hermedomain��selfrrr�test_hermedomain!�zTestConstants.test_hermedomaincC�ttjdg�dS)Nr)rr�	hermezerorrrr�test_hermezero$�zTestConstants.test_hermezerocCr!�Nr	)rr�hermeonerrrr�
test_hermeone'r$zTestConstants.test_hermeonecCr)Nrr	)rr�hermexrrrr�test_hermex*r zTestConstants.test_hermexN)�__name__�
__module__�__qualname__rr#r'r)rrrrrs
rc@sJeZdZe�ddd�Zdd�Zdd�Zdd	�Zd
d�Z	dd
�Z
dd�ZdS)�TestArithmeticrr�dcCs�td�D]H}td�D]A}d|�d|��}t�t||�d�}||d7<||d7<t�dg|dgdg|dg�}tt|�t|�|d�q
qdS�N��At i=�, j=r	r��err_msg)�range�np�zeros�maxr�hermeaddrr�r�i�j�msg�tgt�resrrr�
test_hermeadd1�$��zTestArithmetic.test_hermeaddcCs�td�D]H}td�D]A}d|�d|��}t�t||�d�}||d7<||d8<t�dg|dgdg|dg�}tt|�t|�|d�q
qdSr/)r5r6r7r8r�hermesubrrr:rrr�
test_hermesub;rAzTestArithmetic.test_hermesubcCsvtt�dg�dg�tt�dg�ddg�tdd�D]}dg|dg}dg|d|ddg}tt�|�|�qdS)Nrr	r0)rr�	hermemulxr5)rr;�serr>rrr�test_hermemulxEs�zTestArithmetic.test_hermemulxc
Cs�td�D]R}dg|dg}t�|j|�}td�D]<}d|�d|��}dg|dg}t�|j|�}t�||�}t�|j|�}	tt|�||dk|�t|	|||d�qqdS)Nr0rr	r1r2r3)r5r�hermevalr�hermemulr�lenr)
rr;�pol1�val1r<r=�pol2�val2�pol3�val3rrr�
test_hermemulMs��zTestArithmetic.test_hermemulc
Cs�td�D]D}td�D]=}d|�d|��}dg|dg}dg|dg}t�||�}t�||�\}}t�t�||�|�}	tt|	�t|�|d�q
qdS)Nr0r1r2rr	r3)r5rr9�hermedivrHrr)
rr;r<r=�ci�cjr>�quo�remr?rrr�
test_hermediv[s��zTestArithmetic.test_hermedivcCs|td�D]7}td�D]0}d|�d|��}t�|d�}ttj|g|t�dg��}t�||�}tt	|�t	|�|d�q
qdS)Nr0r1r2r	r3)
r5r6�arangerrrH�array�hermepowrr)rr;r<r=�cr>r?rrr�
test_hermepowfs��zTestArithmetic.test_hermepowN)r*r+r,r6�linspacerr@rCrFrPrVr[rrrrr-.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@�@�@�i,j->ij�
i,j,k->ijk�rr0�r	)g�?r^r_cs�tt�gdg�jd�t�dd���fdd�tD�}td�D]}d|��}||}t��dg|dg�}t|||d�qtd	�D]/}d
g|}t�	|��tt��dg�j
|�tt��ddg�j
|�tt��gd��j
|�qAdS)Nr	rr
csg|]}t�|��qSrr��.0rZrrr�
<listcomp>��z0TestEvaluation.test_hermeval.<locals>.<listcomp>�
r1r3rrc)r	rr)rrrG�sizer6r\�Helistr5rr7�shape)r�yr;r=r>r?�dimsrrr�
test_hermevalzs


�zTestEvaluation.test_hermevalc
Cs�|j\}}}|j\}}}tttj||dd�|j�||}t�|||j�}t||�t�	d�}	t�|	|	|j�}t
|jdk�dS�Nrc�rcr)rrlr�
ValueErrorr�
hermeval2d�c2drr6�onesrrk�
r�x1�x2�x3�y1�y2�y3r>r?�zrrr�test_hermeval2d�s

zTestEvaluation.test_hermeval2dc
Cs�|j\}}}|j\}}}tttj|||dd�|j�|||}t�||||j�}t||�t�	d�}	t�|	|	|	|j�}t
|jdk�dSro)rrlrrqr�
hermeval3d�c3drr6rtrrkrurrr�test_hermeval3d�s

zTestEvaluation.test_hermeval3dc
Csl|j\}}}|j\}}}t�d||�}t�|||j�}t||�t�d�}	t�|	|	|j�}t	|j
dk�dS)Nr`rp)rcrrcr)rrlr6�einsumr�hermegrid2drsrrtrrkrurrr�test_hermegrid2d�s

zTestEvaluation.test_hermegrid2dc
Csr|j\}}}|j\}}}t�d|||�}t�||||j�}t||�t�d�}	t�|	|	|	|j�}t	|j
dk�dS)Nrarp)rcrrcrrcr)rrlr6r�r�hermegrid3drrrtrrkrurrr�test_hermegrid3d�s

zTestEvaluation.test_hermegrid3dN)r*r+r,r6rX�c1dr�rsr�randomrrrlrnr}r�r�r�rrrrr]psr]c@�eZdZdd�Zdd�ZdS)�TestIntegralc
	Cs.tttjdgd�tttjdgd�tttjdgdddg�tttjdgdgd�tttjdgdgd�tttjdgdd�tdd	�D]}dg|ddg}tjdg||d
�}t|ddg�qCtd	�D]7}|d}dg|dg}|gdg|d|g}t�|�}tj|d|gd
�}t�|�}tt	|�t	|��qdtd	�D]&}|d}dg|dg}t�|�}tj|d|gdd�}tt�
d|�|�q�td	�D]8}|d}dg|dg}|gdg|d|g}t�|�}tj|d|gdd�}t�|�}tt	|�t	|��q�td	�D]9}tdd	�D]0}	dg|dg}|dd�}t|	�D]
}tj|dd
�}�q#tj||	d
�}tt	|�t	|���q�qtd	�D]@}tdd	�D]7}	dg|dg}|dd�}t|	�D]}tj|d|gd
�}�qatj||	tt|	��d
�}tt	|�t	|���qM�qFtd	�D]B}tdd	�D]9}	dg|dg}|dd�}t|	�D]
}tj|d|gdd�}�q�tj||	tt|	��dd�}tt	|�t	|���q��q�td	�D]B}tdd	�D]9}	dg|dg}|dd�}t|	�D]
}tj|d|gdd�}�q�tj||	tt|	��dd�}tt	|�t	|���qِq�dS)Nr��?r
r	)�lbnd)�scl��axisrcr0)�m�k)r�r�r�)r�r�r��r�)r�	TypeErrorr�hermeintrqr5r�
poly2herme�
herme2polyrrG�list)
rr;r�r?r��polr>�hermepolr�r<rrr�
test_hermeint�s�




�	�	�	��zTestIntegral.test_hermeintcCs�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�cS�g|]}t�|��qSr�rr�rdrrrrf'rgz3TestIntegral.test_hermeint_axis.<locals>.<listcomp>rr�cSr�rr�rdrrrrf+rgr	cSsg|]	}tj|dd��qS)r)r�r�rdrrrrf/sr)r�r�)r6r��vstack�Trr�r�rrsr>r?rrr�test_hermeint_axis#s

zTestIntegral.test_hermeint_axisN)r*r+r,r�r�rrrrr��sSr�c@r�)�TestDerivativecCstttjdgd�tttjdgd�td�D]}dg|dg}tj|dd�}tt|�t|��qtd�D]'}tdd�D]}dg|dg}tjtj||d�|d�}t	t|�t|��q<q5td�D])}tdd�D]!}dg|dg}tjtj||dd�|dd�}t	t|�t|��qhqadS)	Nrr�r
r0r	r�rc)r�r�)
rr�r�hermederrqr5rrr�r)rr;r>r?r<rrr�
test_hermeder6s*����zTestDerivative.test_hermedercCsltj�d�}t�dd�|jD��j}tj|dd�}t||�t�dd�|D��}tj|dd�}t||�dS)Nr�cSr�r�rr�rdrrrrfTrgz5TestDerivative.test_hermeder_axis.<locals>.<listcomp>rr�cSr�rr�rdrrrrfXrgr	)r6r�r�r�rr�rr�rrr�test_hermeder_axisPs
z!TestDerivative.test_hermeder_axisN)r*r+r,r�r�rrrrr�4sr�c@s8eZdZej�d�ddZdd�Zdd�Zdd	�Zd
S)�
TestVanderrbrcr	cCs�t�d�}t�|d�}t|jdk�td�D]}dg|dg}t|d|ft�||��qt�	ddgddgdd	gg�}t�|d�}t|jd
k�td�D]}dg|dg}t|d|ft�||��qMdS)Nrr�r�rr	.rcr0�)rrcr�)
r6rWr�hermevanderrrkr5rrGrX)rr�vr;�coefrrr�test_hermevanderas
�zTestVander.test_hermevandercCsx|j\}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||�t�|g|gddg�}t	|j
dk�dS)Nrpr	rc)r	r0r�)rr6r�r�
hermevander2drr�dot�flatrrrk�rrvrwrxrZ�vanr>r?rrr�test_hermevander2drs
zTestVander.test_hermevander2dcCs�|j\}}}tj�d�}t�|||gd��}t�||||�}t�||j�}t||�t�|g|g|ggd��}t	|j
dk�dS)N)rcrr�)r	rcr)r	r0�)rr6r�r�
hermevander3dr~r�r�rrrkr�rrr�test_hermevander3ds
zTestVander.test_hermevander3dN)	r*r+r,r6r�rr�r�r�rrrrr�]s

r�c@�eZdZdd�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	rcrrrrr�f�r z$TestFitting.test_hermefit.<locals>.fcSs|d|ddS)Nr�rcr	rrrrr�f2�r z%TestFitting.test_hermefit.<locals>.f2r	r
rrc)�w)rcr
r�rr�)rr	rcrr0)rr	rcrr�)rcrr�r	r)r	y�?r
y��)rrcr�)rrqr�hermefitr�r6r\rrIrrGrXr��
zeros_like�copy)rr�r�rrl�coef3�coef4�coef2dr��yw�wcoef3�wcoef2d�coef1�coef2rrr�
test_hermefit�sp


"zTestFitting.test_hermefitN)r*r+r,r�rrrrr���r�c@s$eZdZdd�Zdd�Zdd�ZdS)�
TestCompanioncCs"tttjg�tttjdg�dSr%)rrqr�hermecompanionrrrr�test_raises�szTestCompanion.test_raisescCs<tdd�D]}dg|dg}tt�|�j||fk�qdS)Nr	r0r)r5rrr�rk)rr;r�rrr�test_dimensions�s�zTestCompanion.test_dimensionscCstt�ddg�ddk�dS)Nr	rc)rr��)rrr�rrrr�test_linear_root�szTestCompanion.test_linear_rootN)r*r+r,r�r�r�rrrrr��sr�c@r�)�	TestGausscCs�t�d�\}}t�|d�}t�|j||�}dt�|���}|dd�df||}t|t�	d��t�dtj
�}t|��|�dS)Nr.�cr	rc)r�
hermegaussr�r6r�r��sqrt�diagonalr�eye�pi�sum)rrr�r��vv�vdr>rrr�test_100�szTestGauss.test_100N)r*r+r,r�rrrrr��r�r�c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�TestMiscc	Cs�t�g�}tt|�dg�tdd�D]=}t�t�tjdd|d�ddd��}t�|�}t�	||�}d}t
t|�|dk�tt�|�dd�t||�qdS)Nr	r0rrcr
)
r�hermefromrootsrrr5r6�cosr\r�rGrrIr�)rr?r;�rootsr�r>rrr�test_hermefromroots�s
*
�zTestMisc.test_hermefromrootscCsltt�dg�g�tt�ddg�dg�tdd�D]}t�dd|�}t�t�|��}tt|�t|��qdS)Nr	r
rcr0)rr�
hermerootsr5r6r\r�r)rr;r>r?rrr�test_hermeroots
s�zTestMisc.test_hermerootscCsbgd�}tttj|d�tt�|�|dd��tt�|d�|dd��tt�|d�dg�dS)N)rcr
r	rr
r	rrcr)rrqrrr)rr�rrr�test_hermetrims
zTestMisc.test_hermetrimcCstt�dd�ddg�dS)Nrr�)rr�	hermelinerrrr�test_hermelineszTestMisc.test_hermelinecCs2td�D]}tt�dg|dg�t|�qdS�Nrhrr	)r5rrr�rj�rr;rrr�test_herme2poly �"�zTestMisc.test_herme2polycCs2td�D]}tt�t|�dg|dg�qdSr�)r5rrr�rjr�rrr�test_poly2herme$r�zTestMisc.test_poly2hermecCs8t�ddd�}t�d|d�}t�|�}t||�dS)N���r0�r�rc)r6r\�expr�hermeweightr)rrr>r?rrr�test_weight(s
zTestMisc.test_weightN)
r*r+r,r�r�r�r�r�r�r�rrrrr��sr�)'�__doc__�	functoolsr�numpyr6�numpy.polynomial.hermite_e�
polynomial�	hermite_er�numpy.polynomial.polynomialr�
numpy.testingrrrrrX�He0�He1�He2�He3�He4�He5�He6�He7�He8�He9rjrrr-r]r�r�r�r�r�r�r�rrrr�<module>s8B^f)0M