%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_polynomial.cpython-39.pyc

a

z[ycO�
@s0dZddlmZddlZddlmmZddlm	Z	m
Z
mZmZm
Z
mZmZdd�ZdgZddgZgd�Zgd	�Zgd
�Zgd�Zgd�Zgd
�Zgd�Zgd�Zeeeeeeeeeeg
ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�Z Gdd�d�Z!Gdd�d�Z"Gdd�d�Z#dS) zTests for polynomial module.

�)�reduceN)�assert_almost_equal�
assert_raises�assert_equal�assert_�assert_warns�assert_array_equal�assert_raises_regexcCstj|dd�S)Ng���ư>)Ztol)�poly�polytrim��x�r�L/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_polynomial.py�trim
sr�)���r�)r���r�)rri����r�)r�ri��r�)rr�ri���r� )ri����r�8ri����r�@)	rri��r�ri���r�)
r�	ri����ri�ri����r�c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�
TestConstantscCsttjddg�dS)Nrr)rr
Z
polydomain��selfrrr�test_polydomain szTestConstants.test_polydomaincCsttjdg�dS)Nr)rr
Zpolyzeror"rrr�
test_polyzero#szTestConstants.test_polyzerocCsttjdg�dS�Nr)rr
Zpolyoner"rrr�test_polyone&szTestConstants.test_polyonecCsttjddg�dS)Nrr)rr
Zpolyxr"rrr�
test_polyx)szTestConstants.test_polyxN)�__name__�
__module__�__qualname__r$r%r'r(rrrrr!sr!c@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�TestArithmeticcCs�td�D]�}td�D]�}d|�d|��}t�t||�d�}||d7<||d7<t�dg|dgdg|dg�}tt|�t|�|d�qqdS�Nr�At i=�, j=rr��err_msg)�range�np�zeros�maxr
�polyaddrr�r#�i�j�msg�tgt�resrrr�test_polyadd/s$zTestArithmetic.test_polyaddcCs�td�D]�}td�D]�}d|�d|��}t�t||�d�}||d7<||d8<t�dg|dgdg|dg�}tt|�t|�|d�qqdSr-)r2r3r4r5r
Zpolysubrrr7rrr�test_polysub9s$zTestArithmetic.test_polysubcCsrtt�dg�dg�tt�dg�ddg�tdd�D]8}dg|dg}dg|ddg}tt�|�|�q4dS)Nrrr)rr
Zpolymulxr2)r#r8Zserr;rrr�
test_polymulxCszTestArithmetic.test_polymulxcCs�td�D]�}td�D]t}d|�d|��}t�||d�}|||d7<t�dg|dgdg|dg�}tt|�t|�|d�qqdSr-)r2r3r4r
�polymulrrr7rrr�test_polymulKs$zTestArithmetic.test_polymulc
Cs�tttjdgdg�t�dgdg�\}}t||fd�t�ddgdg�\}}t||fd�td�D]�}td�D]v}d|�d|��}dg|ddg}dg|ddg}t�||�}t�||�\}}t�t�||�|�}	t|	||d	�qnqbdS)
Nrrr)rr))rrrrr.r/r0)r�ZeroDivisionErrorr
Zpolydivrr2r6r@)
r#ZquoZremr8r9r:ZciZcjr;r<rrr�test_polydivTszTestArithmetic.test_polydivcCs|td�D]n}td�D]`}d|�d|��}t�|d�}ttj|g|t�dg��}t�||�}tt	|�t	|�|d�qqdS)Nrr.r/rr0)
r2r3�arangerr
r@�arrayZpolypowrr)r#r8r9r:�cr;r<rrr�test_polypowiszTestArithmetic.test_polypowN)	r)r*r+r=r>r?rArCrGrrrrr,-s

	r,c@s�eZdZe�gd��Ze�dee�Ze�deee�Zej	�	d�ddZ
e�e
gd��Z
dd�Zd	d
�Zdd�Zd
d�Zdd�Zdd�ZdS)�TestEvaluation)��?g@g@�i,j->ij�
i,j,k->ijk��rrrc
s~tt�gdg�jd�t�dd���fdd�td�D�}td�D].}||}t��dg|dg�}t||�q@��dd}t��gd��}t||�td	�D]^}dg|}t�|��tt��dg�j	|�tt��ddg�j	|�tt��gd
��j	|�q�gd�}tj
jgd�|d
�}t�gd�|�}t|j
|�Gdd�dtj�}t�gd���|�}	ttt�gd�|	��|�dS)Nrrrcsg|]}�|�qSrr��.0r8rrr�
<listcomp>��z/TestEvaluation.test_polyval.<locals>.<listcomp>rr)rrrrrM�rrr)FTF�rrrM)�mask)�rrMc@seZdZdS)z&TestEvaluation.test_polyval.<locals>.CN)r)r*r+rrrr�C�srV�rrMr)rr
�polyval�sizer3�linspacer2rr4�shapeZmarErrTZndarray�view�type)
r#�yr8r;r<�dimsrTZmxrVZcxrrr�test_polyval}s.


