Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/lib/tests/__pycache__/test_function_base.cpython-37.pyc
Назад
B �5�g� � @ s d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dl m Z d dl m Z d dlZd dlmZ d dlmZmZmZmZmZmZmZmZmZmZm Z m Z d dl m 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD dd � ZEd d � ZFG d d � d �ZGG dd� d�ZHG dd� d�ZIG dd� d�ZJG dd� d�ZKG dd� d�ZLG dd� d�ZMG dd� d�ZNG d d � d �ZOG d d � d �ZPG d d!� d!�ZQG d"d#� d#�ZRG d$d%� d%�ZSG d&d'� d'�ZTG d(d)� d)�ZUG d*d+� d+�ZVG d,d-� d-�ZWG d.d/� d/�ZXG d0d1� d1�ZYG d2d3� d3�ZZdpd5d6�Z[dqd8d9�Z\G d:d;� d;�Z]G d<d=� d=�Z^G d>d?� d?�Z_G d@dA� dA�Z`G dBdC� dC�ZaG dDdE� dE�ZbG dFdG� dG�ZcG dHdI� dI�ZdG dJdK� dK�ZeG dLdM� dM�ZfG dNdO� dO�ZgG dPdQ� dQ�ZhG dRdS� dS�ZiG dTdU� dU�ZjG dVdW� dW�ZkG dXdY� dY�ZlG dZd[� d[�ZmG d\d]� d]�Znd^d_� ZoG d`da� da�ZpG dbdc� dc�ZqG ddde� de�ZrG dfdg� dg�ZsG dhdi� di�ZtG djdk� dk�ZuG dldm� dm�ZvG dndo� do�ZwdS )r� N)�Fraction)�arrays)�ma) �assert_� assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal� assert_raises�assert_allclose�IS_PYPY� assert_warns�assert_raises_regex�suppress_warnings� HAS_REFCOUNT)�rand) �add_newdoc_ufunc�angle�average�bartlett�blackman�corrcoef�cov�delete�diff�digitize�extract�flipud�gradient�hamming�hanning�i0�insert�interp�kaiser�meshgrid�msort� piecewise�place�rot90�select�setxor1d�sinc�trapz� trim_zeros�unwrap�unique� vectorizec C s t �| �}t j�||�}|S )N)�np�arange�add�outer)�n�data� r8 �X/home/digitalm-up/venv/lib/python3.7/site-packages/numpy/lib/tests/test_function_base.py�get_mat s r: c C s( t �t �| |�jt j�}| |_||_|S )za Like real + 1j * imag, but behaves as expected when imag contains non-finite values )r2 �zeros� broadcast�shape�complex_�real�imag)r? r@ �retr8 r8 r9 � _make_complex$ s rB c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestRot90c C s� t ttt�d�� t ttt�d�dd� t ttt�d�dd� t ttt�d�dd� t ttt�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�|� �qW x(tddd�D ]}tt||d�|� �q2W x(td dd�D ]}tt||d�|� �q\W ttt|dd�dd�|� tt|d dd�t|ddd�� d S )N� )� rE rE )r � rE )�axes)rE rE )r rE )rF rF )���rF r rF rE � � ���� )�krH ���)r rF )rF r )rM rG )r � ValueErrorr) r2 �ones�ranger )�self�a�b1�b2Zb3Zb4rM r8 r8 r9 � test_basic0 s6 zTestRot90.test_basicc C sR t �d�}tt|�jd� tt|dd�t|dd�� tt|dd�t|dd�� d S )N)�2 �( rI )rX rW rI )r rE )rG )r rN )rF rE )rH rN )r2 rP r r) r= )rR rS r8 r8 r9 � test_axesP 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� )rE rE rE rE rI � � r rF rD rJ )r rF )rG )rF r )rF rE )rE r )rM rG )r2 r3 �reshaper r) rQ )rR rS Z a_rot90_01Z a_rot90_12Z a_rot90_20Z a_rot90_10rM r8 r8 r9 �test_rotation_axesV s. z TestRot90.test_rotation_axesN)�__name__� __module__� __qualname__rV rY r^ r8 r8 r8 r9 rC / s rC c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestFlipc C sl t tjtjt�d�dd� t tjtjt�d�dd� t tjtjt�d�dd� t tjtjt�d�dd� d S )NrD rF )�axis)rD rD rE rK )r rI )r r2 � AxisError�fliprP )rR r8 r8 r9 rY u s zTestFlip.test_axesc C sn t d�}|d d �d d d�f }tt�|d�|� dddgdddgg}dddgdddgg}tt�|d�|� d S )NrD rN rF r rE rI rJ )r: r r2 re )rR rS �br8 r8 r9 � test_basic_lr{ s zTestFlip.test_basic_lrc C sn t d�}|d d d�d d �f }tt�|d�|� dddgdddgg}dddgdddgg}tt�|d�|� d S )NrD rN r rF rE rI rJ )r: r r2 re )rR rS rf r8 r8 r9 � test_basic_ud� s zTestFlip.test_basic_udc C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rF rE rI rD rJ r[ r\ )r2 �arrayr re )rR rS rf r8 r8 r9 �test_3d_swap_axis0� s zTestFlip.test_3d_swap_axis0c C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rF rE rI rD rJ r[ r\ )r2 ri r re )rR rS rf r8 r8 r9 �test_3d_swap_axis1� s zTestFlip.test_3d_swap_axis1c C sb t �ddgddggddgddggg�}t �ddgddggddgddggg�}tt �|d�|� d S ) Nr rF rE rI rD rJ r[ r\ )r2 ri r re )rR rS rf r8 r8 r9 �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 rE rI rD rJ r ) r2 r3 r] rQ �ndimr re r �swapaxes)rR rS �ir8 r8 r9 �test_4d� s zTestFlip.test_4dc C sH t �dddgdddgg�}t �dddgdddgg�}tt �|�|� d S )NrF rE rI rD rJ r[ )r2 ri r re )rR rS rf r8 r8 r9 �test_default_axis� s zTestFlip.test_default_axisc C s� t �ddgddggddgddggg�}tt j|d d �|� t �ddgddggddgddggg�}tt j|d d �|� t �ddgddggddgddggg�}tt j|d d �|� d S ) Nr rF rE rI rD rJ r[ r\ r8 )rc )r rE )rF rE )r2 ri r re )rR rS rf �cr8 r8 r9 �test_multiple_axes� s zTestFlip.test_multiple_axesN) r_ r` ra rY rg rh rj rk rl rq rr rt r8 r8 r8 r9 rb s s rb c @ s e Zd Zdd� Zdd� ZdS )�TestAnyc C sT ddddg}ddddg}ddddg}t t�|�� t t�|�� t t�|� � d S )Nr rF )r r2 �any)rR �y1�y2�y3r8 r8 r9 rV � s zTestAny.test_basicc C sb dddgdddgdddgg}t t�|�� ttj|dd�dddg� ttj|dd�dddg� d S )Nr rF )rc )r r2 rv r Zsometrue)rR rw r8 r8 r9 �test_nd� s zTestAny.test_ndN)r_ r` ra rV rz r8 r8 r8 r9 ru � s ru 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 rF )r r2 �allri )rR rw rx ry r8 r8 r9 rV � s zTestAll.test_basicc C sd dddgdddgdddgg}t t�|� � ttj|dd�dddg� ttj|dd�dddg� d S )Nr rF )rc )r r2 r| r Zalltrue)rR rw r8 r8 r9 rz � s zTestAll.test_ndN)r_ r` ra rV rz r8 r8 r8 r9 r{ � s r{ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestCopyc C sR t �ddgddgg�}t �|�}t||� d|d< t|d d� t|d d� d S )NrF rE rI rD � )r r )r2 ri �copyr r )rR rS �a_copyr8 r8 r9 rV � 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 )NrF rE rI rD �F)�order)r2 ri r �flags� c_contiguous� f_contiguousr )rR rS Za_fortr� Z a_fort_copyr8 r8 r9 � test_order s zTestCopy.test_orderc C sV t �d�}tt �tj|dd�� � tt �tj|dd��� tt �t�|�� � d S )NrJ F)ZsubokT)r rP r Z isMaskedArrayr2 r )rR Zmxr8 r8 r9 � test_subok s zTestCopy.test_subokN)r_ r` ra rV r� r� r8 r8 r8 r9 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�� d S )NrF rE rI r )rc g @g �?g @g )rD rD )r rF )rF r rJ )r2 ri r r rP r �meanr )rR rw rx ry Zy4Zy5r8 r8 r9 rV " 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~ )�weightsrE g �?rF rI rD rJ r[ r )r� rc g @g @g @g @)r2 r3 r �sumr ri r r r �astype�float32�float64r �dtypeZ result_type) rR �y�w�actual�desiredrw �w0�w1�w2ry Zw3r8 r8 r9 � test_weights4 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 )NrF rE rI rD rJ r[ T)�returnedg @r g @g @)r� rc r� g �?)r2 ri r r r )rR r� �avgZsclr� r� r� r8 r8 r9 � test_returnedT s zTestAverage.test_returnedc C sz G dd� dt j�}t �ddgddgg��|�}t �ddgddgg��|�}ttt �|��|� ttt j||d��|� d S )Nc @ s e Zd ZdS )z-TestAverage.test_subclasses.<locals>.subclassN)r_ r` ra r8 r8 r8 r9 �subclasso s r� rF rE rI rD )r� )r2 �ndarrayri �viewr �typer )rR r� rS r� r8 r8 r9 �test_subclassesn 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� rF rE rI rD )r� )r� )r2 ri r r r� )rR Ztyps�at�wt�rtrS r� r8 r8 r9 �test_upcastingw 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 r8 )�decimal�Decimal)�.0�xr8 r8 r9 � <listcomp>� s z1TestAverage.test_object_dtype.<locals>.<listcomp>r~ c S s g | ]}t �d ��qS )rF )r� r� )r� �_r8 r8 r9 r� � s r )r� )r2 ri rQ r� r r� r )rR rS r� r8 r8 r9 �test_object_dtype s z TestAverage.test_object_dtypeN) r_ r` ra rV r� r� r� r� r� r8 r8 r8 r9 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 ) � TestSelectrF rE rI rD rJ r[ r\ rZ � FTr c s@ g }x6t t|��D ]&� |� fdd�t||�D �p4|g7 }qW |S )Nc s g | ]\}}|� r|� �qS r8 r8 )r� �V�C)�mr8 r9 r� � s z&TestSelect._select.<locals>.<listcomp>)rQ �len�zip)rR Zcond�values�default�outputr8 )r� r9 �_select� s &zTestSelect._selectc C sL | j }| j}tt||dd�| j||dd�� tt|�d� tt|�d� d S )N� )r� rI )�choices� conditionsr r* r� r r� )rR r� r� r8 r8 r9 rV � s zTestSelect.test_basicc C sf t �d�t �dddg�g}dt �d��dd�g}tt||�t �d�� ttdgdgdgd �jd � d S ) NTFrF � rD rI )rD rI r )r� )rF ) r2 ri r3 r] r r* rP r r= )rR r� r� r8 r8 r9 �test_broadcasting� 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 r8 )r� r2 �int8)r� �choicer8 r8 r9 r� � s z0TestSelect.test_return_dtype.<locals>.<listcomp>rF rE rI rJ r\ r ) r r* r� r� r� r2 r> r� ri �nan�isnan)rR r� �dr� r8 r8 r9 �test_return_dtype� s z TestSelect.test_return_dtypec C s" t ttg g d� t ttg g � d S )Ny @)r rO r* )rR r8 r8 r9 �test_deprecated_empty� s z TestSelect.test_deprecated_emptyc C sj | j }| jd d � }|d �tj�|d< ttt||� |d �tj�|d< ttt||� ttt||� d S )Nr ) r� r� r� r2 �int_r � TypeErrorr* �uint8)rR r� r� r8 r8 r9 �test_non_bool_deprecation� s z$TestSelect.test_non_bool_deprecationc C s2 t �dg�gd }t �dg�gd }t||� d S )NF�d rF )r2 ri r* )rR r� r� r8 r8 r9 �test_many_arguments� s z TestSelect.test_many_argumentsN)r )r_ r` ra r2 ri r� r� r� rV r� r� r� r� r� r8 r8 r8 r9 r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� 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 )NrF rE rI r rN r� r\ rZ )r� g g �?T)�record�always� rD ) r r" �slicer2 ri r� �warnings�catch_warnings�filterwarnings� FutureWarningr �category)rR rS rf r� r8 r8 r9 rV � 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 ) NrF rE r )rc rI rD � )rE rI rD rN rH ���) r r" r2 ri r3 �repeatr] Z concatenate�Tr rd )rR rS �rrf rs r8 r8 r9 � test_multidim� s> , $ "22"$"$(("$"zTestInsert.test_multidimc C s\ t �d�}t�t j�� t|g ddd� W d Q R X t�t�� t|g ddd� W d Q R X d S )NrF rE r )rc �nonsense)r2 ri �pytest�raisesrd r" r� )rR rS r8 r8 r9 �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` ra r8 r8 r8 r9 �SubClass s r� r~ r rF rE rH rN ) r2 r� r3 r� r � isinstancer" r� ri )rR r� rS r8 r8 r9 � test_subclass s " zTestInsert.test_subclassc C sD t �dddg�}t �dddg|dddg� t|t �dddg�� d S )NrF r rE rI rD rJ )r2 ri r" r )rR r� r8 r8 r9 �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)rF rS )rE rf )rI rs )�foorp )�bar�a1)r� )rD r� r rE rI )r2 ri r" r r� )rR rS �valrf r8 r8 r9 �test_structured_array s z TestInsert.test_structured_arrayc C sp t �t��( t�dddgt�ddg�ddg� W d Q R X t �t��$ t�dddgtjg td�g � W d Q R X d S ) Nr rF rE g �?g @r~ � )r� )r� r� � IndexErrorr2 r" ri �float)rR r8 r8 r9 �test_index_floats& s , z TestInsert.test_index_floatsN) r_ r` ra rV r� r� r� r� r� r� r8 r8 r8 r9 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 )NrI rD rJ r~ rK ���g @g $@g "@g @rZ g @g @r )rc g @rF )r r2 Zamax)rR rS rf r8 r8 r9 rV / s zTestAmax.test_basicN)r_ r` ra rV r8 r8 r8 r9 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 )NrI rD rJ r~ rK r� g @g �g "@g $@g @rZ g @g @r )rc rF g @)r r2 Zamin)rR rS rf r8 r8 r9 rV ; s zTestAmin.test_basicN)r_ r` ra rV r8 r8 r8 r9 r� 9 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� t|jddd�d ddgg� t|jddd�dgg� d S )NrI rD rJ r~ rK r� g @r )rc g .@g "@g $@g @rZ g @g @g @rN T)rc �keepdims)r rF g @)r2 ri r Zptp)rR rS rf r8 r8 r9 rV G s zTestPtp.test_basicN)r_ r` ra rV r8 r8 r8 r9 r� E 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 )NrF rE r~ � r[ rJ rD rI r\ r� rL r� � �# �' r )rc rZ � � � � � � )r2 r� r� �int16�uint16�int32�uint32r� r� � complex64� complex128ri r �cumsum)rR �ba�ba2�ctyperS �a2�tgtr8 r8 r9 rV V s " &&zTestCumsum.test_basicN)r_ r` ra rV r8 r8 r8 r9 r� T 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 t j |� t t t j |d� qZt |j d d �d� t|j d d �t � ddddg|�� t|j dd �t � dddg|�� qZW d S )NrF rE r~ r� r[ rJ rD rI r\ r� )�1rf r )rc i g rW �$ �T � rN r� ib iX )r2 r� r� r� r� r� r� r r ri r �ArithmeticError�prodr r )rR r r r rS r r8 r8 r9 rV m s " zTestProd.test_basicN)r_ r` ra rV r8 r8 r8 r9 r k s r c @ s e Zd Zdd� ZdS )� TestCumprodc C sT dddddddg}ddddgddd d gddddgg}�xt jt jt jt jt jt jt jt jgD ]�}t � ||�}t � ||�}|d kr�t t t j |� t t t j |d� t 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 ) NrF rE r~ r� r[ rJ rD rI r\ r� )r rf rN )rc r� �� i( i� i g r r� � r rW r r r� r� �� ib rm iX )r2 r� r� r� r� r� r� r r ri r r Zcumprodr )rR r r r rS r r8 r8 r9 rV � s* " zTestCumprod.test_basicN)r_ r` ra rV r8 r8 r8 r9 r s r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestDiffc C s� dddddg}t �ddddg�}t �d d dg�}t �d dg�}tt|�|� tt|dd �|� tt|dd �|� d d dddg}t �d dddg�}tt|�|� ddddg}t �dddg�}t �ddg�}tt|�|� tt|dd �|� d S )NrF rD r[ r\ r� rI rE rJ rN r )r6 g�������?g������@g @g������ɿg��������g�������?g������ �g�������?TF)r2 ri r r r )rR r� �out�out2�out3r8 r8 r9 rV � s zTestDiff.test_basicc C s� t �d�}d|d d �dd d�d d �f< t �d�}d|d d �dd d�d d �f< tt|�t �d�� tt|dd�t �d�� tt|dd�t �d �� tt|dd�|� tt|d d�|� tt jt|d d� tt jt|d d� t �d t j�}tt t|� d S )N)r~ r� r� rF rE )r~ � r� rN )r~ r� � )rc r )r� r� r� rH rI r� g��q ��?) r2 r; rP r r r rd ri r� rO )rR r� �expr8 r8 r9 � test_axis� s zTestDiff.test_axisc 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� rF rN rE )r6 r )rc )r6 rc )r r r )rR r� Zout1r r Zout4r8 r8 r9 rz � s 4444zTestDiff.test_ndc s� t td��� ttt� dd� � fdd�tdd�D �}ddgdgg g g}tt� dd�� k� xhtt||�dd �D ]R\}\}}tt|�t j k� t ||� t |j t j� t t|�tdt� �| �� qnW d S ) NrI rN )r6 c s g | ]}t � |d ��qS ))r6 )r )r� r6 )r� r8 r9 r� � s z#TestDiff.test_n.<locals>.<listcomp>rF rJ r )�start)�listrQ r rO r r � enumerater� r� r2 r� r r r� r� r� �max)rR r� �expectedr6 r r8 )r� r9 �test_n� s zTestDiff.test_nc C s� t jddt jd�}t jddgdd�t jdgdd�g}|�t jg dd�gd � x<t|dd�D ],\}}t||d �}t||� t|j |j � q\W d S ) Nz 1066-10-13z 1066-10-16)r� rF ztimedelta64[D]r rI )r )r6 ) r2 r3 Z datetime64ri �extendr r r r r� )rR r� r r6 r r r8 r8 r9 � test_times� s zTestDiff.test_timesc C s� t jddgddgddgddgd d ggd d gd d gd d gd d gd d ggd �}t|�}t|jdgdgdgdgdgg� t|jd gd gd gd gd gg� tt|�t|�k� t|dd�}t|jg g g g g g� t|jg g g g g g� tt|�t|�k� d S )NrF rE rI rD rJ r[ r\ rZ r� r~ FT)�mask)r6 )r ri r r r7 r# r r� )rR r� r r r8 r8 r9 r� � s $ zTestDiff.test_subclassc C sN t �d�d }tt|dd�t �d�� tt|dgd�t �d�� tt �t j|dd��|� tt|ddgd�t �d�� t �d��dd�}t j|ddd �}ddgddgg}t||� t j|ddgdggd �}t||� t j|ddd �}ddgddgg}t||� t j|dddggd �}t||� ttt j|t � d �d� tt j t|dd d � d S ) NrJ rF r )�prependrN r[ rD rE )rc r$ )rI rI rI )r$ rc ) r2 r3 r r rP r r] r rO r; rd )rR r� �resultr r8 r8 r9 � test_prepend� s$ zTestDiff.test_prependc C sB t �d�}t|dd�}dddddg}t||� t|dgd�}t||� t|ddgd�}|dg }t||� t �d��dd�}t j|ddd�}dd gdd gg}t||� t j|ddgdggd�}t||� t j|ddd�}ddgd d gg}t||� t j|dddggd�}t||� ttt j|t �d �d� tt jt|dd d� d S )NrJ r )�appendrF r� rE rD )rc r' rN rK rH )rI rI rI )r' rc ) r2 r3 r r r] r rO r; rd )rR r� r% r r8 r8 r9 � test_append s, zTestDiff.test_appendN) r_ r` ra rV r rz r r" r� r& r( r8 r8 r8 r9 r � s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestDeletec C s, t �d�| _t �d��d��ddd�| _d S )NrJ rE rF )r2 r3 rS r� r] �nd_a)rR r8 r8 r9 �setup( s zTestDelete.setupc C s� t | j|�}t | j|dd�}d| }tt|| j|f �| j|d� t|dd d �df | jd|df �}t|| jdd d �df |d� d S )NrF )rc zDelete failed for obj: %r)�err_msgr )r rS r* r r+ )rR �indicesZa_delZnd_a_del�msg�xorr8 r8 r9 �_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����rH r rF rE rD rJ rK rN rI )r� r0 )rR ZlimsZstepsr �stop�step�sr8 r8 r9 � test_slices5 s zTestDelete.test_slicesc C s� | � t�ddgddgg�� t�t�� t| jdg� W d Q R X t�t�� t| jdg� W d Q R X | � ddddg� | � dddddg� t�t�� t| jd� W d Q R X t�t�� t| jd� W d Q R X t�t�� t| jdgd � W d Q R X d S ) Nr rF rE r� i����rN TFrD ) r0 r2 ri r� r� r� r rS rO )rR r8 r8 r9 � test_fancy> s zTestDelete.test_fancyc C s | � d� | � d� d S )Nr r� )r0 )rR r8 r8 r9 � test_singleS s zTestDelete.test_singlec C sX t �d�}t�t j�� t|g dd� W d Q R X t�t�� t|g dd� W d Q R X d S )NrF r )rc r� )r2 ri r� r� rd r r� )rR rS r8 r8 r9 r� W 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` ra r8 r8 r8 r9 r� _ s r� r rF rE rH )r2 r� rS r� r r� r r� )rR r� rS r8 r8 r9 r� ^ s zTestDelete.test_subclassc C sR t �d�jdddd�}t|tdd �dd�}t|jj|jj� t|jj|jj� d S ) Nr~ rE rJ r� )r� �< rF )rc ) r2 r3 r] r r� r r� r� r� )rR rM r� r8 r8 r9 �test_array_order_preserveh s z$TestDelete.test_array_order_preservec C sh t �t��" t�dddgt�ddg�� W d Q R X t �t��" t�dddgtjg td�� W d Q R X d S )Nr rF rE g �?g @)r� )r� r� r� r2 r ri r� )rR r8 r8 r9 r� r s & z TestDelete.test_index_floatsN) r_ r` ra r+ r0 r4 r5 r6 r� r� r8 r� r8 r8 r8 r9 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 ej�dejejejejg�dd� �Zej�dejejejejg�dd� �Zej�d ejejejejg�d d � �Z ej�d ejejejejg�d d � �Z d!S )"� TestGradientc C sf ddgddgg}t �|�}t �ddgddgg�t �ddgddgg�g}tt|�|� tt|�|� d S )NrF rI rD g @g @g g �?)r2 ri r r )rR �vr� �dxr8 r8 r9 rV { 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 )NrJ g �?g @g @g "@g &@� g @g �?rE rF )rc z.*scalars or 1drN ) r2 r rP r3 r] r ri r rO �stack)rR r; Z dx_uneven�f_2dr8 r8 r9 � 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< rJ rE rF )r rF )rc ) r2 r3 r] r rP r rO r r� )rR r> r� r8 r8 r9 � 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� rK r � �= i� i� ztimedelta64[D])r2 ri r r r r� )rR r� r; r8 r8 r9 �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 |j ddd ddg� d S ) NrF rI rD F)r# r rJ rE )� edge_orderT) r2 r ri r r r� r Z_maskr3 �maskedr r# )rR r� r �x2r8 r8 r9 � 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 rF r~ rE rI rD r[ rZ )rD g� ��Q��?T) r2 �linspace�absr r r| �random�seed�sort)rR r� r; r� Z analyticalZ num_errorr8 r8 r9 �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[ rF rN rF g g �?g �?g @g @g �?g @g �g333333�?g �?g пrE )r rF )rc rD )rF r ) r2 ri Ztiler] r3 r r r r� r r= ) rR �fZx_unevenZx_evenZ fdx_even_ord1Z fdx_even_ord2Zfdx_uneven_ord1Zfdx_uneven_ord2rD Zexp_res�res1�res2�res3r8 r8 r9 � 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 )NrF rI rD g @g @g g �?r )rc rN )rF r rE rK )r2 ri r r r r r� rd )rR r: r� r; r8 r8 r9 �test_specific_axes 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� rK r~ r� rB iA i, ztimedelta64[D])r� rE r\ r< � �w i��)r2 ri r r r r� )rR r� r; r8 r8 r9 �test_timedelta645 s z TestGradient.test_timedelta64c C sJ xDt jt jt jgD ]0}t jdddg|d�}tt|�jt �|�j� qW d S )NrF rE rI )r� ) r2 Zfloat16r� r� ri r r r� r )rR �dtr� r8 r8 r9 �test_inexact_dtypes@ s z TestGradient.test_inexact_dtypesc 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 )NrE rF )rD rI r )r r2 r3 r rO )rR r8 r8 r9 � test_valuesF s zTestGradient.test_values�f_dtypec C s8 t jdddddg|d�}t|�}t|dgt|� � d S )NrJ rD rI rE rF )r� rN )r2 ri r r r� )rR rZ rN �gr8 r8 r9 � test_f_decreasing_unsigned_intR s z+TestGradient.test_f_decreasing_unsigned_intc C sR t �|�j}t �ddg�}t jd|g|d�}t||�}t||d d gd � d S )NrF rI rN )r� rE )r2 �iinfor ri r r )rR rZ �maxintr� rN �dfdxr8 r8 r9 �test_f_signed_int_big_jumpY s z'TestGradient.test_f_signed_int_big_jump�x_dtypec C sF t jdddg|d�}t �dddg�}t||�}t|dgt|� � d S )NrI rE rF )r� r rD rH )r2 ri r r r� )rR ra r� rN r_ r8 r8 r9 �test_x_decreasing_unsignedb s z'TestGradient.test_x_decreasing_unsignedc C sX t �|�j}t �|�j}t jd|g|d�}t �|d dg�}t||�}t|ddg� d S )NrN )r� rE r g �?)r2 r] �minr ri r r )rR ra Zminintr^ r� rN r_ r8 r8 r9 �test_x_signed_int_big_jumpj s z'TestGradient.test_x_signed_int_big_jumpN) r_ r` ra rV r? r@ rC rG rM rR rS rV rX rY r� �mark� parametrizer2 r� r� r� Zuint64r\ r� r� r� Zint64r` rb rd r8 r8 r8 r9 r9 y s& @ r9 c @ s e Zd Zdd� 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|d d�}t �|�d t j }t||d� t||d� d S )Ny �? @rE g @y �?rF rN y � �y �? �y � @g @g �?r T)�degr r� )r2 �sqrtr �arctan�piri r )rR r� r� Zyo�zZzor8 r8 r9 rV w s zTestAngle.test_basicc C s� t j�ddt �d�d d g�}t jj|d< t j�t �d�dt �d�g�}t jj|d< t|�}tt|�t|�� t|j |j � t||� d S )Ny �? @rF rE y �? �?g @r g �?) r2 r ri ri rE rj r r r� r# )rR r� r r� r8 r8 r9 r� � s zTestAngle.test_subclassN)r_ r` ra rV r� r8 r8 r8 r9 rg u s rg c @ s� e Zd Ze�ddddddddg�Ze�e�Ze�e �Z e�e �Z dd� Z dd � Zd d � Zd d � Zdd� Zdd� Zej�de�dddg�e�dddg�e�dddg�g�dd� �Zdd� Zdd� Zd S ) � TestTrimZerosr rF rE rI rD c s d}� fdd�|D �S )N)rS rf rs r� c 3 s | ]}t � |�V qd S )N)�getattr)r� �name)rR r8 r9 � <genexpr>� s z'TestTrimZeros.values.<locals>.<genexpr>r8 )rR Z attr_namesr8 )rR r9 r� � s zTestTrimZeros.valuesc C s: t jdd� }x&| �� D ]}t|�}t||| � qW d S )NrE rN )r2 �s_r� r. r )rR �slc�arr�resr8 r8 r9 rV � s zTestTrimZeros.test_basicc C s> t jd d� }x*| �� D ] }t|dd�}t||| � qW d S )NrN rf )�trim)r2 rq r� r. r )rR rr rs rt r8 r8 r9 �test_leading_skip� s z TestTrimZeros.test_leading_skipc C s> t jdd � }x*| �� D ] }t|dd�}t||| � qW d S )NrE r� )ru )r2 rq r� r. r )rR rr rs rt r8 r8 r9 �test_trailing_skip� s z TestTrimZeros.test_trailing_skipc C s^ xX| � � D ]L}tj||jd�}t|dd�}t|�dks:t�t|dd�}t|�dks t�q W d S )N)r� �B)ru r rN )r� r2 Z zeros_liker� r. r� �AssertionError)rR Z_arrrs rO rP r8 r8 r9 � test_all_zero� s zTestTrimZeros.test_all_zeroc C s t �d�}t|�}t||� d S )Nr )r2 r; r. r )rR rs rt r8 r8 r9 �test_size_zero� s z TestTrimZeros.test_size_zerors l l l c C s( t jdd� }t|�}t||| � d S )NrF rE )r2 rq r. r )rR rs rr rt r8 r8 r9 � test_overflow� s zTestTrimZeros.test_overflowc C s& t �d dd g�}t|�}t||� d S )NrF )r2 ri r. r )rR rs rt r8 r8 r9 � test_no_trim� s zTestTrimZeros.test_no_trimc C s t | j�� �}t|t�s t�d S )N)r. rS �tolistr� r ry )rR rt r8 r8 r9 �test_list_to_list� s z TestTrimZeros.test_list_to_listN)r_ r` ra r2 ri rS r� r� rf �complexrs �objectr� r� rV rv rw rz r{ r� re rf r| r} r r8 r8 r8 r9 rm � s" rm c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestExtinsc C s> t �dddddddg�}t|dk|�}t|dddddg� d S )NrF rI rE )r2 ri r r )rR rS rf r8 r8 r9 rV � 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 )NrF rE rI TFr rD rJ rZ r\ r[ r� z!Cannot insert from an empty arrayc s t � dddddddgg �S )Nr rF )r( r8 )rS r8 r9 �<lambda>� � z'TestExtins.test_place.<locals>.<lambda>Z12Z34�9) r r� r( r2 ri r r; r3 r rO )rR r8 )rS r9 � 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 r# �acrs r8 r8 r9 � test_both� s zTestExtins.test_bothN)r_ r` ra rV r� r� r8 r8 r8 r9 r� � s r� � �?c C s |t �| � S )N)�math�floor)r� r� r8 r8 r9 �_foo1� s r� � c C s |t �| � | S )N)r� r� )r� r� rl r8 r8 r9 �_foo2� 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/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dES )F� TestVectorizec C s@ dd� }t |�}|ddddgdddd g�}t|dddd g� d S ) Nc S s | |kr| | S | | S d S )Nr8 )rS rf r8 r8 r9 � addsubtract s z.TestVectorize.test_simple.<locals>.addsubtractr rI r[ r� rF rJ r\ rE )r1 r )rR r� rN r� r8 r8 r9 � 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 )Nr8 )rS rf r8 r8 r9 r� s z.TestVectorize.test_scalar.<locals>.addsubtractr rI r[ r� rJ rZ rF rD )r1 r )rR r� rN r� r8 r8 r9 � test_scalar s zTestVectorize.test_scalarc C s0 t �ddd�}tdd� �}||�}t||� d S )NrK rE i' c S s | S )Nr8 )r� r8 r8 r9 r� r� z*TestVectorize.test_large.<locals>.<lambda>)r2 rH r1 r )rR r� rN r� r8 r8 r9 � test_large s zTestVectorize.test_largec C sR t tj�}t�ddtj tjdtj dtj g�}||�}t�|�}t||� d S )Nr g �?g �?rE )r1 r� �cosr2 ri rk r )rR rN �args�r1�r2r8 r8 r9 � test_ufunc s ( zTestVectorize.test_ufuncc C sl ddd�}t |�}t�dddg�}||�}t�dddg�}t||� ||d�}t�dddg�}t||� d S ) NrF c S s | | S )Nr8 )rS rf r8 r8 r9 r� ( s z(TestVectorize.test_keywords.<locals>.foorE rI rD rJ )rF )r1 r2 ri r )rR r� rN r� r� r� r8 r8 r9 � test_keywords& s zTestVectorize.test_keywordsc C s: t ttgd�}|t�d�d�}|t�d��}t||� d S )N)�otypesg @g �?)r1 r� r� r2 r3 r )rR rN r� r� r8 r8 r9 � test_keywords_with_otypes_order14 s z.TestVectorize.test_keywords_with_otypes_order1c C s: t ttgd�}|t�d��}|t�d�d�}t||� d S )N)r� g @g �?)r1 r� r� r2 r3 r )rR rN r� r� r8 r8 r9 � test_keywords_with_otypes_order2> s z.TestVectorize.test_keywords_with_otypes_order2c C sT t ttgd�}|t�d��}|t�d�dd�}|t�d��}t||� t||� d S )N)r� g @g �?)r� )r1 r� r� r2 r3 r )rR rN r� r� �r3r8 r8 r9 � test_keywords_with_otypes_order3I s z.TestVectorize.test_keywords_with_otypes_order3c C sd t ttgd�}|ddd�}|ddd�}|d�}t|tddd�� t|tddd�� t|td�� d S )N)r� g������$@r� )rl rN )r� )r1 r� r� r )rR rN r� r� r� r8 r8 r9 �+test_keywords_with_otypes_several_kwd_args1U s z9TestVectorize.test_keywords_with_otypes_several_kwd_args1c C sR t ttgd�}|dddd�}|ddd�}t|tdddd�� t|tddd�� d S ) N)r� r� g������$@rN )rl r� r� rF rE rI )r1 r� r� r )rR rN r� r� r8 r8 r9 �+test_keywords_with_otypes_several_kwd_args2b s z9TestVectorize.test_keywords_with_otypes_several_kwd_args2c C s6 dd l }yt|j� W n tk r0 t� �Y nX d S )Nr )rJ r1 � randrange� Exceptionry )rR rJ r8 r8 r9 �test_keywords_no_func_codem 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 ) NrF c S s | | S )Nr8 )rS rf r8 r8 r9 r� z s z5TestVectorize.test_keywords2_ticket_2100.<locals>.foorE rI )rS rD )rf rS )rf rJ )rF )r1 r2 ri r )rR r� rN r� r� r� r8 r8 r9 �test_keywords2_ticket_2100w 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 )r �pop)r� �p�_prt r8 r8 r9 � mypolyval� s z;TestVectorize.test_keywords3_ticket_2100.<locals>.mypolyvalr� rF )�excludedrI r[ r rE )r� r� )r� )r2 r1 r )rR r� ZvpolyvalZansr8 r8 r9 �test_keywords3_ticket_2100� 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 �?r8 )�kwrt Z_kr8 r8 r9 rN � s z3TestVectorize.test_keywords4_ticket_2100.<locals>.frF rE rI rD )rS rf rZ )r1 r )rR rN r8 r8 r9 �test_keywords4_ticket_2100� s z(TestVectorize.test_keywords4_ticket_2100c C s, t dd� �}t|ddgddg�ddg� d S )Nc W s t �| �S )N)r2 r )r: r8 r8 r9 rN � s z3TestVectorize.test_keywords5_ticket_2100.<locals>.frF rE rI rD rZ )r1 r )rR rN r8 r8 r9 �test_keywords5_ticket_2100� s z(TestVectorize.test_keywords5_ticket_2100c C s dd� }t |�}t|� d� d S )Nc S s dS )NrF r8 r8 r8 r8 r9 r� � s z5TestVectorize.test_coverage1_ticket_2100.<locals>.foorF )r1 r )rR r� rN r8 r8 r9 �test_coverage1_ticket_2100� 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 documentationr8 )r� r8 r8 r9 r� � s z3TestVectorize.test_assigning_docstring.<locals>.foozProvided documentation)�doc)r1 r �__doc__)rR r� rN r� r8 r8 r9 �test_assigning_docstring� 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>.FoorE c S s || j S )N)rf )rR rS r8 r8 r9 r� � s z=TestVectorize.test_UnboundMethod_ticket_1156.<locals>.Foo.barN)r_ r` ra rf r� r8 r8 r8 r9 �Foo� s r� r� rE )r r1 r� r2 r3 )rR r� r8 r8 r9 � 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 )Nr8 )r� r8 r8 r9 r� � r� z@TestVectorize.test_execution_order_ticket_1487.<locals>.<lambda>rI g�������?c S s | S )Nr8 )r� r8 r8 r9 r� � r� )r1 r2 r3 r )rR �f1Zres1aZres1b�f2Zres2bZres2ar8 r8 r9 � 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 )Nr8 )r� r8 r8 r9 r� � r� z7TestVectorize.test_string_ticket_1892.<locals>.<lambda>Zd0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789)r2 r1 r )rR rN r3 r8 r8 r9 �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 rF rE r8 )r� )�_callsr8 r9 rN � s z#TestVectorize.test_cache.<locals>.fTrJ )r1 �cacher2 r3 r r r� )rR rN r� r8 )r� r9 � test_cache� s zTestVectorize.test_cachec C s0 t �dd� �}d|_t �d�}t||�|� d S )Nc S s | S )Nr8 )r� r8 r8 r9 r� � r� z+TestVectorize.test_otypes.<locals>.<lambda>rp rJ )r2 r1 r� r3 r )rR rN r� r8 r8 r9 � 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� r8 z (x,y)->())r� r� z (x),(y)->())r� z(x)->(y)z (x)->(y),()z(),(a,b,c),(d)->(d,e))rS rf rs )r� )r� �ez (x)(y)->()z (x),(y)->z ((x))->(x))r �nfbZ_parse_gufunc_signaturer rO )rR r8 r8 r9 �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 )Nr8 )rS rf r8 r8 r9 r� s z8TestVectorize.test_signature_simple.<locals>.addsubtractz (),()->())� signaturer rI r[ r� rF rJ r\ rE )r1 r )rR r� rN r� r8 r8 r9 �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 r8 r8 r9 r� s z4TestVectorize.test_signature_mean_last.<locals>.meanz(n)->())r� rF rI rE rD )r1 r )rR r� rN r� r8 r8 r9 �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 r8 r8 r9 �center s z3TestVectorize.test_signature_center.<locals>.centerz(n)->(n))r� rF rI rE rD rN )r1 r )rR r� rN r� r8 r8 r9 �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 )Nr8 )r� r8 r8 r9 r� r� z:TestVectorize.test_signature_two_outputs.<locals>.<lambda>z ()->(),())r� rF rE rI r )r1 r r� �tupler� r )rR rN r� r8 r8 r9 �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� rF rE rI rD r[ r )r1 r2 r5 r )rR rN r� r8 r8 r9 �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 )NrN r8 )r� r8 r8 r9 r� 5 r� z<TestVectorize.test_signature_computed_size.<locals>.<lambda>z(n)->(m))r� rF rE rI rD )r1 r )rR rN r� r8 r8 r9 � test_signature_computed_size4 s z*TestVectorize.test_signature_computed_sizec C sV d dd�}t |ddhd�}t|dddg�ddd g� t|dddgd d �dddg� d S ) NrF c S s | | S )Nr8 )rS rf r8 r8 r9 r� > s z2TestVectorize.test_signature_excluded.<locals>.fooz()->()rf )r� r� rE rI rD r )rf )rF )r1 r )rR r� rN r8 r8 r9 �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 )Nr8 )r� r8 r8 r9 r� F r� z5TestVectorize.test_signature_otypes.<locals>.<lambda>z(n)->(n)r� )r� r� rF rE rI )r1 r r� r2 r )rR rN r� r8 r8 r9 �test_signature_otypesE 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 positionalrF rE z does not have enough dimensionsz$inconsistent size for core dimensionrI z()->())r1 �operatorr4 r r� rO )rR rN r8 r8 r9 � test_signature_invalid_inputsK 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 )NrN r8 )r� r8 r8 r9 r� \ r� z>TestVectorize.test_signature_invalid_outputs.<locals>.<lambda>z(n)->(n))r� z$inconsistent size for core dimensionrF rE rI c S s | S )Nr8 )r� r8 r8 r9 r� a r� z ()->(),()zwrong number of outputsc S s | | fS )Nr8 )r� r8 r8 r9 r� e r� z()->())r1 r rO )rR rN r8 r8 r9 � test_signature_invalid_outputsZ 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 )Nr8 )r� r8 r8 r9 r� k r� z5TestVectorize.test_size_zero_output.<locals>.<lambda>r rJ )r� r� rp c S s | S )Nr8 )r� r8 r8 r9 r� s r� z()->())r� c S s | S )Nr8 )r� r8 r8 r9 r� w r� )r� r� c S s | S )Nr8 )r� r8 r8 r9 r� z r� z(n)->(n)c S s | S )Nr8 )r� r8 r8 r9 r� } r� c S s | gS )Nr8 )r� r8 r8 r9 r� � r� z()->(n)znew output dimensions) r2 r1 r; �intr rO r� r r� )rR rN r� r8 r8 r9 �test_size_zero_outputi s$ z#TestVectorize.test_size_zero_outputN)%r_ r` ra 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� r� r� r� r� r8 r8 r8 r9 r� sD r� c @ sL e Zd ZG dd� d�Zejje dd�ej�ddej fdg�dd � ��Z d S ) � TestLeaksc @ s$ e Zd ZdZdd� Zedd� �ZdS )z TestLeaks.Ar� c G s dS )Nr r8 )rR r� r8 r8 r9 �bound� s zTestLeaks.A.boundc G s dS )Nr r8 )r� r8 r8 r9 �unbound� s zTestLeaks.A.unboundN)r_ r` ra �itersr� � staticmethodr� r8 r8 r8 r9 �A� s r� zPython lacks refcounts)�reasonz name, incrr� )r� r c C s� dd l }t| j|�}|�� z�t�|�}xBt| jj�D ]2}| �� }t� t||�dd�|_ |� t� d��}q6W d }t t�|�|| � xtd�D ] }|� � q�W t t�|�|� W d |�� X d S )Nr rF r~ rJ )�gcrn r� �disable�sys� getrefcountrQ r� r2 Z frompyfuncrN r3 r Zcollect�enable) rR ro �incrr� ZA_func�refcountrp rS r r8 r8 r9 �test_frompyfunc_leaks� s z TestLeaks.test_frompyfunc_leaksN) r_ r` ra r� r� re �skipifr rf r� r� r8 r8 r8 r9 r� � s r� c @ sz e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z ejjdd�dd� �ZdS ) � TestDigitizec C s2 t �dd�}t �dd�}tt||�t �d�� d S )Ni����rJ r� r� )r2 r3 r r )rR r� �binsr8 r8 r9 � test_forward� s zTestDigitize.test_forwardc C s6 t �ddd�}t �ddd�}tt||�t �d�� d S )NrJ i����rN r� r� )r2 r3 r r )rR r� r� r8 r8 r9 � test_reverse� s zTestDigitize.test_reversec C s: t d�}t�|�� |�� d�}tt�t||�dk�� d S )Nr~ r )r r2 rH rc r r r| r )rR r� �binr8 r8 r9 � test_random� s zTestDigitize.test_randomc C sf dddddddg}dddg}dddd dd dg}t t||�|� dddddd dg}t t||d �|� d S ) NrF rJ rD r~ rZ r� r rE rI T)r r )rR r� r� Zdefault_answerZ right_answerr8 r8 r9 �test_right_basic� s z TestDigitize.test_right_basicc C s4 t �dd�}t �dd�}tt||d�t �d�� d S )Ni����rJ rD Tr� )r2 r3 r r )rR r� r� r8 r8 r9 �test_right_open� s z TestDigitize.test_right_openc C s8 t �ddd�}t �ddd�}tt||d�t �d�� d S )NrJ i����rN rD Tr� )r2 r3 r r )rR r� r� r8 r8 r9 �test_right_open_reverse� s z$TestDigitize.test_right_open_reversec C s< t d�}t�|�� |�� d�}tt�t||d�dk�� d S )Nr~ T)r r2 rH rc r r r| r )rR r� r� r8 r8 r9 �test_right_open_random� 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 ) NrN r rF rE FrI TrD )r r r rO )rR r� r� r8 r8 r9 �test_monotonic� s zTestDigitize.test_monotonicc C s> dddg}dddg}t tt||� || }}t tt||� d S )NrF rE y @ �?rI )r r� r )rR r� r� r8 r8 r9 �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` ra r8 r8 r8 r9 r� � s r� rJ rF rI FT)r2 r� r3 r� r r� r )rR r� rS rf r8 r8 r9 �test_return_type� s z TestDigitize.test_return_typec C s& d}t t�||d |d g�d� d S )Nl rF )r r2 r )rR r� r8 r8 r9 � test_large_integers_increasing� s z+TestDigitize.test_large_integers_increasingz8gh-11022: np.core.multiarray._monoticity loses precision)r� c C s& d}t t�||d |d g�d� d S )Nl rF )r r2 r )rR r� r8 r8 r9 � test_large_integers_decreasing� s z+TestDigitize.test_large_integers_decreasingN)r_ r` ra r� r� r� r� r� r� r� r� r� r� r� r� re �xfailr� r8 r8 r8 r9 r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestUnwrapc C sH t tdddtj g�ddg� tt�tttd�d ��tjk �� d S )NrF rE r~ r� )r r/ r2 rk r r| r r )rR r8 r8 r9 r� s zTestUnwrap.test_simplec C s� t tddgdd�ddg� tt�tttd�d dd��dk �� t�dd d d d g�}t�|d�}t t|dd�|� t�dd d d d d g�}t�|d�}t|dd�}t |dd d d d dg� t|ddd�}t |dd d d d d g� |j |j ks�t �d S )NrF i � )�periodrE r~ i� r �K � �� i, i� � r � )r� Zdiscont) r r/ r r2 r| r r ri �modr� ry )rR Z simple_seqZwrap_seqZ uneven_seqZ wrap_unevenZ no_discontZ sm_discontr8 r8 r9 � test_period s & zTestUnwrap.test_periodN)r_ r` ra r� r� r8 r8 r8 r9 r� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFilterwindowsc C s0 t d�}t|t|�� ttj|dd�dd� d S )Nr~ r )rc g @rD )r r r r r2 r� )rR r� r8 r8 r9 � test_hanning s z TestFilterwindows.test_hanningc C s0 t d�}t|t|�� ttj|dd�dd� d S )Nr~ r )rc g�(\��@rD )r r r r r2 r� )rR r� r8 r8 r9 � test_hamming& s z TestFilterwindows.test_hammingc C s0 t d�}t|t|�� ttj|dd�dd� d S )Nr~ r )rc g�^)��@rD )r r r r r2 r� )rR r� r8 r8 r9 � test_bartlett- s z TestFilterwindows.test_bartlettc C s0 t d�}t|t|�� ttj|dd�dd� d S )Nr~ r )rc g= ףp=@rD )r r r r r2 r� )rR r� r8 r8 r9 � test_blackman4 s z TestFilterwindows.test_blackmanN)r_ r` ra r� r� r� r� r8 r8 r8 r9 r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestTrapzc C sH t �ddd�}tt �d|d �t �dt j � dd�}t|dd� d S ) Ni����r~ g�������?g �rE )r; rF r\ )r2 r3 r- r ri rk r )rR r� r� r8 r8 r9 r� > 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 rF rI rE rZ rL rN )rc )r� rc )r2 rH Z ones_liker� r- r ) rR r� r� rl ZwxZwyZwz�qZqxZqyZqzr� r8 r8 r9 � test_ndimD 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 )NrJ rE )r# g *@)r2 r3 r ri r r- )rR r� r� r# Zymr� Zxmr8 r8 r9 rG i s zTestTrapz.test_maskedN)r_ r` ra r� r rG r8 r8 r8 r9 r� < s %r� c @ s e Zd Zdd� Zdd� ZdS )�TestSincc C s6 t td�dk� tt�ddd��}t|t|�d� d S )Nr rF rN r� r\ )r r, r2 rH r r )rR r� r8 r8 r9 r� | s zTestSinc.test_simplec C sF ddg}t t�|��}t t|��}t t|��}t||� t||� d S )Nr g �?)r, r2 ri r r� r )rR r� rw rx ry r8 r8 r9 �test_array_like� s zTestSinc.test_array_likeN)r_ r` ra r� r r8 r8 r8 r9 r z 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 )NrD rI rE rF r ZwidgetZhamr� r� y @ @y �? �?y �? $@r~ )r2 ri r r| r0 )rR r� r8 r8 r9 r� � s ( zTestUnique.test_simpleN)r_ r` ra r� r8 r8 r8 r9 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 )NrF rE rI )r2 �infr� �lib�asarray_chkfiniter rO )rR rS rf rs r8 r8 r9 r� � s zTestCheckFinite.test_simplec C s2 dddg}t jj|dt jd�}t|jt jk� d S )NrF rE rI r� )r� r� )r2 r r r� r r� )rR rS r8 r8 r9 �test_dtype_order� s z TestCheckFinite.test_dtype_orderN)r_ r` ra r� r r8 r8 r8 r9 r � s r c @ s6 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� Zej�d3ejejejejg�d4d5� �Zd6S )7� 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 rF g �?g �)r r2 r )rR r8 r8 r9 �test_non_array� 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 rO r r2 r| rI rx rP )rR Ztgt1Ztgt2r8 r8 r9 r� � 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� rN )�ddofrI ) r r� � simplefilterr �DeprecationWarningr r� �filterr rO rx rP )rR �supr8 r8 r9 � test_ddof� 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� rF r )Zbias) r r� r r r r r� rx r r rO )rR r r8 r8 r9 � test_bias� s zTestCorrCoef.test_biasc C s^ t �dddgdddgg�}t|�}t �ddgddgg�}t||� tt �t �|�dk�� d S ) NrF rE rI y �?y @y @g �?y � �)r2 ri r r r r| rI )rR r� rt r r8 r8 r9 � test_complex� s zTestCorrCoef.test_complexc C sL t �dddgg�}t �dddgg�}tt �||�t �ddgddgg�� d S ) NrF rE rI y �?y @y @g �?y � �)r2 ri r r )rR r� r� r8 r8 r9 �test_xy� 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 rE ) r� r� r �RuntimeWarningr r r2 ri r� r] )rR r8 r8 r9 � test_empty� 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 �)r2 Zerrstater r ri r r| rI )rR r� rs r8 r8 r9 � test_extreme� s zTestCorrCoef.test_extreme� test_typec C s* | j �|�}t||d�}||jks&t�d S )N)r� )r� r� r r� ry )rR r Zcast_Art r8 r8 r9 �test_corrcoef_dtype� s z TestCorrCoef.test_corrcoef_dtypeN)r_ r` ra r2 ri r� rx rO rP r r� r r r r r r r� re rf �half�single�double� longdoubler r8 r8 r8 r9 r � s8 r c @ sh 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 e j � d*ej ej!ej"ej#g�d+d,� �Z$d-S ).�TestCovr rE rF g �?g �g g @)�ndminrD g�������?g������ٿrI )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 �x1rO )rR r8 r8 r9 rV s zTestCov.test_basicc C sZ t �dddgdddgg�}t �ddgddgg�}tt|�|� tt|t �d�d �|� d S ) NrF rE rI y �?y @y @g �?y � �)�aweights)r2 ri r r rP )rR r� rt r8 r8 r9 r s zTestCov.test_complexc C sJ t �dddgg�}t �dddgg�}tt||�t �ddgddgg�� d S ) NrF rE rI y �?y @y @g �?y � �)r2 ri r r )rR r� r� r8 r8 r9 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 rE ) r� r� r r r r r2 ri r� r] )rR r8 r8 r9 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� rJ )r ) r� r� r r r r r r2 ri r )rR r8 r8 r9 �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 )NF)Zrowvarg+����?g4��@���?g��&S�?g�o_��?g�J�4�?)r r �x3r2 ri )rR r� r8 r8 r9 �test_1D_rowvar. s zTestCov.test_1D_rowvarc C s$ t t| jdd�tj| jdd�� d S )NrF )r )r r r# r2 �var)rR r8 r8 r9 �test_1D_variance3 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 �?)rE rI )r� rE rN rI )r r rF � frequencies� x2_repeatsr rP �unit_frequenciesrO r r� r2 rP r� � RuntimeErrorrO )rR ZnonintrN r8 r8 r9 � test_fweights6 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)r! g @)rE rI rE g �rI ) r r r r� rQ � unit_weightsrO r2 rP r r+ rO )rR r� r8 r8 r9 � test_aweightsF 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 rF r( r- r) r rP r* rO r� rQ )rR r8 r8 r9 � test_unit_fweights_and_aweightsR s$ z'TestCov.test_unit_fweights_and_aweightsr c C s* | j �|�}t||d�}||jks&t�d S )N)r� )r r� r r� ry )rR r Zcast_x1rt r8 r8 r9 �test_cov_dtypef s zTestCov.test_cov_dtypeN)%r_ r` ra r2 ri r� r rO rF r( r) rP rP r� r* r� rQ r- r# rV r r r r" r$ r& r, r. r/ r� re rf r r r r r0 r8 r8 r8 r9 r s, $ r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Test_I0c C s� t td�t�d�� t�dddddg�}t�dd d d d g�}t t|�|� t t| �|� t�d dgddgddgddgddgg�}t t|�t�ddgddgdd gd d gd d gg�� t�d!g�}t|jd"� tt�d!g�t�d#g�� d S )$Ng �?g�ǒ#�?gC��7��?g�m�M�?g��3a�,�?g�$WV?ۏ?g $@g�ֿN^�?gE � �?g閡�1�?g�%�m? �?gD��n��@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�?g )rF g �?)r r! r2 ri r r= r )rR r� r rx Zi0_0r8 r8 r9 r� o s, zTest_I0.test_simplec sV t �d�� G � fdd�d�}tt �|� �|�s2t�t �� �}t �|� �}t||� d S )NrD c s e Zd Z� jZdd� ZdS )z*Test_I0.test_non_array.<locals>.array_likec S s | S )Nr8 )rR rs r8 r8 r9 �__array_wrap__� s z9Test_I0.test_non_array.<locals>.array_like.__array_wrap__N)r_ r` ra Z__array_interface__r2 r8 )rS r8 r9 � array_like� s r3 )r2 r3 r� rI ry r! r )rR r3 r rt r8 )rS r9 r � s zTest_I0.test_non_arrayc C s4 t �ddg�}tjtdd�� t|�}W d Q R X d S )Nr y �? @z#i0 not supported for complex values)�match)r2 ri r� r� r� r! )rR rS rt r8 r8 r9 r � s zTest_I0.test_complexN)r_ r` ra r� r r r8 r8 r8 r9 r1 m s r1 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 ) NrF g �?r rE g ��oF�?rJ g�vT�=�?gg���?g������?g���Y�J�?)r r2 �isfiniter$ r ri )rR r8 r8 r9 r� � s zTestKaiser.test_simplec C s t dd� d S )NrI rD )r$ )rR r8 r8 r9 � test_int_beta� s zTestKaiser.test_int_betaN)r_ r` ra r� r7 r8 r8 r8 r9 r5 � s r5 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��?)r2 ri r r& )rR r� r8 r8 r9 r� � s zTestMsort.test_simpleN)r_ r` ra r� r8 r8 r8 r9 r8 � s r8 c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )� TestMeshgridc C s� t dddgddddg�\}}t|t�dddgdddgdddgdddgg�� t|t�dddgdddgdddgdddgg�� d S )NrF rE rI rD rJ r[ r\ )r% r r2 ri )rR �X�Yr8 r8 r9 r� � s zTestMeshgrid.test_simplec C s. t ddddg�\}t|t�ddddg�� d S )NrF rE rI rD )r% r r2 ri )rR r: r8 r8 r9 �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 r� r8 r8 r9 � 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 )NrF rE rI rD rJ r[ r\ �ij)�indexingrZ r� r )rD rI )rI rD )rD rI rE )rI rD rE Znotvalid)r% r r2 ri r r= r rO )rR r� r� r: r; rl r8 r8 r9 � test_indexing� s zTestMeshgrid.test_indexingc C s\ t dddgddddgdd �\}}t|t�dddgg�� t|t�dgdgdgdgg�� d S ) NrF rE rI rD rJ r[ r\ T)�sparse)r% r r2 ri )rR r: r; r8 r8 r9 � test_sparse� s zTestMeshgrid.test_sparsec C s$ t ttdddgddddgdd � d S ) NrF rE rI rD rJ r[ r\ r> )r- )r r� r% )rR r8 r8 r9 �test_invalid_arguments� 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 )rA )r2 r3 r� r� r% r r� )rR r� r� r: r; r8 r8 r9 r� � 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)rA r r rF )r2 ri r% r )rR r: r; r� r� r8 r8 r9 �test_writeback s zTestMeshgrid.test_writebackc C sh t jdd� tdd�D �� \}}}}}d}t|j|� t|j|� t|j|� t|j|� t|j|� d S )Nc s s | ]}d g| V qdS )r Nr8 )r� rp r8 r8 r9 rp s z-TestMeshgrid.test_nd_shape.<locals>.<genexpr>rF r[ )rE rF rI rD rJ )r2 r% rQ r r= )rR rS rf rs r� r� Zexpected_shaper8 r8 r9 � test_nd_shape s $ zTestMeshgrid.test_nd_shapec C s~ t �dgddgdddg�\}}}t|dddggdddggg� t|dddggdddggg� t|dddggdddggg� d S )Nr rF rE rI rD rJ )r2 r% r )rR rS rf rs r8 r8 r9 �test_nd_values! s zTestMeshgrid.test_nd_valuesc C s| t jdgddgdddgdd�\}}}t|dddgdddggg� t|dddgdddggg� t|dddgdddggg� d S ) Nr rF rE rI rD rJ r> )r? )r2 r% r )rR rS rf rs r8 r8 r9 �test_nd_indexing' s $ z TestMeshgrid.test_nd_indexingN)r_ r` ra r� r< r= r@ rB rC r� rD rE rF rG r8 r8 r8 r9 r9 � s r9 c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestPiecewisec C s6 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� ttdt ddgddggg � ttdt ddgddggdd d g� d S ) Nr TFrF c S s dS )NrN r8 )r� r8 r8 r9 r� E r� z+TestPiecewise.test_simple.<locals>.<lambda>rN z 1 or 2 functions are expectedrE rI )r' r r2 ri r rO )rR r� r8 r8 r9 r� 0 s zTestPiecewise.test_simplec C s2 t ddgddgddggddg�}t|ddg� d S )NrF rE TFrI rD )r' r )rR r� r8 r8 r9 �test_two_conditionsM s z!TestPiecewise.test_two_conditionsc C s& t ddddgdddg�}t|d� d S )NrI TFrD rE r )r' r )rR r� r8 r8 r9 �$test_scalar_domains_three_conditionsQ s z2TestPiecewise.test_scalar_domains_three_conditionsc C sN t ddgddgdg�}t|ddg� t ddgddgddg�}t|ddg� d S )NrF rE TFr rI )r' r )rR r� r8 r8 r9 � test_defaultU s zTestPiecewise.test_defaultc C s� t �d�}t||dkddg�}t|jdk� t|dk� d}t|ddgddg�}t|jdk� t|dk� t|dddgdddg�}t|d� d S ) NrI rD r rJ TFrF rE )r2 ri r' r rn r )rR r� r� r8 r8 r9 r� ^ s zTestPiecewise.test_0dc C s� d}t ||dk|dkgddg�}t|d� d}t ||dk|dk|dk |dkgdddg�}t|d� ttdt ||dk|dkgdg� ttdt ||dk|dkgddddg� d S )NrI rD r rJ rF rE z 2 or 3 functions are expected)r' r r r rO )rR r� r� r8 r8 r9 �test_0d_comparisonm s , z TestPiecewise.test_0d_comparisonc C s8 t �d�}t �|dk�}t||gddg�}t|d� d S )NrI rF rE )r2 ri r' r )rR r� rs r� r8 r8 r9 �test_0d_0d_condition| s z"TestPiecewise.test_0d_0d_conditionc 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 rE rN rF rI g �g @g �?)r2 ri r' r )rR r� r� r8 r8 r9 � test_multidimensional_extrafunc� s z-TestPiecewise.test_multidimensional_extrafuncc C sf G dd� dt j�}t �d��|�}t||dk |dkgdddg�}tt|�|� t|dddddg� d S ) Nc @ s e Zd ZdS )z/TestPiecewise.test_subclasses.<locals>.subclassN)r_ r` ra r8 r8 r8 r9 r� � s r� g @g @rD g �g �?g )r2 r� r3 r� r' r r� )rR r� r� r� r8 r8 r9 r� � s z TestPiecewise.test_subclassesN) r_ r` ra r� rI rJ rK r� rL rM rN r� r8 r8 r8 r9 rH . s rH c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e jje dd�dd� �Ze j�dddggdg�dd � �Zd S ) � TestBincountc C s$ t �t �d��}t|t �d�� d S )NrD )r2 �bincountr3 r rP )rR r� r8 r8 r9 r� � s zTestBincount.test_simplec C s: t �t �dddddg��}t|t �ddddddg�� d S )NrF rJ rE rD r )r2 rP ri r )rR r� r8 r8 r9 � test_simple2� s zTestBincount.test_simple2c C s6 t �d�}t �ddddg�}t �||�}t||� d S )NrD g�������?g333333�?g �?g�������?)r2 r3 ri rP r )rR r� r� r� r8 r8 r9 �test_simple_weight� s z TestBincount.test_simple_weightc C sT t �dddddg�}t �dddddg�}t �||�}t|t �d ddd ddg�� d S ) NrF rE rD rJ g�������?g333333�?g �?g�������?r )r2 ri rP r )rR r� r� r� r8 r8 r9 �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 rF rI )� minlengthrE )r2 ri rP r )rR r� r� r8 r8 r9 �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 rF rE rI )rT )r2 ri rP r )rR r� r� r8 r8 r9 �)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 ) NrF rE rD rJ g�������?g333333�?g �?g�������?rZ r )r2 ri rP r )rR r� r� r� r8 r8 r9 � test_with_minlength_and_weights� s z,TestBincount.test_with_minlength_and_weightsc C s& t jg td�}t �|�}t||� d S )N)r� )r2 ri r� rP r )rR r� r� r8 r8 r9 r � s zTestBincount.test_emptyc C s4 t jg td�}t j|dd�}t|t jdtd�� d S )N)r� rJ )rT )r2 ri r� rP r r; )rR r� r� r8 r8 r9 �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)rT )r2 rP r8 )r� r8 r9 r� � r� z<TestBincount.test_with_incorrect_minlength.<locals>.<lambda>zmust not be negativec s t j� dd�S )NrN )rT )r2 rP r8 )r� r8 r9 r� � r� rJ c s t j� dd�S )NrY )rT )r2 rP r8 )r� r8 r9 r� � r� c s t j� dd�S )NrN )rT )r2 rP r8 )r� r8 r9 r� � r� )r2 ri r� r r� rO r3 )rR r8 )r� r9 � test_with_incorrect_minlength� s z*TestBincount.test_with_incorrect_minlengthzPython lacks refcounts)r� c 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~ rF rE rI rD rJ r[ ) r� r� r2 r� Zintpr rQ rP r )rR Z intp_refcountZdouble_refcount�jr8 r8 r9 �test_dtype_reference_leaks� s z'TestBincount.test_dtype_reference_leaks�valsrE c C sJ t �|�}tt�� t �|� W d Q R X tt�� t �|� W d Q R X d S )N)r2 �asarrayr rO rP )rR r] Zvals_arrr8 r8 r9 �test_error_not_1d� s z TestBincount.test_error_not_1dN)r_ r` ra r� rQ rR rS rU rV rW r rX rZ r� re r� r r\ rf r_ r8 r8 r8 r9 rO � s rO c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej d d � d d � dd � dd � gddddgd�dd� �Z dd� Z dd� Z dd � Z d d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*� TestInterpc C st t ttdg g � t ttddgddg� t ttdddgddgdd� t ttdg g dd� t ttddgddgdd� d S )Nr rF rE )r� ih )r rO r# )rR r8 r8 r9 �test_exceptions� s zTestInterp.test_exceptionsc C sB t �ddd�}t �ddd�}t �ddd�}tt �|||�|� d S )Nr rF rJ rW )r2 rH r r# )rR r� r� �x0r8 r8 r9 rV � s zTestInterp.test_basicc C s� �x�t dd�D �]�}tj|tjd�}tj|tjd�}tjdd|d |gtjd�}|d d d� }t|||�}t|||�}tjddddgtd�}|d d d� } t||� t|| � t|||dd�}t|||dd�}tjddddgtd�}|d d d� } t||� t|| � t|||dd�}t|||dd�}tjddddgtd�}|d d d� } t||� t|| � t|||ddd �}t|||ddd �}tjddddgtd�}|d d d� } t||� t|| � qW d S ) NrF r~ )r� rN r )�leftrE )�right)rc rd ) rQ r2 r3 r rP ri r# r� r ) rR �size�xpZypZincptsZdecptsZincresZdecresZinctgtZdectgtr8 r8 r9 �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 rF rJ g333333�?)r2 rH r r# r� r� r� )rR r� r� rb r8 r8 r9 � test_scalar_interpolation_point' s z*TestInterp.test_scalar_interpolation_pointc C s~ dddddg}ddddg}ddt jdg}tt �|||�ddt jt jdg� ddt jdg}tt �|||�ddt jt jdg� d S )NrF rE g @rI rD )r2 r r r# r� )rR r� rf �fpr8 r8 r9 � test_non_finite_behavior_exact_x5 s "z+TestInterp.test_non_finite_behavior_exact_xc C s t �| �S )N)r2 Zfloat_)r� r8 r8 r9 r� > r� zTestInterp.<lambda>c C s t | d�S )Nr )rB )r� r8 r8 r9 r� ? r� c C s t d| �S )Nr )rB )r� r8 r8 r9 r� @ r� c C s t | t�| d��S )NrH )rB r2 �multiply)r� r8 r8 r9 r� A r� r? z complex-realz complex-imagz complex-both)�paramsZidsc C s |j S )z( scale function used by the below tests )�param)rR �requestr8 r8 r9 �sc= s z TestInterp.scc C s� t t�dtjdg|ddg��|tj�� t t�ddtjg|ddg��|tj�� t t�dddg|tjdg��|tj�� t t�dddg|dtjg��|tj�� dS )z test that nans are propagated g �?rF r r~ N)r r2 r# r� )rR ro r8 r8 r9 �test_non_finite_any_nanL s (((z"TestInterp.test_non_finite_any_nanc C s� t t�dtj tj g|ddg��|tj�� t t�dddg|tj tj g��|tj�� t t�dddg|tj tj g��|tj�� t t�dtj tj g|ddg��|d�� dS )z2 Test that interp between opposite infs gives nan g �?r r~ rF N)r r2 r# r r� )rR ro r8 r8 r9 �test_non_finite_infS s ...z TestInterp.test_non_finite_infc C st t t�dtj dg|tj dg��|tj�� t t�dtj dg|tj dg��|tj�� t t�dtj dg|dtj g��|tj�� t t�dtj dg|dtj g��|tj�� t t�ddtj g|tj dg��|tj�� t t�ddtj g|tj dg��|tj�� t t�ddtj g|dtj g��|tj�� t t�ddtj g|dtj g��|tj�� dS )z@ Test that interp where both axes have a bound at inf gives nan g �?rF r~ r N)r r2 r# r r� )rR ro r8 r8 r9 �test_non_finite_half_inf_xf\ s .......z&TestInterp.test_non_finite_half_inf_xfc C s� t t�dtj tj g|ddg��|d�� t t�dtj dg|ddg��|d�� t t�ddtj g|ddg��|d�� t t�dtj tj g|ddg��|d�� dS )z1 Test interp where the x axis has a bound at inf g �?r r~ rF N)r r2 r# r )rR ro r8 r8 r9 �test_non_finite_half_inf_xg s ,((z%TestInterp.test_non_finite_half_inf_xc C s t t�dddg|dtj g��|tj �� t t�dddg|dtj g��|tj �� t t�dddg|tj dg��|tj �� t t�dddg|tj dg��|tj �� t t�dddg|tj tj g��|tj �� t t�dddg|tj tj g��|tj �� dS )z1 Test interp where the f axis has a bound at inf g �?r rF r~ N)r r2 r# r )rR ro r8 r8 r9 �test_non_finite_half_inf_fn s ,,,,0z%TestInterp.test_non_finite_half_inf_fc C sH t �ddd�}t �ddd�dt �ddd� d }d}|d| d }tt �|||�|� d}d}tt j||||d�|� d }d}tt j||||d �|� dd d d dg}dd d dg}ddt jdg}ddt jd t jdg}tt �|||�|� ddddddddg}ddddg}d d d d g}d d d!d"d d#d$d%g}tt j|||d&d'�|� d S )(Nr rF rJ y �?g333333�?rN y @ @)rc g @)rd rE g @rI rD y @ �?y �?iL���iV���iG���� i����r� im � iB���i^ i����y @ �?y $@ @y @ @y @ @y @ �?y �!@ �?y @ �?y @ @y @ @y @ @ih )r� )r2 rH r r# r ) rR r� r� rb Zy0rc rd rf ri r8 r8 r9 �test_complex_interpw s, $ z TestInterp.test_complex_interpc C s� t �ddd�}t �ddd�}t �d�}tt �|||�|� t �dddg�}t �dddg�}t �t �d�||�}t|d� tt|t j�� t jt �d�||dd �}t|d � tt|t j�� d S ) Nr rF rJ g333333�?rE rD rN g @)r� g �?) r2 rH ri r r# r r r� r� )rR r� r� rb rf ri r� r8 r8 r9 �)test_zero_dimensional_interpolation_point� s z4TestInterp.test_zero_dimensional_interpolation_pointc C s2 t �ddd�}t �|�}tt �t j||�d� d S )Nr r~ g-C ��6?g )r2 r3 �sinr r# rk )rR rf ri r8 r8 r9 �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���ru i����r� r im rv iB���i^ i����rJ r~ rI rD g @g @g �!@g @g @g @g @g @ih )r� r� )r� rE rN r� )r r2 r# ri r] )rR r� rf ri r� r8 r8 r9 r� � s zTestInterp.test_periodN)r_ r` ra ra rV rg rh rj r� Zfixturero rp rq rr rs rt rw rx rz r� r8 r8 r8 r9 r` � s. & r` c C s, x&t t|��D ]}t| | || � qW d S )N)rQ r� r )rt r� rp r8 r8 r9 �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/d0� Zd1d2� Zd3S )4�TestPercentilec C s~ t �d�d }tt �|d�d� tt �|d�d� tt �|d�d� t j|d < tt �|d�t j� tt j|dd d �t j� d S ) NrZ g �?r g r� g @rW g �?rF �nearest)� interpolation)r2 r3 r � percentiler� )rR r� r8 r8 r9 rV � s zTestPercentile.test_basicc C s� dd� t d�D �}t�|td��}t|td�� tt|�t� t�|td��}t|tdd�� tt|�t� t�|td��}t|tdd �� tt|�t� t�|td�g�}t|t�tdd �g�� tt|�tj� d S ) Nc S s g | ]}t |d ��qS )rE )r )r� rp r8 r8 r9 r� � s z0TestPercentile.test_fraction.<locals>.<listcomp>rZ r r� r\ rE rW rD )rQ r2 r r r r� ri r� )rR r� r� r8 r8 r9 � test_fraction� s z TestPercentile.test_fractionc C sR t �d�}t �|dd d d� t �|dd d dd� t �d�}t �|dd |dd� d S )NrJ F�linear)rF )r2 rP r )rR r� �or8 r8 r9 �test_api� s zTestPercentile.test_apic C sR t �dddgdddgdddgdddgdddgg�}tt j|ddd�dddg� d S )NrF rD rI rW r )rc )r2 ri r r )rR r� r8 r8 r9 �test_2D� s zTestPercentile.test_2Dc C s4 t t�td�d�d� t tjtd�ddd�d� d S )Nr~ rW g @r� )r~ )r r2 r rQ )rR r8 r8 r9 � test_linear� s zTestPercentile.test_linearc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nr~ rW �lower)r~ rD �higherrJ )r r2 r rQ )rR r8 r8 r9 �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 @rW rJ )r r2 r rQ )rR r8 r8 r9 � test_midpoint� s z TestPercentile.test_midpointc C s8 t tjtd�ddd�d� t tjtd�ddd�d� d S )Nr~ r� r} )r~ rJ �1 rD )r r2 r rQ )rR r8 r8 r9 � test_nearest� s zTestPercentile.test_nearestc C s0 t �d�d }tt �|dddg�dddg� d S )NrZ g �?r r� rW g @g �?)r2 r3 r r )rR r� r8 r8 r9 � 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� rI rD )r< rW r� g @g @g &@rE rJ r[ r\ rZ r� r~ r� r )rc g �?g �?g @g �!@g #@rF ih )r< rW )rE )r< rW r� )rI )rE rD rJ r[ )rE rI rJ r[ )rE rI rD r[ )rE rI rD rJ )rI rI rJ r[ r� )r~ )rc r~ )r2 r3 r] r r ri r� r= )rR r� �r0r� r8 r8 r9 r s: " zTestPercentile.test_axisc C s� t �d��dd�}tt �|d�d� tt �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� tt �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� rI rD rW g @g @g @g @g @r )rc g �?g #@rF )r )rc r r� )r~ )r~ rc g �?g "@r8 )r� )r~ r rJ )r~ rc r ) r2 r3 r] r r r Zisscalarri r= r �emptyr� )rR r� r� r� r Zc0�c1rs r8 r8 r9 � 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 ) NrF rE �8 rY )r~ �e rN rW g��������)r rO r2 r r rQ )rR r8 r8 r9 �test_exception] s z TestPercentile.test_exceptionc C s t t�dddgd�d� d S )NrF rE rI r )r r2 r )rR r8 r8 r9 �test_percentile_liste 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 )%NrF rE rI )rI )rF rE rI )r rD rJ r[ )rI rI r )rc r )rc )rI rE r� g @g @g @g @g @g @)rE rD )r< rW g �?g @g �!@g �?g @g #@)rE rI r\ )r� r� )r~ rc r rZ r� ) r2 ri r; r r r3 r] r� r� )rR r� r� r� r� r r� rs r8 r8 r9 �test_percentile_outh 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� rF rE rW r )rc )rF rE rF )r� rE rF )r� rF rF rI )r� rF rE rN rH rK r� r� )rc r~ r~ )rE rF rE rF )rE r� rE rF )rE r� rF rF )rE r� rF rE )r2 r3 r] r r r= ri )rR r� r8 r8 r9 �test_percentile_empty_dim� 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 )NrE rI rD rF rW F)�overwrite_input)r2 ri r r )rR rS r8 r8 r9 � test_percentile_no_overwrite� 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@rJ )�numr� )r~ )r2 rH r r3 r r~ )rR r� r8 r8 r9 �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 ) NrE rI rD rF rW T)r� g @)r2 ri r r )rR rS rf r8 r8 r9 �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)�G � )re r~ r� )r rF )rc rN r )rH rN rF )r rN r< r7 )r rF rE )r i� )rI rJ r\ r� �Z )r rF rI )rI rF r� rE )rI rF rE )rI rE )rE rF )rF rH )rF rI )rE rE )r2 rJ �normal�dstackr r �moveaxisro r r3 r] �shuffle�ravel�flatten)rR r� r� r� r8 r8 r9 �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)rI rJ r\ r� r� r< )rc r� )r r� rD )r rD )rF rF )rN rN )rI rN )r2 rP r rd r rO )rR r� r8 r8 r9 �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)rI rJ r\ r� r\ T)rc r� )rF rF rF rF )r rF )rF rF r\ r� )r rI )rF rJ r\ rF )rF )rI rF r\ r� )r rF rE rI )r� )r rF rI )rF rF r\ rF rF )rE rF rF r\ rF )rE rF rJ r\ rF )r2 rP r r r= )rR r� r8 r8 r9 � 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)rD )rI rD r )r r} )r~ r )rI rF r8 rE )r2 r; rP r r )rR r� r� r8 r8 r9 �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� )rD )rI rD )rE rF r )r r} )r~ r )rI rF r8 ) r� r� r� r r2 r; rP r� r r )rR r� r� r8 r8 r9 � test_out_nan s zTestPercentile.test_out_nanc C s: t jdtd�}t j|d< tt �|d�t j� tt j|ddd�t j� tt j|ddgdd�t �t jgd �� t jdtd��ddd �}t j|d < t j|d < tt �|d�t j� tt �|d�jd� t �t jdtd��ddd �dd�}t j|d < t j|d < tt �|dd�|� t �t jdtd��ddd �ddgd�}t j|d d �ddf< t j|d d �ddf< tt �|ddgd�|� t �t jdtd��ddd �dd�}t j|d< t j|d < tt �|dd�|� t �t jdtd��ddd �ddgd�}t j|d d �ddf< t j|d d �ddf< tt �|ddgd�|� t �t jdtd��ddd �dd�}t j|d< t j|d< tt �|dd�|� t �t jdtd��ddd �ddgd�}t j|d d �df< t j|d d �df< tt �|ddgd�|� t jt jdtd��ddd �ddgddd�}t j|d d �df< t j|d d �df< tt j|ddgddd�|� d S )Nr� )r� rE g333333�?r )rc g333333�?rI rD )rF rE rI )rF rF rE )rE rI )rF rE rF )rF rI )r rE r} )r~ ) r2 r3 r� r� r r ri r] rn )rR rS rf r8 r8 r9 �test_nan_behavior$ sZ " " " z TestPercentile.test_nan_behaviorc C s� t jtdd�� t�ddddgtj� W d Q R X t jtdd�� t�ddddgtjg� W d Q R X t�ddd �}tj|d < t jtdd�� t�ddddg|� W d Q R X d S ) NzPercentiles must be in)r4 rF rE rI g @g �?g �X@r� r )r� r� rO r2 r r� rH )rR r� r8 r8 r9 � test_nan_qa s " zTestPercentile.test_nan_qN) r_ r` ra rV r� r� r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r8 r8 r8 r9 r| � s2 $0) =r| c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej e e j e jd d d �e jddddd�d�d�dd� �ZdS )� TestQuantilec C sH t �d�d }tt �|d�d� tt �|d�d� tt �|d�d� d S )NrZ g �?r g rF g @g �?)r2 r3 r �quantile)rR r� r8 r8 r9 rV o s zTestQuantile.test_basicc C sn t �dg�}t �dd�}t||d � tt|�|j� t �dddg�}t �||�}t||� t|j|j� d S )NTFr )r2 ri r� r r� r� r )rR rS Ztf_quantZ quant_resr8 r8 r9 �test_correct_quantile_valueu s z(TestQuantile.test_correct_quantile_valuec C s8 dd� t d�D �}t�|d�}t|d� tt|�t� t�|d�}t|tdd�� tt|�t� t�|tdd��}t|tdd�� tt|�t� t�|tdd�g�}t|t�tdd�g�� tt|�tj� t�|tdd�gg�}t|t�tdd�gg�� tt|�tj� t�d�}tt�|tdd��tdd�� d S ) Nc S s g | ]}t |d ��qS )rE )r )r� rp r8 r8 r9 r� � s z.TestQuantile.test_fraction.<locals>.<listcomp>rZ r rF r\ rE rD ) rQ r2 r� r r� r ri r� r3 )rR r� r� r8 r8 r9 r� s$ zTestQuantile.test_fractionc C sp t �dddddg�}|�� }t jt �d�|dd� t||� |�� }|�� }t jt �d�|dd� t||� d S ) Nr g �?g �?g �?g �?g Y@r� )r~ )r2 ri r r� r3 r r~ )rR �p0r� r8 r8 r9 r� � s z TestQuantile.test_no_p_overwritec C s\ t �ddd�}t �t �dddddddddddddddd d d g�d |�}tt �|�|� d S ) Nr rF g{�G�z�?rE rI rD rJ r� rZ r\ g�������?)r2 r3 r� ri r rL )rR r� r� r8 r8 r9 �test_quantile_monotonic� s &z$TestQuantile.test_quantile_monotonicrI i� )� min_value� max_valueFg�u �<�7�g�u �<�7~)�allow_infinity� allow_nanr� r� )r� r= �elements)rs c C s. t �ddd�}t �||�}tt �|�|� d S )Nr rF g{�G�z�?)r2 r3 r� r rL )rR rs r� r� r8 r8 r9 � test_quantile_monotonic_hypo� s z)TestQuantile.test_quantile_monotonic_hypoN)r_ r` ra rV r� r� r� r� � hypothesis�givenr r2 r� �stZintegers�floatsr� r8 r8 r8 r9 r� l s r� c @ s� e Zd Zejejddddd�ejddddd�ejddddd�ejddddd�d�dd � �Zejejddddd�ejddddd�ejddddd�d �d d � �Zejejddddd�ejddddd�ejddddd�d �d d� �Z dd� Z dS )�TestLerpFr rF )r� r� r� r� g�u �<�7�g�u �<�7~)�t0�t1rS rf c C sp t jj�|||�}t jj�|||�}||ks4||krB||kslt�n*||k ||k kr`||kslt�n ||kslt�d S )N)r2 r � function_base�_lerpry )rR r� r� rS rf Zl0�l1r8 r8 r9 �test_lerp_monotonic� s z TestLerp.test_lerp_monotonic)�trS rf c C s^ ||kr2|t jj�|||� kr*|ksZn t�n(|t jj�|||� krT|ksZn t�d S )N)r2 r r� r� ry )rR r� rS rf r8 r8 r9 �test_lerp_bounded� s *zTestLerp.test_lerp_boundedc C s@ t jj�||dd| �}t jj�||d| �}||ks<t�d S )NrF )r2 r r� r� ry )rR r� rS rf rc rd r8 r8 r9 �test_lerp_symmetric� s z TestLerp.test_lerp_symmetricc C s< t �d�}t �d�}t �d�}t jj�|||�dks8t�d S )NrE rJ g�������?g������@)r2 ri r r� r� ry )rR rS rf r� r8 r8 r9 �test_lerp_0d_inputs� s z TestLerp.test_lerp_0d_inputsN) r_ r` ra r� r� r� r� r� r� r� r� r8 r8 r8 r9 r� � s* 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 sH 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< tt �|�jd� d S )NrF rE r[ rI g �?g @r )rc g �?g @rD g�6-¦?g� �0���?g���}r�?g��4 �?g @rN ) r2 ri r3 r] r �medianr r rn r� )rR �a0r� r rS r8 r8 r9 rV � 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 )NrE rI r rF r[ r\ rD rJ r� )rE rI rD )re )rc g @) r2 ri rJ �randintr r� rQ rn r r r� )rR �a3rS �origZaxr8 r8 r9 �test_axis_keyword 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 )NrE rI r rF r[ r\ rD rJ T)r� g �?g @)r� rc g �?g @r7 )r� )rI rD rJ )rc )rc r� ) r2 ri r3 r] r r� r r� r� rJ r� r� )rR r� r� r� r Za4r8 r8 r9 �test_overwrite_keyword s< z!TestMedian.test_overwrite_keywordc C sH dddg}t t�|�d� |g}t t�|�d� ttj|dd�|� d S )NrF rE rI r )rc )r r2 r� r )rR r� rF r8 r8 r9 r 7 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)r2 r^ r� �info)�clsZ input_arrayr� �objr8 r8 r9 �__new__B s z4TestMedian.test_subclass.<locals>.MySubClass.__new__c S s dS )Ni����r8 )rR rc r� r r8 r8 r9 r� G s z1TestMedian.test_subclass.<locals>.MySubClass.mean)N)NNN)r_ r` ra r� r� r8 r8 r8 r9 � MySubClass@ s r� rF rE rI i����)r2 r� r r� )rR r� rS r8 r8 r9 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)rD )rI rD r )r )rI rF r8 )r2 r; rP r r� )rR r� r� r8 r8 r9 r� M 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� )rD )rI rD )rE rF r )r )rI rF r8 ) r� r� r� r r2 r; rP r� r r� )rR r� r� r8 r8 r9 r� V s zTestMedian.test_out_nanc C sf t jdtd�}t j|d< tt �|�t j� tt j|dd�t j� t jdtd��ddd�}t j|d< t j|d < tt �|�t j� tt �|�jd� t �t jdtd��ddd�d�}t j|d < t j|d < tt �|d�|� t �t jdtd��ddd�d �}t j|d < t j|d < tt �|d �|� t �t jdtd��ddd�d�}t j|d < t j|d< tt �|d�|� d S )Nr� )r� rE r )rc rI rD )rF rE rI )rF rF rE )rE rI )rF rE rF )rF rI )r rE )r2 r3 r� r� r r� r] rn )rR rS rf r8 r8 r9 r� b s* z TestMedian.test_nan_behaviorc C sZ t jg td�}tjdd��F}t�ddt� tt �|�t j � t |d j tk� tt |�d� 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 tdd �}tt j|dd �|� tt j|d d �|� t jt j tdd �}tjdd��:}t�ddt� tt j|dd �|� t |d j tk� W d Q R X d S ) N)r� T)r� r� r� r rE rI )r� r )rc rF ) r2 ri r� r� r� r� r r r� r� r r� r� )rR rS r� rf r8 r8 r9 r � s& zTestMedian.test_emptyc C sL t �d�}ttt �|�t���t� t j|d< ttt �|�t���t� d S )Ng @rE ) r2 r3 r r� r� r� r� r� r� )rR r� r8 r8 r9 � 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� )re r~ )r rF )rc rN r )rH rN rF )r rN )r rF rE )r )rN i� )rI rJ r\ r� )r rF rI )rI rF r� rE )rI rF rE )rI rE )rE rF )rF rH )rF rI )rE rE )r2 rJ r� r� r r� r� ro r r3 r] r� r� r� )rR r� r� r� r8 r8 r9 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)rI rJ r\ r� r� )rc )r r� rD )r rD )rF rF )r2 rP r rd r� rO )rR r� r8 r8 r9 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)rI rJ r\ r� T)rc r� )rF rF rF rF )r rF )rF rF r\ r� )r rI )rF rJ r\ rF )rF )rI rF r\ r� )r rF rE rI )r rF rI )rF rF r\ rF )r2 rP r r� r= )rR r� r8 r8 r9 r� � s zTestMedian.test_keepdimsN)r_ r` ra rV r� r� r r� r� r� r� r r� r� r� r� r8 r8 r8 r9 r� � s ! r� c @ s e Zd Zdd� Zdd� ZdS )�TestAdd_newdoc_ufuncc C s" t ttdd� t tttjd� d S )NrE Zblah)r r� r rO r2 r4 )rR r8 r8 r9 �test_ufunc_arg� s z#TestAdd_newdoc_ufunc.test_ufunc_argc C s t tttjd� d S )NrI )r r� r r2 r4 )rR r8 r8 r9 �test_string_arg� s z$TestAdd_newdoc_ufunc.test_string_argN)r_ r` ra r� r� r8 r8 r8 r9 r� � s r� c @ s\ e Zd Zejjejjdkdd�ejj e dd�dd� ��Z ejjejjdkdd�dd� �Z d S ) �TestAdd_newdocrE zPython running -OO)r� zPyPy does not modify tp_docc C sV d}t tjjjjd t|�� |� tttjjj j�dk� tttj j j j�dk� d S )Nz"Current flat index into the array.i, ) r r2 �core�flatiter�indexr� r� r Zufunc�identityr Z index_tricksZmgrid)rR r r8 r8 r9 � test_add_doc� s zTestAdd_newdoc.test_add_docc C s2 t jjjj}t �ddd� |t jjjjks.t�d S )Nz numpy.corer� )r� z bad docstring)r2 r� r� r� r� Z add_newdocry )rR Zprev_docr8 r8 r9 �test_errors_are_ignored� s z&TestAdd_newdoc.test_errors_are_ignoredN) r_ r` ra r� re r� r� r� �optimizer� r r� r� r8 r8 r8 r9 r� � s r� c @ s\ e Zd Zejjejjdkdd�ejje dd�dd� ��Z ejjejjdkdd�dd� �Z d S ) �TestAddDocstringrE zPython running -OO)r� zPyPy does not modify tp_docc C s0 t �t jjt jjj� dd� }t �||j� d S )Nc S s dS )� docstringNr8 r8 r8 r8 r9 �func s z6TestAddDocstring.test_add_same_docstring.<locals>.func)r2 � add_docstringr� �flatr� )rR r� r8 r8 r9 �test_add_same_docstring� s z(TestAddDocstring.test_add_same_docstringc C sP t t�� t�tjjd� W d Q R X dd� }t t�� t�|d� W d Q R X d S )Nzdifferent docstringc S s dS )r� Nr8 r8 r8 r8 r9 r� s z=TestAddDocstring.test_different_docstring_fails.<locals>.func)r r+ r2 r� r� r� )rR r� r8 r8 r9 � test_different_docstring_fails s z/TestAddDocstring.test_different_docstring_failsN) r_ r` ra r� re r� r� r� r� r r� r� r8 r8 r8 r9 r� � s r� c @ s6 e Zd Zej�dddddddg�dd � �Zd d � Zd S ) �TestSortComplexztype_in, type_out)�l�D)�hr� )�Hr� )rf r� )rx r� )r[ �Gc C sN t jdddddg|d�}t �|�}t �|��|�}t||� t|j|j� d S )NrJ rI r[ rE rF )r� )r2 ri � sort_complexrL r� r r� )rR Ztype_inZtype_outrS r� r r8 r8 r9 �test_sort_real s z TestSortComplex.test_sort_realc C sR t jddddgdd�}t jddddgdd�}t �|�}t||� t|j|j� d S )Ny @ @y �? �y �? �y @ �?r� )r� )r2 ri r� r r� )rR rS r r� r8 r8 r9 �test_sort_complex( s z!TestSortComplex.test_sort_complexN)r_ r` ra r� re rf r� r� r8 r8 r8 r9 r� s r� )r� )r� r� )xr� r� r� r� � fractionsr r� r� r� Zhypothesis.extra.numpyr Zhypothesis.strategies� strategiesr� �numpyr2 r Z numpy.testingr r r r r r r r r r r r Znumpy.lib.function_baser r� r� 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, r- r. r/ r0 r1 r: rB rC rb ru r{ r} r� r� r� r� r� r� r� r r r r) r9 rg rm r� r� r� r� r� r� r� r� r� r r r r r r1 r5 r8 r9 rH rO r` r{ r| r� r� r� r� r� r� r� r8 r8 r8 r9 �<module> s� 8 � Dg#e>j S }A) (U > [j3 nd` C 6N2 p
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка