Edit File: test_function_base.cpython-37.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? ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/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�a�b1�b2�b3�b4rL 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 )rZ rY 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__rX r\ ra 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@ r\ 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 rg )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 rg )rR rS rh 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 r^ r_ )r9 �arrayr rg )rR rS rh 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 r^ r_ )r9 rk r rg )rR rS rh 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 r^ r_ )r9 rk r rg )rR rS rh 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 rg r �swapaxes)rR rS �ir? r? r@ �test_4d� s zTestFlip.test_4dN) rb rc rd r\ ri rj rl rm rn rs r? r? r? r@ re g s re 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@ rX � s zTestAny.test_basicc C sb dddgdddgdddgg}t t�|�� ttj|dd�dddg� ttj|dd�dddg� d S )Nr rE )rf )r r9 ru r �sometrue)rR rv r? r? r@ �test_nd� s zTestAny.test_ndN)rb rc rd rX rz r? r? r? r@ rt � s rt 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 �allrk )rR rv rw rx r? r? r@ rX � s zTestAll.test_basicc C sd dddgdddgdddgg}t t�|� � ttj|dd�dddg� ttj|dd�dddg� d S )Nr rE )rf )r r9 r| r �alltrue)rR rv r? r? r@ rz � s zTestAll.test_ndN)rb rc rd rX rz r? r? r? r@ r{ � s r{ 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 rk �copyr r )rR rS �a_copyr? r? r@ rX � 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 rk r �flags� c_contiguous� f_contiguousr� )rR rS �a_fortr� Z a_fort_copyr? r? r@ � test_order� s zTestCopy.test_orderN)rb rc rd rX r� r? r? r? r@ r~ � s r~ 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 )rf g @g �?g @g )rC rC )r rE )rE r rI ) r9 rk r r rP r �meanr �matrixr )rR rv rw rx Zy4Zy5Zy6r? r? r@ rX � 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 )Nr )�weightsrD g �?rE rH rC rI r^ r )r� rf g @g @g @g @)r9 r: r �sumr rk r r r �astype�float32�float64r �dtype� result_type) rR �y�w�actual�desiredrv �w0�w1�w2rx 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 r^ T)�returnedg @r g @g @)r� rf r� g �?)r9 rk r r r )rR r� �avg�sclr� 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)rb rc rd r? r? r? r@ �subclassA s r� rE rD rH rC )r� r )rf r� g @g������ @)r9 �ndarrayrk �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 rk r r r� )rR �types�at�wt�rtrS 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>r c S s g | ]}t �d ��qS )rE )r� r� )r� �_r? r? r@ r� [ s r )r� )r9 rk rQ r� r r� r )rR rS r� r? r? r@ �test_object_dtypeY s z TestAverage.test_object_dtypeN) rb rc rd rX 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 r^ r_ r] � 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 �cond�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@ rX 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 rk r: r` r r/ rP r r[ )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� �choicer? r? r@ r� � s z0TestSelect.test_return_dtype.<locals>.<listcomp>rE rD rH rI r_ r ) r r/ r� r� r� r9 �complex_r� rk �nan�isnan)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 �int_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 rk r/ )rR r� r� r? r? r@ �test_many_arguments� s z TestSelect.test_many_argumentsN)r )rb rc rd r9 rk r� r� r� rX 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_ r] )r� g g �?T)r� r� � rC ) r r' �slicer9 rk r� r� r� r� � FutureWarningr �category)rR rS rh r� r? r? r@ rX � 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 )rf rH rC � )rD rH rC rM rG ���) r r' r9 rk r: �repeatr` � concatenate�Tr � AxisError)rR rS r� rh �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 )rf ) r9 rk 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)rb rc rd r? r? r? r@ �SubClass� s r� r r rE rD rG rM ) r9 r� r: r� r � isinstancer' r� rk )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 rk 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 rh )rH r� )�foorr )�bar�a1)r� )rC r� r rD rH )r9 rk r' r r� )rR rS �valrh r? r? r@ �test_structured_array s z TestInsert.test_structured_arrayN) rb rc rd rX 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 r rJ ���g @g $@g "@g @r] g @g @r )rf g @rE )r r9 �amax)rR rS rh r? r? r@ rX s zTestAmax.test_basicN)rb rc rd rX 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 r rJ r� g @g �g "@g $@g @r] g @g @r )rf rE g @)r r9 �amin)rR rS rh r? r? r@ rX s zTestAmin.test_basicN)rb rc rd rX 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 r rJ r� g @r )rf g .@g "@g $@g @r] g @g @g @rM )r9 rk r �ptp)rR rS rh r? r? r@ rX & s zTestPtp.test_basicN)rb rc rd rX 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 r � r^ rI rC rH r_ r� rK r� � �# �' r )rf r] � � � � � � )r9 r� r� �int16�uint16�int32�uint32r� r� � complex64� complex128rk r �cumsum)rR �ba�ba2�ctyperS �a2�tgtr? r? r@ rX 2 s " &&zTestCumsum.test_basicN)rb rc rd rX 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 r r� r^ rI rC rH r_ r� )�1rh r )rf i g rY �$ �T � rM r� ib iX )r9 r r r r r� r� r r rk rN �ArithmeticError�prodr r )rR r r r rS r r? r? r@ rX I s " zTestProd.test_basicN)rb rc rd rX 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 r r� r^ rI rC rH r_ r� )r rh rM )rf � �� i( i� i g r r� � r rY r r r� r� �� ib ro iX )r9 r r r r r� r� r r rk rN r �cumprodr )rR r r r rS r r? r? r@ rX ] s* " zTestCumprod.test_basicN)rb rc rd rX 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 r^ r_ r� rH rD rI rM r )r= )r9 rk r r )rR r� �out�out2�out3r? r? r@ rX 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 r r� rE rM rD )r= r )rf )r= rf )r r r )rR r� �out1r r Zout4r? r? r@ rz � s 4444zTestDiff.test_ndN)rb rc rd rX rz 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 )rf 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 Zlims�steps�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 rk 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 )rf ) r9 rk 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)rb rc rd 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 ) Nr rD rI r� )r� �< rE )rf ) 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) rb rc rd r# r, r2 r5 r6 r� r� r8 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 rk r r! )rR �vr� �dxr? r? r@ rX � 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 )rf z.*scalars or 1drM ) r9 r rP r: r` r! rk 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 )rf ) 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 rk 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 �mark r! r r� r rE r: �maskedr )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 r rD rH rC r^ r] )rF g� ��Q��?T) r9 �linspace�absr! r r| �random�seed�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 @)r^ rE rM rE g g �?g �?g @g @g �?g @g �g333333�?g �?g пrD )r rE )rf rF )rE r ) r9 rk �tiler` r: r! r r r� r r[ ) rR �fZx_unevenZx_evenZ fdx_even_ord1Z fdx_even_ord2Zfdx_uneven_ord1Zfdx_uneven_ord2rF 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 )rf rM )rE r rD rJ )r9 rk 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 r r� rC iA i, ztimedelta64[D])r� rD r_ r< � �w i��)r9 rk 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 )rF rH r )r! r9 r: r rO )rR r? r? r@ � test_values� s zTestGradient.test_valuesN) rb rc rd rX r? rA rD rJ rP rV rW rZ r[ r? r? r? r@ r9 � s @ r9 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 )�degr r� )r9 �sqrtr �arctan�pirk r )rR r� r� �yo�zZzor? r? r@ rX � s zTestAngle.test_basicN)rb rc rd rX r? r? r? r@ r\ � s r\ 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 rk r4 r )rR rS �resr? r? r@ rX � s zTestTrimZeros.test_basicc C s@ t �ddddddddg�}t|�}t|t �dddddg�� d S )Nr rE rD rH rC )r9 rk r4 r )rR rS rd 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 rk r4 r )rR rS rd r? r? r@ �test_trailing_skip� s z TestTrimZeros.test_trailing_skipN)rb rc rd �__doc__rX re rf r? r? r? r@ rc � s rc c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestExtinsc C s> t �dddddddg�}t|dk|�}t|dddddg� d S )NrE rH rD )r9 rk r r )rR rS rh r? r? r@ rX � 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 r] r_ r^ 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>�12Z34�9) r r@ r- r9 rk 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 )Nr g �?r )r r� r r- r )rR rS rE �acr� r? r? r@ � test_both� s zTestExtins.test_bothN)rb rc rd rX rn rp r? r? r? r@ rh � s rh 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 rh r? r? r@ � addsubtract� s z.TestVectorize.test_simple.<locals>.addsubtractr rH r^ r� rE rI r_ rD )r7 r )rR rr rR 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 rh r? r? r@ rr s z.TestVectorize.test_scalar.<locals>.addsubtractr rH r^ r� rI r] rE rC )r7 r )rR rr rR 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@ ri rj z*TestVectorize.test_large.<locals>.<lambda>)r9 rK r7 r )rR r� rR 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 �cosr9 rk r` r )rR rv rR �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 rh r? r? r@ r� " s z(TestVectorize.test_keywords.<locals>.foorD rH rC rI )rE )r7 r9 rk r )rR r� rR rx ry rz r? r? r@ � test_keywords s zTestVectorize.test_keywordsc C s. dd l }yt|j� W n t� �Y nX d S )Nr )rM r7 � randrange�AssertionError)rR rM 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 rh r? r? r@ r� ; s z5TestVectorize.test_keywords2_ticket_2100.<locals>.foorD rH )rS rC )rh rS )rh rI )rE )r7 r9 rk r )rR r� rR rx ry rz 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_prd r? r? r@ � mypolyvalK s z;TestVectorize.test_keywords3_ticket_2100.<locals>.mypolyvalr� rE )�excludedrH r^ r rD )r� r� )r� )r9 r7 r )rR r� 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? )�kwrd �_kr? r? r@ rR Z s z3TestVectorize.test_keywords4_ticket_2100.<locals>.frE rD rH rC )rS rh r] )r7 r )rR rR 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@ rR e s z3TestVectorize.test_keywords5_ticket_2100.<locals>.frE rD rH rC r] )r7 r )rR rR 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� rR 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 rg )rR r� rR r� 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)rh )rR rS r? r? r@ r� � s z=TestVectorize.test_UnboundMethod_ticket_1156.<locals>.Foo.barN)rb rc rd rh r� r? r? r? r@ �Foo� s r� r� rD )r r7 r� r9 r: )rR r� 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@ ri � rj z@TestVectorize.test_execution_order_ticket_1487.<locals>.<lambda>rH g�������?c S s | S )Nr? )r� r? r? r@ ri � rj )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@ ri � rj z7TestVectorize.test_string_ticket_1892.<locals>.<lambda>Zd0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r9 r7 r )rR rR r1 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@ rR � s z#TestVectorize.test_cache.<locals>.fTrI )r7 �cacher9 r: r r r� )rR rR r� r? )r� 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@ ri � rj z+TestVectorize.test_otypes.<locals>.<lambda>rr rI )r9 r7 �otypesr: r )rR rR 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 rh r� )r� )r� �ez (x)(y)->()z (x),(y)->z ((x))->(x))r �nfb�_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 rh r? r? r@ rr � s z8TestVectorize.test_signature_simple.<locals>.addsubtractz (),()->())� signaturer rH r^ r� rE rI r_ rD )r7 r )rR rr rR 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)->())r� rE rH rD rC )r7 r )rR r� rR 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))r� rE rH rD rC rM )r7 r )rR r� rR 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@ ri � rj z:TestVectorize.test_signature_two_outputs.<locals>.<lambda>z ()->(),())r� rE rD rH r )r7 r r� �tupler� r )rR rR 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))r� rE rD rH rC r^ r )r7 r9 r< r )rR rR 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@ ri � rj z<TestVectorize.test_signature_computed_size.<locals>.<lambda>z(n)->(m))r� rE rD rH rC )r7 r )rR rR 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 rh r? r? r@ r� � s z2TestVectorize.test_signature_excluded.<locals>.fooz()->()rh )r� r� rD rH rC r )rh )rE )r7 r )rR r� rR 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@ ri rj z5TestVectorize.test_signature_otypes.<locals>.<lambda>z(n)->(n)r� )r� r� rE rD rH )r7 r r� r9 r )rR rR 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))r� zwrong number of positionalrE rD z does not have enough dimensionsz$inconsistent size for core dimensionrH z()->())r7 �operatorr; r r@ rO )rR rR 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@ ri rj z>TestVectorize.test_signature_invalid_outputs.<locals>.<lambda>z(n)->(n))r� z$inconsistent size for core dimensionrE rD rH c S s | S )Nr? )r� r? r? r@ ri " rj z ()->(),()zwrong number of outputsc S s | | fS )Nr? )r� r? r? r@ ri & rj z()->())r7 r rO )rR rR 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@ ri , rj z5TestVectorize.test_size_zero_output.<locals>.<lambda>r rI )r� r� rr c S s | S )Nr? )r� r? r? r@ ri 4 rj z()->())r� c S s | S )Nr? )r� r? r? r@ ri 8 rj )r� r� c S s | S )Nr? )r� r? r? r@ ri ; rj z(n)->(n)c S s | S )Nr? )r� r? r? r@ ri > rj c S s | gS )Nr? )r� r? r? r@ ri A rj z()->(n)znew output dimensions) r9 r7 rm r% r rO r� r r� )rR rR r� r? r? r@ �test_size_zero_output* s$ z#TestVectorize.test_size_zero_outputN) rb rc rd rs rt ru 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� 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 )Nr r )r r9 rK �min�maxr r| 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 r r] 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 )Nr T)r r9 rK r� r� r r| 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)rb rc rd r? r? r? r@ �A� s r� rI rE rH FT)r9 r� r: r� r r� r )rR r� rS rh r? r? r@ �test_return_type� s z TestDigitize.test_return_typeN) rb rc rd 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 r r� )r r5 r9 r` r r| r r )rR r? r? r@ rs � s zTestUnwrap.test_simpleN)rb rc rd 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 )Nr r_ r )rf 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 )Nr r_ r )rf 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 )Nr r_ r )rf 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 )Nr r_ r )rf g= ףp=@rC )r r r r r9 r� )rR r� r? r? r@ � test_blackman� s z TestFilterwindows.test_blackmanN)rb rc rd 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����r g�������?g �rD )r; rE r_ )r9 r: r3 �expr^ r` 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 r] rK rM )rf )r� rf )r9 rK � ones_liker� r3 r ) rR r� r� rb �wxZwy�wz�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 )rE g *@)r9 r: rG rk r r3 )rR r� r� rE �ymr� �xmr? r? r@ rJ � s zTestTrapz.test_maskedc C sJ t �dd�}|| }t||�}t �|�}t �|�}t||�}t||� d S )Nr rI )r9 rK r3 r� r )rR r� r� r� �mx�my�mrr? r? r@ � test_matrix� s zTestTrapz.test_matrixN)rb rc rd rs r� rJ 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 rK 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 rk r� r� r )rR r� rv rw rx r? r? r@ �test_array_like s zTestSinc.test_array_likeN)rb rc rd 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 )rf r )r r$ r r9 r� rK r )rR r= r: rS rh 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 rk r )rR �hist�edges�hr� 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 r r rI r� )r r$ r9 r� r r r: )rR r= r: rS rh �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 r r rH r^ g�������?g rD rC g �?g �?g �?) r r$ r9 r� r r r: r r �inf) rR r= r: rS rh r� r� �countsZdmyr? 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 )Nr g �?r r� )rQ rE T)rQ r� r� )r� )rQ r� r� r] )r� rQ r� rM )r9 r: r$ r r� r r )rR rS r� rh 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 )Nr g �?T)r� )r� ) r9 r: r$ r � issubdtyper� r% �floatrP )rR rS r� rh 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 rk r� � histogram2dr r� )rR r� r� Z counts_hist�xedges�yedgesr? 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 r r� )r� r� rE rD rC rH )r� r� r� g $@g @r� r^ )r� r� g�������?g�������?g333333�?) r r9 rP r$ r rK r� rm r: r rk r ) rR r: r� rS rh �na�nb�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 rk 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����r r� )rQ r� )r9 rk 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 rk )rR rS rh 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 rK 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 rK 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 rk r$ r� �assertGreaterEqual� assertLess) rR �arrr� r� rE 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 rk 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 �uint64)r� )r� )r9 rk r rO r$ )rR r� r� r� r� r? r? r@ � test_unsigned_monotonicity_check� s z.TestHistogram.test_unsigned_monotonicity_checkN)rb rc rd 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�doaner^ �auto)r� r rE )r$ r r9 rk )rR �estimator_list� estimatorrS rh 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 r] r_ )r� r� r� r� r r^ r r� r r� � r r� r� �G )rY i� i� i����rM rI rD rE rH z*For the {0} estimator with datasize of {1})r$ N)�itemsr9 rK r� r$ r r� �format) rR � basic_test�testlen�expectedResults�x1rI r� r �numbinsrS rh 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 r^ 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 rh 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 r^ r z {0} estimator, No Variance test)r$ N)r9 rP r r$ r r� r )rR Z novar_datasetZnovar_resultdictr r rS rh 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����r rY i����i����rI r r� )r� r� r N)r9 rK �hstackr r$ r r� )rR ZxcenterZoutlier_datasetZoutlier_resultdictr r rS rh 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. r] r� r� )r� r� r� r� r r� � r �! �E r )rY i� i� i����rM rI rD rE r rH i����i���)i��r )rQ zFor the {0} estimatorz with datasize of {0})r$ N)r r9 rK r r$ r r r� ) rR r r r r rI �x3r� r r rS rh 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) rb rc rd rg 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 �squeeze)r� r� r? r? r@ r� � s z/TestHistogramdd.test_simple.<locals>.<listcomp>)rf )rC rH rD )r� rQ )rI rI rI g �?rI ) r9 rk r% r r r| r r2 rm r� rQ r: )rR r� �Hr� �answer�edrb �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 r^ )r^ rC rI )rI r^ rC )rC r^ rI )r^ rI rC )rC rI r^ r rH )r r% r r[ )rR r� r� rh 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 r^ )rC rI r_ r^ )rI r^ rC r_ )r_ r^ rI rC )rI r_ r^ rC )rC r^ r_ rI )r^ rI r_ rC )r_ rI rC r^ )r_ rC r^ rI )r^ rC r_ rI )r^ r_ rI rC )rC r^ rI r_ )rC r_ rI r^ )rI rC r^ r_ )rI r_ rC r^ )r^ r_ rC rI )r^ rI rC r_ )rC r_ r^ rI )rC rI r^ r_ )r_ r^ rC rI )rI rC r_ r^ )rI r^ r_ rC )r^ rC rI r_ )r_ rI r^ rC r rC )r r% r r[ )rR r� r� rh 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)r rD rD )r� r g �g g �?)r9 rm r% r% r rk )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 rk rm )rR rS rh 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 ) Nr] 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)�invalidr^ rH rD rE r r )r� rM )r9 �errstater: r` rk r% r� r )rR r� �expectedr� r� 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 rM r% r rO r� r� )rR r� r? r? r@ r� s z!TestHistogramdd.test_finite_rangeN) rb rc rd 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 �? $@r )r9 rk r r| r6 )rR r� r? r? r@ rs s ( zTestUnique.test_simpleN)rb rc rd 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 rh 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)rb rc rd 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 rS r r9 r| rL �BrT )rR �tgt1�tgt2r? 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 rS r/ rT )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 )�bias) r r� r� r r� r r� r/ r3 r rS )rR r4 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 rk r r r r| rL )rR r� rd 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 rk 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 rk 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)r| g �?g �)r9 r$ r r rk r r| rL )rR r� r� r? r? r@ � test_extremez s zTestCorrCoef.test_extremeN)rb rc rd r9 rk r� r/ rS rT r. rs r5 r7 r8 r9 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 rS )rR r? r? r@ rX � s zTestCov.test_basicc C s> t �dddgdddgg�}tt|�t �ddgddgg�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rk r r )rR r� r? r? r@ r8 � s zTestCov.test_complexc C sJ t �dddgg�}t �dddgg�}tt||�t �ddgddgg�� d S ) NrE rD rH y �?y @y @g �?y � �)r9 rk r r )rR r� r� r? r? r@ r9 � 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 rk 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 )r2 ) r� r� r� r: r r r r9 rk 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 )�rowvarg+����?g4��@���?g��&S�?g�o_��?g�J�4�?)r r r r9 rk )rR r� r? r? r@ �test_1D_rowvar� s zTestCov.test_1D_rowvarc C s$ t t| jdd�tj| jdd�� d S )NrE )r2 )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 rI � frequencies� x2_repeatsr rT �unit_frequenciesrS r r@ r9 rP r& � RuntimeErrorrO )rR ZnonintrR 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� rU � unit_weightsrS r9 rP r rH 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)rD rJ g @)rJ ) r r rI rE rK rF r rT rG rS r� rU )rR r? r? r@ � test_unit_fweights_and_aweights� s$ z'TestCov.test_unit_fweights_and_aweightsN) rb rc rd r9 rk r� r rS rI rE rF rT rP r& rG r� rU rK r rX r8 r9 r� r? rA rC rI rL rM 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 rk )rR r� r/ r? r? r@ rs � s& zTest_I0.test_simpleN)rb rc rd rs r? r? r? r@ rN � s rN 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 �isfiniter) r rk )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)rb rc rd rs rQ r? r? r? r@ rO � s rO 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 rk r r+ )rR r� r? r? r@ rs s zTestMsort.test_simpleN)rb rc rd rs r? r? r? r@ rR s rR 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 r^ r_ )r* r r9 rk )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 rk )rR rT r? r? r@ �test_single_input+ s z TestMeshgrid.test_single_inputc C s* g }t g t|� � t g t|ddi�� d S )Nr� F)r r* )rR rx 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 r^ r_ �ij)�indexingr] r� r )rC rH )rH rC )rC rH rD )rH rC rD Znotvalid)r* r r9 rk r r[ r rO )rR r� r� rT rU rb 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 r^ r_ T)�sparse)r* r r9 rk )rR rT rU r? r? r@ � test_sparseH s zTestMeshgrid.test_sparsec C s$ t ttdddgddddgdd � d S ) NrE rD rH rC rI r^ r_ rX )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 r )r� r T)r� )r[ )r9 r: r� r� r* r r� )rR r� r� rT rU 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[ r� r rE )r9 rk r* r )rR rT rU r� r� r? r? r@ �test_writebackl s zTestMeshgrid.test_writebackN) rb rc rd rs rV rW rZ r\ r] r� r^ r? r? r? r@ rS s rS 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@ ri � rj z+TestPiecewise.test_simple.<locals>.<lambda>rM )r, r r9 rk )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 rk r, r rp 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 rk r, r )rR r� r� r? r? r@ � test_multidimensional_extrafunc� s z-TestPiecewise.test_multidimensional_extrafuncN) rb rc rd rs r` ra rb r� rc rd 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 rf rk 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: rk rf 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 rk rf 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 rk rf 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 )rj )r9 rk rf 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�������?r] r )r9 rk rf 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 rk r% rf 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 )rj )r9 rk r% rf r rm )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)rj )r9 rf r? )r� r? r@ ri � rj z<TestBincount.test_with_incorrect_minlength.<locals>.<lambda>zmust be non-negativec s t j� dd�S )NrM )rj )r9 rf r? )r� r? r@ ri � rj rI c s t j� dd�S )Nro )rj )r9 rf r? )r� r? r@ ri rj z minlength must be non-negativec s t j� dd�S )NrM )rj )r9 rf r? )r� r? r@ ri rj )r9 rk 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 )Nr rE rD rH rC rI r^ ) �sys� getrefcountr9 r� r( �doublerQ rf r )rR Z intp_refcountZdouble_refcount�jr? r? r@ �test_dtype_reference_leaks s z'TestBincount.test_dtype_reference_leaksN)rb rc rd rs rg rh ri rk rl rm r� rn rp r �skipifr ru r? r? r? r@ re � s re 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 rY )r9 rK r r( )rR r� r� �x0r? r? r@ rX # 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 r )r� rM r )r� rD )r� )r� r� ) rQ r9 r: rs rP rk r( r� r ) rR �size�xp�ypZincptsZdecptsZincresZdecresZinctgtZdectgtr? 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 rK r r( r� r� r� )rR r� r� rz 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 )rx )r9 rK r r( ) rR r� r� rz �y0r� 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 rK rk r r( �object)rR r� r� rz 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 r g-C ��6?g )r9 r: �sinr r( r` )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 r rH rC g @g @g �!@g @g @g @g @g @ih )rx r� )r� rD rM r� )r r9 r( rk r` )rR r� r| r� r� r? r? r@ � test_period� s zTestInterp.test_periodN) rb rc rd ry rX r~ r r� r� r� r� r? r? r? r@ rw s &rw c C s, x&t t|��D ]}t| | || � qW d S )N)rQ r� r )rd r� rr r? r? r@ �compare_results� 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%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 )Nr] g �?r g r� g @rY g �?rE T)r� r� r� �nearest)� interpolation) r9 r: r � percentiler� r� r� r� r: r r� )rR r� r� r? r? r@ rX � 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 r� )rR r� �or? r? r@ �test_api� s zTestPercentile.test_apic C sR t �dddgdddgdddgdddgdddgg�}tt j|ddd�dddg� d S )NrE rC rH rY r )rf )r9 rk r r� )rR r� r? r? r@ �test_2D� s zTestPercentile.test_2Dc C s4 t t�td�d�d� t tjtd�ddd�d� d S )Nr rY g @r� )r� )r r9 r� rQ )rR r? r? r@ � test_linear� s zTestPercentile.test_linearc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nr rY �lower)r� rC �higherrI )r r9 r� 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 ) Nr �3 �midpoint)r� g @r� g @rY rI )r r9 r� rQ )rR r? r? r@ � test_midpoint� s z TestPercentile.test_midpointc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nr r� r� )r� rI �1 rC )r r9 r� rQ )rR r? r? r@ � test_nearest� s zTestPercentile.test_nearestc C s0 t �d�d }tt �|dddg�dddg� d S )Nr] g �?r r� rY g @g �?)r9 r: r r� )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< rY r� g @g @g &@rD rI r^ r_ r] r� r r� r )rf g �?g �?g @g �!@g #@rE ih )r< rY )rD )r< rY �K )rH )rD rC rI r^ )rD rH rI r^ )rD rH rC r^ )rD rH rC rI )rH rH rI r^ r� )r� )rf r� )r9 r: r` r r� rk r� r[ )rR r� �r0ry 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 rY g @g @g @g @g @r )rf g �?g #@rE )r )rf r r� )r� )r� rf g �?g "@r? )r� )r� r rI )r� rf r ) r9 r: r` r r� � assertTrue�isscalarrk r[ r �emptyr� )rR r� r� ry r �c0�c1r� 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 ro )r� �e rM rY g��������)r rO r9 r� 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 r� )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 r^ )rH rH r )rf r )rf )rH rD r� g @g @g @g @g @g @)rD rC )r< rY g �?g @g �!@g �?g @g #@)rD rH r_ )r� r� )r� rf r r] r� ) r9 rk rm r� r r: r` r� r� )rR r� r� r� r� r ry 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 rY r )rf )rE rD rE )r� rD rE )r� rE rE rH )r� rE rD rM rG rJ r� r� )rf r� r )rD rE rD rE )rD r� rD rE )rD r� rE rE )rD r� rE rD )r9 r: r` r r� r[ rk )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 rY F)�overwrite_input)r9 rk r� 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� r� )r� )r9 rK r� r: r �tolist)rR r� 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 rY T)r� g @)r9 rk r� r )rR rS rh 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 )r{ r r� )r rE )rf rM r )rG rM rE )r rM r< r7 )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 rM �normal�dstackr r� �rollaxisrq r� r: r` �shuffle�ravel�flatten)rR r� 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< )rf r� )r r� rC )r rC )rE rE )rM rM )rH rM )r9 rP r r� r� 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)rf �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 )r� )r rE rH )rE rE r_ rE rE )rD rE rE r_ rE )rD rE rI r_ rE )r9 rP r r� r[ )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 r� )r� r )rH rE r? rD )r9 rm rP r r� )rR r� 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 r� )r� r )rH rE r? ) r� r� r� r: r9 rm rP r� r r� )rR r� 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 )rf g333333�?rE rH rC )rE rD rH )rE rE rD )rD rH )rE rD )rE rH )r rD r� )r� )r9 r: r� r� r� r� r� r: r r� rk r r� r` rp )rR rS r� rh r? r? r@ �test_nan_behavior� s� " "" "" "z TestPercentile.test_nan_behaviorN)rb rc rd rX 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@ r� � s. $0) r� 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 r^ rH g �?g @r )rf g �?g @rC g�6-¦?g� �0���?g���}r�?g��4 �?g @rM T)r� r� r� )r9 rk r: r` r �medianr r rp r� r� r� r� r: r r� )rR �a0r� r rS r� r? r? r@ rX 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 r^ r_ rC rI r� )rD rH rC )r{ )rf g @) r9 rk rM �randintr� r� rQ rp r r r� )rR �a3rS �orig�axr? 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 r^ r_ rC rI T)r� g �?g @)r� rf g �?g @r7 )r� )rH rC rI )rf )rf r� ) r9 rk r: r` r r� r� r� r� rM r� r� )rR r� r� 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 )rf )r r9 r� r )rR r� rI 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� r4 r? r? r@ �__new__� s z4TestMedian.test_subclass.<locals>.MySubClass.__new__c S s dS )Ni����r? )rR rf r� r r? r? r@ r� � s z1TestMedian.test_subclass.<locals>.MySubClass.mean)N)NNN)rb rc rd r� r� r? r? r? r@ � MySubClass� s r� rE rD rH i����)r9 r� r r� )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 rm rP r r� )rR r� r� r? r? r@ r� � 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 rm rP r� r r� )rR r� r� r? r? r@ r� � 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 )rf 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 r� r r� r` rp r� )rR rS r� rh r? r? r@ r� � 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 )rf rE ) r9 rk r� r� r� r� r: r r� r� r r� )rR rS r� rh 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� r� r� r� r� r� )rR r� 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 )r{ r )r rE )rf 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 rM r� r� r r� r� rq r� r: r` r� r� r� )rR r� r� r� r? r? r@ r� 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� )rf )r r� rC )r rC )rE rE )r9 rP r r� r� rO )rR r� r? r? r@ r� ( 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)rf r� )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 r� r[ )rR r� r? r? r@ r� 0 s zTestMedian.test_keepdimsN)rb rc rd rX r� r� r� r� r� r� r� r� r� r� r� r� r? r? r? r@ r� ? s ! 0 r� c @ s e Zd Zdd� Zdd� ZdS )�TestAdd_newdoc_ufuncc C s" t ttdd� t tttjd� d S )NrD �blah)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)rb rc rd 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 �core�flatiter�indexrg r� r� �ufunc�identityr* � index_tricks�mgrid)rR r r? r? r@ � test_add_docL s zTestAdd_newdoc.test_add_docN) rb rc rd r rv rq r� �optimizer� r? r? r? r@ r� J s r� �__main__)v� __future__r r r r� r� rq r� �numpyr9 Z numpy.testingr r r r r r r r r r r r r r Znumpy.testing.utilsr Znumpy.lib.function_baser* � function_baser� Z numpy.randomr � 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 � numpy.compatr8 rA rB re rt r{ r~ r� r� r� r� r� r� r� r r r r! r9 r\ rc rh rq r� r� r� r� r� r� r� r r( r) r- r= rN rO rR rS r_ re rw r� r� r� r� r� rb r? r? r? r@ �<module> s� @ � DI pHe K U' NI H e Ub YLWq 2