Edit File: test_hermite.cpython-37.pyc
B ��FdyG � @ s d Z ddlmZmZmZ ddlZddlm m Z ddl m Z ddl mZmZmZmZmZmZ e�dg�Ze�ddg�Ze�ddd g�Ze�dd dd g�Ze�d dd ddg�Ze�ddddddg�Ze�dddddddg�Ze�ddddddddg�Z e�ddddd dd dd g �Z e�dd dd dd!dd"dd#g �Z eeeeeeee e e g Z d$d%� Z G d&d'� d'e�Z!G d(d)� d)e�Z"G d*d+� d+e�Z#G d,d-� d-e�Z$G d.d/� d/e�Z%G d0d1� d1e�Z&G d2d3� d3e�Z'G d4d5� d5e�Z(G d6d7� d7e�Z)G d8d9� d9e�Z*e+d:k�r e� dS );zTests for hermite module. � )�division�absolute_import�print_functionN)�polyval)�TestCase�assert_almost_equal� assert_raises� assert_equal�assert_�run_module_suite� � ���� i�� � i���� �x i`���� i����i� i ����@ ip���i i����� i� i���i�4 i �� i v i ��i � i ��i c C s t j| dd�S )Ng���ư>)�tol)�herm�hermtrim)�x� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/polynomial/tests/test_hermite.py�trim s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestConstantsc C s t tjddg� d S )N���r )r r � hermdomain)�selfr r r �test_hermdomain! s z TestConstants.test_hermdomainc C s t tjdg� d S )Nr )r r �hermzero)r" r r r � test_hermzero$ s zTestConstants.test_hermzeroc C s t tjdg� d S )Nr )r r �hermone)r" r r r � test_hermone' s zTestConstants.test_hermonec C s t tjddg� d S )Nr g �?)r r �hermx)r" r r r � test_hermx* s zTestConstants.test_hermxN)�__name__� __module__� __qualname__r# r% r' r) r r r r r s r c @ sB e Zd Ze�ddd�Zdd� Zdd� Zdd � Zd d � Z d d � Z dS )�TestArithmetic���� �d c C s� x�t d�D ]�}x�t d�D ]~}d||f }t�t||�d �}|| d7 < || d7 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )N� z At i=%d, j=%dr r )�err_msg)�range�np�zeros�maxr �hermaddr r )r" �i�j�msg�tgt�resr r r � test_hermadd1 s $zTestArithmetic.test_hermaddc C s� x�t d�D ]�}x�t d�D ]~}d||f }t�t||�d �}|| d7 < || d8 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )Nr1 z At i=%d, j=%dr r )r2 )r3 r4 r5 r6 r �hermsubr r )r" r8 r9 r: r; r< r r r � test_hermsub; s $zTestArithmetic.test_hermsubc C sz t t�dg�dg� t t�dg�ddg� xJtdd�D ]<}dg| dg }dg|d |ddg }t t�|�|� q6W d S )Nr r g �?r1 )r r �hermmulxr3 )r" r8 �serr; r r r � test_hermmulxE s z TestArithmetic.test_hermmulxc C s� x�t d�D ]�}dg| dg }t�| j|�}x�t d�D ]t}d||f }dg| dg }t�| j|�}t�||�}t�| j|�} tt|�|| d k|� t| || |d� q6W q W d S )Nr1 r r z At i=%d, j=%d)r2 )r3 r �hermvalr �hermmulr �lenr ) r" r8 Zpol1�val1r9 r: Zpol2�val2Zpol3Zval3r r r � test_hermmulM s zTestArithmetic.test_hermmulc C s� x�t d�D ]�}x�t d�D ]v}d||f }dg| dg }dg| dg }t�||�}t�||�\}}t�t�||�|�} tt| �t|�|d� qW q W d S )Nr1 z At i=%d, j=%dr r )r2 )r3 r r7 �hermdivrD r r ) r" r8 r9 r: �ci�cjr; �quo�remr< r r r � test_hermdiv[ s zTestArithmetic.test_hermdivN) r* r+ r, r4 �linspacer r= r? rB rH rN r r r r r- . s r- c @ s� e Zd Ze�dddg�Ze�dee�Ze�deee�Zej � d�d d Z e e dd d g�Z d d � Z d d� Zdd� Zdd� Zdd� ZdS )�TestEvaluationg @g �?g �?zi,j->ijz i,j,k->ijk)r/ r1 r r g @g @c s� t t�g dg�jd� t�dd�� � fdd�tD �}xFtd�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 )�.0�c)r r r � <listcomp>w s z/TestEvaluation.test_hermval.<locals>.<listcomp>� zAt i=%d)r2 r/ r ) r r rC �sizer4 rO �Hlistr3 r r5 �shape)r" �yr8 r: r; r< �dimsr )r r � test_hermvalq s zTestEvaluation.test_hermvalc C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t |j dk� d S )Nr )r r/ ) r rX r � ValueErrorr � hermval2d�c2dr r4 �onesr rW ) r" �x1�x2�x3�y1�y2�y3r; r< �zr r r �test_hermval2d� s z TestEvaluation.test_hermval2dc C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t |j dk� d S )Nr )r r/ ) r rX r r[ r � hermval3d�c3dr r4 r^ r rW ) r" r_ r` ra rb rc rd r; r< re r r r �test_hermval3d� s z TestEvaluation.test_hermval3dc 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 rX r4 �einsumr � hermgrid2dr] r r^ r rW ) r" r_ r` ra rb rc rd r; r< re r r r �test_hermgrid2d� s z TestEvaluation.test_hermgrid2dc 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 rX r4 rj r � hermgrid3drh r r^ r rW ) r" r_ r` ra rb rc rd r; r< re r r r �test_hermgrid3d� s z TestEvaluation.test_hermgrid3dN)r* r+ r, r4 �array�c1drj r] rh �randomr r rX rZ rf ri rl rn r r r r rP g s rP c @ s e Zd Zdd� Zdd� ZdS )� TestIntegralc C s2 t ttjdgd� t ttjdgd� t ttjdgdddg� xFtdd�D ]8}dg|d dg }tjdg||d�}t|ddg� qHW 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|�|� �q W x~td�D ]r}|d }dg| dg }|gdg| d| g }t�|�}tj|d|gdd �}t�|�}tt|�t|�� �qfW x�td�D ]z}xrtdd�D ]d} dg| dg }|d d � }x t| �D ]}tj|dd �}�q W tj|| d �}tt|�t|�� �q�W �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 �qnW 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|�� �qW �qW 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|�� �q�W �q�W d S ) Nr g �?r r r r1 )�m�k)rs rt �lbnd)rs rt �scl)rs ) r r[ r �hermintr3 r � poly2herm� herm2polyr rC �list) r" r8 rt r< rv �polr; Zhermpolrw r9 r r r � test_hermint� sz zTestIntegral.test_hermintc 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/ r c S s g | ]}t �|��qS r )r rw )rQ rR r r r rS s z2TestIntegral.test_hermint_axis.<locals>.<listcomp>r )�axisc S s g | ]}t �|��qS r )r rw )rQ rR r r r rS s r c S s g | ]}t j|d d��qS )r/ )rt )r rw )rQ rR r r r rS # s r/ )rt r} )r4 rq �vstack�Tr rw r )r" r] r; r< r r r �test_hermint_axis s z TestIntegral.test_hermint_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�}tt|�t|�� q�W qpW xbtd�D ]V}xPtdd�D ]B}dg| dg }tjtj||dd�|dd�}tt|�t|�� q�W q�W d S ) Nr g �?r r1 r )rs r )rs rv ) r r[ r �hermderr3 r r rw r )r" r8 r; r< r9 r r r � test_hermder* s zTestDerivative.test_hermderc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )N)r/ r c S s g | ]}t �|��qS r )r r� )rQ rR r r r rS G s z4TestDerivative.test_hermder_axis.<locals>.<listcomp>r )r} c S s g | ]}t �|��qS r )r r� )rQ rR r r r rS K s r )r4 rq r~ r r r� r )r" r] r; r< r r r �test_hermder_axisC s z TestDerivative.test_hermder_axisN)r* r+ r, r� r� r r r r r� ( s r� c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd S ) � TestVander)r/ r1 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/ r r r r .r r1 � )r/ r r ) r4 �aranger � hermvanderr rW r3 r rC ro )r" r �vr8 �coefr r r �test_hermvanderT s zTestVander.test_hermvanderc 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 r1 r� ) r r4 rq r � hermvander2dr\ �dot�flatr r rW )r" r_ r` ra rR �vanr; r< r r r �test_hermvander2de s z TestVander.test_hermvander2dc C s� | j \}}}tj�d�}t�|||dddg�}t�||||�}t�||j�}t||� t�|g|g|gdddg�}t |j dk� d S )N)r r/ r r r r/ )r r1 � ) r r4 rq r � hermvander3drg r� r� r r rW )r" r_ r` ra rR r� r; r< r r r �test_hermvander3dr s z TestVander.test_hermvander3dN) r* r+ r, r4 rq r r� r� r� r r r r r� P s r� 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_hermfit.<locals>.fc S s | d | d d S )Nr r r r )r r r r �f2� s z$TestFitting.test_hermfit.<locals>.f2r r r r )�wr� r/ r r1 y �?y � �g �?)r r[ r �hermfit� TypeErrorr4 rO r rE r rC ro r � zeros_like�copy)r" r� r� r rX �coef3�coef4�coef2dr� �yw�wcoef3�wcoef2d�coef1�coef2r r r � test_hermfit� sp " & zTestFitting.test_hermfitN)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 r[ r � hermcompanion)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 r1 r )r3 r r r� rW )r" r8 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� ZdS )� TestGaussc C s� t �d�\}}t �|d�}t�|j| |�}dt�|�� � }|d d �d f | | }t|t� d�� t�tj �}t|� � |� d S )Nr0 �c r ) r � hermgaussr� r4 r� r �sqrt�diagonalr �eye�pi�sum)r" r r� r� �vv�vdr; 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 r1 r r r ) r � hermfromrootsr r r3 r4 �cosrO r� rC r rE ry )r" r< r8 �rootsr{ r; r r r �test_hermfromroots� s * zTestMisc.test_hermfromrootsc C sp t t�dg�g � t t�ddg�dg� xBtdd�D ]4}t�dd|�}t�t�|��}t t|�t|�� q4W d S )Nr g �r r1 r )r r � hermrootsr3 r4 rO r� r )r" r8 r; r< r r r �test_hermroots� s zTestMisc.test_hermrootsc C sf ddddg}t ttj|d� tt�|�|d d� � tt�|d�|d d� � tt�|d�dg� d S )Nr r r r r. )r r[ r r r )r" r� r r r � test_hermtrim s zTestMisc.test_hermtrimc C s t t�dd�ddg� d S )Nr/ r r )r r �hermline)r" r r r � test_hermline s zTestMisc.test_hermlinec C s6 x0t d�D ]$}tt�dg| dg �t| � q W d S )NrT r r )r3 r r ry rV )r" r8 r r r �test_herm2poly s zTestMisc.test_herm2polyc C s6 x0t d�D ]$}tt�t| �dg| dg � q W d S )NrT r r )r3 r r rx rV )r" r8 r r r �test_poly2herm s zTestMisc.test_poly2hermc C s6 t �ddd�}t �|d �}t�|�}t||� d S )N���r1 � r )r4 rO �expr � hermweightr )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� �__main__),�__doc__� __future__r r r �numpyr4 Znumpy.polynomial.hermite� polynomial�hermiter Znumpy.polynomial.polynomialr Z numpy.testingr r r r r r ro ZH0ZH1ZH2ZH3ZH4ZH5ZH6ZH7ZH8ZH9rV r r r- rP rr r� r� r� r� r� r� r* r r r r �<module> s: 9^c(0M3