zTestEvaluation.test_polyvalcs�tttjdgdgdd�tt�gdg�jd�tt�gdg�jdk�tt�gdgdg�jd�tt�gdgdg�jdk�tt�dd�d�tt�dt�	d��jd	k�t�
d
d���fdd�td�D�}tdd�D](}||}t��dg|�}t||�qڈ�d�d}t��gd
��}t||�td�D]`}dg|}t�
|��tt��dg�j|�tt��ddg�j|�tt��gd��j|��q:gd�}t�|�}t�
d
d��tt��|�t��|��d}t�dd��tjjdd|d�}tj�|dd�}t�|jdd��}t|j�D]&}	t��|	|dd�|	f�||	<�q$t||�t��d�g��tj�|dd�}t�|jdd��j�}t|jd�D]F}	t�jd�D]0}
t��|
|dd�|	f�||	|
dd�f<�q��q�t||�dS)NrF)Ztensorr)rr)rr)rMrM)rMrcsg|]}�|�qSrrrNrrrrP�rQz8TestEvaluation.test_polyvalfromroots.<locals>.<listcomp>)rrrrMrrR)�ri����rrLr���)rYT)r�
ValueErrorr
ZpolyvalfromrootsrrYrr[r3�onesrZr2rr4�	polyrootsrXrD�randomZrandint�empty�vstack)r#r^r8r;r<r_Zptest�rZrshape�iiZjjrrr�test_polyvalfromroots�sZ
�




�$
2z$TestEvaluation.test_polyvalfromrootsc
Cs�|j\}}}|j\}}}ttdtj||dd�|j�||}t�|||j�}t||�t�	d�}	t�|	|	|j�}t
|jdk�dS�NZincompatibler�rrM)r
r^r	rdr
�	polyval2d�c2drr3rerr[�
r#�x1�x2�x3Zy1Zy2Zy3r;r<�zrrr�test_polyval2d�s�

zTestEvaluation.test_polyval2dc
	Cs�|j\}}}|j\}}}ttdtj|||dd�|j�|||}t�||||j�}t||�t�	d�}	t�|	|	|	|j�}t
|jdk�dSrm)r
r^r	rdr
�	polyval3d�c3drr3rerr[rqrrr�test_polyval3d�s�

zTestEvaluation.test_polyval3dc
Csl|j\}}}|j\}}}t�d||�}t�|||j�}t||�t�d�}	t�|	|	|j�}t	|j
dk�dS)NrJrn)rrMrrM)r
r^r3�einsumr
Z
polygrid2drprrerr[rqrrr�test_polygrid2ds

zTestEvaluation.test_polygrid2dc
Csr|j\}}}|j\}}}t�d|||�}t�||||j�}t||�t�d�}	t�|	|	|	|j�}t	|j
dk�dS)NrKrn)rrMrrMrrM)r
r^r3rzr
Z
polygrid3drxrrerr[rqrrr�test_polygrid3ds

zTestEvaluation.test_polygrid3dN)r)r*r+r3rEZc1drzrprxrgr
r
rXr^r`rlrvryr{r|rrrrrHss$@rHc@seZdZdd�Zdd�ZdS)�TestIntegralc	Cs:tttjdgd�tttjdgd�tttjdgdddg�tttjdgdgd�tttjdgdgd�tttjdgdd�tt�� t�ddgd�Wd�n1s�0Ytd	d
�D]8}dg|d	dg}tjdg||d�}t|ddg�q�td
�D]\}|d}dg|dg}|gdg|d|g}tj|d|gd�}tt	|�t	|���qtd
�D]D}|d}dg|dg}tj|d|gdd�}tt�
d|�|��qftd
�D]^}|d}dg|dg}|gdg|d	|g}tj|d|gd	d
�}tt	|�t	|���q�td
�D]r}td	d
�D]`}dg|dg}|dd�}t|�D]}tj|dd�}�qRtj||d�}tt	|�t	|���q*�qtd
�D]�}td	d
�D]n}dg|dg}|dd�}t|�D]}tj|d|gd�}�q�tj||tt|��d�}tt	|�t	|���q��q�td
�D]�}td	d
�D]r}dg|dg}|dd�}t|�D]}tj|d|gdd�}�qXtj||tt|��dd�}tt	|�t	|���q0�q"td
�D]�}td	d
�D]r}dg|dg}|dd�}t|�D]}tj|d|gd	d
�}�q�tj||tt|��d	d
�}tt	|�t	|���q��q�dS)Nr��?rr)�lbnd)�scl��axisrIrr)�m�k)r�r�r)r�r�r��r�)r�	TypeErrorr
�polyintrdr�DeprecationWarningr2rrrX�list)r#r8r�r<r�Zpolr;r9rrr�test_polyint$sz
.zTestIntegral.test_polyintcCs�tj�d�}t�dd�|jD��j}tj|dd�}t||�t�dd�|D��}tj|dd�}t||�t�dd�|D��}tj|d	dd
�}t||�dS)N�rMrcSsg|]}t�|��qSr�r
r��rOrFrrrrPxrQz2TestIntegral.test_polyint_axis.<locals>.<listcomp>rr�cSsg|]}t�|��qSrr�r�rrrrP|rQrcSsg|]}tj|dd��qS)rM)r�r�r�rrrrP�rQrM)r�r�)r3rgri�Tr
r�r�r#rpr;r<rrr�test_polyint_axists

zTestIntegral.test_polyint_axisN)r)r*r+r�r�rrrrr}"sPr}c@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~rrrr�r)r�r�)
rr�r
�polyderrdr2rrr�r)r#r8r;r<r9rrr�test_polyder�s zTestDerivative.test_polydercCsltj�d�}t�dd�|jD��j}tj|dd�}t||�t�dd�|D��}tj|dd�}t||�dS)Nr�cSsg|]}t�|��qSr�r
r�r�rrrrP�rQz4TestDerivative.test_polyder_axis.<locals>.<listcomp>rr�cSsg|]}t�|��qSrr�r�rrrrP�rQr)r3rgrir�r
r�rr�rrr�test_polyder_axis�s
z TestDerivative.test_polyder_axisN)r)r*r+r�r�rrrrr��sr�c@s@eZdZej�d�ddZdd�Zdd�Zdd	�Zd
d�Z	dS)
�
TestVanderrLrrcCs�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)NrMr�rrr.rr�)rMrr)
r3rDr
�
polyvanderrr[r2rrXrE)r#r
�vr8�coefrrr�test_polyvander�s
zTestVander.test_polyvandercCsx|j\}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||�t�|g|gddg�}t	|j
dk�dS)Nrnrr)rrr�)r
r3rgr
Zpolyvander2dro�dot�flatrrr[�r#rrrsrtrFZvanr;r<rrr�test_polyvander2d�s
zTestVander.test_polyvander2dcCs�|j\}}}tj�d�}t�|||gd��}t�||||�}t�||j�}t||�t�|g|g|ggd��}t	|j
dk�dS)NrWrS)rr�)r
r3rgr
Zpolyvander3drwr�r�rrr[r�rrr�test_polyvander3d�s
zTestVander.test_polyvander3dcCst�d�}tttj|d�dS)NrMr)r3rDrrdr
r�)r#r
rrr�test_polyvandernegdeg�s
z TestVander.test_polyvandernegdegN)
r)r*r+r3rgr
r�r�r�r�rrrrr��s


r�c@s$eZdZdd�Zdd�Zdd�ZdS)�
TestCompanioncCs"tttjg�tttjdg�dSr&)rrdr
�
polycompanionr"rrr�test_raises�szTestCompanion.test_raisescCs<tdd�D],}dg|dg}tt�|�j||fk�q
dS)Nrrr)r2rr
r�r[)r#r8r�rrr�test_dimensions�szTestCompanion.test_dimensionscCstt�ddg�ddk�dS)Nrr)rr��)rr
r�r"rrr�test_linear_root�szTestCompanion.test_linear_rootN)r)r*r+r�r�r�rrrrr��sr�c@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�TestMiscc	Cs�t�g�}tt|�dg�tdd�D]^}t�t�tjdd|d�ddd��}t	|}t�|�d|d}tt|�t|��q$dS)Nrrrr)
r
�
polyfromrootsrrr2r3ZcosrZZpi�Tlist)r#r<r8�rootsr;rrr�test_polyfromroots�s
*zTestMisc.test_polyfromrootscCsltt�dg�g�tt�ddg�dg�tdd�D]4}t�dd|�}t�t�|��}tt|�t|��q2dS)Nrrr�rr)rr
rfr2r3rZr�r)r#r8r;r<rrr�test_polyroots�szTestMisc.test_polyrootscCs�dd�}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�|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)Nrrrrrrr�fsz TestMisc.test_polyfit.<locals>.fcSs|d|ddS)Nrrrrrrrr�f2sz!TestMisc.test_polyfit.<locals>.f2rrrr)�w)rrr�rMr)rrrrMr)rrrrMr)ry�?ry��)rrr)rrdr
Zpolyfitr�r3rZr�lenrrXrEr�Z
zeros_like�copy)r#r�r�r
r^Zcoef3Zcoef4Zcoef2dr�ZywZwcoef3Zwcoef2dZcoef1Zcoef2rrr�test_polyfitsj


"zTestMisc.test_polyfitcCsbgd�}tttj|d�tt�|�|dd��tt�|d�|dd��tt�|d�dg�dS)N)rrrrrrrrr)rrdr
rr)r#r�rrr�
test_polytrimIs
zTestMisc.test_polytrimcCstt�dd�ddg�dS)NrMr�rr
Zpolyliner"rrr�
test_polylineTszTestMisc.test_polylinecCstt�dd�dg�dS)NrMrr�r"rrr�test_polyline_zeroWszTestMisc.test_polyline_zeroN)	r)r*r+r�r�r�r�r�r�rrrrr��s	Fr�)$�__doc__�	functoolsrZnumpyr3Znumpy.polynomial.polynomialZ
polynomialr
Z
numpy.testingrrrrrrr	rZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9r�r!r,rHr}r�r�r�r�rrrr�<module>s2$F0c(4

Zerion Mini Shell 1.0