Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_polyutils.cpython-39.pyc
a
z[yc�
� @ sV d Z ddlZddlm mZ ddlmZm Z m
Z
m Z G dd� d�Z G dd� d�Z
dS )z Tests for polyutils module.
� N)�assert_almost_equal�
assert_raises� assert_equal�assert_c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestMiscc C s8 t d�D ]*}dg}t�dgdgd �}t||� qd S )N� � r )�range�puZtrimseqr )�self�i�tgt�res� r �K/usr/lib64/python3.9/site-packages/numpy/polynomial/tests/test_polyutils.py� test_trimseq
s zTestMisc.test_trimseqc C s� t ttjg g� t ttjddggg� t ttjdgdgg� g d�}tt|��D ]l}t|�D ]^}t�d|| �}t�d|| �}t�||g�\}}t|j j
|j j
k� t|j j
|| k� q\qPd S )Nr � �a)r �d�O) r �
ValueErrorr
Z as_seriesr �len�npZonesr Zdtype�char)r �typesr �jZciZcjZresiZresjr r r �test_as_series s zTestMisc.test_as_seriesc 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
Ztrimcoefr )r Zcoefr r r �
test_trimcoef" s
zTestMisc.test_trimcoefc C s> t ttjdddg� t ttjdddg� t ttjddg � d S )Nr �r r � �Z g������V@)r r r
Z
_vander_nd�r r r r �test_vander_nd_exception+ s z!TestMisc.test_vander_nd_exceptionc C s t ttjtjddg� d S )Nr r )r �ZeroDivisionErrorr
Z_divr# r r r �test_div_zerodiv3 s zTestMisc.test_div_zerodivc C s t ttjdg d�dd� d S )Nr r r � )r r r
Z_powr# r r r �test_pow_too_large7 s zTestMisc.test_pow_too_largeN) �__name__�
__module__� __qualname__r r r r$ r& r( r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
TestDomainc C sL g d�}ddg}t �|�}t||� g d�}ddg}t �|�}t||� d S )N)r �
r! r r r- )y �? �?� �? �r r y � � @ �?)r
Z getdomainr )r �xr
r r r r �test_getdomain= s
zTestDomain.test_getdomainc C s� ddg}ddg}|}t �|||�}t||� ddg}ddg}|}|}t �|||�}t||� ddg}ddg}t�||g�}t�||g�}t �|||�}t||� G d d
� d
tj�}ddg}ddg}t�||g��|�}t �|||�}tt||�� d S ) Nr r' r r! � �r/ ���r c @ s e Zd ZdS )z,TestDomain.test_mapdomain.<locals>.MyNDArrayN)r) r* r+ r r r r � MyNDArrayc s r4 ) r
Z mapdomainr r ZarrayZndarray�viewr �
isinstance)r �dom1�dom2r
r r0 r4 r r r �test_mapdomainJ s.
zTestDomain.test_mapdomainc C s` ddg}ddg}ddg}t �||�}t||� ddg}dd g}d
d g}t �||�}t||� d S ) Nr r' r r! g �?r2 r/ r3 r y � �?r. )r
Zmapparmsr )r r7 r8 r
r r r r �
test_mapparmsl s
zTestDomain.test_mapparmsN)r) r* r+ r1 r9 r: r r r r r, ; s
"r, )�__doc__Znumpyr Znumpy.polynomial.polyutilsZ
polynomialZ polyutilsr
Z
numpy.testingr r r r r r, r r r r �<module> s
0
Mini Shell