Edit File: test_chebyshev.cpython-37.pyc
B ��FdOL � @ s� d Z ddlmZmZmZ ddlZddlm m Z ddl m Z ddl mZmZmZmZmZmZ dd� ZdgZddgZd dd gZdd dd gZddd ddgZddddddgZd ddddddgZddddddddgZ ddddddddd g Z dd dd dd dd dd!g Z eeeeeeee e e g 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 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*e+d8k�r�e� dS )9z Tests for chebyshev module. � )�division�absolute_import�print_functionN)�polyval)�TestCase�assert_almost_equal� assert_raises� assert_equal�assert_�run_module_suitec C s t j| dd�S )Ng���ư>)�tol)�cheb�chebtrim)�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_chebyshev.py�trim s r � ���� ���� i����� � i��� � i���� i�����8 i�����@ i��� i ���� � i����i� i����� c @ s e Zd Zdd� Zdd� ZdS )� TestPrivatec C sh xbt d�D ]V}t�dgdg| tj�}t�dg| dg dg| tj�}t�|�}t||� q W d S )Nr r r g �?)�range�np�array�doubler �_cseries_to_zseriesr )�self�i�inp�tgt�resr r r �test__cseries_to_zseries! s $ z$TestPrivate.test__cseries_to_zseriesc C sh xbt d�D ]V}t�dg| dg dg| tj�}t�dgdg| tj�}t�|�}t||� q W d S )Nr g �?r r )r$ r% r& r' r �_zseries_to_cseriesr )r) r* r+ r, r- r r r �test__zseries_to_cseries( s $ z$TestPrivate.test__zseries_to_cseriesN)�__name__� __module__� __qualname__r. r0 r r r r r# s r# c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestConstantsc C s t tjddg� d S )Nr r )r r � chebdomain)r) r r r �test_chebdomain2 s z TestConstants.test_chebdomainc C s t tjdg� d S )Nr )r r �chebzero)r) r r r � test_chebzero5 s zTestConstants.test_chebzeroc C s t tjdg� d S )Nr )r r �chebone)r) r r r � test_chebone8 s zTestConstants.test_chebonec C s t tjddg� d S )Nr r )r r �chebx)r) r r r � test_chebx; s zTestConstants.test_chebxN)r1 r2 r3 r6 r8 r: r< r r r r r4 0 s r4 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestArithmeticc 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 )Nr z At i=%d, j=%dr r )�err_msg)r$ r% �zeros�maxr �chebaddr r )r) r* �j�msgr, r- r r r � test_chebaddA s $zTestArithmetic.test_chebaddc 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 )Nr z At i=%d, j=%dr r )r> )r$ r% r? r@ r �chebsubr r )r) r* rB rC r, r- r r r � test_chebsubK s $zTestArithmetic.test_chebsubc C sz t t�dg�dg� t t�dg�ddg� xJtdd�D ]<}dg| dg }dg|d dddg }t t�|�|� q6W d S )Nr r r g �?)r r �chebmulxr$ )r) r* Zserr, r r r � test_chebmulxU s z TestArithmetic.test_chebmulxc C s� x�t d�D ]�}x�t d�D ]�}d||f }t�|| d �}||| d7 < |t|| � d7 < t�dg| dg dg| dg �}tt|�t|�|d� qW q W d S )Nr z At i=%d, j=%dr g �?r )r> )r$ r% r? �absr �chebmulr r )r) r* rB rC r, r- r r r � test_chebmul] s $zTestArithmetic.test_chebmulc 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 )Nr z At i=%d, j=%dr r )r> )r$ r rA �chebdivrJ r r ) r) r* rB rC ZciZcjr, �quo�remr- r r r � test_chebdivg s zTestArithmetic.test_chebdivN)r1 r2 r3 rD rF rH rK rO 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 dd g�Z d d � Z d d� Zdd� Zdd� Zdd� ZdS )�TestEvaluationg @g @g �?zi,j->ijz i,j,k->ijk)� r 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>� s z/TestEvaluation.test_chebval.<locals>.<listcomp>� zAt i=%d)r> rQ r ) r r �chebval�sizer% �linspace�Tlistr$ r r? �shape)r) �yr* rC r, r- �dimsr )r r � test_chebval} s zTestEvaluation.test_chebvalc C s� | j \}}}| j\}}}tttj||d d� | j� || }t�||| j�}t||� t� d�} t�| | | j�}t |j dk� d S )Nr )r rQ ) r r[ r � ValueErrorr � chebval2d�c2dr r% �onesr rZ ) r) �x1�x2�x3�y1�y2�y3r, r- �zr r r �test_chebval2d� s z TestEvaluation.test_chebval2dc C s� | j \}}}| j\}}}tttj|||d d� | j� || | }t�|||| j�}t||� t� d�} t�| | | | j�}t |j dk� d S )Nr )r rQ ) r r[ r r^ r � chebval3d�c3dr r% ra r rZ ) r) rb rc rd re rf rg r, r- rh r r r �test_chebval3d� s z TestEvaluation.test_chebval3dc C sl | j \}}}| j\}}}t�d||�}t�||| j�}t||� t�d�} t�| | | j�}t |j dk� d S )Nzi,j->ij)r rQ )r rQ r rQ ) r r[ r% �einsumr � chebgrid2dr` r ra r rZ ) r) rb rc rd re rf rg r, r- rh r r r �test_chebgrid2d� s z TestEvaluation.test_chebgrid2dc C sr | j \}}}| j\}}}t�d|||�}t�|||| j�}t||� t�d�} t�| | | | j�}t |j dk� d S )Nz i,j,k->ijk)r rQ )r rQ r rQ r rQ ) r r[ r% rm r � chebgrid3drk r ra r rZ ) r) rb rc rd re rf rg r, r- rh r r r �test_chebgrid3d� s z TestEvaluation.test_chebgrid3dN)r1 r2 r3 r% r& Zc1drm r` rk �randomr r r[ r] ri rl ro rq r r r r rP s 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 r )�m�k)rt ru �lbnd)rt ru �scl)rt ) r r^ r �chebintr$ r � poly2cheb� cheb2polyr rV �list) r) r* ru r- rw �polr, Zchebpolrx rB r r r � test_chebint� sz zTestIntegral.test_chebintc 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)rQ r c S s g | ]}t �|��qS r )r rx )rR rS r r r rT ' s z2TestIntegral.test_chebint_axis.<locals>.<listcomp>r )�axisc S s g | ]}t �|��qS r )r rx )rR rS r r r rT + s r c S s g | ]}t j|d d��qS )rQ )ru )r rx )rR rS r r r rT / s rQ )ru r~ )r% rr �vstack�Tr rx r )r) r` r, r- r r r �test_chebint_axis# s z TestIntegral.test_chebint_axisN)r1 r2 r3 r} r� r r r r rs � s Prs 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 r r )rt r )rt rw ) r r^ r �chebderr$ r r rx r )r) r* r, r- rB r r r � test_chebder6 s zTestDerivative.test_chebderc C sl t j�d�}t �dd� |jD ��j}tj|dd�}t||� t �dd� |D ��}tj|dd�}t||� d S )N)rQ r c S s g | ]}t �|��qS r )r r� )rR rS r r r rT S s z4TestDerivative.test_chebder_axis.<locals>.<listcomp>r )r~ c S s g | ]}t �|��qS r )r r� )rR rS r r r rT W s r )r% rr r r� r r� r )r) r` r, r- r r r �test_chebder_axisO s z TestDerivative.test_chebder_axisN)r1 r2 r3 r� r� r r r r r� 4 s r� c @ s8 e Zd Zej�d�d d Zdd� Zdd� Zdd � Zd S ) � TestVander)rQ 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 ) NrQ )rQ r r r r .r r � )rQ r r ) r% �aranger � chebvanderr rZ r$ r rV r& )r) r �vr* �coefr r r �test_chebvander` s zTestVander.test_chebvanderc C sx | j \}}}tj�d�}t�||ddg�}t�|||�}t�||j�}t||� t�|g|gddg�}t |j dk� d S )N)r rQ r r )r r r� ) r r% rr r � chebvander2dr_ �dot�flatr r rZ )r) rb rc rd rS �vanr, r- r r r �test_chebvander2dq s z TestVander.test_chebvander2dc C s� | j \}}}tj�d�}t�|||dddg�}t�||||�}t�||j�}t||� t�|g|g|gdddg�}t |j dk� d S )N)r rQ r r r rQ )r r � ) r r% rr r � chebvander3drj r� r� r r rZ )r) rb rc rd rS r� r, r- r r r �test_chebvander3d~ s z TestVander.test_chebvander3dN) r1 r2 r3 r% rr r r� r� r� r r r r r� \ 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_chebfit.<locals>.fc S s | d | d d S )Nr r r r )r r r r �f2� s z$TestFitting.test_chebfit.<locals>.f2r r r r )�wr� rQ r r y �?y � �)r r^ r �chebfit� TypeErrorr% rX r �lenr rV r& r� � zeros_like�copy)r) r� r� r r[ Zcoef3Zcoef4Zcoef2dr� ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r � test_chebfit� sp " & zTestFitting.test_chebfitN)r1 r2 r3 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 � chebcompanion)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� rZ )r) r* 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)r1 r2 r3 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�� tj }t|� � |� d S )N�d �c r ) r � chebgaussr� r% r� r� �sqrt�diagonalr �eye�pi�sum)r) r r� r� �vvZvdr, r r r �test_100� s zTestGauss.test_100N)r1 r2 r3 r� r r r r r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestMiscc C s� t �g �}tt|�dg� xttdd�D ]f}t�t�tj dd| d �dd d� �}dg| dg }t �|�d|d }tt|�t|�� q&W d S )Nr r r r ) r � chebfromrootsr r r$ r% �cosrX r� )r) r- r* �rootsr, r r r �test_chebfromroots� s *zTestMisc.test_chebfromrootsc 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 � chebrootsr$ r% rX r� r )r) r* r, r- r r r �test_chebroots s zTestMisc.test_chebrootsc 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_chebtrim s zTestMisc.test_chebtrimc C s t t�dd�ddg� d S )NrQ r )r r �chebline)r) r r r � test_chebline s zTestMisc.test_cheblinec C s6 x0t d�D ]$}tt�dg| dg �t| � q W d S )NrU r r )r$ r r rz rY )r) r* r r r �test_cheb2poly s zTestMisc.test_cheb2polyc C s6 x0t d�D ]$}tt�t| �dg| dg � q W d S )NrU r r )r$ r r ry rY )r) r* r r r �test_poly2cheb s zTestMisc.test_poly2chebc C sN t �ddd�dd� }dt �d| �t �d| � }t�|�}t||� d S )Nr r � g �?)r% rX r� r � chebweightr )r) r r, r- r r r � test_weight$ s zTestMisc.test_weightc C s� t ttjd� t ttjd� dg}tt�d�|� ddg}tt�d�|� dddg}tt�d �|� d d d d g}tt�d�|� d S )Ng �?r r g�;f���g�;f���?r g�LX�z��g�LX�z��?rQ g( 1�k��gŜ��}ؿgŜ��}�?g( 1�k��?r )r r^ r �chebpts1r )r) r, r r r � test_chebpts1* s zTestMisc.test_chebpts1c C s� t ttjd� t ttjd� ddg}tt�d�|� dddg}tt�d�|� ddddg}tt�d �|� d d dd d g}tt�d�|� d S )Ng �?r r r r rQ g �g �?r g �g�Kf���g�Kf���?g �?r )r r^ r �chebpts2r )r) r, r r r � test_chebpts29 s zTestMisc.test_chebpts2N) r1 r2 r3 r� r� r� r� r� r� r� r� r� r r r r r� � s r� �__main__),�__doc__� __future__r r r �numpyr% Znumpy.polynomial.chebyshev� polynomial� chebyshevr Znumpy.polynomial.polynomialr Z numpy.testingr r r r r r r ZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9rY r# r4 r= rP rs r� r� r� r� r� r� r1 r r r r �<module> s< 4^c(0MM