Submit
Path:
~
/
/
opt
/
alt
/
python37
/
lib64
/
python3.7
/
site-packages
/
numpy
/
lib
/
tests
/
__pycache__
/
File Content:
test_function_base.cpython-37.opt-1.pyc
B <�Fd=� � @ s* d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZ d dl m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZ d dlmZ d dlm m Z d dl m Z d dl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD d dlEmFZF dd � ZGG d d � d e �ZHG d d � d e �ZIG dd� de �ZJG dd� de �ZKG dd� de �ZLG dd� de �ZMG dd� de �ZNG dd� de �ZOG dd� de �ZPG d d � d e �ZQG d d � d e �ZRG d d!� d!e �ZSG d"d#� d#e �ZTG d$d%� d%e �ZUG d&d'� d'e �ZVG d(d)� d)e �ZWG d*d+� d+e �ZXG d,d-� d-e �ZYG d.d/� d/e �ZZG d0d1� d1e �Z[G d2d3� d3e �Z\G d4d5� d5e �Z]G d6d7� d7e �Z^G d8d9� d9e �Z_G d:d;� d;e �Z`G d<d=� d=e �ZaG d>d?� d?e �ZbG d@dA� dAe �ZcG dBdC� dCe �ZdG dDdE� dEe �ZeG dFdG� dGe �ZfG dHdI� dIe �ZgG dJdK� dKe �ZhG dLdM� dMe �ZiG dNdO� dOe �ZjG dPdQ� dQe �ZkG dRdS� dSe �ZlG dTdU� dUe �ZmG dVdW� dWe �ZnG dXdY� dYe �ZodZd[� ZpG d\d]� d]e �ZqG d^d_� d_e �ZrG d`da� dae �ZsG dbdc� dce �Zteuddk�r&e � dS )e� )�division�absolute_import�print_functionN)�run_module_suite�TestCase�assert_� assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal� assert_raises�assert_allclose�assert_array_max_ulp� assert_warns�assert_raises_regex�dec�suppress_warnings)� HAS_REFCOUNT)�rand)#�add_newdoc_ufunc�angle�average�bartlett�blackman�corrcoef�cov�delete�diff�digitize�extract�flipud�gradient�hamming�hanning� histogram� histogramdd�i0�insert�interp�kaiser�meshgrid�msort� piecewise�place�rot90�select�setxor1d�sinc�split�trapz� trim_zeros�unwrap�unique� vectorize)�longc C s t �| �}t j�||�}|S )N)�np�arange�add�outer)�n�data� r? �U/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_function_base.py�get_mat s rA c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestRot90c C s� | � ttt�d�� tttt�d�dd� tttt�d�dd� tttt�d�dd� tttt�d�dd� d d d gd dd gg}d d gd dgd d gg}d dd gd d d gg}d d gdd gd d gg}d d d gd dd gg}x&tddd�D ]}tt||d�|� q�W x(tddd�D ]}tt||d�|� �q W x(tddd�D ]}tt||d�|� �q4W x(td dd�D ]}tt||d�|� �q^W ttt|dd�dd�|� tt|d dd�t|ddd�� d S )N� )� rD rD )r � rD )�axes)rD rD )r rD )rE rE )���rE r rE rD � � ���� )�krG ���)r rE )rE r )rL rF )� assertRaises� ValueErrorr. r9 �onesr �ranger )�self�aZb1Zb2Zb3Zb4rL r? r? r@ � test_basic$ s6 zTestRot90.test_basicc C sR t �d�}tt|�jd� tt|dd�t|dd�� tt|dd�t|dd�� d S )N)�2 �( rH )rV rU rH )r rD )rF )r rM )rE rD )rG rM )r9 rP r r. �shape)rR rS r? r? r@ � test_axesD s zTestRot90.test_axesc C s t �d��d�}ddgddggddgd d ggg}ddgddggd dgd dggg}d dgddggd dgddggg}d d gddggddgddggg}tt|d d �|� tt|d d �|� tt|dd �|� x4tdd �D ]&}tt||dd�t||d dd�� q�W d S )N� )rD rD rD rD rH � � r rE rC rI )r rE )rF )rE r )rE rD )rD r )rL rF )r9 r: �reshaper r. rQ )rR rS Z a_rot90_01Z a_rot90_12Z a_rot90_20Z a_rot90_10rL r? r? r@ �test_rotation_axesJ s. z TestRot90.test_rotation_axesN)�__name__� __module__� __qualname__rT rX r] r? r? r? r@ rB # s rB c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestFlipc C sR | j ttjt�d�dd� | j ttjt�d�dd� | j ttjt�d�dd� d S )NrC rE )�axis)rC rC rD rJ )rN rO r9 �fliprP )rR r? r? r@ rX i s zTestFlip.test_axesc C sn t d�}|d d �d d d�f }tt�|d�|� dddgdddgg}dddgdddgg}tt�|d�|� d S )NrC rM rE r rD rH rI )rA r r9 rc )rR rS �br? r? r@ � test_basic_lrn s zTestFlip.test_basic_lrc C sn t d�}|d d d�d d �f }tt�|d�|� dddgdddgg}dddgdddgg}tt�|d�|� d S )NrC rM r rE rD rH rI )rA r r9 rc )rR rS rd r? r? r@ � test_basic_udx s zTestFlip.test_basic_udc C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rE rD rH rC rI rZ r[ )r9 �arrayr rc )rR rS rd r? r? r@ �test_3d_swap_axis0� s zTestFlip.test_3d_swap_axis0c C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rE rD rH rC rI rZ r[ )r9 rg r rc )rR rS rd r? r? r@ �test_3d_swap_axis1� s zTestFlip.test_3d_swap_axis1c C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rE rD rH rC rI rZ r[ )r9 rg r rc )rR rS rd r? r? r@ �test_3d_swap_axis2� s zTestFlip.test_3d_swap_axis2c C sV t �d��dddd�}x:t|j�D ],}tt �||�t �|�d|���|d�� q"W d S )N�x rD rH rC rI r ) r9 r: r\ rQ �ndimr rc r �swapaxes)rR rS �ir? r? r@ �test_4d� s zTestFlip.test_4dN) r^ r_ r` rX re rf rh ri rj ro r? r? r? r@ ra g s ra c @ s e Zd Zdd� Zdd� ZdS )�TestAnyc C sT ddddg}ddddg}ddddg}t t�|�� t t�|�� t t�|� � d S )Nr rE )r r9 �any)rR �y1�y2�y3r? r? r@ rT � s zTestAny.test_basicc C sb dddgdddgdddgg}t t�|�� ttj|dd�dddg� ttj|dd�dddg� d S )Nr rE )rb )r r9 rq r Zsometrue)rR rr r? r? r@ �test_nd� s zTestAny.test_ndN)r^ r_ r` rT ru r? r? r? r@ rp � s rp c @ s e Zd Zdd� Zdd� ZdS )�TestAllc C sl ddddg}ddddg}ddddg}t t�|� � t t�|�� t t�|� � t t�t�|� �� d S )Nr rE )r r9 �allrg )rR rr rs rt r? r? r@ rT � s zTestAll.test_basicc C sd dddgdddgdddgg}t t�|� � ttj|dd�dddg� ttj|dd�dddg� d S )Nr rE )rb )r r9 rw r Zalltrue)rR rr r? r? r@ ru � s zTestAll.test_ndN)r^ r_ r` rT ru r? r? r? r@ rv � s rv c @ s e Zd Zdd� Zdd� ZdS )�TestCopyc C sR t �ddgddgg�}t �|�}t||� d|d< t|d d� t|d d� d S )NrE rD rH rC � )r r )r9 rg �copyr r )rR rS �a_copyr? r? r@ rT � s zTestCopy.test_basicc C s� t �ddgddgg�}t|jj� t|jj � t jddgddggdd�}t|jj � t|jj� t �|�}t|jj� t|jj � t �|�}t|jj � t|jj� d S )NrE rD rH rC �F)�order)r9 rg r �flags� c_contiguous� f_contiguousrz )rR rS Za_fortr{ Z a_fort_copyr? r? r@ � test_order� s zTestCopy.test_orderN)r^ r_ r` rT r� r? r? r? r@ rx � s rx c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestAveragec C s t �dddg�}tt|dd�dk� t �dddg�}tt|dd�dk� d d d g}tt|dd�d k� t �d �}d|d < d|d < t|�d�t|d�� t|�d�t|d�� td d �}t|�d�t|d�� t|�d�t|d�� t �td d ��}t |�d�t|d�� d S )NrE rD rH r )rb g @g �?g @g )rC rC )r rE )rE r rI ) r9 rg r r rP r �meanr �matrixr )rR rr rs rt Zy4Zy5Zy6r? r? r@ rT � s zTestAverage.test_basicc C sT t �d�}t �d�}t||d�}t �d�d �� d t �d��� }t||� t �dddgddd gg�}ddg}t||d d �}t �d d dg�}t||� d d dg}t||dd �}t �d dg�}t||� d d dgd d dgg}t �d dg�}tt||dd �|� tt||d�d� td�� t j �} td�� t j �} t t j| | d�j t �| | �k� d S )Nry )�weightsrD g �?rE rH rC rI rZ r )r� rb g @g @g @g @)r9 r: r �sumr rg r r r �astype�float32�float64r �dtypeZ result_type) rR �y�w�actual�desiredrr �w0�w1�w2rt Zw3r? r? r@ � test_weights s* $ zTestAverage.test_weightsc C s& t �dddgdddgg�}t|dd�\}}t|d � t|d dd�\}}t|t �d d d g�� t|ddd�\}}t|t �d d g�� ddg}t||d dd �\}}t|t �d d d g�� dddg}t||ddd �\}}t|t �d d g�� d d dgdddgg}t||ddd �\}}t|t �dd g�� d S )NrE rD rH rC rI rZ T)�returnedg @r g @g @)r� rb r� g �?)r9 rg r r r )rR r� ZavgZsclr� r� r� r? r? r@ � test_returned& s zTestAverage.test_returnedc C s� G dd� dt j�}t �ddgddgg��|�}t �ddgddgg��|�}ttt �|��|� ttt j||d��|� t �ddgddgg�}t �ddgddgg�}t j|d|d �}tt|�t j� t|d d gg� d S ) Nc @ s e Zd ZdS )z-TestAverage.test_subclasses.<locals>.subclassN)r^ r_ r` r? r? r? r@ �subclassA s r� rE rD rH rC )r� r )rb r� g @g������ @)r9 �ndarrayrg �viewr �typer r� )rR r� rS r� �rr? r? r@ �test_subclasses@ s zTestAverage.test_subclassesc C sv dddddg}xb|D ]Z\}}}t jddgdd gg|d �}t jddgdd gg|d �}tt j||d �jt �|�� qW d S ) N)�i4r� �f8)r� �f4r� )r� r� r� )r� r� r� )r� r� r� rE rD rH rC )r� )r� )r9 rg r r r� )rR �typesZatZwtZrtrS r� r? r? r@ �test_upcastingQ s zTestAverage.test_upcastingc C sX t �dd� td�D ��}t �dd� td�D ��}||�� }t|�d�t||d�� d S )Nc S s g | ]}t �|��qS r? )�decimal�Decimal)�.0�xr? r? r@ � <listcomp>Z s z1TestAverage.test_object_dtype.<locals>.<listcomp>ry c S s g | ]}t �d ��qS )rE )r� r� )r� �_r? r? r@ r� [ s r )r� )r9 rg rQ r� r r� r )rR rS r� r? r? r@ �test_object_dtypeY s z TestAverage.test_object_dtypeN) r^ r_ r` rT r� r� r� r� r� r? r? r? r@ r� � s r� c @ s� e Zd Ze�dddg�e�dddg�e�ddd g�gZe�d d d g�e�d d d g�e�d d d g�gZd d d�Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dS ) � TestSelectrE rD rH rC rI rZ r[ rY � FTr c s@ g }x6t t|��D ]&� |� fdd�t||�D �p4|g7 }qW |S )Nc s g | ]\}}|� r|� �qS r? r? )r� �V�C)�mr? r@ r� j s z&TestSelect._select.<locals>.<listcomp>)rQ �len�zip)rR Zcond�values�default�outputr? )r� r@ �_selectg s &zTestSelect._selectc C sL | j }| j}tt||dd�| j||dd�� tt|�d� tt|�d� d S )N� )r� rH )�choices� conditionsr r/ r� r r� )rR r� r� r? r? r@ rT m s zTestSelect.test_basicc C sf t �d�t �dddg�g}dt �d��dd�g}tt||�t �d�� ttdgdgdgd �jd � d S ) NTFrE � rC rH )rC rH r )r� )rE ) r9 rg r: r\ r r/ rP r rW )rR r� r� r? r? r@ �test_broadcastingv s z TestSelect.test_broadcastingc C s� t t| j| jd�jtj� dd� | jD �}t t| j|�jtj� t�dddtj ddg�}t� |�}t t|g|g�d d d tj d d g� d S ) Ny �?c S s g | ]}|� tj��qS r? )r� r9 �int8)r� Zchoicer? r? r@ r� � s z0TestSelect.test_return_dtype.<locals>.<listcomp>rE rD rH rI r[ r ) r r/ r� r� r� r9 Zcomplex_r� rg �nanZisnan)rR r� �dr� r? r? r@ �test_return_dtype} s z TestSelect.test_return_dtypec C s| t jdd��" t �d� ttg g d�d� W d Q R X t �� �6 t �d� tttg g � t �d� tttg g � W d Q R X d S )NT)�record�alwaysy @�error)�warnings�catch_warnings� simplefilterr r/ r �DeprecationWarningr )rR r? r? r@ �test_deprecated_empty� s z TestSelect.test_deprecated_emptyc C s� | j }| jd d � }t�� �l t�d� |d �tj�|d< tt t ||� |d �tj �|d< tt t ||� t�d� t t t ||� W d Q R X d S )Nr� r r� ) r� r� r� r� �filterwarningsr� r9 Zint_r r� r/ �uint8r )rR r� r� r? r? r@ �test_non_bool_deprecation� s z$TestSelect.test_non_bool_deprecationc C s2 t �dg�gd }t �dg�gd }t||� d S )NF�d rE )r9 rg r/ )rR r� r� r? r? r@ �test_many_arguments� s z TestSelect.test_many_argumentsN)r )r^ r_ r` r9 rg r� r� r� rT 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 )� TestInsertc C s� dddg}t t|dd�ddddg� t t|dd�ddddg� t t|dddgdddg�ddddddg� t t|ddddg�ddddddg� t t|dddgd�ddddddg� t t|tdd d�d�ddddddg� t t|dddgdddg�ddddddg� tjddgtjd �}t t|d|d �d d d g� t t|g g �|� tjd d ��R}t�ddt � t t|t�d gd �d�dddddddg� t |d j t k� W d Q R X d S )NrE rD rH r rM r� r[ rY )r� g g �?T)r� r� � rC ) r r' �slicer9 rg r� r� r� r� � FutureWarningr �category)rR rS rd r� r? r? r@ rT � s *$$&* *zTestInsert.test_basicc C s� dddgg}dddgdddgg}t t|ddg�ddddg� t t|ddddgdd�|� t t|dddd�|� t t|dddd�ddddgg� t�ddgddgddgg�}t�dd��d��dd�}tj|d d �dd�f t�dd��d��dd�j|d d �dd�f fdd�}t t|dgdgdgdggdd�|� t t|dgdddgdd�|� t t|ddddgdd�|� t t|ddgdgdggdd�|� t�d��dd�}t t|d d �d d�f d|d d �df dd�|� t t|d d�d d �f d|dd d �f dd�|� t�d��d�}t t|d|d d �d d �df d d�t|d|d d �d d �df dd�� t t|d|d d �dd d �f d d�t|d|d d �dd d �f dd�� t tj t|d|d d �dd d �f dd� t tj t|d|d d �dd d �f d d� t�d��d�}t t|d|d d �d d �df d d�t|d|d d �d d �df dd�� t t|d|d d �dd d �f d d�t|d|d d �dd d �f dd�� d S ) NrE rD r )rb rH rC � )rD rH rC rM rG ���) r r' r9 rg r: �repeatr\ � concatenate�Tr � AxisError)rR rS r� rd �cr? r? r@ � test_multidim� s> , $ "22"$"$(("$"zTestInsert.test_multidimc C sb t �d�}tjdd��B}t�ddt� tt|g ddd�t �d�� t|d j tk� W d Q R X d S ) NrE T)r� r� r� rD r )rb ) r9 rg r� r� r� r� r r' r r� )rR rS r� r? r? r@ �test_0d� s zTestInsert.test_0dc C s� G dd� dt j�}t �d��|�}ttt �|ddg�|�� ttt �|g g �|�� ttt �|ddgddg�|�� ttt �|tdd�ddg�|�� ttt �|tddd�g �|�� t �d��|�}ttt �|ddg�|�� d S ) Nc @ s e Zd ZdS )z*TestInsert.test_subclass.<locals>.SubClassN)r^ r_ r` r? r? r? r@ �SubClass� s r� ry r rE rD rG rM ) r9 r� r: r� r � isinstancer' r� rg )rR r� rS r? r? r@ � test_subclass� s " zTestInsert.test_subclassc C sD t �dddg�}t �dddg|dddg� t|t �dddg�� d S )NrE r rD rH rC rI )r9 rg r' r )rR r� r? r? r@ �test_index_array_copied� s z"TestInsert.test_index_array_copiedc C s� t jdddgddgd�}d}t �|d|�}t|d t j||jd�� dgd }t �|dd g|�}t|dd g t j||jd�� d S ) N)rE rS )rD rd )rH r� )�foorn )�bar�a1)r� )rC r� r rD rH )r9 rg r' r r� )rR rS �valrd r? r? r@ �test_structured_array s z TestInsert.test_structured_arrayN) r^ r_ r` rT r� r� r� r� r� r? r? r? r@ r� � s * r� c @ s e Zd Zdd� ZdS )�TestAmaxc C sv dddddddg}t t�|�d� ddd gddd gd d d gg}t tj|dd�ddd g� t tj|dd�d ddg� d S )NrH rC rI ry rJ ���g @g $@g "@g @rY g @g @r )rb g @rE )r r9 Zamax)rR rS rd r? r? r@ rT s zTestAmax.test_basicN)r^ r_ r` rT r? r? r? r@ r� s r� c @ s e Zd Zdd� ZdS )�TestAminc C sv dddddddg}t t�|�d� ddd gdd d gd d dgg}t tj|dd�d d dg� t tj|dd�d ddg� d S )NrH rC rI ry rJ r� g @g �g "@g $@g @rY g @g @r )rb rE g @)r r9 Zamin)rR rS rd r? r? r@ rT s zTestAmin.test_basicN)r^ r_ r` rT r? r? r? r@ r� s r� c @ s e Zd Zdd� ZdS )�TestPtpc C s� t �dddddddg�}t|jdd �d � t �ddd gdd d gdddgg�}t|jdd �d ddg� t|jdd �dddg� d S )NrH rC rI ry rJ r� g @r )rb g .@g "@g $@g @rY g @g @g @rM )r9 rg r Zptp)rR rS rd r? r? r@ rT & s zTestPtp.test_basicN)r^ r_ r` rT r? r? r? r@ r� $ s r� c @ s e Zd Zdd� ZdS )� TestCumsumc C s0 dddddddg}ddddgddd d gddddgg}x�t jt jt jt jt jt jt jt jt j t j g D ]�}t � ||�}t � ||�}t � ddd d d ddg|�}t t j |dd�|� t � ddddgdddd gddddgg|�}t t j |dd�|� t � ddddgddddgdd ddgg|�}t t j |dd�|� qbW d S )NrE rD ry � rZ rI rC rH r[ r� rK r� � �# �' r )rb rY � � � � � � )r9 r� r� �int16�uint16�int32�uint32r� r� � complex64� complex128rg r �cumsum)rR �ba�ba2�ctyperS �a2�tgtr? r? r@ rT 2 s " &&zTestCumsum.test_basicN)r^ r_ r` rT r? r? r? r@ r� 0 s r� c @ s e Zd Zdd� ZdS )�TestProdc C s� dddddddg}ddddgddd d gddddgg}x�t jt jt jt jt jt jt jt jgD ]�}t � ||�}t � ||�}|d kr�| � t t j |� | � t t j |d� qZt |j d d �d� t|j d d �t � ddddg|�� t|j dd �t � dddg|�� qZW d S )NrE rD ry r� rZ rI rC rH r[ r� )�1rd r )rb i g rU �$ �T � rM r� ib iX )r9 r� r� r� r� r� r� r� r� rg rN �ArithmeticError�prodr r )rR r� r� r� rS r� r? r? r@ rT I s " zTestProd.test_basicN)r^ r_ r` rT r? r? r? r@ r� G s r� c @ s e Zd Zdd� ZdS )� TestCumprodc C sZ dddddddg}ddddgddd d gddddgg}�x t jt jt jt jt jt jt jt jgD ]�}t � ||�}t � ||�}|d kr�| � t t j |� | � t t j |d� | � t t j |� q\t t j |d d �t � dddddddg|�� t t j |dd �t � ddddgddddgddddgg|�� t t j |d d �t � ddddgddd d gddd d gg|�� q\W d S ) NrE rD ry r� rZ rI rC rH r[ r� )r� rd rM )rb � �� i( i� i g r r� � r� rU r� r� r� r� �� ib rk iX )r9 r� r� r� r� r� r� r� r� rg rN r� Zcumprodr )rR r� r� r� rS r� r? r? r@ rT ] s* " zTestCumprod.test_basicN)r^ r_ r` rT r? r? r? r@ r [ s r c @ s e Zd Zdd� Zdd� ZdS )�TestDiffc C st dddddg}t �ddddg�}t �d d dg�}t �d dg�}tt|�|� tt|dd �|� tt|dd �|� d S ) NrE rC rZ r[ r� rH rD rI rM r )r= )r9 rg r r )rR r� �out�out2�out3r? r? r@ rT x s zTestDiff.test_basicc C s* dt ddd� }|d d �d d �dd �f |d d �d d �d d�f }|d d �d d �dd �f |d d �d d �d d�f }|dd �d d �d d �f |d d�d d �d d �f }|dd �d d �d d �f |d d�d d �d d �f }tt|�|� tt|dd�|� tt|dd �|� tt|ddd �|� d S ) Nr ry r� rE rM rD )r= r )rb )r= rb )r r r )rR r� Zout1r r Zout4r? r? r@ ru � s 4444zTestDiff.test_ndN)r^ r_ r` rT ru r? r? r? r@ r v s r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestDeletec C s, t �d�| _t �d��d��ddd�| _d S )NrI rD rE )r9 r: rS r� r\ �nd_a)rR r? r? r@ �setUp� s zTestDelete.setUpc C s� t | j|�}t | j|dd�}d| }t|ttttjf�s\tj |tj d�}||dk|dk @ }t t || j|f �| j|d� t |dd d �df | jd|df �}t || jdd d �df |d� d S )NrE )rb zDelete failed for obj: %r)r� r rI )�err_msg) r rS r r� r� �intr8 r9 �integer�asarray�intpr r0 )rR �indicesZa_delZnd_a_del�msg�xorr? r? r@ �_check_inverse_of_slicing� s $z$TestDelete._check_inverse_of_slicingc C sb dddddddg}dd dd g}x>|D ]6}x0|D ](}x"|D ]}t |||�}| �|� q8W q.W q$W d S ) Ni����rG r rE rD rC rI rJ rM rH )r� r )rR ZlimsZsteps�start�stop�step�sr? r? r@ � test_slices� s zTestDelete.test_slicesc C s� | � t�ddgddgg�� t�� �8 tjdtd� ttt| j dg� ttt| j dg� W d Q R X tjdd ��h}tjd t d� | � dd ddg� tjdd d gt d �}| � |� t |d j t k� t |d j t k� W d Q R X d S )Nr rE rD r� )r� r� i����T)r� r� rM F)r� )r r9 rg r� r� r� r� r r rS r� �boolr r� )rR r� �objr? r? r@ � test_fancy� s zTestDelete.test_fancyc C s | � d� | � d� d S )Nr r� )r )rR r? r? r@ � test_single� s zTestDelete.test_singlec C sZ t �d�}tjdd��:}t�ddt� tt|g dd�|� t|d j tk� W d Q R X d S )NrE T)r� r� r� r )rb ) r9 rg r� r� r� r� r r r r� )rR rS r� r? r? r@ r� � s zTestDelete.test_0dc C s� G dd� dt j�}| j�|�}ttt|d�|�� ttt|g �|�� ttt|ddg�|�� ttt|tdd��|�� ttt|tdd��|�� d S )Nc @ s e Zd ZdS )z*TestDelete.test_subclass.<locals>.SubClassN)r^ r_ r` r? r? r? r@ r� � s r� r rE rD rG )r9 r� rS r� r r� r r� )rR r� rS r? r? r@ r� � s zTestDelete.test_subclassc C sR t �d�jdddd�}t|tdd �dd�}t|jj|jj� t|jj|jj� d S ) Nry rD rI r| )r} �< rE )rb ) r9 r: r\ r r� r r~ r r� )rR rL r� r? r? r@ �test_array_order_preserve� s z$TestDelete.test_array_order_preserveN) r^ r_ r` r r 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 d� Z dd� Z dd� Z dd� Z dS )� TestGradientc C sf ddgddgg}t �|�}t �ddgddgg�t �ddgddgg�g}tt|�|� tt|�|� d S )NrE rH rC g @g @g g �?)r9 rg r r! )rR �vr� �dxr? r? r@ rT � s zTestGradient.test_basicc C s� t �t �d��}dddddg}t �d��dd�}tt �d�d� tt �d�t �d�� tt �d�|� t|d � t|t �d �� t|||� t||d � t||d d � ttd t|t j |gd dd �d � d S )NrI g �?g @g @g "@g &@� g @g �?rD rE )rb z.*scalars or 1drM ) r9 r� rP r: r\ r! rg r rO �stack)rR r# Z dx_uneven�f_2dr? r? r@ � test_args� s zTestGradient.test_argsc C s� t �d��dd�}t �t �d��}ttt||t �d�� ttt|dt �d�� ttt|t �d�t �d�� ttt||� ttt||dd� ttt||||� ttt|ddd� ttt|||dd� ttt|dddd� d S )Nr$ rI rD rE )r rE )rb ) r9 r: r\ r� rP r rO r! � TypeError)rR r& r� r? r? r@ � test_badargs� s zTestGradient.test_badargsc C s^ t jdddddddgdd �}t jd d d d dddgdd �}tt|�|� t|jt �d�k� d S )Nz 1910-08-16z 1910-08-11z 1910-08-10z 1910-08-12z 1910-10-12z 1910-12-12z 1912-12-12z datetime64[D])r� r� rJ r � �= i� i� ztimedelta64[D])r9 rg r r! r r� )rR r� r# r? r? r@ �test_datetime64 s z TestGradient.test_datetime64c C s� t jjddgddggddgddggd�}t|�d }tt|�t|�� t|j|jk � t j�d�}t jj |d< t j|dd � t |jddd ddg� d S ) NrE rH rC F)�maskr rI rD )� edge_orderT) r9 �marg r! r r� r r- r: Zmaskedr )rR r� r �x2r? r? r@ � test_masked s zTestGradient.test_maskedc C s t �ddd�}|d |d }d|d d|d d| }d|d d| d }t �t j||dd �| d �}tt �|d k �d k� t j�d� t �t j�d��}d|d d|d d| }d|d d| d }t �t j||dd �| d �}tt �|d k �d k� d S ) Nr rE ry rD rH rC rZ rY )r. g� ��Q��?T) r9 �linspace�absr! r rw �randomZseed�sort)rR r� r# r� Z analyticalZ num_errorr? r? r@ �test_second_order_accurate& s z'TestGradient.test_second_order_accuratec C s t �ddddddg�}t �|d�|�dd� }t �d d d ddd g�}t �d �}t �ddd d d d gd�}t �ddd d d dgd�}t �dddd dd gd�}t �dddd ddgd�}x�d|fd|fgD ]�\}} t|d d|d�} t|||d|d�} t|||d |d�} t| | � t| | � t| d | j� t| d | � t|d d|d�} t||d|d�} t | j | j k� t| | j� t|d d|d�} t||d|d�} t | j | j k� t| | � q�W x�d|fd|fgD ]�\}} t|||d|d�} t|||d |d�} t| | � t| d | j� t| d | � t||d|d�} t| | j� t||d|d�} t| | � �q�W t|||ddd�} t|||ddd�} t| d | d � t| d | d � t| d |j� t| d |� t|||ddd�} t|||ddd�} t| d | d � t| d | d � t| d |j� t| d |� d S )Nr g @g @g @g @)rZ rE rM rE g g �?g �?g @g @g �?g @g �g333333�?g �?g пrD )r rE )rb r. )rE r ) r9 rg Ztiler\ r: r! r r r� r rW ) rR �fZx_unevenZx_evenZ fdx_even_ord1Z fdx_even_ord2Zfdx_uneven_ord1Zfdx_uneven_ord2r. Zexp_res�res1�res2�res3r? r? r@ � test_spacing: sd zTestGradient.test_spacingc C s@ ddgddgg}t �|�}t �ddgddgg�t �ddgddgg�g}tt|dd �|d � tt|dd �|d � tt|d d �|d � tt|d d �|d |d g� tt|d d �|d |d g� tt|d d �t|�� tt|d dd d �|d d |d d g� ttt|dd dd � tt jt|dd � tt jt|d d � d S )NrE rH rC g @g @g g �?r )rb rM )rE r rD rJ )r9 rg r r! r r r( r� )rR r" r� r# r? r? r@ �test_specific_axesz s z TestGradient.test_specific_axesc C s^ t jdddddddgdd �}t jd d d d d ddgdd �}tt|�|� t|jt �d�k� d S )Nr� rJ ry r� r+ iA i, ztimedelta64[D])r� rD r[ r$ � �w i��)r9 rg r r! r r� )rR r� r# r? r? r@ �test_timedelta64� s z TestGradient.test_timedelta64c C s� t t�d�dd� t t�d�dd� ttt t�d�dd� ttt t�d�dd� ttt t�d�dd� ttt t�d�dd� ttt t�d�dd� d S )NrD rE )r. rH r )r! r9 r: r rO )rR r? r? r@ � test_values� s zTestGradient.test_valuesN) r^ r_ r` rT r' r) r, r1 r6 r; r< r? r@ r? r? r? r@ r! � s @ r! c @ s e Zd Zdd� ZdS )� TestAnglec C s� dt �d�d dt �d� d dddddd g}t|�}t �d �t �d �d t jd t jt j d t �d � t jt �d � g}t|dd �}t �|�d t j }t||d� t||d� d S )Ny �? @rD g @y �?rE rM y � �y �? �y � @g @g �?r )Zdegr� r� )r9 �sqrtr Zarctan�pirg r )rR r� r� Zyo�zZzor? r? r@ rT � s zTestAngle.test_basicN)r^ r_ r` rT r? r? r? r@ rA � s rA c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � TestTrimZerosz+ Only testing for integer splits. c C s< t �dddddddg�}t|�}t|t �ddddg�� d S )Nr rE rD rH rC )r9 rg r4 r )rR rS �resr? r? r@ rT � s zTestTrimZeros.test_basicc C s@ t �ddddddddg�}t|�}t|t �dddddg�� d S )Nr rE rD rH rC )r9 rg r4 r )rR rS rF r? r? r@ �test_leading_skip� s z TestTrimZeros.test_leading_skipc C sD t �dddddddddg �}t|�}t|t �ddddddg�� d S )Nr rE rD rH rC )r9 rg r4 r )rR rS rF r? r? r@ �test_trailing_skip� s z TestTrimZeros.test_trailing_skipN)r^ r_ r` �__doc__rT rG rH r? r? r? r@ rE � s rE c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestExtinsc C s> t �dddddddg�}t|dk|�}t|dddddg� d S )NrE rH rD )r9 rg r r )rR rS rd r? r? r@ rT � s zTestExtins.test_basicc s t ttdddgddgddg� t�dddddd d g�� t� dddddddgddd g� t� dddddd d g� t� t�d �g � t� t�dd �� t� dddddddgd d g� t� d dd dd d d g� tt d � fdd�� t�ddg�� t� ddgd� t� ddg� d S )NrE rD rH TFr rC rI rY r[ rZ r� z!Cannot insert from an empty arrayc s t � dddddddgg �S )Nr rE )r- r? )rS r? r@ �<lambda>� � z'TestExtins.test_place.<locals>.<lambda>Z12Z34�9) r r( r- r9 rg r �zerosr: r rO )rR r? )rS r@ � test_place� s zTestExtins.test_placec C sH t d�}|dk}|�� }t||�}t||d� t|||� t||� d S )Nry g �?r )r rz r r- r )rR rS r- Zacr� r? r? r@ � test_both� s zTestExtins.test_bothN)r^ r_ r` rT rO rP r? r? r? r@ rJ � s rJ c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;S )<� TestVectorizec C s@ dd� }t |�}|ddddgdddd g�}t|dddd g� d S ) Nc S s | |kr| | S | | S d S )Nr? )rS rd r? r? r@ � addsubtract� s z.TestVectorize.test_simple.<locals>.addsubtractr rH rZ r� rE rI r[ rD )r7 r )rR rR r7 r� r? r? r@ � test_simple� s zTestVectorize.test_simplec C s8 dd� }t |�}|ddddgd�}t|ddd d g� d S ) Nc S s | |kr| | S | | S d S )Nr? )rS rd r? r? r@ rR s z.TestVectorize.test_scalar.<locals>.addsubtractr rH rZ r� rI rY rE rC )r7 r )rR rR r7 r� r? r? r@ � test_scalar s zTestVectorize.test_scalarc C s0 t �ddd�}tdd� �}||�}t||� d S )NrJ rD i' c S s | S )Nr? )r� r? r? r@ rK rL z*TestVectorize.test_large.<locals>.<lambda>)r9 r2 r7 r )rR r� r7 r� r? r? r@ � test_large s zTestVectorize.test_largec C sZ dd l }t|j�}t�ddtj tjdtj dtj g�}||�}t�|�}t||� d S )Nr g �?g �?rD )�mathr7 Zcosr9 rg rC r )rR rV r7 �args�r1�r2r? r? r@ � test_ufunc s ( zTestVectorize.test_ufuncc C sl ddd�}t |�}t�dddg�}||�}t�dddg�}t||� ||d�}t�dddg�}t||� d S ) NrE c S s | | S )Nr? )rS rd r? r? r@ r� " s z(TestVectorize.test_keywords.<locals>.foorD rH rC rI )rE )r7 r9 rg r )rR r� r7 rW rX rY r? r? r@ � test_keywords s zTestVectorize.test_keywordsc C s. dd l }yt|j� W n t� �Y nX d S )Nr )r4 r7 Z randrange�AssertionError)rR r4 r? r? r@ �test_keywords_no_func_code. s z(TestVectorize.test_keywords_no_func_codec C s� d dd�}t |�}t�dddg�}||d�}t�dddg�}t||� |d|d�}t||� ||dd �}t�ddd g�}t||� d S ) NrE c S s | | S )Nr? )rS rd r? r? r@ r� ; s z5TestVectorize.test_keywords2_ticket_2100.<locals>.foorD rH )rS rC )rd rS )rd rI )rE )r7 r9 rg r )rR r� r7 rW rX rY r? r? r@ �test_keywords2_ticket_21008 s z(TestVectorize.test_keywords2_ticket_2100c C sx dd� }t j|ddgd�}ddg}t||ddgdd dgd �� t||ddgdd dgd �� t||ddgdd dg�� d S ) Nc S s2 t |�}|�d�}x|r,|| |�d� }qW |S )Nr )�list�pop)r� �pZ_prF r? r? r@ � mypolyvalK s z;TestVectorize.test_keywords3_ticket_2100.<locals>.mypolyvalra rE )�excludedrH rZ r rD )r� ra )ra )r9 r7 r )rR rb ZvpolyvalZansr? r? r@ �test_keywords3_ticket_2100I s z(TestVectorize.test_keywords3_ticket_2100c C s. t dd� �}t|ddgddgd�ddg� d S ) Nc [ s" d}x| D ]}|| | 9 }q W |S )Ng �?r? )�kwrF Z_kr? r? r@ r7 Z s z3TestVectorize.test_keywords4_ticket_2100.<locals>.frE rD rH rC )rS rd rY )r7 r )rR r7 r? r? r@ �test_keywords4_ticket_2100X s z(TestVectorize.test_keywords4_ticket_2100c C s, t dd� �}t|ddgddg�ddg� d S )Nc W s t �| �S )N)r9 r )r" r? r? r@ r7 e s z3TestVectorize.test_keywords5_ticket_2100.<locals>.frE rD rH rC rY )r7 r )rR r7 r? r? r@ �test_keywords5_ticket_2100c s z(TestVectorize.test_keywords5_ticket_2100c C s dd� }t |�}t|� d� d S )Nc S s dS )NrE r? r? r? r? r@ r� l s z5TestVectorize.test_coverage1_ticket_2100.<locals>.foorE )r7 r )rR r� r7 r? r? r@ �test_coverage1_ticket_2100k s z(TestVectorize.test_coverage1_ticket_2100c C s> dd� }t |�}t|j|j� d}t ||d�}t|j|� d S )Nc S s | S )zOriginal documentationr? )r� r? r? r@ r� s s z3TestVectorize.test_assigning_docstring.<locals>.foozProvided documentation)�doc)r7 r rI )rR r� r7 ri r? r? r@ �test_assigning_docstringr s z&TestVectorize.test_assigning_docstringc C s` G dd� d�}t t|� j�t�d��t�d�d � t t|j�|� t�d��t�d�d � d S )Nc @ s e Zd ZdZdd� ZdS )z9TestVectorize.test_UnboundMethod_ticket_1156.<locals>.FoorD c S s || j S )N)rd )rR rS r? r? r@ r� � s z=TestVectorize.test_UnboundMethod_ticket_1156.<locals>.Foo.barN)r^ r_ r` rd r� r? r? r? r@ �Foo� s rk r� rD )r r7 r� r9 r: )rR rk r? r? r@ � test_UnboundMethod_ticket_1156~ s z,TestVectorize.test_UnboundMethod_ticket_1156c C sl t dd� �}|t�d��}|t�dd��}t dd� �}|t�dd��}|t�d��}t||� t||� d S )Nc S s | S )Nr? )r� r? r? r@ rK � rL z@TestVectorize.test_execution_order_ticket_1487.<locals>.<lambda>rH g�������?c S s | S )Nr? )r� r? r? r@ rK � rL )r7 r9 r: r )rR �f1Zres1aZres1b�f2Zres2bZres2ar? r? r@ � test_execution_order_ticket_1487� s z.TestVectorize.test_execution_order_ticket_1487c C s$ t �dd� �}d}t|||�� d S )Nc S s | S )Nr? )r� r? r? r@ rK � rL z7TestVectorize.test_string_ticket_1892.<locals>.<lambda>Zd0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r9 r7 r )rR r7 r r? r? r@ �test_string_ticket_1892� s z%TestVectorize.test_string_ticket_1892c sN dg� t � fdd��}d|_t�d�}t||�|| � t� d t|�� d S )Nr c s � d d7 < | d S )Nr rE rD r? )r� )�_callsr? r@ r7 � s z#TestVectorize.test_cache.<locals>.fTrI )r7 �cacher9 r: r r r� )rR r7 r� r? )rq r@ � test_cache� s zTestVectorize.test_cachec C s0 t �dd� �}d|_t �d�}t||�|� d S )Nc S s | S )Nr? )r� r? r? r@ rK � rL z+TestVectorize.test_otypes.<locals>.<lambda>rn rI )r9 r7 �otypesr: r )rR r7 r� r? r? r@ � test_otypes� s zTestVectorize.test_otypesc C s� t t�d�dgdgf� t t�d�dgdgf� t t�d�ddgdgf� t t�d�dgdgf� t t�d �dgddgf� t t�d �dd d gd gf� tt�� t�d� W d Q R X tt�� t�d� W d Q R X tt�� t�d� W d Q R X d S )Nz(x)->())r� r? z (x,y)->())r� r� z (x),(y)->())r� z(x)->(y)z (x)->(y),()z(),(a,b,c),(d)->(d,e))rS rd r� )r� )r� �ez (x)(y)->()z (x),(y)->z ((x))->(x))r �nfbZ_parse_gufunc_signaturer rO )rR r? r? r@ �test_parse_gufunc_signature� s" z)TestVectorize.test_parse_gufunc_signaturec C sD dd� }t |dd�}|ddddgd dd d g�}t|d dd d g� d S ) Nc S s | |kr| | S | | S d S )Nr? )rS rd r? r? r@ rR � s z8TestVectorize.test_signature_simple.<locals>.addsubtractz (),()->())� signaturer rH rZ r� rE rI r[ rD )r7 r )rR rR r7 r� r? r? r@ �test_signature_simple� s z#TestVectorize.test_signature_simplec C s: dd� }t |dd�}|ddgddgg�}t|ddg� d S ) Nc S s | � � S )N)r� )rS r? r? r@ r� � s z4TestVectorize.test_signature_mean_last.<locals>.meanz(n)->())ry rE rH rD rC )r7 r )rR r� r7 r� r? r? r@ �test_signature_mean_last� s z&TestVectorize.test_signature_mean_lastc C sB dd� }t |dd�}|ddgddgg�}t|d dgd dgg� d S ) Nc S s | | � � S )N)r� )rS r? r? r@ �center� s z3TestVectorize.test_signature_center.<locals>.centerz(n)->(n))ry rE rH rD rC rM )r7 r )rR r| r7 r� r? r? r@ �test_signature_center� s z#TestVectorize.test_signature_centerc C sd t dd� dd�}|dddg�}tt|t�o4t|�dk� t|d dddg� t|d dddg� d S ) Nc S s | | fS )Nr? )r� r? r? r@ rK � rL z:TestVectorize.test_signature_two_outputs.<locals>.<lambda>z ()->(),())ry rE rD rH r )r7 r r� �tupler� r )rR r7 r� r? r? r@ �test_signature_two_outputs� s z(TestVectorize.test_signature_two_outputsc C s t tjdd�}|ddgdddg�}t|dddgdddgg� |ddgggdddg�}t|dddgdddgggg� |ddgddggdddg�}t|dddgdddggdddgdddggg� |ddgdddgdddgg�}t|dddgdddggdddgdddggg� d S ) Nz(a),(b)->(a,b))ry rE rD rH rC rZ r )r7 r9 r< r )rR r7 r� r? r? r@ �test_signature_outer� s z"TestVectorize.test_signature_outerc C s^ t dd� dd�}|dddg�}t|ddg� |dddgdddgg�}t|ddgddgg� d S ) Nc S s | d d� S )NrM r? )r� r? r? r@ rK � rL z<TestVectorize.test_signature_computed_size.<locals>.<lambda>z(n)->(m))ry rE rD rH rC )r7 r )rR r7 r� r? r? r@ � test_signature_computed_size� s z*TestVectorize.test_signature_computed_sizec C sV d dd�}t |ddhd�}t|dddg�ddd g� t|dddgd d �dddg� d S ) NrE c S s | | S )Nr? )rS rd r? r? r@ r� � s z2TestVectorize.test_signature_excluded.<locals>.fooz()->()rd )ry rc rD rH rC r )rd )rE )r7 r )rR r� r7 r? r? r@ �test_signature_excluded� s z%TestVectorize.test_signature_excludedc C sH t dd� ddgd�}|dddg�}t|jt�d�� t|dddg� d S ) Nc S s | S )Nr? )r� r? r? r@ rK rL z5TestVectorize.test_signature_otypes.<locals>.<lambda>z(n)->(n)r� )ry rt rE rD rH )r7 r r� r9 r )rR r7 r� r? r? r@ �test_signature_otypes s z#TestVectorize.test_signature_otypesc C s� t tjdd�}ttd�� |ddg� W d Q R X ttd�� |dd� W d Q R X ttd�� |ddgdddg� W d Q R X t tjd d�}ttd�� |dd� W d Q R X d S ) Nz (n),(n)->(n))ry zwrong number of positionalrE rD z does not have enough dimensionsz$inconsistent size for core dimensionrH z()->())r7 �operatorr; r r( rO )rR r7 r? r? r@ � test_signature_invalid_inputs s z+TestVectorize.test_signature_invalid_inputsc C s� t dd� dd�}ttd�� |dddg� W d Q R X t d d� d d�}ttd �� |d� W d Q R X t d d� d d�}ttd �� |ddg� W d Q R X d S )Nc S s | d d� S )NrM r? )r� r? r? r@ rK rL z>TestVectorize.test_signature_invalid_outputs.<locals>.<lambda>z(n)->(n))ry z$inconsistent size for core dimensionrE rD rH c S s | S )Nr? )r� r? r? r@ rK " rL z ()->(),()zwrong number of outputsc S s | | fS )Nr? )r� r? r? r@ rK & rL z()->())r7 r rO )rR r7 r? r? r@ � test_signature_invalid_outputs s z,TestVectorize.test_signature_invalid_outputsc C s t �dd� �}t jddgtd�}ttd�� ||� W d Q R X d|_t||�|� t jdd� d d �}ttd�� ||� W d Q R X t jd d� d dd �}t||�|� t jd d� ddd �}t||�|� t jdd� dd �}t||j�|j� t jdd� ddd �}ttd�� ||� W d Q R X d S )Nc S s | S )Nr? )r� r? r? r@ rK , rL z5TestVectorize.test_size_zero_output.<locals>.<lambda>r rI )r� rt rn c S s | S )Nr? )r� r? r? r@ rK 4 rL z()->())ry c S s | S )Nr? )r� r? r? r@ rK 8 rL )ry rt c S s | S )Nr? )r� r? r? r@ rK ; rL z(n)->(n)c S s | S )Nr? )r� r? r? r@ rK > rL c S s | gS )Nr? )r� r? r? r@ rK A rL z()->(n)znew output dimensions) r9 r7 rN r r rO rt r r� )rR r7 r� r? r? r@ �test_size_zero_output* s$ z#TestVectorize.test_size_zero_outputN) r^ r_ r` rS rT rU rZ r[ r] r^ rd rf rg rh rj rl ro rp rs ru rx rz r{ r} r r� r� r� r� r� r� r� r? r? r? r@ rQ � s: rQ c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� TestDigitizec C s2 t �dd�}t �dd�}tt||�t �d�� d S )Ni����rI r� r� )r9 r: r r )rR r� �binsr? r? r@ � test_forwardH s zTestDigitize.test_forwardc C s6 t �ddd�}t �ddd�}tt||�t �d�� d S )NrI i����rM r� r� )r9 r: r r )rR r� r� r? r? r@ � test_reverseM s zTestDigitize.test_reversec C s: t d�}t�|�� |�� d�}tt�t||�dk�� d S )Nry r )r r9 r2 �min�maxr rw r )rR r� �binr? r? r@ � test_randomR s zTestDigitize.test_randomc C sf dddddddg}dddg}dddd dd dg}t t||�|� dddddd dg}t t||d �|� d S ) NrE rI rC ry rY r� r rD rH T)r r )rR r� r� Zdefault_answerZ right_answerr? r? r@ �test_right_basicW s z TestDigitize.test_right_basicc C s4 t �dd�}t �dd�}tt||d�t �d�� d S )Ni����rI rC Tr� )r9 r: r r )rR r� r� r? r? r@ �test_right_open_ s z TestDigitize.test_right_openc C s8 t �ddd�}t �ddd�}tt||d�t �d�� d S )NrI i����rM rC Tr� )r9 r: r r )rR r� r� r? r? r@ �test_right_open_reversed s z$TestDigitize.test_right_open_reversec C s< t d�}t�|�� |�� d�}tt�t||d�dk�� d S )Nry T)r r9 r2 r� r� r rw r )rR r� r� r? r? r@ �test_right_open_randomi s z#TestDigitize.test_right_open_randomc C s ddddg}dddg}t t||d�ddddg� t t||d�ddddg� dddg}t t||d�ddddg� t t||d�ddddg� ddddg}t t||d�ddddg� t t||d�ddddg� ddddg}ttt||� ddddg}ttt||� d S ) NrM r rE rD FrH TrC )r r r rO )rR r� r� r? r? r@ �test_monotonicn s zTestDigitize.test_monotonicc C s> dddg}dddg}t tt||� || }}t tt||� d S )NrE rD y @ �?rH )r r( r )rR r� r� r? r? r@ �test_casting_error~ s z TestDigitize.test_casting_errorc C sh G dd� dt j�}t �d��|�}t �dd��|�}ttt||d�|� � ttt||d�|� � d S )Nc @ s e Zd ZdS )z(TestDigitize.test_return_type.<locals>.AN)r^ r_ r` r? r? r? r@ �A� s r� rI rE rH FT)r9 r� r: r� r r� r )rR r� rS rd r? r? r@ �test_return_type� s z TestDigitize.test_return_typeN) r^ r_ r` r� r� r� r� r� r� r� r� r� r� r? r? r? r@ r� F s r� c @ s e Zd Zdd� ZdS )� TestUnwrapc C sH t tdddtj g�ddg� tt�tttd�d ��tjk �� d S )NrE rD ry r� )r r5 r9 rC r rw r r )rR r? r? r@ rS � s zTestUnwrap.test_simpleN)r^ r_ r` rS r? r? r? r@ r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFilterwindowsc C s2 t d�}t|t|�d� ttj|dd�dd� d S )Nry r[ r )rb g @rC )r# r r r r9 r� )rR r� r? r? r@ � test_hanning� s z TestFilterwindows.test_hanningc C s2 t d�}t|t|�d� ttj|dd�dd� d S )Nry r[ r )rb g�(\��@rC )r" r r r r9 r� )rR r� r? r? r@ � test_hamming� s z TestFilterwindows.test_hammingc C s2 t d�}t|t|�d� ttj|dd�dd� d S )Nry r[ r )rb g�^)��@rC )r r r r r9 r� )rR r� r? r? r@ � test_bartlett� s z TestFilterwindows.test_bartlettc C s2 t d�}t|t|�d� ttj|dd�dd� d S )Nry r[ r )rb g= ףp=@rC )r r r r r9 r� )rR r� r? r? r@ � test_blackman� s z TestFilterwindows.test_blackmanN)r^ r_ r` r� r� r� r� r? r? r? r@ r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestTrapzc C sH t �ddd�}tt �d|d �t �dt j � dd�}t|dd� d S ) Ni����ry g�������?g �rD )r# rE r[ )r9 r: r3 ZexprB rC r )rR r� r� r? r? r@ rS � s *zTestTrapz.test_simplec C s& t �ddd�}t �ddd�}t �ddd�}t �|�|d |d }|d d < |d d < t �|�|d |d }|d d < |d d < t �|�|d |d }|d d < |d d < |d d �d d f |d d d �d f |d d d d �f }||d d �d d f jdd�}||d d d �d f jdd�} ||d d d d �f jdd�} t||d d �d d f dd �} t| |� t||d d d �d f dd �} t| | � t||d d d d �f dd �} t| | � t||dd �} t| |� t||dd �} t| | � t||dd �} t| | � d S ) Nr rE rH rD rY rK rM )rb )r� rb )r9 r2 Z ones_liker� r3 r ) rR r� r� rD ZwxZwyZwz�qZqxZqyZqzr� r? r? r@ � test_ndim� s8 6 zTestTrapz.test_ndimc C s� t �d�}|| }|dk}t jj||d�}d}tt||�|� t jj||d�}tt||�|� t jj||d�}tt||�|� d S )NrI rD )r- g *@)r9 r: r/ rg r r3 )rR r� r� r- Zymr� Zxmr? r? r@ r1 � s zTestTrapz.test_maskedc C sJ t �dd�}|| }t||�}t �|�}t �|�}t||�}t||� d S )Nr rI )r9 r2 r3 r� r )rR r� r� r� ZmxZmyZmrr? r? r@ � test_matrix� s zTestTrapz.test_matrixN)r^ r_ r` rS r� r1 r� r? r? r? r@ r� � s %r� c @ s e Zd Zdd� Zdd� ZdS )�TestSincc C s6 t td�dk� tt�ddd��}t|t|�d� d S )Nr rE rM r� r[ )r r1 r9 r2 r r )rR r� r? r? r@ rS s zTestSinc.test_simplec C sF ddg}t t�|��}t t|��}t t|��}t||� t||� d S )Nr g �?)r1 r9 rg r_ r~ r )rR r� rr rs rt r? r? r@ �test_array_like s zTestSinc.test_array_likeN)r^ r_ r` rS r� r? r? r? r@ r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&� TestHistogramc C s d S )Nr? )rR r? r? r@ r s zTestHistogram.setUpc C s d S )Nr? )rR r? r? r@ �tearDown s zTestHistogram.tearDownc C sP d}t |�}t|�\}}ttj|dd�|� tt�ddd��\}}t|d� d S )Nr� r )rb ry )r r$ r r9 r� r2 r )rR r= r" rS rd r? r? r@ rS s zTestHistogram.test_simplec C s� t ddddgddg�\}}t|dg� t|ddg� ttt ddgdd� t ddgdd�\}}t|t�dg�� t|t�ddg�� d S ) NrE rD rH rC r )r� g �?g @)r$ r r rO r r9 rg r )rR �hist�edges�hrv r? r? r@ � test_one_bin# s zTestHistogram.test_one_binc C s� d}t |�}t|dd�\}}t�|t|� �}t|d� t�d�}dddddg}t||dd�\}}t�|t|� �}t|d� d S ) Nr� T)�normedrE ry r rI r� )r r$ r9 r� r r r: )rR r= r" rS rd �arear� r? r? r@ � test_normed- s zTestHistogram.test_normedc C s� d}t |�}t|dd�\}}t�|t|� �}t|d� t�d�}dddddg}t||dd�\}}t|d � tt�|t|� �d� t�d�}ddddtj g}t||dd�\}}t|d d d d g� tjdd dd gd dtj gdd�\}}t|ddg� d S )Nr� T)�densityrE ry r rH rZ g�������?g rD rC g �?g �?g �?) r r$ r9 r� r r r: r r �inf) rR r= r" rS rd r� r� ZcountsZdmyr? r? r@ � test_density= s" zTestHistogram.test_densityc C s� t �d�d }t|ddgd�\}}t|�� d� t|ddgd�\}}t|�� d� t|ddgdd�\}}t|t|� �� dd d � t �d�d }t|ddg|dd �\}}t|t|� �� d� t|d ddg|d �\}}t||dd� � d S )Nry g �?r r� )rQ rE T)rQ r� r� )r� )rQ r� r� rY )r� rQ r� rM )r9 r: r$ r r� r r )rR rS r� rd r� r? r? r@ � test_outliersY s zTestHistogram.test_outliersc C s� t �d�d }t|�\}}tt �|jt�� t|dd�\}}tt �|jt�� t|t �dt�d�\}}tt �|jt�� t|t �dt�d�\}}tt �|jt�� d S )Nry g �?T)r� )r� ) r9 r: r$ r Z issubdtyper� r �floatrP )rR rS r� rd r? r? r@ � test_typeq s zTestHistogram.test_typec C sT t jdddgt jd�}t jdddgt jd�}t j||dd �\}}}t|�� d � d S ) Ng2V��Eq@g��x>fQ�g-��VT5@)r� g!����@g���S��@g���^z�@r� )r� g @)r9 rg r� Z histogram2dr r� )rR r� r� Z counts_histZxedgesZyedgesr? r? r@ �test_f32_rounding� s z TestHistogram.test_f32_roundingc C s~ t d�}t�d�d }t|�\}}t|dd�\}}t||d�\}}t||dd�\} } t|d |� t|| � t�ddd�}t�t�d�t�d�f�}t|t�d �|d �\}}t||� td d d d gd d dd d gd �\}}t |d ddd g� td d d d gd d dd d gdd�\}}t|t� d ddd g�d d d � tt�d�dd dddgd d d d d d d d d g dd�\}}t |ddddg� d S )Nr� rI T)r� )r� )r� r� r ry r� )r� r� rE rD rC rH )r� r� r� g $@g @r� rZ )r� r� g�������?g�������?g333333�?) r r9 rP r$ r r2 r� rN r: r rg r ) rR r" r� rS rd ZnaZnb�wa�wbZnwaZnwbr? r? r@ r� � s* ""$ zTestHistogram.test_weightsc C sH t �dddg�}t �dddg�dt �dddg� }t|ddd g|d �\}}t|t �ddg�dt �dd g� � t|ddd g|d �\}}t|t �ddg�dt �dd g� � dd lm} t �dddg�}t �|d�|d�|d �g�}t|ddd g|d �\}}t||d�|d �g� t|ddd g|d �\}}t||d�|d �g� d S )Ng�������?g @gffffff@rE rM rD y �?r rH )r� r� )r� rQ r� )r� rI )r9 rg r$ r r� r� )rR r� r� r� r� r� r? r? r@ �test_exotic_weights� s $&& z!TestHistogram.test_exotic_weightsc C s8 t �dddg�}t j|ddgdd� t|dddg� d S )Ng�������?g @gffffff@i����ry r� )rQ r� )r9 rg r$ r )rR r� r? r? r@ �test_no_side_effects� s z"TestHistogram.test_no_side_effectsc C s> t g ddgd�\}}t|t�dg�� t|t�ddg�� d S )Nr rE )r� )r$ r r9 rg )rR rS rd r? r? r@ � test_empty� s zTestHistogram.test_emptyc C s, t jdddd�}t|d� ttt|d� d S )Ng g �?r� )�numrI g333333@)r9 r2 r$ r r( )rR �valsr? r? r@ �test_error_binnum_type� s z$TestHistogram.test_error_binnum_typec C sP t jdddd�}t|ddgd� ttt|t jdgd� ttt|dt jgd� d S )Ng g �?r� )r� g �?g �?)rQ )r9 r2 r$ r rO r� r� )rR r� r? r? r@ �test_finite_range� s z TestHistogram.test_finite_rangec C s� t �dddddddg�}t j|dd d �\}}|d k}|d d � | }|d d � | }x2t|||�D ]"\}}} | �||� | �|| � qbW d S )NiQ i� i� i& i� i i� ih )rD i� )r� rQ r rM rE )r9 rg r$ r� ZassertGreaterEqualZ assertLess) rR �arrr� r� r- Z left_edgesZ right_edgesr� �left�rightr? r? r@ �test_bin_edge_cases� s z!TestHistogram.test_bin_edge_casesc C sD t �dddddddddg �}t j|ddd �\}}| �|d d � d S ) Ng g �?g @g @g @g @r� )g �rI )r� rQ rM rE )r9 rg r$ � assertEqual)rR r� r� r� r? r? r@ � test_last_bin_inclusive_range� s z+TestHistogram.test_last_bin_inclusive_rangec C sJ t �dg�}t jdddgdd�}tt�� t j||d�\}}W d Q R X d S )NrD rE rH Zuint64)r� )r� )r9 rg r rO r$ )rR r� r� r� r� r? r? r@ � test_unsigned_monotonicity_check� s z.TestHistogram.test_unsigned_monotonicity_checkN)r^ r_ r` r r� rS r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r? r? r? r@ r� s$ r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�TestHistogramOptimBinNumsz] Provide test coverage when using provided estimators for optimal number of bins c C sZ dddddddg}xB|D ]:}t g |d�\}}t|t�d g�� t|t�d d g�� qW d S ) N�fd�scott�rice�sturges�doanerB �auto)r� r rE )r$ r r9 rg )rR �estimator_list� estimatorrS rd r? r? r@ r� � s z$TestHistogramOptimBinNums.test_emptyc C s� dddddddd�dddddddd�d d d d d d d d�d �}x�|� � D ]�\}}t�dd|d d �}t�dd|d d �}t�||f�}x>|� � D ]2\}}t�||�\} } tt| �|d�||�d� q�W qFW dS )z� Straightforward testing with a mixture of linspace data (for consistency). All test values have been precomputed and the values shouldn't change rC rY r[ )r� r� r� r� r� rB r� r� ry r� � r� r� r� �G )rU i� i� i����rM rI rD rE rH z*For the {0} estimator with datasize of {1})r N)�itemsr9 r2 r� r$ r r� �format) rR � basic_test�testlen�expectedResults�x1r0 r� r� �numbinsrS rd r? r? r@ rS s z%TestHistogramOptimBinNums.test_simplec C s� ddddddd�ddddddd�ddddddd�d�}x^|� � D ]R\}}t�|�}x>|� � D ]2\}}t�||�\}}tt|�|d�||�d� q\W q@W dS ) z� Smaller datasets have the potential to cause issues with the data adaptive methods, especially the FD method. All bin numbers have been precalculated. rE )r� r� r� r� r� rB rD rH )rE rD rH z*For the {0} estimator with datasize of {1})r N)r� r9 r: r$ r r� r� ) rR Z small_datr� r� Ztestdatr� ZexpbinsrS rd r? r? r@ � test_small s z$TestHistogramOptimBinNums.test_smallc C s2 ddddg}x |D ]}t ttdddg|� qW dS ) zS Check a Value Error is thrown when an unknown string is passed in ZmadZfreemanZ histogramsZIQRrE rD rH N)r rO r$ )rR Z check_listr� r? r? r@ �test_incorrect_methods1 s z0TestHistogramOptimBinNums.test_incorrect_methodsc C s` t �d�}dddddddd�}x<|�� D ]0\}}t �||�\}}tt|�|d�|�d� q(W dS )z� Check that methods handle no variance in data Primarily for Scott and FD as the SD and IQR are both 0 in this case r� rE )r� r� r� r� r� rB r� z {0} estimator, No Variance test)r N)r9 rP r� r$ r r� r� )rR Z novar_datasetZnovar_resultdictr� r� rS rd r? r? r@ �test_novariance9 s z)TestHistogramOptimBinNums.test_novariancec C sj t �ddd�}t �t �ddd�|f�}dddd �}x2|�� D ]&\}}t �||�\}}tt|�|� q<W d S ) a� Check the FD, Scott and Doane with outliers. The FD estimates a smaller binwidth since it's less affected by outliers. Since the range is so (artificially) large, this means more bins, most of which will be empty, but the data of interest usually is unaffected. The Scott estimator is more affected and returns fewer bins, despite most of the variance being in one area of the data. The Doane estimator lies somewhere between the other two. i����ry rU i����i����rI r r� )r� r� r� N)r9 r2 �hstackr� r$ r r� )rR ZxcenterZoutlier_datasetZoutlier_resultdictr� r� rS rd r? r? r@ � test_outlierG s z&TestHistogramOptimBinNums.test_outlierc C s� dddddd�dddddd�ddd d dd�d �}x�|� � D ]�\}}t�d d |d d �}t�dd|d d �}t�dd|�}t�|||f�}xR|� � D ]F\}} tj||dd�\} } d�|�} | d�|�7 } tt| �| | d� q�W q:W dS )z� Straightforward testing with a mixture of linspace data (for consistency). Adding in a 3rd mixture that will then be completely ignored. All test values have been precomputed and the shouldn't change. rY r� r� )r� r� r� r� r� r� � r �! �E r� )rU i� i� i����rM rI rD rE ry rH i����i���)i��r )rQ zFor the {0} estimatorz with datasize of {0})r N)r� r9 r2 r� r$ r� r r� ) rR r� r� r� r� r0 �x3r� r� r� rS rd r r? r? r@ �test_simple_range[ s z+TestHistogramOptimBinNums.test_simple_rangec C s> dddddg}x*|D ]"}t ttdddg|dddgd � qW d S ) z= Check that weighted data raises a TypeError r� r� r� r� r� rE rD rH )r� N)r r( r$ )rR r� r� r? r? r@ �test_simple_weightedz s z.TestHistogramOptimBinNums.test_simple_weightedN) r^ r_ r` rI r� rS 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 d� Z dd� Z dd� Z dd� Z dS )�TestHistogramddc C sr t �dddgdddgdddgdddgdddgdddgg�}t|dddgdd gdd ggd �\}}t �dddgdddgdddggdddgdddgdddggg�}t||� d dd gddd d gddd d gg}t||d d�\}}tt �||d k�� t|dddgdd gddggd d�\}}t �ddddgddddgddddggddddgddddgddddggg�}t||d d� dd� t|d dd�D �}t|dd d gdd gdd ggd�\}}t �ddgddgddggddgddgddggddgddgddggddgddgddggg�}t||� t �d�}d|t t d��t t d��t t d��f< tt � d�t � d�t � d�gd�\}}t||� d S ) Ng �g �?g �?g @)rD rH rH rM rE r rH )rQ rG rD T)r� r� g (@)rD rH rC rC )rQ r� g @c S s g | ]}t �|��qS r? )r9 Zsqueeze)r� r� r? r? r@ r� � s z/TestHistogramdd.test_simple.<locals>.<listcomp>)rb )rC rH rD )r� rQ )rI rI rI g �?rI ) r9 rg r% r r rw r r2 rN r_ rQ r: )rR r� �Hr� ZanswerZedrD �Zr? r? r@ rS � s8 $&" &&zTestHistogramdd.test_simplec C s< d}t dd�}x(|D ] }t||�\}}t|j|k� qW d S )N))rI rC rZ )rZ rC rI )rI rZ rC )rC rZ rI )rZ rI rC )rC rI rZ ry rH )r r% r rW )rR r� r� rd r� r� r? r? r@ � test_shape_3d� s z TestHistogramdd.test_shape_3dc C s< d}t dd�}x(|D ] }t||�\}}t|j|k� qW d S )N))r[ rC rI rZ )rC rI r[ rZ )rI rZ rC r[ )r[ rZ rI rC )rI r[ rZ rC )rC rZ r[ rI )rZ rI r[ rC )r[ rI rC rZ )r[ rC rZ rI )rZ rC r[ rI )rZ r[ rI rC )rC rZ rI r[ )rC r[ rI rZ )rI rC rZ r[ )rI r[ rC rZ )rZ r[ rC rI )rZ rI rC r[ )rC r[ rZ rI )rC rI rZ r[ )r[ rZ rC rI )rI rC r[ rZ )rI rZ r[ rC )rZ rC rI r[ )r[ rI rZ rC ry rC )r r% r rW )rR r� r� rd r� r� r? r? r@ � test_shape_4d� s z TestHistogramdd.test_shape_4dc C s� t dd�}t|�\}}t|dd�\}}t|t�d�d�\}}t||� t|t�d�d dd�\}}t||� t|t�dt�d d�\}}t|d| � d S )Nr� rD T)r� )r� )r� r� )r r% r9 rP r r )rR r" r� r� Zn_histZw_histr? r? r@ r� � s z TestHistogramdd.test_weightsc C s: t �dt�}t|dd�\}}t|d t �dddg�� d S )N)ry rD rD )r� r g �g g �?)r9 rN r r% r rg )rR r� r� r� r? r? r@ �test_identical_samples� s z&TestHistogramdd.test_identical_samplesc C s` t g g gddgddgfd�\}}t|t�dgg�� tj g g g gdd�\}}t|t�d�� d S )Nr rE )r� g rD )rD rD rD )r% r r9 rg rN )rR rS rd r? r? r@ r� � s zTestHistogramdd.test_emptyc C s� t �d��dd�}ttt j|ddddgd� ttt j|ddddgd� ttt j|ddddddd ggd� ttt j|dddddd d ggd� tt j|dddddd dggd�� d S ) NrY rD rC rM rI )r� rE g�G�z��?rH rJ )r9 r: r\ r rO r% r )rR r� r? r? r@ �test_bins_errors� s z TestHistogramdd.test_bins_errorsc C s� t jdd��� t �d��dd�}t �ddgddgddgg�}t j|dt j ddggd �\}}t||� t j|dt �d dt jg�gd �\}}t||� t j|dt j dt jggd �\}}t||� W d Q R X d S ) N�ignore)ZinvalidrZ rH rD rE r ry )r� rM )r9 �errstater: r\ rg r% r� r )rR r� Zexpectedr� rv r? r? r@ �test_inf_edges� s $ "z TestHistogramdd.test_inf_edgesc C s dg}dddgg}t ||d�\}}t|d dk� t|d dk� dg}dddgg}t ||d�\}}t|d dk� t|d dk� dg}dddgg}t ||d�\}}t|d dk� t|d dk� d g}dddgg}t ||d�\}}t|d dk� t|d dk� d S ) Ng�G�����?g g �?g �?)r� r rE g8� �?gq���h �?)r% r )rR r� r� r� r� r? r? r@ �test_rightmost_binedge� s( z&TestHistogramdd.test_rightmost_binedgec C sv t j�d�}t|ddgddgddggd� ttt|ddgddgdt jggd� ttt|ddgt jdgddggd� d S )N)r� rH g g �?g �?g �?g �?)rQ )r9 r4 r% r rO r� r� )rR r� r? r? r@ r� s z!TestHistogramdd.test_finite_rangeN) r^ r_ r` rS r� r� r� r� r� r� r� r� r� r? r? r? r@ r� � s % r� c @ s e Zd Zdd� ZdS )� TestUniquec C s� t �dddddddddg �}tt �t|�dddddgk�� ttt �dddddg��t �dg�k� dddd ddg}tt �t|�d dddgk�� t �d d d d d g�}tt �t|�d d d d gk�� d S )NrC rH rD rE r ZwidgetZhamr� r� y @ @y �? �?y �? $@ry )r9 rg r rw r6 )rR r� r? r? r@ rS s ( zTestUnique.test_simpleN)r^ r_ r` rS r? r? r? r@ r� s r� c @ s e Zd Zdd� Zdd� ZdS )�TestCheckFinitec C sR dddg}ddt jg}ddt jg}t j�|� ttt jj|� ttt jj|� d S )NrE rD rH )r9 r� r� �lib�asarray_chkfiniter rO )rR rS rd r� r? r? r@ rS s zTestCheckFinite.test_simplec C s2 dddg}t jj|dt jd�}t|jt jk� d S )NrE rD rH r| )r} r� )r9 r� r� r� r r� )rR rS r? r? r@ �test_dtype_order& s z TestCheckFinite.test_dtype_orderN)r^ r_ r` rS r� r? r? r? r@ r� s r� c @ s e Zd Ze�dddgdddgddd gg�Ze�d d d gd ddgdddgg�Ze�dddgdddgdddgg�Ze�ddddddgdddddd gdddd d d gddd dd d!gddd d dd"gdd d d!d"dgg�Zd#d$� Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Zd/d0� Zd1d2� Zd3S )4� TestCorrCoefg����^��?g�.{�<�?g ���$,�?g���4��?gp�)��?g�J��G��?gY�����?g������?g�ވ0v��?gة�� ��?gB� F�U�?g �G)t��?g3�TB��?gL�\���?g��A��?g> ���?gq�F�E`�?g�5� �?g �?gޛ����?g���r@��gp��]�4�?g?�'L�?g��(��8�?g��4V�}�?g��w�{��g,H��5�?gjS�, ��?g������g�)h����?g�5#A���?g��t��o�gB C��,�g@��02v�?c C s. t t�dddgdddg�ddgddgg� d S )Nr rE g �?g �)r r9 r )rR r? r? r@ �test_non_arrayB s zTestCorrCoef.test_non_arrayc C sd t | j�}t|| j� tt�t�|�dk�� t | j| j�}t|| j � tt�t�|�dk�� d S )Ng �?) r r� r r8 r r9 rw r3 �Br9 )rR Ztgt1Ztgt2r? r? r@ rS F s zTestCorrCoef.test_simplec C s� t � ��}t�d� ttt| jdd� |�t� tt| jdd�| j � tt| j| j dd�| j � tt| jdd�| j � tt| j| j dd�| j � W d Q R X d S )Nr� rM )�ddofrH ) r r� r� r r� r r� �filterr r8 r� r9 )rR �supr? r? r@ � test_ddofO s zTestCorrCoef.test_ddofc C sh t � �X}t�d� ttt| j| jdd� ttt| jdd� |�t� t t| jdd�| j � W d Q R X d S )Nr� rE r )Zbias) r r� r� r r� r r� r� r� r r8 )rR r� r? r? r@ � test_bias[ s zTestCorrCoef.test_biasc C s^ t �dddgdddgg�}t|�}t �ddgddgg�}t||� tt �t �|�dk�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rg r r r rw r3 )rR r� rF r� r? r? r@ � test_complexe s zTestCorrCoef.test_complexc C sL t �dddgg�}t �dddgg�}tt �||�t �ddgddgg�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rg r r )rR r� r� r? r? r@ �test_xyl s zTestCorrCoef.test_xyc C s� t jdd��� t �dt� ttt�g ��tj� ttt�g �� dd��t�g �� dd�� ttt�g �� dd��t�tjtjgtjtjgg�� W d Q R X d S )NT)r� r� r rD ) r� r� r� �RuntimeWarningr r r9 rg r� r\ )rR r? r? r@ r� q s zTestCorrCoef.test_emptyc C sh ddgddgg}t jdd�� t|�}W d Q R X t|t �ddgddgg�� tt �t �|�dk�� d S )Ng0��.�++g}Ô%�I�T�raise)rw g �?g �)r9 r� r r rg r rw r3 )rR r� r� r? r? r@ � test_extremez s zTestCorrCoef.test_extremeN)r^ r_ r` r9 rg r� r� r8 r9 r� rS r� r� r� r� r� r� r? r? r? r@ r� - s6 r� c @ sB e Zd Ze�ddgddgddgg�jZe�ddgddgg�Zejdddgdd�Ze�dd dg�Z e�dgdgdgdgdgdgg�jZ e�d d gd d gg�Z ej d ej d �Ze�dddg�Ze�ddgddgg�Ze� d �Ze�dddddg�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*S )+�TestCovr rD rE g �?g �g g @)�ndminrC g�������?g������ٿrH )r� g @gUUUUUU�?gUUUUUU�g&S��:�?g�o_��?g#�~j��?g���ZӼ�?g�Q�|�?c C s t t| j�| j� d S )N)r r r� r8 )rR r? r? r@ rT � s zTestCov.test_basicc C s> t �dddgdddgg�}tt|�t �ddgddgg�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rg r r )rR r� r? r? r@ r� � s zTestCov.test_complexc C sJ t �dddgg�}t �dddgg�}tt||�t �ddgddgg�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rg r r )rR r� r� r? r? r@ r� � s zTestCov.test_xyc C s� t jdd��� t �dt� ttt�g ��tj� ttt�g �� dd��t�g �� dd�� ttt�g �� dd��t�tjtjgtjtjgg�� W d Q R X d S )NT)r� r� r rD ) r� r� r� r� r r r9 rg r� r\ )rR r? r? r@ r� � s zTestCov.test_emptyc C sZ t jdd��D t �dt� tt| jdd�t�tj tj gtj tj gg�� W d Q R X d S )NT)r� r� rI )r� ) r� r� r� r� r r r� r9 rg r� )rR r? r? r@ �test_wrong_ddof� s zTestCov.test_wrong_ddofc C sP t t| j�t| jdd�� t�dddddg�}t t| j|�t| j|dd�� d S )Nr )Zrowvarg+����?g4��@���?g��&S�?g�o_��?g�J�4�?)r r r� r9 rg )rR r� r? r? r@ �test_1D_rowvar� s zTestCov.test_1D_rowvarc C s$ t t| jdd�tj| jdd�� d S )NrE )r� )r r r� r9 �var)rR r? r? r@ �test_1D_variance� s zTestCov.test_1D_variancec C s� t t| j| jd�t| j�� t t| j| jd�| j� t t| j| jd�| j� | jd }t t t| j|d� t j dt j d�}t tt| j|d� t j dt j d�}t tt| j|d� dt j dt j d� }t tt| j|d� d S )N)�fweightsg �?)rD rH )r� rD rM rH )r r r0 � frequencies� x2_repeatsr� r9 �unit_frequenciesr8 r r( r9 rP r � RuntimeErrorrO )rR Znonintr7 r? r? r@ � test_fweights� s zTestCov.test_fweightsc C s� t t| j| jd�| j� t t| jd| j d�t| j| jd�� t t| j| jd�| j� t�d�}t t t| j|d� t�d�}t t t| j|d� dt�d� }t t t| j|d� d S )N)�aweightsg @)rD rH rD g �rH ) r r r� r� r: � unit_weightsr8 r9 rP r r rO )rR r� r? r? r@ � test_aweights� s zTestCov.test_aweightsc C s� t t| j| j| jd�t| j�� t t| j| j| jd�| j� t t| j| j| jd�| j � t t| j| j| j d�| j � t t| j| jd| j d�t| j| j d�� t t| j| j| jd�| j � d S )N)r r g @)r ) r r r0 r r r r� r9 r r8 r� r: )rR r? r? r@ � test_unit_fweights_and_aweights� s$ z'TestCov.test_unit_fweights_and_aweightsN) r^ r_ r` r9 rg r� r� r8 r0 r r r9 rP r r r� r: r r� rT r� r� r� r r r r r r r? r? r? r@ r� � s* $ r� c @ s e Zd Zdd� ZdS )�Test_I0c C s� t td�t�d�� t�ddddg�}t t|�t�ddd d g�� t�d d gd dgddgddgddgg�}t t|�t�ddgddgddgdd gd d gg�� d S ) Ng �?g�ǒ#�?gC��7��?g�m�M�?g��3a�,�?g�$WV?ۏ?g�ֿN^�?gE � �?g閡�1�?g�%�m? �?g>@���v�?g�u�ͥ��?gV�X�˳�?g�ua�&�?gB- �sJ�?g-^ ���?g�_��V�?g �X��.�?gV��aX��?g�j ���?gp����?g~-��@�?g� �D/b�?gݦ�����?g,�Y1ؔ�?g�[!���?g�Q�L��?gW�sU+�?g���@��?g�fx�w�?)r r& r9 rg )rR r� r� r? r? r@ rS � s& zTest_I0.test_simpleN)r^ r_ r` rS r? r? r? r@ r � s r c @ s e Zd Zdd� Zdd� ZdS )� TestKaiserc C s� t t�tdd��� ttdd�t�g �� ttdd�t�ddg�� ttdd�t�dddddg�� ttdd�t�d d dd d g�� d S ) NrE g �?r rD g ��oF�?rI g�vT�=�?gg���?g������?g���Y�J�?)r r9 Zisfiniter) r rg )rR r? r? r@ rS s zTestKaiser.test_simplec C s t dd� d S )NrH rC )r) )rR r? r? r@ � test_int_beta s zTestKaiser.test_int_betaN)r^ r_ r` rS r r? r? r? r@ r � s r c @ s e Zd Zdd� ZdS )� TestMsortc C sR t �dddgdddgddd gg�}tt|�t �ddd gdddgdddgg�� d S ) Ng�\��?gb��C Q�?gה�s��?gw�����?g� �kl��?g�*dta��?g��ӯ��?g6H����?gܘC��?)r9 rg r r+ )rR r� r? r? r@ rS s zTestMsort.test_simpleN)r^ r_ r` rS r? r? r? r@ r s r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestMeshgridc C s� t dddgddddg�\}}t|t�dddgdddgdddgdddgg�� t|t�dddgdddgdddgdddgg�� d S )NrE rD rH rC rI rZ r[ )r* r r9 rg )rR �X�Yr? r? r@ rS s zTestMeshgrid.test_simplec C s. t ddddg�\}t|t�ddddg�� d S )NrE rD rH rC )r* r r9 rg )rR r r? r? r@ �test_single_input+ s z TestMeshgrid.test_single_inputc C s* g }t g t|� � t g t|ddi�� d S )Nrz F)r r* )rR rW r? r? r@ � test_no_input/ s zTestMeshgrid.test_no_inputc C s dddg}ddddg}t ||dd �\}}t|t�ddddgddddgddddgg�� t|t�ddddgddddgddddgg�� d d g}tt ||�d jd k� tt ||dd �d jdk� tt |||�d jdk� tt |||dd �d jdk� ttt ||dd � d S )NrE rD rH rC rI rZ r[ �ij)ZindexingrY r� r )rC rH )rH rC )rC rH rD )rH rC rD Znotvalid)r* r r9 rg r rW r rO )rR r� r� r r rD r? r? r@ � test_indexing4 s zTestMeshgrid.test_indexingc C s\ t dddgddddgdd �\}}t|t�dddgg�� t|t�dgdgdgdgg�� d S ) NrE rD rH rC rI rZ r[ T)�sparse)r* r r9 rg )rR r r r? r? r@ � test_sparseH s zTestMeshgrid.test_sparsec C s$ t ttdddgddddgdd � d S ) NrE rD rH rC rI rZ r[ r )r )r r( r* )rR r? r? r@ �test_invalid_argumentsM s z#TestMeshgrid.test_invalid_argumentsc C s� t jddt jd�}t jddt jd�}t �||�\}}t|j|jk� t|j|jk� t j||dd�\}}t|j|jk� t|j|jk� t j||dd�\}}t|j|jk� t|j|jk� d S )Nr ry )r� r T)rz )r )r9 r: r� r� r* r r� )rR r� r� r r r? r? r@ r� T s z TestMeshgrid.test_return_typec C sr t �ddg�}t �ddg�}t j||ddd�\}}d|dd d �f< t|dd d �f d� t|d d d �f |� d S ) Ng�������?g������@gffffff @g������@FT)r rz r rE )r9 rg r* r )rR r r r� r� r? r? r@ �test_writebackl s zTestMeshgrid.test_writebackN) r^ r_ r` rS r r r 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 )� TestPiecewisec C s� t ddgddgdg�}t|ddg� t ddgddggdg�}t|ddg� t ddgt�ddg�dg�}t|ddg� t ddgt�ddg�dg�}t|ddg� t ddgt�ddg�gdg�}t|ddg� t ddgddggdd� g�}t|ddg� d S )Nr TFrE c S s dS )NrM r? )r� r? r? r@ rK � rL z+TestPiecewise.test_simple.<locals>.<lambda>rM )r, r r9 rg )rR r� r? r? r@ rS y s zTestPiecewise.test_simplec C s2 t ddgddgddggddg�}t|ddg� d S )NrE rD TFrH rC )r, r )rR r� r? r? r@ �test_two_conditions� s z!TestPiecewise.test_two_conditionsc C s& t ddddgdddg�}t|d� d S )NrH TFrC rD r )r, r )rR r� r? r? r@ �$test_scalar_domains_three_conditions� s z2TestPiecewise.test_scalar_domains_three_conditionsc C sN t ddgddgdg�}t|ddg� t ddgddgddg�}t|ddg� d S )NrE rD TFr rH )r, r )rR r� r? r? r@ � test_default� s zTestPiecewise.test_defaultc C s� t �d�}t||dkddg�}t|jdk� t|dk� d}t|dgdggddg�}t|jdk� t|dk� t|dddgdddg�}t|d� d S ) NrH rC r rI TFrE rD )r9 rg r, r rl r )rR r� r� r? r? r@ r� � s zTestPiecewise.test_0dc C sh d}t ||dk|dkgddg�}t|d� d}t ||dk|dk|dk |dkgdddg�}t|d� d S )NrH rC r rI rE rD )r, r r )rR r� r� r? r? r@ �test_0d_comparison� s ,z TestPiecewise.test_0d_comparisonc C s\ t �dddgdddgg�}t||dk |dkgd d d g�}t|t �d d d gd d dgg�� d S )Ng �g ��g �g �?g �?g @r rD rM rE rH g �g @g �?)r9 rg r, r )rR r� r� r? r? r@ � test_multidimensional_extrafunc� s z-TestPiecewise.test_multidimensional_extrafuncN) r^ r_ r` rS r r r r� r! r" r? r? r? r@ r w s r c @ sr e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e �e d�dd� �ZdS )� TestBincountc C s$ t �t �d��}t|t �d�� d S )NrC )r9 �bincountr: r rP )rR r� r? r? r@ rS � s zTestBincount.test_simplec C s: t �t �dddddg��}t|t �ddddddg�� d S )NrE rI rD rC r )r9 r$ rg r )rR r� r? r? r@ � test_simple2� s zTestBincount.test_simple2c C s6 t �d�}t �ddddg�}t �||�}t||� d S )NrC g�������?g333333�?g �?g�������?)r9 r: rg r$ r )rR r� r� r� r? r? r@ �test_simple_weight� s z TestBincount.test_simple_weightc C sT t �dddddg�}t �dddddg�}t �||�}t|t �d ddd ddg�� d S ) NrE rD rC rI g�������?g333333�?g �?g�������?r )r9 rg r$ r )rR r� r� r� r? r? r@ �test_simple_weight2� s z TestBincount.test_simple_weight2c C s^ t �dddddg�}t j|dd�}t|t �dddg�� g }t j|dd�}t|t �g �� d S )Nr rE rH )� minlengthrD )r9 rg r$ r )rR r� r� r? r? r@ �test_with_minlength� s z TestBincount.test_with_minlengthc C sh t �dddddddg�}t j|dd�}t|t �ddddg�� t j|dd�}t|t �ddddg�� d S )Nr rE rD rH )r( )r9 rg r$ r )rR r� r� r? r? r@ �)test_with_minlength_smaller_than_maxvalue� s z6TestBincount.test_with_minlength_smaller_than_maxvaluec C sZ t �dddddg�}t �dddddg�}t �||d �}t|t �d ddd ddd d g�� d S ) NrE rD rC rI g�������?g333333�?g �?g�������?rY r )r9 rg r$ r )rR r� r� r� r? r? r@ � test_with_minlength_and_weights� s z,TestBincount.test_with_minlength_and_weightsc C s& t jg td�}t �|�}t||� d S )N)r� )r9 rg r r$ r )rR r� r� r? r? r@ r� � s zTestBincount.test_emptyc C s4 t jg td�}t j|dd�}t|t jdtd�� d S )N)r� rI )r( )r9 rg r r$ r rN )rR r� r� r? r? r@ �test_empty_with_minlength� s z&TestBincount.test_empty_with_minlengthc sl t jg td�� ttd� fdd�� ttd� fdd�� t �d�� ttd� fdd�� ttd � fd d�� d S ) N)r� z"'str' object cannot be interpretedc s t j� dd�S )N�foobar)r( )r9 r$ r? )r� r? r@ rK � rL z<TestBincount.test_with_incorrect_minlength.<locals>.<lambda>zmust be non-negativec s t j� dd�S )NrM )r( )r9 r$ r? )r� r? r@ rK � rL rI c s t j� dd�S )Nr- )r( )r9 r$ r? )r� r? r@ rK rL z minlength must be non-negativec s t j� dd�S )NrM )r( )r9 r$ r? )r� r? r@ rK rL )r9 rg r r r( rO r: )rR r? )r� r@ � test_with_incorrect_minlength� s z*TestBincount.test_with_incorrect_minlengthz python has no sys.getrefcountc C s� t �t�tj��}t �t�tj��}x td�D ]}t�dddg� q.W tt �t�tj��|� tt �t�tj��|� x(td�D ] }t�dddgdddg� q�W tt �t�tj��|� tt �t�tj��|� d S )Nry rE rD rH rC rI rZ ) �sys� getrefcountr9 r� r �doublerQ r$ r )rR Z intp_refcountZdouble_refcount�jr? r? r@ �test_dtype_reference_leaks s z'TestBincount.test_dtype_reference_leaksN)r^ r_ r` rS r% r& r' r) r* r+ r� r, r. r �skipifr r3 r? r? r? r@ r# � s r# c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestInterpc C st t ttdg g � t ttddgddg� t ttdddgddgdd� t ttdg g dd� t ttddgddgdd� d S )Nr rE rD )�periodih )r rO r( )rR r? r? r@ �test_exceptions s zTestInterp.test_exceptionsc C sB t �ddd�}t �ddd�}t �ddd�}tt �|||�|� d S )Nr rE rI rU )r9 r2 r r( )rR r� r� �x0r? r? r@ rT # s zTestInterp.test_basicc C s� �x�t dd�D �]�}tj|tjd�}tj|tjd�}tjdd|d |gtjd�}|d d d� }t|||�}t|||�}tjddddgtjd�}|d d d� } t||� t|| � t|||dd�}t|||dd�}tjddddgtjd�}|d d d� } t||� t|| � t|||dd�}t|||dd�}tjddddgtjd�}|d d d� } t||� t|| � t|||ddd �}t|||ddd �}tjddddgtjd�}|d d d� } t||� t|| � qW d S ) NrE ry )r� rM r )r� rD )r� )r� r� ) rQ r9 r: r1 rP rg r( r� r ) rR �size�xpZypZincptsZdecptsZincresZdecresZinctgtZdectgtr? r? r@ �test_right_left_behavior) s: z#TestInterp.test_right_left_behaviorc C s� t �ddd�}t �ddd�}d}tt �|||�|� d}tt �|||�|� t �d�}tt �|||�|� t �d�}tt �|||�|� t j}tt �|||�|� d S )Nr rE rI g333333�?)r9 r2 r r( r� r� r� )rR r� r� r8 r? r? r@ � test_scalar_interpolation_pointO s z*TestInterp.test_scalar_interpolation_pointc C s� t �ddd�}t �ddd�dt �ddd� d }d}|d| d }tt �|||�|� d}d}tt j||||d�|� d }d}tt j||||d �|� d d d dddddg}ddddg}ddddg}dddd dd d d g}tt j|||d d!�|� d S )"Nr rE rI y �?g333333�?rM y @ @)r� g @)r� iL���iV���iG���� i����r� im � iB���i^ i����y @ �?y $@ @y @ @y @ @y @ �?y �!@ �?y @ �?y @ @y @ @y @ @ih )r6 )r9 r2 r r( ) rR r� r� r8 Zy0r� r� r: �fpr? r? r@ �test_complex_interp] s" $ z TestInterp.test_complex_interpc C s` t �ddd�}t �ddd�}t �d�}tt �|||�|� t jdtd�}tt �|||�d� d S )Nr rE rI g333333�?)r� )r9 r2 rg r r( �object)rR r� r� r8 r? r? r@ �)test_zero_dimensional_interpolation_points s z4TestInterp.test_zero_dimensional_interpolation_pointc C s2 t �ddd�}t �|�}tt �t j||�d� d S )Nr ry g-C ��6?g )r9 r: Zsinr r( rC )rR r: r? r? r? r@ �test_if_len_x_is_small{ s z!TestInterp.test_if_len_x_is_smallc C s� ddddddddg}d d d d g}d dddg}ddddddddg}t tj|||dd�|� tj|dd ��d d �}tj|d d ��d d �}t tj|||dd�|� d S ) NiL���iV���iG���r= i����r� r im r> iB���i^ i����rI ry rH rC g @g @g �!@g @g @g @g @g @ih )r6 r| )r} rD rM r� )r r9 r( rg r\ )rR r� r: r? r� r? r? r@ � test_period� s zTestInterp.test_periodN) r^ r_ r` r7 rT r; r< r@ rB rC rD r? r? r? r@ r5 s &r5 c C s, x&t t|��D ]}t| | || � qW d S )N)rQ r� r )rF r� rn r? r? r@ �compare_results� s rE c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0�TestPercentilec C s� t �d�d }tt �|d�d� tt �|d�d� tt �|d�d� t j|d < tjd d ��R}t�d d t� tt �|d�t j� tt j|ddd�t j� t |d j tk� W d Q R X d S )NrY g �?r g r� g @rU g �?rE T)r� r� r� �nearest)� interpolation) r9 r: r � percentiler� r� r� r� r� r r� )rR r� r� r? r? r@ rT � s zTestPercentile.test_basicc C sR t �d�}t �|dd d d� t �|dd d dd� t �d�}t �|dd |dd� d S )NrI F�linear)rE )r9 rP rI )rR r� �or? r? r@ �test_api� s zTestPercentile.test_apic C sR t �dddgdddgdddgdddgdddgg�}tt j|ddd�dddg� d S )NrE rC rH rU r )rb )r9 rg r rI )rR r� r? r? r@ �test_2D� s zTestPercentile.test_2Dc C s4 t t�td�d�d� t tjtd�ddd�d� d S )Nry rU g @rJ )rH )r r9 rI rQ )rR r? r? r@ � test_linear� s zTestPercentile.test_linearc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nry rU �lower)rH rC �higherrI )r r9 rI rQ )rR r? r? r@ �test_lower_higher� s z TestPercentile.test_lower_higherc C sR t tjtd�ddd�d� t tjtd�ddd�d� t tjtd�ddd�d � d S ) Nry �3 �midpoint)rH g @r� g @rU rI )r r9 rI rQ )rR r? r? r@ � test_midpoint� s z TestPercentile.test_midpointc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nry rR rG )rH rI �1 rC )r r9 rI rQ )rR r? r? r@ � test_nearest� s zTestPercentile.test_nearestc C s0 t �d�d }tt �|dddg�dddg� d S )NrY g �?r r� rU g @g �?)r9 r: r rI )rR r� r? r? r@ � test_sequence� s z TestPercentile.test_sequencec C s t �d��dd�}tt �|d�dddg� dddd gdd d d gd d ddgg}tt j|ddd�|� dddgddd gdddgg}tt j|ddd�t �|�j� t �d��ddd d �}tt �|d�jd� tt �|d�jd � tt j|ddd�jd � tt j|ddd�jd � tt j|ddd�jd � tt j|ddd�jd � tt j|ddd�jd!� tt j|dd"d#�jd� tt j|dd"d#�jd � tt j|ddd"d$�jd � tt j|ddd"d$�jd � tt j|ddd"d$�jd � tt j|ddd"d$�jd � tt j|ddd"d$�jd!� d S )%Nr� rH rC )r$ rU r� g @g @g &@rD rI rZ r[ rY r� ry r� r )rb g �?g �?g @g �!@g #@rE ih )r$ rU )rD )r$ rU �K )rH )rD rC rI rZ )rD rH rI rZ )rD rH rC rZ )rD rH rC rI )rH rH rI rZ rP )rH )rb rH )r9 r: r\ r rI rg r� rW )rR r� �r0rX r? r? r@ � test_axis� s: " zTestPercentile.test_axisc C s� t �d��dd�}tt �|d�d� | �t �t �|d��� t �dddd g�}tt j|dd d �|� tt j|dd d �j|j� t �d dd g�}t t j|ddd �|� tt j|ddd �j|j� t � d�}tt j|d|d�d� t|d� t � d�}tt j|dd |d�|� t||� t � d�}tt j|dd|d�|� t||� t �d��dd�}tt j|ddd�d� | �t �t �|d��� t �dddd g�}t j|ddd d�}t||� t|j|j� t �dddg�}t j|dddd�}t ||� t|j|j� t j d|j d�}t j|dd|d�}t|d� t|d� t j d|j d�}t j|ddd |d�}t||� t||� t j d|j d�}t j|ddd|d�}t||� t||� d S )Nr� rH rC rU g @g @g @g @g @r )rb g �?g #@rE )r )rb r rO )rH )rH rb g �?g "@r? )r� )rH r rI )rH rb r ) r9 r: r\ r rI � assertTrueZisscalarrg rW r �emptyr� )rR r� rY rX r Zc0Zc1r� r? r? r@ � test_scalar_q� sR z TestPercentile.test_scalar_qc C s� t ttjddgddd� t ttjdgd� t ttjdgd� t ttjdgttd��dg � t ttjdgttd��d g � d S ) NrE rD �8 r- )rH �e rM rU g��������)r rO r9 rI r_ rQ )rR r? r? r@ �test_exception$ s z TestPercentile.test_exceptionc C s t t�dddgd�d� d S )NrE rD rH r )r r9 rI )rR r? r? r@ �test_percentile_list, s z#TestPercentile.test_percentile_listc C s t �dddg�}t �d�}d}t j|||d� t|t �||�� t �dddgddd gg�}t �d �}t j||d |d � t|t j||d d �� t �d�}t j||d|d � t|t j||dd �� t �d��dd�}t �ddddgddddgg�}t �d�}tt j|dd |d �|� t||� t �dddgdd d gg�}t �d �}tt j|dd|d �|� t||� t �d dddgddd d gg�}t jd|jd �}t j|dd!d |d"�}t||� t||� t �d dd#gddd$gg�}t jd |jd �}t j|dd!d|d"�}t||� t||� d S )%NrE rD rH )rH )rE rD rH )r rC rI rZ )rH rH r )rb r )rb )rH rD r� g @g @g @g @g @g @)rD rC )r$ rU g �?g @g �!@g �?g @g #@)rD rH r[ )r� rO )rH rb r rY r� ) r9 rg rN rI r r: r\ r\ r� )rR r� r� ra rY r rX r� r? r? r@ �test_percentile_out/ s@ z"TestPercentile.test_percentile_outc C s� t �d��dddd�}tt j|ddd�jd� tt j|ddd�jd � tt j|ddd�jd � tt j|dd d�jd � tt j|dd d�jd � tt j|ddd�jd � tt j|ddd�jd � tt j|ddd�jd� tt j|dddd�jd � tt j|dddd�jd � tt �t j|ddgdd��jd� tt �t j|ddgdd��jd� tt �t j|ddgdd��jd� tt �t j|ddgd d��jd� d S )Nr� r� rE rD rU r )rb )rE rD rE )r� rD rE )r� rE rE rH )r� rE rD rM rG rJ r� rS )rb rH ry )rD rE rD rE )rD r� rD rE )rD r� rE rE )rD r� rE rD )r9 r: r\ r rI rW rg )rR r� r? r? r@ �test_percentile_empty_dimX s. z(TestPercentile.test_percentile_empty_dimc C sx t �ddddg�}t j|dgdd� t|t �ddddg�� t �ddddg�}t �|dg� t|t �ddddg�� d S )NrD rH rC rE rU F)�overwrite_input)r9 rg rI r )rR rS r? r? r@ � test_percentile_no_overwritet s z+TestPercentile.test_percentile_no_overwritec C s� t jdddd�}t jt �d�|dd� t|t jdddd�� t jdddd��� }t jt �d�|dd� t|t jdddd��� � d S )Ng g Y@rI )r� rS )rH )r9 r2 rI r: r �tolist)rR ra r? r? r@ �test_no_p_overwrite} s z"TestPercentile.test_no_p_overwritec C sf t �ddddg�}t j|dgdd�}t|t �dg�� t jddddgdgdd�}t|t �dg�� d S ) NrD rH rC rE rU T)rd g @)r9 rg rI r )rR rS rd r? r? r@ �test_percentile_overwrite� s z(TestPercentile.test_percentile_overwritec C s� t jjdd�}t �|gd �}tt j|ddd�t �|d�� t �|dd�}tt j|dd d�t �|d�� |�dd ��� }tt j|dd d�t �|d�� |�dd ��� }tt j|d d gdd�t j|d d gd d�� tt j|d d gdd�t j|d d gdd�� t � d�� d�}t j� |� � � tt j|d dd�d t �|d d �d d �d d �df � � d �� tt j|ddgdd�d d �d f t �|d d �d d �d d d �f � � ddg�� tt j|d dd�d t �|d d �d d �dd d �f � � d �� tt j|d dd�d t �|dd d �d d �d d �f � � d �� tt j|d dd�d t �|dd d d �d d �f � � d �� tt j|d dd�d t �|dd d �d d �d f � � d �� tt j|d dd�d t �|dd d �dd d �f � � d �� d S ) N)r� r� )r9 ry r� )r rE )rb rM r )rG rM rE )r rM r$ r )r rE rD )r i� )rH rI r[ r� �Z )r rE rH )rH rE r� rD )rH rE rD )rH rD )rD rE )rE rG )rE rH )rD rD )r9 r4 �normal�dstackr rI �rollaxisrm rz r: r\ �shuffle�ravel�flatten)rR rK r� r� r? r? r@ �test_extended_axis� s8 * .**&&z!TestPercentile.test_extended_axisc C s� t �d�}tt jt j|ddd� tt jt j|ddd� tt jt j|ddd� tt jt j|ddd� ttt j|ddd� ttt j|d dd� ttt j|d dd� d S ) N)rH rI r[ r� r� r$ )rb r� )r r� rC )r rC )rE rE )rM rM )rH rM )r9 rP r r� rI rO )rR r� r? r? r@ �test_extended_axis_invalid� s z)TestPercentile.test_extended_axis_invalidc C s� t �d�}tt j|dd dd�jd� tt j|dddd�jd� tt j|dddd�jd � tt j|dd dd�jd � tt j|dd dd �jd� tt j|dddd�jd� tt j|ddgddd�jd� tt j|ddgddd�jd� d S )N)rH rI r[ r� r[ T)rb �keepdims)rE rE rE rE )r rE )rE rE r[ r� )r rH )rE rI r[ rE )rE )rH rE r[ r� )r rE rD rH )rr )r rE rH )rE rE r[ rE rE )rD rE rE r[ rE )rD rE rI r[ rE )r9 rP r rI rW )rR r� r? r? r@ � test_keepdims� s" z TestPercentile.test_keepdimsc C s� t �d�}t �d�}tt j|dd|d�|� tt j|ddd|d�|� t �d�}tt j|dd|d�|� tt j|ddd|d�|� t �d �}tt j|d |d�|� tt j|d d|d�|� d S ) N)rC )rH rC r )r rG )rH r )rH rE r? rD )r9 rN rP r rI )rR rK r� r? r? r@ �test_out� s zTestPercentile.test_outc C s� t jdd��� t �ddt� t�d�}t�d�}tj|d< ttj |dd|d �|� ttj |ddd |d �|� t�d �}ttj |d d |d �|� ttj |d d d |d �|� t�d�}ttj |d |d �|� ttj |d d |d �|� W d Q R X d S )NT)r� r� r� )rC )rH rC )rD rE r )r rG )rH r )rH rE r? ) r� r� r� r� r9 rN rP r� r rI )rR rK r� r? r? r@ � test_out_nan� s zTestPercentile.test_out_nanc C s� t jdtd�}t j|d< tjdd���}t�ddt� tt � |d�t j� tt j |dd d �t j� tt j |dd gd d �t � t jgd �� t |d j tk� t |d j tk� t |d j tk� W d Q R X t jdtd�� dd d�}t j|d< t j|d< tjdd��N}t�ddt� tt � |d�t j� tt � |d�jd � t |d j tk� W d Q R X t � t jdtd�� dd d�dd �}t j|d< t j|d< tjdd��(}t�ddt� tt � |dd �|� W d Q R X t � t jdtd�� dd d�dd gd �}t j|d d �dd f< t j|d d �d df< tjdd��,}t�ddt� tt � |dd gd �|� W d Q R X t � t jdtd�� dd d�dd �}t j|d< t j|d< tjdd��(}t�ddt� tt � |dd �|� W d Q R X t � t jdtd�� dd d�dd gd �}t j|d d �d d f< t j|d d �d df< tjdd��,}t�ddt� tt � |dd gd �|� W d Q R X t � t jdtd�� dd d�dd�}t j|d < t j|d< tjdd��(}t�ddt� tt � |dd�|� W d Q R X t � t jdtd�� dd d�dd gd�}t j|d d �d f< t j|d d �df< tjdd��,}t�ddt� tt � |dd gd�|� W d Q R X t j t jdtd�� dd d�dd gddd�}t j|d d �d f< t j|d d �df< tjdd��0}t�ddt� tt j |dd gddd�|� W d Q R X d S )Nr� )r� rD T)r� r� r� g333333�?r )rb g333333�?rE rH rC )rE rD rH )rE rE rD )rD rH )rE rD )rE rH )r rD rG )rH )r9 r: r� r� r� r� r� r� r rI rg r r� r\ rl )rR rS r� rd r? r? r@ �test_nan_behavior� s� " "" "" "z TestPercentile.test_nan_behaviorN)r^ r_ r` rT rL rM rN rQ rT rV rW rZ r] r` ra rb rc re rg rh rp rq rs rt ru rv r? r? r? r@ rF � s. $0) rF c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) � TestMedianc C s� t �d�}t �d�}t �d��dd�}tt �|�d� tt �|�d� tt �|�d� tt j|dd�d dd g� tt j|dd�dd g� tt j|d d�d� t �d d ddg�}t|d |d d t �|�� t �d d dg�}t|d t �|�� t �d dd g�}t|d t �|�� tt �|�jd� t j |d< t j dd��8}t � ddt � tt �|�jd� t|d jt k� W d Q R X d S )NrE rD rZ rH g �?g @r )rb g �?g @rC g�6-¦?g� �0���?g���}r�?g��4 �?g @rM T)r� r� r� )r9 rg r: r\ r �medianr r rl r� r� r� r� r� r r� )rR �a0r� r� rS r� r? r? r@ rT A s* zTestMedian.test_basicc C s� t �ddgddgddgddgg�}x`|t jjdd d d �gD ]F}|�� }t j|d d � x t|j�D ]}t j||d � q`W t||� q:W t t j|dd �ddg� t t j|j dd �ddg� t t �|�d � t t j|d d �d � t t �|j �d � d S )NrD rH r rE rZ r[ rC rI r� )rD rH rC )r9 )rb g @) r9 rg r4 Zrandintrz rx rQ rl r r r� )rR �a3rS ZorigZaxr? r? r@ �test_axis_keywordZ s z TestMedian.test_axis_keywordc C s� t �ddgddgddgddgg�}t �d�}t �d�}t �d��dd�}tt j|�� d d �d� tt j|�� d d �d � tt j|�� d d �d � tt j|�� d dd �dd dg� tt j|�� d dd �ddg� tt j|�� d d d �d � tt j|�� d dd �ddg� tt j|j�� d dd �ddg� t jdt jd��d�}t j � |� � � tt j|d d�t j|�� d d d�� tt j|dd�t j|�� dd d�� tt j|dd�t j|�� dd d�� tt j|dd�t j|�� dd d�� d S )NrD rH r rE rZ r[ rC rI T)rd g �?g @)rd rb g �?g @r )r� )rH rC rI )rb )rb rd ) r9 rg r: r\ r rx rz r� r� r4 rm rn )rR rz ry r� r� Za4r? r? r@ �test_overwrite_keywordl s< z!TestMedian.test_overwrite_keywordc C sH dddg}t t�|�d� |g}t t�|�d� ttj|dd�|� d S )NrE rD rH r )rb )r r9 rx r )rR r� r0 r? r? r@ r� � s zTestMedian.test_array_likec C s4 G dd� dt j�}|dddg�}tt �|�d� d S )Nc @ s e Zd Zddd�Zddd�ZdS )z,TestMedian.test_subclass.<locals>.MySubClassNc S s t �|��| �}||_|S )N)r9 r r� �info)�clsZ input_arrayr} r r? r? r@ �__new__� s z4TestMedian.test_subclass.<locals>.MySubClass.__new__c S s dS )Ni����r? )rR rb r� r r? r? r@ r� � s z1TestMedian.test_subclass.<locals>.MySubClass.mean)N)NNN)r^ r_ r` r r� r? r? r? r@ � MySubClass� s r� rE rD rH i����)r9 r� r rx )rR r� rS r? r? r@ r� � s zTestMedian.test_subclassc C sl t �d�}t �d�}tt j|d|d�|� t �d�}tt j|d|d�|� t �d�}tt j||d�|� d S )N)rC )rH rC r )r )rH rE r? )r9 rN rP r rx )rR rK r� r? r? r@ rt � s zTestMedian.test_outc C s� t jdd��� t �ddt� t�d�}t�d�}tj|d< ttj |d|d �|� t�d �}ttj |d |d �|� t�d �}ttj ||d �|� W d Q R X d S ) NT)r� r� r� )rC )rH rC )rD rE r )r )rH rE r? ) r� r� r� r� r9 rN rP r� r rx )rR rK r� r? r? r@ ru � s zTestMedian.test_out_nanc C s� t jdtd�}t j|d< tjdd��`}t�ddt� tt � |�t j� tt j |dd �t j� t |d j tk� t |d j tk� W d Q R X t jdtd�� dd d �}t j|d < t j|d< tjdd��J}t�ddt� tt � |�t j� tt � |�j d� t |d j tk� W d Q R X t � t jdtd�� dd d �d�}t j|d< t j|d< tjdd��4}t�ddt� tt � |d�|� tt|�d � W d Q R X t � t jdtd�� dd d �d �}t j|d< t j|d< tjdd��4}t�ddt� tt � |d �|� tt|�d � W d Q R X t � t jdtd�� dd d �d�}t j|d < t j|d< tjdd��4}t�ddt� tt � |d�|� tt|�d � W d Q R X d S )Nr� )r� rD T)r� r� r� r )rb rE rH rC )rE rD rH )rE rE rD )rD rH )rE rD )rE rH )r rD )r9 r: r� r� r� r� r� r� r rx r r� r\ rl r� )rR rS r� rd r? r? r@ rv � sJ z TestMedian.test_nan_behaviorc C sL t jg td�}tjdd��8}t�ddt� tt �|�t j � t |d j tk� W d Q R X t jg tdd�}tjdd��8}t�ddt� tt �|�t j � t |d j tk� W d Q R X t jg td d�}tt j|dd �|� tt j|d d �|� t jt j td d�}tjdd��:}t�ddt� tt j|d d �|� t |d j tk� W d Q R X d S ) N)r� T)r� r� r� r rH )r� r� rD )rb rE ) r9 rg r� r� r� r� r� r rx r� r r� )rR rS r� rd r? r? r@ r� � s$ zTestMedian.test_emptyc C sL t �d�}ttt �|�t���t� t j|d< ttt �|�t���t� d S )Ng @rD ) r9 r: r r� rx r� rA r� r� )rR rK r? r? r@ � test_object s zTestMedian.test_objectc C s� t jjdd�}t �|gd �}tt j|dd�t �|�� t �|dd�}tt j|dd�t �|�� |�dd ��� }tt j|d d�t �|�� tt j|d d�t j|d d�� tt j|d d�t j|dd�� tt j|d d�t j|dd�� t � d�� d�}t j� |� � � tt j|d d�d t �|d d �d d �d d �df � � �� tt j|dd�d t �|d d �d d �d d d �f � � �� tt j|dd�d t �|d d �d d �dd d �f � � �� tt j|dd�d t �|dd d �d d �d d �f � � �� tt j|dd�d t �|dd d d �d d �f � � �� tt j|dd�d t �|dd d �d d �d f � � �� tt j|dd�d t �|dd d �dd d �f � � �� d S )N)r� r� )r9 ry )r rE )rb rM r )rG rM rE )r rM )r rE rD )r )rM i� )rH rI r[ r� )r rE rH )rH rE r� rD )rH rE rD )rH rD )rD rE )rE rG )rE rH )rD rD )r9 r4 rj rk r rx rl rm rz r: r\ rm rn ro )rR rK r� r� r? r? r@ rp s4 (((($$z TestMedian.test_extended_axisc C sp t �d�}tt jt j|dd� tt jt j|dd� tt jt j|dd� tt jt j|dd� ttt j|dd� d S )N)rH rI r[ r� r� )rb )r r� rC )r rC )rE rE )r9 rP r r� rx rO )rR r� r? r? r@ rq ( s z%TestMedian.test_extended_axis_invalidc C s� t �d�}tt j|d dd�jd� tt j|ddd�jd� tt j|ddd�jd� tt j|d dd�jd � tt j|d dd�jd� tt j|d dd�jd � d S )N)rH rI r[ r� T)rb rr )rE rE rE rE )r rE )rE rE r[ r� )r rH )rE rI r[ rE )rE )rH rE r[ r� )r rE rD rH )r rE rH )rE rE r[ rE )r9 rP r rx rW )rR r� r? r? r@ rs 0 s zTestMedian.test_keepdimsN)r^ r_ r` rT r{ r| r� r� rt ru rv r� r� rp rq rs r? r? r? r@ rw ? s ! 0 rw c @ s e Zd Zdd� Zdd� ZdS )�TestAdd_newdoc_ufuncc C s" t ttdd� t tttjd� d S )NrD Zblah)r r( r rO r9 r; )rR r? r? r@ �test_ufunc_argB s z#TestAdd_newdoc_ufunc.test_ufunc_argc C s t tttjd� d S )NrH )r r( r r9 r; )rR r? r? r@ �test_string_argF s z$TestAdd_newdoc_ufunc.test_string_argN)r^ r_ r` r� r� r? r? r? r@ r� @ s r� c @ s& e Zd Ze�ejjdk�dd� �ZdS )�TestAdd_newdocrD c C s\ d}| � tjjjjd t|�� |� | �ttjjj j�dk� | �ttj j j j�dk� d S )Nz"Current flat index into the array.i, ) r� r9 ZcoreZflatiter�indexrI r� r[ ZufuncZidentityr� Z index_tricksZmgrid)rR r� r? r? r@ � test_add_docL s zTestAdd_newdoc.test_add_docN) r^ r_ r` r r4 r/ r~ �optimizer� r? r? r? r@ r� J s r� �__main__)vZ __future__r r r r� r� r/ r� Znumpyr9 Z numpy.testingr r r r r r r r r r r r r r Znumpy.testing.utilsr Znumpy.lib.function_baser� Z function_baserw Z numpy.randomr Z numpy.libr r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 Z numpy.compatr8 rA rB ra rp rv rx r� r� r� r� r� r� r� r� r r r r! rA rE rJ rQ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r# r5 rE rF rw r� r� r^ r? r? r? r@ �<module> s� @ � DI pHe K U' NI H e Ub YLWq 2
Submit
FILE
FOLDER
Name
Size
Permission
Action
test__datasource.cpython-37.opt-1.pyc
12541 bytes
0644
test__datasource.cpython-37.pyc
12541 bytes
0644
test__iotools.cpython-37.opt-1.pyc
11670 bytes
0644
test__iotools.cpython-37.pyc
11670 bytes
0644
test__version.cpython-37.opt-1.pyc
2520 bytes
0644
test__version.cpython-37.pyc
2520 bytes
0644
test_arraypad.cpython-37.opt-1.pyc
48281 bytes
0644
test_arraypad.cpython-37.pyc
48281 bytes
0644
test_arraysetops.cpython-37.opt-1.pyc
12026 bytes
0644
test_arraysetops.cpython-37.pyc
12124 bytes
0644
test_arrayterator.cpython-37.opt-1.pyc
1829 bytes
0644
test_arrayterator.cpython-37.pyc
1829 bytes
0644
test_financial.cpython-37.opt-1.pyc
5494 bytes
0644
test_financial.cpython-37.pyc
5494 bytes
0644
test_format.cpython-37.opt-1.pyc
13304 bytes
0644
test_format.cpython-37.pyc
13304 bytes
0644
test_function_base.cpython-37.opt-1.pyc
119918 bytes
0644
test_function_base.cpython-37.pyc
119918 bytes
0644
test_index_tricks.cpython-37.opt-1.pyc
13318 bytes
0644
test_index_tricks.cpython-37.pyc
13318 bytes
0644
test_io.cpython-37.opt-1.pyc
72333 bytes
0644
test_io.cpython-37.pyc
72333 bytes
0644
test_mixins.cpython-37.opt-1.pyc
7171 bytes
0644
test_mixins.cpython-37.pyc
7171 bytes
0644
test_nanfunctions.cpython-37.opt-1.pyc
31910 bytes
0644
test_nanfunctions.cpython-37.pyc
31910 bytes
0644
test_packbits.cpython-37.opt-1.pyc
7841 bytes
0644
test_packbits.cpython-37.pyc
7841 bytes
0644
test_polynomial.cpython-37.opt-1.pyc
5913 bytes
0644
test_polynomial.cpython-37.pyc
5913 bytes
0644
test_recfunctions.cpython-37.opt-1.pyc
25950 bytes
0644
test_recfunctions.cpython-37.pyc
25950 bytes
0644
test_regression.cpython-37.opt-1.pyc
10789 bytes
0644
test_regression.cpython-37.pyc
10789 bytes
0644
test_shape_base.cpython-37.opt-1.pyc
22967 bytes
0644
test_shape_base.cpython-37.pyc
22967 bytes
0644
test_stride_tricks.cpython-37.opt-1.pyc
10755 bytes
0644
test_stride_tricks.cpython-37.pyc
10755 bytes
0644
test_twodim_base.cpython-37.opt-1.pyc
15207 bytes
0644
test_twodim_base.cpython-37.pyc
15207 bytes
0644
test_type_check.cpython-37.opt-1.pyc
17919 bytes
0644
test_type_check.cpython-37.pyc
17919 bytes
0644
test_ufunclike.cpython-37.opt-1.pyc
3641 bytes
0644
test_ufunclike.cpython-37.pyc
3641 bytes
0644
test_utils.cpython-37.opt-1.pyc
2474 bytes
0644
test_utils.cpython-37.pyc
2474 bytes
0644
N4ST4R_ID | Naxtarrr
