Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/polynomial/tests/__pycache__/test_hermite_e.cpython-37.pyc
Назад
B �5�g�I � @ s� d Z ddlmZ ddlZddlm mZ ddl m Z ddl m Z m Z mZmZ e�dg�Ze�ddg�Ze�dddg�Ze�ddddg�Ze�d dd ddg�Ze�dd dd ddg�Ze�d dddd ddg�Ze�ddddddddg�Ze�dddddddddg �Ze�ddddddddddg �Zeeeeeeeeeeg Zdd� Z G dd � 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)d*� d*�Z$G d+d,� d,�Z%G d-d.� d.�Z&dS )/z Tests for hermite_e module. � )�reduceN)�polyval)�assert_almost_equal� assert_raises� assert_equal�assert_� ������� i����� i����i��- i�����i i��i\����� i��i� i���iz i���c C s t j| dd�S )Ng���ư>)Ztol)�herme� hermetrim)�x� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/polynomial/tests/test_hermite_e.py�trim s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestConstantsc C s t tjddg� d S )Nr r )r r Z hermedomain)�selfr r r �test_hermedomain! s z TestConstants.test_hermedomainc C s t tjdg� d S )Nr )r r Z hermezero)r r r r �test_hermezero$ s z TestConstants.test_hermezeroc C s t tjdg� d S )Nr )r r Zhermeone)r r r r � test_hermeone' s zTestConstants.test_hermeonec C s t tjddg� d S )Nr r )r r Zhermex)r r r r � test_hermex* s zTestConstants.test_hermexN)�__name__� __module__� __qualname__r r r r r r r r r s r c @ sJ e Zd Ze�ddd�Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )�TestArithmeticr r �d c 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 )N� zAt i=z, j=r r )�err_msg)�range�np�zeros�maxr �hermeaddr r )r �i�j�msg�tgt�resr r r � test_hermeadd1 s $z TestArithmetic.test_hermeaddc 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 Zhermesubr r )r r( r) r* r+ r, r r r � test_hermesub; s $z TestArithmetic.test_hermesubc C sz t t�dg�dg� t t�dg�ddg� xJtdd�D ]<}dg| dg }dg|d |ddg }t t�|�|� q6W d S )Nr r r! )r r Z hermemulxr# )r r( Zserr+ r r r �test_hermemulxE s z TestArithmetic.test_hermemulxc C s� x�t d�D ]�}dg| dg }t�| j|�}x�t d�D ]x}d|� d|� �}dg| dg }t�| j|�}t�||�}t�| j|�} tt|�|| d k|� t| || |d� q6W q W d S )Nr! r r zAt i=z, j=)r" )r# r �hermevalr �hermemulr �lenr ) r r( Zpol1Zval1r) r* Zpol2Zval2Zpol3Zval3r r r � test_hermemulM s z TestArithmetic.test_hermemulc C s� x�t d�D ]�}x�t d�D ]z}d|� d|� �}dg| dg }dg| dg }t�||�}t�||�\}}t�t�||�|�} tt| �t|�|d� qW q W d S )Nr! zAt i=z, j=r r )r" )r# r r' Zhermedivr1 r r ) r r( r) r* �ci�cjr+ Zquo�remr, r r r � test_hermediv[ s z TestArithmetic.test_hermedivc 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 r1 �arrayZhermepowr r )r r( r) r* �cr+ r, r r r � test_hermepowf s z TestArithmetic.test_hermepowN) r r r r$ �linspacer r- r. r/ r3 r7 r; r r r r r . s r c @ s� e Zd Ze�dddg�Ze�dee�Ze�deee�Zej � d�d d Z e e d ddg�Z d d � Z d d � Zdd� Zdd� Zdd� ZdS )�TestEvaluationg @g @g @zi,j->ijz i,j,k->ijk)r r! � r g �?c s� t t�g dg�jd� t�dd�� � fdd�tD �}xHtd�D ]<}d|� �}|| }t�� dg| dg �}t|||d� q>W xltd �D ]`}d g| }t� |�� t t�� dg�j |� t t�� ddg�j |� t t�� dddg�j |� q�W d S ) Nr r r c s g | ]}t � |��qS r )r )�.0r: )r r r � <listcomp>� s z0TestEvaluation.test_hermeval.<locals>.<listcomp>� zAt i=)r" r r> ) r r r0 �sizer$ r< �Helistr# r r% �shape)r �yr( r* r+ r, Zdimsr )r r � test_hermevalz s z TestEvaluation.test_hermevalc C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t |j dk� d S )Nr> )r> r ) r rE r � ValueErrorr � hermeval2d�c2dr r$ �onesr rD ) r �x1�x2�x3�y1�y2�y3r+ r, �zr r r �test_hermeval2d� s z TestEvaluation.test_hermeval2dc C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t |j dk� d S )Nr> )r> r ) r rE r rG r � hermeval3d�c3dr r$ rJ r rD ) r rK rL rM rN rO rP r+ r, rQ r r r �test_hermeval3d� s z TestEvaluation.test_hermeval3dc C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j dk� d S )Nzi,j->ij)r> r )r> r r> r ) r rE r$ �einsumr Z hermegrid2drI r rJ r rD ) r rK rL rM rN rO rP r+ r, rQ r r r �test_hermegrid2d� s z TestEvaluation.test_hermegrid2dc C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j dk� d S )Nz i,j,k->ijk)r> r )r> r r> r r> r ) r rE r$ rV r Z hermegrid3drT r rJ r rD ) r rK rL rM rN rO rP r+ r, rQ r r r �test_hermegrid3d� s z TestEvaluation.test_hermegrid3dN)r r r r$ r9 Zc1drV rI rT �randomr r rE rF rR rU rW rX r r r r r= p s r= c @ s e Zd Zdd� Zdd� ZdS )� TestIntegralc C sr t ttjdgd� t ttjdgd� t ttjdgdddg� t ttjdgdgd� t ttjdgdgd� t ttjdgdd� xFtdd �D ]8}dg|d dg }tjdg||d �}t|ddg� q�W xztd �D ]n}|d }dg| dg }|gdg| d| g }t�|�}tj|d|gd �}t�|�}tt |�t |�� q�W xZtd �D ]N}|d }dg| dg }t�|�}tj|d|gdd �}tt� d|�|� �qJW x~td �D ]r}|d }dg| dg }|gdg| d| g }t�|�}tj|d|gdd �}t�|�}tt |�t |�� �q�W x�td �D ]z}xrtdd �D ]d} dg| dg }|d d � }x t| �D ]}tj|dd �}�q`W tj|| d �}tt |�t |�� �q6W �q&W x�td �D ]�}x�tdd �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�tdd �D ]v} dg| dg }|d d � }x&t| �D ]}tj|d|gdd �}�q~W tj|| t t| ��dd �}tt |�t |�� �qTW �qDW x�td �D ]�}x�tdd �D ]v} dg| dg }|d d � }x&t| �D ]}tj|d|gdd �}�qW tj|| t t| ��dd �}tt |�t |�� �q�W �q�W d S )Nr g �?r r )�lbnd)�scl)�axisr> r! )�m�k)r^ r_ r[ )r^ r_ r\ )r^ ) r � TypeErrorr �hermeintrG r# r � poly2herme� herme2polyr r0 �list) r r( r_ r, r\ �polr+ Zhermepolra r) r r r � test_hermeint� s� zTestIntegral.test_hermeintc 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)r � c S s g | ]}t �|��qS r )r ra )r? r: r r r r@ ' s z3TestIntegral.test_hermeint_axis.<locals>.<listcomp>r )r] c S s g | ]}t �|��qS r )r ra )r? r: r r r r@ + s r c S s g | ]}t j|d d��qS )r )r_ )r ra )r? r: r r r r@ / s r )r_ r] )r$ rY �vstack�Tr ra r )r rI r+ r, r r r �test_hermeint_axis# s z TestIntegral.test_hermeint_axisN)r r r rf rj r r r r rZ � s SrZ 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 )r^ r> )r^ r\ ) r r` r �hermederrG r# r r ra r )r r( r+ r, r) r r r � test_hermeder6 s" z TestDerivative.test_hermederc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )N)r rg c S s g | ]}t �|��qS r )r rl )r? r: r r r r@ T s z5TestDerivative.test_hermeder_axis.<locals>.<listcomp>r )r] c S s g | ]}t �|��qS r )r rl )r? r: r r r r@ X s r )r$ rY rh ri r rl r )r rI r+ r, r r r �test_hermeder_axisP s z!TestDerivative.test_hermeder_axisN)r r r rm rn r r r r rk 4 s rk c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd S ) � TestVander)r 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 ) Nr )r rg rg r r .r> r! � )r r> rg ) r$ r8 r � hermevanderr rD r# r r0 r9 )r r �vr( �coefr r r �test_hermevandera s zTestVander.test_hermevanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j dk� d S )N)r> r r r> )r r! rp ) r r$ rY r Z hermevander2drH �dot�flatr r rD )r rK rL rM r: �vanr+ r, r r r �test_hermevander2dr s z TestVander.test_hermevander2dc C s� | j \}}}tj�d�}t�|||dddg�}t�||||�}t�||j�}t||� t�|g|g|gdddg�}t |j dk� d S )N)r> r rg r r> r )r r! � ) r r$ rY r Z hermevander3drS ru rv r r rD )r rK rL rM r: rw r+ r, r r r �test_hermevander3d s z TestVander.test_hermevander3dN) r r r r$ rY r rt rx rz r r r r ro ] s ro c @ s e Zd Zdd� ZdS )� TestFittingc 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�||dd d ddg�}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$TestFitting.test_hermefit.<locals>.fc S s | d | d d S )Nrg r> r r )r r r r �f2� s z%TestFitting.test_hermefit.<locals>.f2r r r r> )�wrp r rg r! y �?y � �)r rG r Zhermefitr` r$ r< r r2 r r0 r9 ri Z zeros_like�copy)r r| r} r rE Zcoef3Zcoef4Zcoef2dr~ ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r � test_hermefit� sp " & zTestFitting.test_hermefitN)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 rG r �hermecompanion)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� rD )r r( rs 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� ZdS )� TestGaussc C s� t �d�\}}t �|d�}t�|j| |�}dt�|�� � }|d d �d f | | }t|t� d�� t�dtj �}t|� � |� d S )Nr �c r r> ) r Z hermegaussrq r$ ru ri �sqrtZdiagonalr �eye�pi�sum)r r r~ rr �vvZvdr+ r r r �test_100� s zTestGauss.test_100N)r r r r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestMiscc C s� t �g �}tt|�dg� x�tdd�D ]z}t�t�tj dd| d �dd d� �}t �|�}t � ||�}d}t t |�|d k� tt � |�d d� t||� q&W d S )Nr r! r r> r ) r �hermefromrootsr r r# r$ �cosr< r� r0 r r2 rc )r r, r( �rootsre r+ r r r �test_hermefromroots� s * z TestMisc.test_hermefromrootsc 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 r> r! )r r Z hermerootsr# r$ r< r� r )r r( r+ r, r r r �test_hermeroots s zTestMisc.test_hermerootsc C sf ddddg}t ttj|d� tt�|�|d d� � tt�|d�|d d� � tt�|d�dg� d S )Nr> r r r r )r rG r r r )r rs r r r �test_hermetrim s zTestMisc.test_hermetrimc C s t t�dd�ddg� d S )Nr rg )r r Z hermeline)r r r r �test_hermeline s zTestMisc.test_hermelinec C s6 x0t d�D ]$}tt�dg| dg �t| � q W d S )NrA r r )r# r r rc rC )r r( r r r �test_herme2poly s zTestMisc.test_herme2polyc C s6 x0t d�D ]$}tt�t| �dg| dg � q W d S )NrA r r )r# r r rb rC )r r( r r r �test_poly2herme$ s zTestMisc.test_poly2hermec C s8 t �ddd�}t �d|d �}t�|�}t||� d S )N���r! � g �r> )r$ r< �expr Z hermeweightr )r r r+ r, r r r � test_weight( s zTestMisc.test_weightN) r r r r� r� r� r� r� r� r� r r r r r� � s r� )'�__doc__� functoolsr �numpyr$ Znumpy.polynomial.hermite_eZ polynomialZ hermite_er Znumpy.polynomial.polynomialr Z numpy.testingr r r r r9 ZHe0ZHe1ZHe2ZHe3ZHe4ZHe5ZHe6ZHe7ZHe8ZHe9rC r r r r= rZ rk ro r{ r� r� r� r r r r �<module> s6 B^f)0M
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка