Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/polynomial/tests/__pycache__/test_polynomial.cpython-37.pyc
Назад
B �5�gO � @ sx d Z ddlmZ ddlZddlm mZ ddlm Z m Z m Z m Z m Z mZmZ dd� ZdgZddgZddd gZdd dd gZddd dd gZddddddgZdddddddgZddddddddgZdddddddddg Zdd dd dd dd dd g Zeeeeeeeeeeg ZG d!d"� d"�Z G d#d$� d$�Z G d%d&� d&�Z G d'd(� d(�Z G d)d*� d*�Z G d+d,� d,�Z!G d-d.� d.�Z"G d/d0� d0�Z#dS )1z Tests for polynomial module. � )�reduceN)�assert_almost_equal� assert_raises� assert_equal�assert_� assert_warns�assert_array_equal�assert_raises_regexc C s t j| dd�S )Ng���ư>)Ztol)�poly�polytrim)�x� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/polynomial/tests/test_polynomial.py�trim s r � ���� ���� i����� � i��� � i���� i�����8 i�����@ i��� i ���� � i����i� i����� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestConstantsc C s t tjddg� d S )Nr r )r r Z polydomain)�selfr r r �test_polydomain s z TestConstants.test_polydomainc C s t tjdg� d S )Nr )r r Zpolyzero)r! r r r � test_polyzero# s zTestConstants.test_polyzeroc C s t tjdg� d S )Nr )r r Zpolyone)r! r r r � test_polyone& s zTestConstants.test_polyonec C s t tjddg� d S )Nr r )r r Zpolyx)r! r r r � test_polyx) s zTestConstants.test_polyxN)�__name__� __module__� __qualname__r" r# r$ r% r r r r r s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestArithmeticc C s� x�t d�D ]�}x�t d�D ]�}d|� d|� �}t�t||�d �}|| d7 < || d7 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )Nr zAt i=z, j=r r )�err_msg)�range�np�zeros�maxr �polyaddr r )r! �i�j�msg�tgt�resr r r � test_polyadd/ s $zTestArithmetic.test_polyaddc C s� x�t d�D ]�}x�t d�D ]�}d|� d|� �}t�t||�d �}|| d7 < || d8 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )Nr zAt i=z, j=r r )r* )r+ r, r- r. r Zpolysubr r )r! r0 r1 r2 r3 r4 r r r � test_polysub9 s $zTestArithmetic.test_polysubc C sv t t�dg�dg� t t�dg�ddg� xFtdd�D ]8}dg| dg }dg|d dg }t t�|�|� q6W d S )Nr r r )r r Zpolymulxr+ )r! r0 Zserr3 r r r � test_polymulxC s z TestArithmetic.test_polymulxc C s� x�t d�D ]�}x�t d�D ]t}d|� d|� �}t�|| d �}||| d7 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )Nr zAt i=z, j=r r )r* )r+ r, r- r �polymulr r )r! r0 r1 r2 r3 r4 r r r � test_polymulK s $zTestArithmetic.test_polymulc C s� t ttjdgdg� t�dgdg�\}}t||fd� t�ddgdg�\}}t||fd� x�td�D ]�}x�td�D ]v}d|� d|� �}dg| ddg }dg| ddg }t�||�}t�||�\}}t�t�||�|�} t| ||d � qrW qdW d S ) Nr r r )r r ))r r r r zAt i=z, j=)r* )r �ZeroDivisionErrorr Zpolydivr r+ r/ r8 ) r! Zquo�remr0 r1 r2 �ci�cjr3 r4 r r r � test_polydivT s zTestArithmetic.test_polydivc C s� x~t d�D ]r}xlt d�D ]`}d|� d|� �}t�|d �}ttj|g| t�dg��}t�||�}tt |�t |�|d� qW q W d S )Nr zAt i=z, j=r )r* ) r+ r, �aranger r r8 �arrayZpolypowr r )r! r0 r1 r2 �cr3 r4 r r r � test_polypowi s zTestArithmetic.test_polypowN) r& r' r( r5 r6 r7 r9 r> rB r r r r r) - s r) c @ s� e Zd Ze�dddg�Ze�dee�Ze�deee�Zej � d�d d Z e � e dddg�Z d d � Zd d � Zd d� Zdd� Zdd� Zdd� ZdS )�TestEvaluationg �?g @g @zi,j->ijz i,j,k->ijk)� r r r c s� t t�g dg�jd� t�dd�� � fdd�td�D �}x:td�D ].}|| }t�� dg| dg �}t||� qBW � � d d }t�� ddddg�}t||� xltd�D ]`}dg| }t�|�� t t�� dg�j |� t t�� ddg�j |� t t�� dddg�j |� q�W d d d g}tj j dddg|d �}t�d ddg|�}t |j |� G d d� dtj�}t� dddg��|�} t tt�dddg| ��|� d S )Nr r r c s g | ] }� | �qS r r )�.0r0 )r r r � <listcomp>� s z/TestEvaluation.test_polyval.<locals>.<listcomp>r r rD FT)�mask� c @ s e Zd ZdS )z&TestEvaluation.test_polyval.<locals>.CN)r& r' r( r r r r �C� s rI r )r r �polyval�sizer, �linspacer+ r r- �shape�mar@ r rG Zndarray�view�type) r! �yr0 r3 r4 �dimsrG ZmxrI Zcxr )r r � test_polyval} s. zTestEvaluation.test_polyvalc s t ttjdgdgdd� tt�g dg�jd� tt�g dg�jdk� tt�g dgd g�jd� tt�g dgd g�jdk� tt�dd�d� tt�dt� d��jd k� t� d d�� � fd d �t d�D �}x6t dd�D ](}|| }t�� dg| �}t ||� q�W � � d � d }t�� d ddg�}t ||� xnt d �D ]b}dg| }t� |�� tt�� dg�j|� tt�� ddg�j|� tt�� dddg�j|� �qBW dddddg}t�|�}t� d d�� t t�� |�t�� |�� d}t�dd�� tjjdd|d�}tj� |dd�}t�|jdd � �}x4t |j�D ]&} t�� | |d d �| f �|| < �q8W t||� t�� d� g�� tj� |dd�}t�|jdd � � j �}x\t |jd �D ]J} xBt � jd �D ]0} t�� | |d d �| f �|| | d d �f< �q�W �q�W t||� d S )Nr F)Ztensorr )r r )r r )rD rD )rD r c s g | ] }� | �qS r r )rE r0 )r r r rF � s z8TestEvaluation.test_polyvalfromroots.<locals>.<listcomp>rD r � i����)rD r r ���)rK T)r � ValueErrorr Zpolyvalfromrootsr rK r rM r, �onesrL r+ r r- � polyrootsrJ r? �random�randint�empty�vstack) r! rQ r0 r3 r4 rR Zptest�rZrshape�iiZjjr )r r �test_polyvalfromroots� sV & 6z$TestEvaluation.test_polyvalfromrootsc C s� | j \}}}| j\}}}ttdtj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t |j dk� d S )N� incompatibler )r rD ) r rQ r rW r � polyval2d�c2dr r, rX r rM ) r! �x1�x2�x3�y1�y2�y3r3 r4 �zr r r �test_polyval2d� s z TestEvaluation.test_polyval2dc C s� | j \}}}| j\}}}ttdtj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t |j dk� d S )Nra r )r rD ) r rQ r rW r � polyval3d�c3dr r, rX r rM ) r! rd re rf rg rh ri r3 r4 rj r r r �test_polyval3d� s z TestEvaluation.test_polyval3dc C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j dk� d S )Nzi,j->ij)r rD )r rD r rD ) r rQ r, �einsumr Z polygrid2drc r rX r rM ) r! rd re rf rg rh ri r3 r4 rj r r r �test_polygrid2d s z TestEvaluation.test_polygrid2dc C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j dk� d S )Nz i,j,k->ijk)r rD )r rD r rD r rD ) r rQ r, ro r Z polygrid3drm r rX r rM ) r! rd re rf rg rh ri r3 r4 rj r r r �test_polygrid3d s z TestEvaluation.test_polygrid3dN)r& r' r( r, r@ Zc1dro rc rm rZ r r rJ rQ rS r` rk rn rp rq r r r r rC s s $@rC c @ s e Zd Zdd� Zdd� ZdS )� TestIntegralc C sd t ttjdgd� t ttjdgd� t ttjdgdddg� t ttjdgdgd� t ttjdgdgd� t ttjdgdd� tt�� t�ddgd� W d Q R X xFtd d �D ]8}dg|d dg }tjdg||d �}t|ddg� q�W xftd �D ]Z}|d }dg| dg }|gdg| d| g }tj|d|gd �}tt |�t |�� q�W xPtd �D ]D}|d }dg| dg }tj|d|gdd �}tt� d|�|� �qZW xjtd �D ]^}|d }dg| dg }|gdg| d | g }tj|d|gd d �}tt |�t |�� �q�W x�td �D ]z}xrtd d �D ]d}dg| dg }|d d � }x t|�D ]}tj|dd�}�qRW tj||d�}tt |�t |�� �q(W �qW x�td �D ]�}x�td d �D ]r}dg| dg }|d d � }x$t|�D ]}tj|d|gd �}�q�W tj||t t|��d �}tt |�t |�� �q�W �q�W x�td �D ]�}x�td d �D ]v}dg| dg }|d d � }x&t|�D ]}tj|d|gdd �}�qpW tj||t t|��dd �}tt |�t |�� �qFW �q6W x�td �D ]�}x�td d �D ]v}dg| dg }|d d � }x&t|�D ]}tj|d|gd d �}�q W tj||t t|��d d �}tt |�t |�� �q�W �q�W d S )Nr g �?r r )�lbnd)�scl)�axisg �?r r )�m�k)rv rw rs )rv rw rt )rv ) r � TypeErrorr �polyintrW r �DeprecationWarningr+ r r rJ �list)r! r0 rw r4 rt Zpolr3 r1 r r r � test_polyint$ sz zTestIntegral.test_polyintc 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)rD r c S s g | ]}t �|��qS r )r ry )rE rA r r r rF x s z2TestIntegral.test_polyint_axis.<locals>.<listcomp>r )ru c S s g | ]}t �|��qS r )r ry )rE rA r r r rF | s r c S s g | ]}t j|d d��qS )rD )rw )r ry )rE rA r r r rF � s rD )rw ru )r, rZ r] �Tr ry r )r! rc r3 r4 r r r �test_polyint_axist s z TestIntegral.test_polyint_axisN)r& r' r( r| r~ r r r r rr " s Prr c @ s e Zd Zdd� Zdd� ZdS )�TestDerivativec C s. t ttjdgd� t ttjdgd� x@td�D ]4}dg| dg }tj|dd�}tt|�t|�� q.W x^td�D ]R}xLtdd�D ]>}dg| dg }tjtj||d�|d�}t t|�t|�� q�W qpW xbtd�D ]V}xPtdd�D ]B}dg| dg }tjtj||dd�|dd�}t t|�t|�� q�W q�W d S ) Nr g �?r r r )rv r )rv rt ) r rx r �polyderrW r+ r r ry r )r! r0 r3 r4 r1 r r r � test_polyder� s zTestDerivative.test_polyderc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )N)rD r c S s g | ]}t �|��qS r )r r� )rE rA r r r rF � s z4TestDerivative.test_polyder_axis.<locals>.<listcomp>r )ru c S s g | ]}t �|��qS r )r r� )rE rA r r r rF � s r )r, rZ r] r} r r� r )r! rc r3 r4 r r r �test_polyder_axis� s z TestDerivative.test_polyder_axisN)r& r' r( r� r� r r r r r � s r c @ s@ e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd d � Z d S ) � TestVander)rD r r r c C s� t �d�}t�|d�}t|jdk� x:td�D ].}dg| dg }t|d|f t�||�� q.W t � ddgddgdd gg�}t�|d�}t|jd k� x:td�D ].}dg| dg }t|d|f t�||�� q�W d S ) NrD )rD r r r r .r r � )rD r r ) r, r? r � polyvanderr rM r+ r rJ r@ )r! r �vr0 �coefr r r �test_polyvander� s zTestVander.test_polyvanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j dk� d S )N)r rD r r )r r r� ) r r, rZ r Z polyvander2drb �dot�flatr r rM )r! rd re rf rA �vanr3 r4 r r r �test_polyvander2d� s z TestVander.test_polyvander2dc C s� | j \}}}tj�d�}t�|||dddg�}t�||||�}t�||j�}t||� t�|g|g|gdddg�}t |j dk� d S )N)r rD r r r rD )r r � ) r r, rZ r Z polyvander3drl r� r� r r rM )r! rd re rf rA r� r3 r4 r r r �test_polyvander3d� s z TestVander.test_polyvander3dc C s t �d�}tttj|d� d S )NrD r )r, r? r rW r r� )r! r r r r �test_polyvandernegdeg� s z TestVander.test_polyvandernegdegN) r& r' r( r, rZ r 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 )Nr )r rW r � polycompanion)r! r r r � test_raises� s zTestCompanion.test_raisesc C s@ x:t dd�D ],}dg| dg }tt�|�j||fk� q W d S )Nr r r )r+ r r r� rM )r! r0 r� r r r �test_dimensions� s z TestCompanion.test_dimensionsc C s t t�ddg�d dk� d S )Nr r )r r g �)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� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestMiscc C s� t �g �}tt|�dg� xltdd�D ]^}t�t�tj dd| d �dd d� �}t | }t �|�d|d }tt|�t|�� q&W d S )Nr r r r ) r � polyfromrootsr r r+ r, �cosrL �pi�Tlist)r! r4 r0 �rootsr3 r r r �test_polyfromroots� s *zTestMisc.test_polyfromrootsc C sp t t�dg�g � t t�ddg�dg� xBtdd�D ]4}t�dd|�}t�t�|��}t t|�t|�� q4W d S )Nr r g �r r )r r rY r+ r, rL r� r )r! r0 r3 r4 r r r �test_polyroots� s zTestMisc.test_polyrootsc C s� dd� }dd� }t ttjdgdgd� t ttjdggdgd� t ttjg dgd� t ttjdgdgggd� t ttjddgdgd� t ttjdgddgd� t ttjdgdgddggd � t ttjdgdgdddgd � t ttjdgdgdg� t ttjdgdgddd g� t ttjdgdgg � t�dd�}||�}t�||d �}tt|�d � t t� ||�|� t�||dddd g�}tt|�d � t t� ||�|� t�||d �}tt|�d � t t� ||�|� t�||dddd d g�}tt|�d � t t� ||�|� t�|t� ||g�j d �}t |t� ||g�j � t�|t� ||g�j dddd g�}t |t� ||g�j � t� |�}|�� } d|dd d�< d| dd d�< tj|| d |d �} t | |� tj|| dddd g|d �} t | |� tj|t� | | g�j d |d �} t | t� ||g�j � tj|t� | | g�j dddd g|d �} t | t� ||g�j � ddddg}t t�||d�ddg� t t�||ddg�ddg� t�dd�}||�}t�||d �} t t� || �|� t�||ddd g�} t t� || �|� t | | � d S )Nc S s | | d | d S )Nr r r )r r r r �f s z TestMisc.test_polyfit.<locals>.fc S s | d | d d S )Nr r r r )r r r r �f2 s z!TestMisc.test_polyfit.<locals>.f2r r r r )�wr� rD r r y �?y � �)r rW r Zpolyfitrx r, rL r �lenr rJ r@ r} Z zeros_like�copy)r! r� r� r rQ Zcoef3Zcoef4Zcoef2dr� ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r � test_polyfit sj " & zTestMisc.test_polyfitc C sf ddddg}t ttj|d� tt�|�|d d� � tt�|d�|d d� � tt�|d�dg� d S )Nr r r r r )r rW r r r )r! r� r r r � test_polytrimI s zTestMisc.test_polytrimc C s t t�dd�ddg� d S )NrD r )r r �polyline)r! r r r � test_polylineT s zTestMisc.test_polylinec C s t t�dd�dg� d S )NrD r )r r r� )r! r r r �test_polyline_zeroW s zTestMisc.test_polyline_zeroN) r& r' r( r� r� r� r� r� r� r r r r r� � s F r� )$�__doc__� functoolsr �numpyr, Znumpy.polynomial.polynomialZ polynomialr Z numpy.testingr r r r r r r r ZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9r� r r) rC rr r r� r� r� r r r r �<module> s2 $ F 0c(4
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка