OwlCyberSecurity - MANAGER
Edit File: test_ufunc.cpython-38.pyc
U �p�]�< � @ s, d dl mZmZmZ d dlZd dlZd dlZd dlZd dl m m Z d dl m mZ d dlm mZ d dlm mZ d dlmZmZmZmZmZm Z m Z m Z d dl m Z G dd� de!�Z"G dd� de!�Z#G d d � d e!�Z$ej%�&d d d � e'e�D ��dd� �Z(ej%�&d dd � e'e�D ��dd� �Z)dS )� )�division�absolute_import�print_functionN)�assert_� assert_equal� assert_raises�assert_array_equal�assert_almost_equal�assert_array_almost_equal�assert_no_warnings�assert_allclose)�picklec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestUfuncKwargsc C s� t ttjdddd� t ttjddtd� t ttjdddgd� t ttjddd d� t ttjddd d � t ttjddd d � t ttjddd d � t ttjdddgd� d S )N� � Zsafe)Zcastingx)Zdtypex� )Zextobjx)Zoutx�ii->i)Zsigx)Z signaturexF)ZsubokxT)Zwherex)r � TypeError�np�add�int��self� r �=/usr/lib/python3/dist-packages/numpy/core/tests/test_ufunc.py�test_kwarg_exact s z TestUfuncKwargs.test_kwarg_exactc C s t ttjddddd� d S )Nr r r )�sig� signature)r � ValueErrorr r r r r r �test_sig_signature s �z"TestUfuncKwargs.test_sig_signaturec C s0 t ttjdddtd� t ttjdddtd� d S )Nr r r )r �dtype)r r )r � RuntimeErrorr r r r r r r �test_sig_dtype$ s ��z TestUfuncKwargs.test_sig_dtypec C s t ttjdddgdd� d S )Nr r r T)Zextobj�parrot)r r r r r r r r �test_extobj_refcount* s z$TestUfuncKwargs.test_extobj_refcountN)�__name__� __module__� __qualname__r r r" r$ r r r r r s r c @ s� e Zd ZdZejejfejejfejejfejejfejejfej ej fejejfej ej fgZ e j �de �ejddfdd��Zdd� Ze j �de �eddfd d ��ZG d d � d e�Zd d� Zefdd�Zdd� Zefdd�Zefdd�ZdS )�TestUfuncGenericLoopsab Test generic loops. The loops to be tested are: PyUFunc_ff_f_As_dd_d PyUFunc_ff_f PyUFunc_dd_d PyUFunc_gg_g PyUFunc_FF_F_As_DD_D PyUFunc_DD_D PyUFunc_FF_F PyUFunc_GG_G PyUFunc_OO_O PyUFunc_OO_O_method PyUFunc_f_f_As_d_d PyUFunc_d_d PyUFunc_f_f PyUFunc_g_g PyUFunc_F_F_As_D_D PyUFunc_F_F PyUFunc_D_D PyUFunc_G_G PyUFunc_O_O PyUFunc_O_O_method PyUFunc_On_Om Where: f -- float d -- double g -- long double F -- complex float D -- complex double G -- complex long double O -- python object It is difficult to assure that each of these loops is entered from the Python level as the special cased loops are a moving target and the corresponding types are architecture dependent. We probably need to define C level testing ufuncs to get at them. For the time being, I've just looked at the signatures registered in the build directory to find relevant functions. zinput_dtype,output_dtyper r c C s@ t jd||�|d�}||�d d d� }t||� t|j|� d S �N� �r r �r �fullr r r �r Z input_dtypeZ output_dtype�f�x�yZxsZysr r r �test_unary_PyUFuncb s z(TestUfuncGenericLoops.test_unary_PyUFuncc C s | | S �Nr )r0 r1 r r r �f2i s zTestUfuncGenericLoops.f2c C sB t jd||�|d�}|||�d d d� }t||� t|j|� d S r) r, r. r r r �test_binary_PyUFuncl s z)TestUfuncGenericLoops.test_binary_PyUFuncc @ s e Zd Zdd� Zdd� ZdS )zTestUfuncGenericLoops.fooc C s t �d�S �Nr �r �bool_r r r r � conjugateu s z#TestUfuncGenericLoops.foo.conjugatec C s t �d�S r6 r7 )r �objr r r � logical_xorx s z%TestUfuncGenericLoops.foo.logical_xorN)r% r& r' r9 r; r r r r �foot s r< c C s* t jdtd�}tt �t �|�dk�� d S )Nr* r+ r )r �ones�objectr �all�abs�r r0 r r r �test_unary_PyUFunc_O_O{ s z,TestUfuncGenericLoops.test_unary_PyUFunc_O_Oc C s. t jd|� td�}tt �t �|�dk�� d S )Nr* r+ T)r r- r> r r? r9 �r r< r0 r r r � test_unary_PyUFunc_O_O_method s z3TestUfuncGenericLoops.test_unary_PyUFunc_O_O_methodc C s, t jdtd�}tt �t �||�dk�� d S r) )r r= r> r r? r rA r r r �test_binary_PyUFunc_OO_O� s z.TestUfuncGenericLoops.test_binary_PyUFunc_OO_Oc C s, t jd|� td�}tt �t �||��� d S )Nr* r+ �r r- r> r r? r; rC r r r � test_binary_PyUFunc_OO_O_method� s z5TestUfuncGenericLoops.test_binary_PyUFunc_OO_O_methodc C s, t jd|� td�}tt �t �||��� d S )N)r* r � r+ rF rC r r r � test_binary_PyUFunc_On_Om_method� s z6TestUfuncGenericLoops.test_binary_PyUFunc_On_Om_methodN)r% r& r' �__doc__r Zsingle�doubleZcsingleZcdouble� longdouble� clongdoubleZ np_dtypes�pytest�mark� parametrizeZexpr2 r4 r5 r> r< rB rD rE rG rI r r r r r( / s* - � r( c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd Zd Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMdN� Z*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3dadb� Z4dcdd� Z5dedf� Z6dgdh� Z7didj� Z8dkdl� Z9dmdn� Z:dodp� Z;dqdr� Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBeCjD�Edd��eCjD�Ed�eF�Gd�d�d�g�eF�Gd�gd�gd�gg�eF�Gd�d�d�gd�d�d�gd�d�d�gg�f�d�d�� ��ZHeCjD�Ed�d�eF�Gd�d�d�g�fd�d�d�d�gfd�f�eCjD�Ed�eFjI d�f�d�d�� ��ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\eCjD�Ed�eF�]d��eF�^d��g�d�d�� �Z_eCjD�Ed�eF�]d��eF�^d��g�d�d�� �Z`eCjD�Ed�eF�]d��eF�^d��g�d�d�� �Zad�S )�� TestUfuncc C s^ t dtjd �D ]H}tt�tjtj|d��tjk� t�tjtj |d��}t|tj k� qd S )Nr r )Zprotocol) �ranger ZHIGHEST_PROTOCOLr �loads�dumpsr Zsin�_rational_tests�test_add)r �proto�resr r r � test_pickle� s �� �zTestUfunc.test_picklec C s d}t t�|�tjk� d S )NsM cnumpy.core _ufunc_reconstruct p0 (S'numpy.core.umath' p1 S'cos' p2 tp3 Rp4 .)r r rS r Zcos)r Zastringr r r �test_pickle_withstring� s z TestUfunc.test_pickle_withstringc C sh d}t �|�}t �ttt �|d �t �|d �d ����� }tt j�||�d d d� ddddg� d S )N� r r rH � � ) r �arange�array�list�zipZravelr r �reduceat)r �Lr0 �idxr r r �test_reduceat_shifting_sum� s 0z$TestUfunc.test_reduceat_shifting_sumc C s dS )aL Try to check presence and results of all ufuncs. The list of ufuncs comes from generate_umath.py and is as follows: ===== ==== ============= =============== ======================== done args function types notes ===== ==== ============= =============== ======================== n 1 conjugate nums + O n 1 absolute nums + O complex -> real n 1 negative nums + O n 1 sign nums + O -> int n 1 invert bool + ints + O flts raise an error n 1 degrees real + M cmplx raise an error n 1 radians real + M cmplx raise an error n 1 arccos flts + M n 1 arccosh flts + M n 1 arcsin flts + M n 1 arcsinh flts + M n 1 arctan flts + M n 1 arctanh flts + M n 1 cos flts + M n 1 sin flts + M n 1 tan flts + M n 1 cosh flts + M n 1 sinh flts + M n 1 tanh flts + M n 1 exp flts + M n 1 expm1 flts + M n 1 log flts + M n 1 log10 flts + M n 1 log1p flts + M n 1 sqrt flts + M real x < 0 raises error n 1 ceil real + M n 1 trunc real + M n 1 floor real + M n 1 fabs real + M n 1 rint flts + M n 1 isnan flts -> bool n 1 isinf flts -> bool n 1 isfinite flts -> bool n 1 signbit real -> bool n 1 modf real -> (frac, int) n 1 logical_not bool + nums + M -> bool n 2 left_shift ints + O flts raise an error n 2 right_shift ints + O flts raise an error n 2 add bool + nums + O boolean + is || n 2 subtract bool + nums + O boolean - is ^ n 2 multiply bool + nums + O boolean * is & n 2 divide nums + O n 2 floor_divide nums + O n 2 true_divide nums + O bBhH -> f, iIlLqQ -> d n 2 fmod nums + M n 2 power nums + O n 2 greater bool + nums + O -> bool n 2 greater_equal bool + nums + O -> bool n 2 less bool + nums + O -> bool n 2 less_equal bool + nums + O -> bool n 2 equal bool + nums + O -> bool n 2 not_equal bool + nums + O -> bool n 2 logical_and bool + nums + M -> bool n 2 logical_or bool + nums + M -> bool n 2 logical_xor bool + nums + M -> bool n 2 maximum bool + nums + O n 2 minimum bool + nums + O n 2 bitwise_and bool + ints + O flts raise an error n 2 bitwise_or bool + ints + O flts raise an error n 2 bitwise_xor bool + ints + O flts raise an error n 2 arctan2 real + M n 2 remainder ints + real + O n 2 hypot real + M ===== ==== ============= =============== ======================== Types other than those listed will be accepted, but they are cast to the smallest compatible type for which the function is defined. The casting rules are: bool -> int8 -> float32 ints -> double Nr r r r r �test_all_ufunc� s QzTestUfunc.test_all_ufuncr � c C sR t �ddd�\}}}}}t|d� t|d� t|d� t|| jf� t|d� d S )Nr r � (i),(i)->())r r r �r r ������umt�test_signaturer � size_inferred�r ZenabledZnum_dimsZixs�flagsZsizesr r r �test_signature0� s � zTestUfunc.test_signature0c C sN t �ddd�\}}}}}t|d� t|d� t|d� t|d� t|d� d S )Nr r z (),()->()r �r r r r �rm rn r rp r r r �test_signature1 s � zTestUfunc.test_signature1c C sV t �ddd�\}}}}}t|d� t|d� t|d� t|| jfd � t|d� d S )Nr r z(i1,i2),(J_1)->(_kAB)�r r r )r r r rH rg �rk rk rk rk rl rp r r r �test_signature2 s � zTestUfunc.test_signature2c C sV t �ddd�\}}}}}t|d� t|d� t|d� t|| jfd � t|d� d S )Nr r z (i1, i12), (J_1)->(i12, i2)�r r r )r r r r rH rg rw rl rp r r r �test_signature3 s � zTestUfunc.test_signature3c C sV t �ddd�\}}}}}t|d� t|d� t|d� t|| jfd � t|d� d S )Nr r z(n,k),(k,m)->(n,m)�r r r �r r r r r r rH �rk rk rk rl rp r r r �test_signature4$ s � zTestUfunc.test_signature4c C sf t �ddd�\}}}}}t|d� t|d� t|d� t|| j| jB | j| j| jB f� t|d� d S )Nr r z(n?,k),(k,m?)->(n?,m?)r{ r| r} )rm rn r ro � can_ignorerp r r r �test_signature5. s � �zTestUfunc.test_signature5c C sN t �ddd�\}}}}}t|d� t|d� t|d� t|d� t|d� d S )Nr z(3)->())r r )r �rH rt rp r r r �test_signature6: s � zTestUfunc.test_signature6c C sV t �ddd�\}}}}}t|d� t|d� t|d� t|d| jdf� t|d� d S )NrH r z(3),(03,3),(n)->(9)�r r r r �r r r r r r �rH rk � rl rp r r r �test_signature7C s � zTestUfunc.test_signature7c C sX t �ddd�\}}}}}t|d� t|d� t|d� t|| j| jdf� t|d� d S )NrH r z(3?),(3?,3?),(n)->(9)r� r� r r� )rm rn r r ro rp r r r �test_signature8L s � zTestUfunc.test_signature8c C s& t t�� t�ddd� W 5 Q R X d S )Nr r z ((i)),(i)->()�r r rm rn r r r r �(test_signature_failure_extra_parenthesisU s z2TestUfunc.test_signature_failure_extra_parenthesisc C s& t t�� t�ddd� W 5 Q R X d S )Nr r z (i),)i(->()r� r r r r �.test_signature_failure_mismatching_parenthesisY s z8TestUfunc.test_signature_failure_mismatching_parenthesisc C s& t t�� t�ddd� W 5 Q R X d S )Nr r z(i),->()r� r r r r �2test_signature_failure_signature_missing_input_arg] s z<TestUfunc.test_signature_failure_signature_missing_input_argc C s& t t�� t�ddd� W 5 Q R X d S )Nr rh r� r r r r �3test_signature_failure_signature_missing_output_arga s z=TestUfunc.test_signature_failure_signature_missing_output_argc C s t tjjd� d S )Nrh )r rm �inner1dr r r r r �test_get_signaturee s z TestUfunc.test_get_signaturec C s� dt jddd� }tt �|d�dddg� tt j|dddd �d d dg� tt j|dd dd �d d dg� tt j|dd dd �d d dg� tt j|dd dd �d d dg� t jddd�}t j|d|d� t|dddg� d |d d �< t j|dd|dd� t|d d dg� d |d d �< t j|dd |dd� t|d d dg� d |d d �< t j|dd |dd� t|d d dg� d |d d �< t j|dd |dd� t|d d dg� d S )N� �?rH �f8r+ r � �?�i�unsafe)r �castingr r )�i4)r� r� r� r� ��out)r r� r� )r r^ r r �zeros�r �a�br r r �test_forced_sigh s2 �� zTestUfunc.test_forced_sigc C s� t �d�}t �d�}t �d�}dD �]J}t �|�}|�|�}|�|�}t�|| g|| g�D �] \}} |dkr�t|�t| � }nt|�t| � }t �|| �} t t � | �j d�} t | || d� |dkr�t | jjd k� nt | jj|jk� dD ]"} t �| �} ttt j|| | d � q�d D ]�} t �| �} |dk�rDttt j|| | d � n�t|�t| � }t t � | �j d�} t t � | �jd �}t jd d�� t j|| | d �} W 5 Q R X t �| ��s�| dk�r��qt | || |d� t | jj| jk� �qdD ]�} t �| �} t|�t| � }t t � | �j d�} t t � | �jd �}t j|| | d �} t �| ��sF�q�t | || |d� t | jj| jk� �q�q^q"t jdt jd �}t �||�} t | dk� t | jjd k� t �| |�} t | dk� t | jjd k� d S )Nr* � r� ZbhilqBHILQefdgFDGZFDGgV瞯�<)�rtolZ bhilqBHILQ�float64r+ Zefdggw�Mh�� �ignore��invalid�e)r� �atolr � �?� )r r_ r Zastype� itertools�product�complex�float� true_divide�max�finfoZ resolutionr r �namer r ZtinyZerrstate�isfiniter= r8 )r r� r� �tgtZtc�dtZaaZbbr0 r1 rX r� ZtcoutZdtoutr� r r r �test_true_divide� sd " zTestUfunc.test_true_dividec C s` t jdt jd�}t|d �� |jd dd� t jdt jd�}t|d �� |jd dd� d S )N�� r+ � $@r rg � )r r= �float32r �sum�sizer� �r r� r r r �test_sum_stability� s z TestUfunc.test_sum_stabilityc C s� t tjtjtjtjfD �]�}dD ]�}|||d d �}tjd|d |d�}t�|� }tj dd��^}t� d� t t� |�|� t t|�d| � t t� |d d d� �|� t t|�d| � W 5 Q R X q tjd |d�}t t� |d d d� �d � t t� |dd d� �d � t t� |d d d � �d � t t� |dd d � �d � t t� |d d d � �d � t t� |dd d � �d � t t� |d d d� �d � t t� |dd d� �d � tjd|d�}||7 }t |d� qd S )N� r r r r] � r� � � � � � i i� r r r+ T��record�alwaysrk r� g @o@rH g �d@�������r � @)r r Zfloat16r� r� rL r^ r� �warnings�catch_warnings� simplefilterr r� r �lenr= )r r� �vr� �dZoverflow�wr r r �test_sum� s. zTestUfunc.test_sumc C s� t jt jt jfD �]�}dD ]�}|||d d �|||d d d � }t j||d�}t �d|d �|_t �d|d � |_tt � |�|� tt � |d d d� �|� qt j d|d�d }tt � |d d d� �d� tt � |dd d� �d� tt � |d d d � �d � tt � |dd d � �d � tt � |d d d � �d� tt � |dd d � �d� tt � |d d d � �d � tt � |dd d � �d � t j d |d�d }||7 }t|d� qd S )Nr� r r y �?r+ rk r� y @o@ @o@rH y �d@ �d@r� r� r� y @ @) r Z complex64Z complex128rM �emptyr^ �real�imagr r� r= )r r� r� r� r� r r r �test_sum_complex� s( , zTestUfunc.test_sum_complexc C sX t tjdgdd�d� ttjdgdd�d� t tjtjdtjd �d dd �d d d g� d S ) NrH r ��initialr\ g�������?皙�����?g333333�?)r rH r\ r+ �r r )�axisr� � )r r r� r r= �int64r r r r �test_sum_initial s �zTestUfunc.test_sum_initialc C sT t tjddgddggddgd�d� t tjddgddggdd ddgd �d d g� d S ) Nr� r� � @� @TF��wherer � @)r� r� r� � "@)r r r� r r r r �test_sum_where s $��zTestUfunc.test_sum_wherec C sZ t �d��d�}tt�||�t j|| dd�� t �d�}tt�||�t �|| �� d S )Nr[ �r rH rk �r� )r r^ �reshaper rm r� r� r� r r r � test_inner1d s zTestUfunc.test_inner1dc C s� d}t �d��d�}t �d��d�}tt�||�t j|| dd�|d� d}t �d��d �}tt�||�t j|| dd�|d� t �d ��d �}t �d��d �}tttj||� t �d ��d �}t � d �}tttj||� t �d��d�}t �d��d�}tttj||� t �d�}t �d��d �}t � ||�\}}t |j d d� || }t jdd��:}t �d� ||7 }t t|�d� t|d |d k� W 5 Q R X d S )NZ broadcastrg ry )r r r rk r� ��err_msgz"extend & broadcast loop dimensions�r r r� )rg r )rg r r] r rv rH )rH r r r Tr� r� r ri )r r^ r� r rm r� r� r r r_ Zbroadcast_arraysr �stridesr� r� r� r� r )r �msgr� r� �ur� r0 r� r r r �test_broadcast" s4 $$ zTestUfunc.test_broadcastc C s� d}t jddd��d�}tt�||�t j|| dd�|d� d }t �d��d�}|d }tt�||�t j|| dd�|d� d S ) Nz type castr[ Zshortr+ r� rk r� r� ztype cast on one argumentr� )r r^ r� r rm r� r� r )r r� r� r� r r r �test_type_castC s � �zTestUfunc.test_type_castc C sP d}t jddd��d�}tt�||�t j|| dd�|d� d }t jdd d��d�}tt�||�t j|| dd�|d� t jd d d�}t jd d d�}t|| jt �d�� t|| jt �d�� t|| jt �d�� t|| jt �d�� tt � |�jt �d�� tt � |�jt �d�� tt � |�jt �d�� tt � |�jt �d�� d S )Nz big endianr[ z>i4r+ r� rk r� r� z little endianz<i4r z>f8z<f8r� ) r r^ r� r rm r� r� r r Zabsolute�negative)r r� r� ZBaZLar r r � test_endianN s( � �zTestUfunc.test_endianc C s� d}t �d��d�}|d d �dd d �dd d �df }|d d �dd d �dd d �df }d|d< d}t|d d|d � tt�||�t j|| dd �|d � t �d ��d dd�}|j}|j}d|d< t|d d|d � tt�||�t j|| dd �|d � d S )Nz#incontiguous memory layout of array�@ )r r r r r r r r rk rs z-make sure it references to the original array)r r r r r r r� r� � r rH rg ) r r^ r� r r rm r� r� �T)r r� r0 r� r� Zmsg2r r r �test_incontiguous_arraye s $z!TestUfunc.test_incontiguous_arrayc C s� d}t �d��d�}t �d��d�d }t jddd �}t�|||� t|t j|| d d �|d � d |d d �< tj|||d � t|t j|| d d �|d � d}t jddd �}t�|||� t|t j|| d d �|d � d |d d �< tj|||d � t|t j|| d d �|d � d}t jddd �}t�|||d � t|d t j|| d d �|d � d |d d �< tj|||d d � t|d t j|| d d �|d � d S )Nzoutput argumentr� �r rH r rg ry r r� r r+ rk r� r� r� z output argument with type cast�int16z(output argument with incontiguous layout�r rH rg ).r )r r^ r� r� rm r� r r� )r r� r� r� �cr r r �test_output_argumentu s. z TestUfunc.test_output_argumentc C s, t j}t�d��d�}t�dd��d�}|||�}t||| �d�� |||dddgd �}t||| �d�� |||dddgd �}t||| �d�� |||dddgd �}t||| �d�� |||ddgd �}t||| �d�� |||t�d�tjdtj d �gd �}t||| �d�� |||d d gd �}t||| �d �� |||d d gd �}t||� d d d �| �d�� t t |||dd � t t |||dgd � t t |||ddgd � t t |||ddgd � t t |||d d gd � t t |||dddgd � t t |||dddgd � t t |||dddgd � t t |||d d gd � t j}t�d��d�}t�d��d�d }|||�}t|t�||�� |||dddgd �}t|t�||�� |||dddgd �}t|t�||�� |||dddgd �}t|t�|� d d d �|� d dd d ��� t�d�}||||dddgd�}t||k� t|t�||�� t�d�}||||ddd gd�}t||k� t|t�||�� dd d d �� t t |||d d � t t |||d d � t t |||d d � t t |||g d � t t |||dgd � t t |||dddgd � t t |||dddgd � t t |||d dgd dgd dggd � t t |||ddd dggd � t t |||ddd gd � t t |||ddd gd � t�d!�}t t |||d � t t ||||d d �d f d"� t t ||d |d d gd � t t ||||d d d gd� t t tjd#d#d gd � t t |||d d gd$d%� d S )&N� ;@�rH rH rH r� � 3@�rH r rH rk rj r )�axesr+ r r r � �)r �r� rk r� r� r� �r r r r �r r r� )r rk rH )r r rH r )r� r )r r r rH )rH r r� )r r r r� )r� r� r� r� r� T)r r# )rm r� r r^ r� r r� Zint8r_ Zint32� transposer r r �matrix_multiply�matmulr� r r� r )r r� r� r� r� �mmr� �zr r r �test_axes_argument� s� $ � � � z TestUfunc.test_axes_argumentc C s> t j}t�d��d�}t�dd��d�}|||�}t||| �d�� |||dd�}t||| �d�� t�|�}|||d|d�}t||k� t||� |||d d�}t||| �d �� t�d ��d �}t�d d ��d �}t�dd��d �}tt j |||d d�tj|| | d d�� tt j |d d�tj |d d�� tt j |dd�tj |dd�� t� |�}t j ||d d�}t||k� t|tj |d d�� t j ||dd�}t||k� t|tj |dd�� t t |||d d d gd� t t |||d gd� t j}t t |||dd� tjd|jd�}t tt j ||d d� t t tjddd d� d S )Nr� r� r� r� r� rk r� )r� r� r r[ r� r* r� r� � �r� r� r )r� r �r r rH r+ r� )rm r� r r^ r� r r� � zeros_liker �innerwt�cumsumZ empty_liker r r r� r r r ) r r� r� r� r� r� r� r� r r r r �test_axis_argument� sF � z TestUfunc.test_axis_argumentc C s t j}t�d��d�}t�dd��d�}|||�}t||| �d�� |||dd�}t||| �d�� |||d d�}t||| jdd d�� t�|�}|||d |d �}t||k� t||� |||ddd �}t||| jddd�� |||dd d �}t||| jdd d�� |||d dd �}t||| jd dd�� |||d d d �}t||| jd d d�� |||d d dgdd�}t||| �d�� |||d d d gd d�}t||| jdd d�� |||d d gdd�}t||| �d �� |||d d d gd d�}t||| jd d d�� |||d dgdd�}t||� ddd �| �d�� |||d dgd d�}t||� ddd �| jdd d�� |||d ddgd d�}t||� ddd �| jdd d�� |||d dd gd d�}t|||� dd d� jd d d�� |||d ddgd d�}t||� dd d�|� d dd� jdd d�� t� d�t�d�d d �tj tj f }t � |�}t j |dd�}t||� t j |d d�}t||d d �tj tj f � t� d�t�d�d d �tj tj f }t �|�\}} t j|dd�\} } t| |� t| | � t j|d d�\} } t| |d d �tj tj f � t| | d d �tj tj f � t�d��d�}t�dd��d�}t�dd��d�} tt j||| d d�tj|| | dd d �� tt j||| d d d �tj|| | d d d �� tt|||dd� t j} tt| ||d d� tt| ||dd� tttjdddd� d S ) Nr� r� r� r� r� rk F��keepdimsT)r r� )r� r r rj r )r r r r rH r� r[ r� r* r� r� r �truer� )rm r� r r^ r� r r� r r r �eye�newaxis�umlZdetZslogdetr r r r r )r r� r� r� r� r� r� �expectedZ expected_sZ expected_lZcsZclr� r r r r �test_keepdims_argument# s� ��" �& & ��z TestUfunc.test_keepdims_argumentc C s� t �d��d�}t �dd��d�}t �dd��d�}tt�|||�t j|| | dd�� t �d d ��d �}t �d d ��d �}t �dd��d �}tt�|||�t j|| | dd�� d S )Nr[ r� r* r� r� r rk r� �d �| r� �� �� �, iD )r r^ r� r rm r r� �r r� r� r� r r r � test_innerwtr s &zTestUfunc.test_innerwtc C sT t jg dd�}t jg dd�}t jg dd�}tt�|||�t j|| | dd�� dS )z/Test generalized ufunc with zero-sized operandsr� r+ rk r� N)r r_ r rm r r� r r r r �test_innerwt_empty| s z TestUfunc.test_innerwt_emptyc C s� t �d�}tt�||�t �d�� t �d�}t�|d ||�}t||k� t|t �t �d�|d |d f�� tt tjt �d�t �d�� tt tj|t � d�� tt tj|t � d�t �d �� d S ) z Test with fixed-sized signature.rH �rH rH r r r rg r� r� )rH rg N) r r r rm Zcross1dr� r Zvstackr r r^ )r r� r� �resultr r r � test_cross1d� s & zTestUfunc.test_cross1dc C sR t �d��d�}t �d�}|d d �t jf }t �d��d�d }t�||�}t�||�}t||� tt tj||� t�||�}t||� � � t�||�}t�||�}t||� t �d�}|t jd d �f }t �d��d �d } t�||�} t�||�} t| | � tt tj||� t�||�} t| | � � � t�| |�} t�| |�} t| | � |j }| � d d �}t�||�} t�||�}t|| � tt tj||� t�||�}t|| � � � t�| |�}t�| |�}t||� t � |�}tj| ||d �}t||� d |d d �< tj| ||d �}t||� t � | �}tt tj|||� t�|||�}t|| � � � d S )Nr� r� r r� r r rH r� )rg r r r rH r� rk r� r )r r^ r� r rm r r r r r �squeezer� �swapaxesr )r ZmatZ single_vecZcol_vecZ col_vec_arrayZ mm_col_vecZmatmul_col_vecZ matmul_colZrow_vecZ row_vec_arrayZ mm_row_vecZmatmul_row_vecZ matmul_rowZmm_row_col_vecZmatmul_row_col_vecZmatmul_row_colZmm_row_col_arrayZmatmul_row_col_arrayr� r r r �test_can_ignore_signature� s` �z#TestUfunc.test_can_ignore_signaturec C s | � tj� | � tj� d S r3 )� compare_matrix_multiply_resultsr ZlongrK r r r r �test_matrix_multiply� s z TestUfunc.test_matrix_multiplyc C sT t �t�d�t�d��}t|t�d�� t �t�d�t�d��}t|t�d�� d S )N)r r* )r* r ri )r* r* )rm r r r= r r� )r rX r r r � test_matrix_multiply_umath_empty� s z*TestUfunc.test_matrix_multiply_umath_emptyc s� t jt j�ddd�|d�}t jt j�ddd�|d�}d|jj }� fdd�� �fdd ��d d � }� d�}�d�td d d �fd f }d }|D ]�} |D ]�} |D ]�} |D ]�} |�| �| } |�| �| }|o�| jd k }|o�|jd k }| j d |j d kr�|| j d |j d �r�t t � | |�t j |dt jf �dd �| dt jd d �f d d�|dt| j �t|j �f d� q�q�q�q�t|d dd� d S )Nr rH rg r+ zmatrix multiply on type %sc sp | dkrdgfS d}� | d �}|D ]F}t | �D ]8}|| d g }|| || d < | d ||< ||f7 }q0q$|S )Nr r r )rR )�n�ret�baseZpermr� �new)� permute_nr r r/ � s z<TestUfunc.compare_matrix_multiply_results.<locals>.permute_nc sT | dkr dS d}� | d �}|D ].}||t d �f f7 }||t dd�f f7 }q |S )Nr )r r r )�slice)r+ r, r- Zsl)�slice_nr r r1 � s z:TestUfunc.compare_matrix_multiply_results.<locals>.slice_nc S s | |kp| dkp|dkS r6 r )�s1�s2r r r � broadcastable� s z@TestUfunc.compare_matrix_multiply_results.<locals>.broadcastablerk Tr� r .r� r� z %s %sr� zreference check)r r_ ZrandomZrandr r� r0 r r- �shaper rm r r� r r& �strr )r �tpZd1Zd2r� r4 Z permute_3Zslice_3�refZp1Zp2r2 r3 Za1Za2r )r/ r1 r r( � s@ � �� ��z)TestUfunc.compare_matrix_multiply_resultsc C s� t jdtd��dd�}t j|jd |jd d d f|jd�}t�||� t � t j |d d �d f | d dd ��}|t j |jd t d� }t ||� tttj|� d S ) Nr� r+ rg rH r r r rk r� )r r^ r� r� r� r5 r rm Zeuclidean_pdist�sqrtr� Ztri�boolr r r )r r� r� r� r r r �test_euclidean_pdist s * ( z TestUfunc.test_euclidean_pdistc C s& t �d�}t�|�}t||�� � d S )Nr* )r r^ rm r r )r r� r# r r r � test_cumsum s zTestUfunc.test_cumsumc C s� t jdd ddddgtd�}tt �|d �t jdd� |D �td�� tt �|d�t jd d� |D �td�� tt �|d �t jd d� |D �td�� tt �|d �t jd d� |D �td�� tt �|d �t jdd� |D �td�� tt �|d�t jdd� |D �td�� tt �|d �t jdd� |D �td�� tt �|d �t jdd� |D �td�� tt �|�t jdd� |D �td�� tt j�|�d� tt j�|�d � d S )NrH TFZtest� r+ c S s g | ] }|pd �qS r3 r ��.0r0 r r r � <listcomp>( s z1TestUfunc.test_object_logical.<locals>.<listcomp>c S s g | ] }|pd �qS �Tr r> r r r r@ * s r� c S s g | ] }|pd �qS �r� r r> r r r r@ , s �blahc S s g | ] }|pd �qS �rC r r> r r r r@ . s c S s g | ] }|od �qS r3 r r> r r r r@ 1 s c S s g | ] }|od �qS rA r r> r r r r@ 3 s c S s g | ] }|od �qS rB r r> r r r r@ 5 s c S s g | ] }|od �qS rD r r> r r r r@ 7 s c S s g | ] }| �qS r r r> r r r r@ : s )r r_ r> r � logical_or� logical_andZ logical_not�reducer� r r r �test_object_logical% s< � � � � � � � � �z TestUfunc.test_object_logicalc C s� G dd� dt �}t�|� �}t||kd� tt�||�d� ttj||t d�d� t�|� g�}t||kt�dg�� tt�||�t�dg�� ttj||t d�t�dg�� d S )Nc @ s e Zd Zdd� ZdS )z8TestUfunc.test_object_comparison.<locals>.HasComparisonsc S s dS )N�==r �r �otherr r r �__eq__A s z?TestUfunc.test_object_comparison.<locals>.HasComparisons.__eq__N)r% r& r' rL r r r r �HasComparisons@ s rM Tr+ rI )r> r r_ r �equal)r rM Zarr0dZarr1dr r r �test_object_comparison? s z TestUfunc.test_object_comparisonc C sP t jdddgtd�}tt �|�d� tt �|�d� tt �|�d� t jdddgtd�}tt �|�d� tt �|�d � tt �|�d� tt � |�d� tt �|�d� tt �|�d� tt jd ggtd��� d � tt jd dgggtd��d �d dg� tt jd gtd�jd d �d� tt jd gdd ggtd�jd gddgd�d dd g� d S )Nr� r� r� r+ �abcTFr r r �r r r� rH )r� r� ) r r_ r> r r� r� �min�prod�anyr? r� r r r �test_object_array_reductionN s( & ��z%TestUfunc.test_object_array_reductionc C s� t jdtd�}dd� td�D �|d d �< t jj||d� t jj||d� t|t �dd� dD ��� t jdtd�}d d� td�D �|d d d �f< t jj||d d � t jj||d d � t|d d d �f t �d d� dD ��� d S )Nrg r+ c S s g | ] }d g�qS r� r �r? r� r r r r@ d s zBTestUfunc.test_object_array_accumulate_inplace.<locals>.<listcomp>r� c S s g | ]}d g| �qS r� r rV r r r r@ h s )r rH r[ r* �r rg c S s g | ] }d g�qS �r r rV r r r r@ l s r rk r c S s g | ]}d g| �qS rX r rV r r r r@ o s )r r= r> rR r � accumulater r_ )r �arrr r r �$test_object_array_accumulate_inplacea s z.TestUfunc.test_object_array_accumulate_inplacec C s$ t jdtd�}dd� td�D �|d d �< t jdtd�}dd� td�D �|d d �< t jj|t �d�|d� t jj|t �d�|d� t||� t jdtd�}dd� td�D �|d d d �f< t jdtd�}d d� td�D �|d d d �f< t jj|t �d�|d d � t jj|t �d�|d d � t||� d S ) Nrg r+ c S s g | ] }d g�qS r� r rV r r r r@ t s z@TestUfunc.test_object_array_reduceat_inplace.<locals>.<listcomp>c S s g | ] }d g�qS r� r rV r r r r@ v s r� rW c S s g | ] }d g�qS rX r rV r r r r@ } s r c S s g | ] }d g�qS rX r rV r r r r@ s rk r ) r r� r> rR r rb r^ r r= )r rZ r� r r r �"test_object_array_reduceat_inplaceq s z,TestUfunc.test_object_array_reduceat_inplacec C sx g t jg td�fD ]`}tt �|�d� tt �|�d� tt �|�d� tt �|�d� tt t j |� tt t j |� qd S )Nr+ r r FT) r r_ r> r r� rS rT r? r r r� rR r� r r r �test_zerosize_reduction� s z!TestUfunc.test_zerosize_reductionc C s� t �ddg�}tt j|jdd� t �ddg�}tt j|jdd� t �ddg�}tt j|jdd� t �ddg�}tt j|jdd� d S )NFr r� r� )r r_ r Z AxisErrorr? rT r� r r r �test_axis_out_of_bounds� s z!TestUfunc.test_axis_out_of_boundsc C sj t tjddd�d� t tjddd�d� t tjddd�d� t tjddd�d� t tjddd�d� t tjddd�d� t tj� d�d� t t tjt� d�dd��tj k� t t tjt� d�dd��tj k� t t tjt� d�dd��tj k� t t tjt� d�dd��tj k� t t tjddd��tj k� G dd � d tj�}t�d ��|�}t t t�|��|k� d S ) NrH r r� � @TFg @c @ s e Zd ZdS )z0TestUfunc.test_scalar_reduction.<locals>.MyArrayN)r% r& r' r r r r �MyArray� s r` r )r r r� rS rT r? r� rR �powerrG r �typer� r8 �ndarrayr_ �view)r r` r� r r r �test_scalar_reduction� s """" z TestUfunc.test_scalar_reductionc C s| t �dt j�}t �dt j�}t �dt j�}t j|||d� t|d� t �d�}t jdt jd�}t j|d|d� t|d� d S )N)r r r� r i r+ r ) r r= r� r� r r r� r� �subtract�r r� r� r� r r r �test_casting_out_param� s z TestUfunc.test_casting_out_paramc C s� t �d�}t �d�}t �d�}t j||||d dkd� t|dddddddg� t �d��dd�d }t j|ddg|ddgddggd� t|dd gd d gg� t j|d|d d gd� t|dd gdd gg� d S )Nr] r r �r� r� r rg r[ rH � r� r\ TF� ) r r^ r= r� r r r� ra rf rg r r r �test_where_param� s "zTestUfunc.test_where_paramc C sv t �dt j�}t �dt j�}dt �dt j� }t j|||ddddddddddg d� t|ddddddddddg � d S )Nr* r� r r ri r )r r= r� r� r r rg r r r � test_where_param_buffer_output� s &z(TestUfunc.test_where_param_buffer_outputc C st t jdgt jd�}t jdgtd�}tt j||d�dg� t jdgt jd�}t jdgtd�}tt j||d�dg� d S )Nr r+ Tr� )r r_ r� r: r r9 r� �r r� �mr r r �test_where_param_alloc� s z TestUfunc.test_where_param_allocc C s@ d|d< d|d< t tjj|d d�d� t tjj|dd�ddddg� t tjj|dd�dddg� t tjj|dd�ddg� t tjj|dd�ddddgddddgddddgg� t tjj|dd�ddddgddddgg� t tjj|d d�dddgdddgg� t tjj|d d�|� d|d< d|d < t tjj|d d�d� t tjj|dd�ddddg� t tjj|dd�dddg� t tjj|dd�ddg� t tjj|dd�ddddgddddgddddgg� t tjj|dd�ddddgddddgg� t tjj|d d�dddgdddgg� t tjj|d d�|� d|d< d|d < t tjj|d d�d� t tjj|dd�ddddg� t tjj|dd�dddg� t tjj|dd�ddg� t tjj|dd�ddddgddddgddddgg� t tjj|dd�ddddgddddgg� t tjj|d d�dddgdddgg� t tjj|d d�|� d S ) Nr .r )r r r r� rQ r� r r r )r r r )r r r )r r �minimumrG r� r r r � check_identityless_reduction� s` ��� ��� ���z&TestUfunc.check_identityless_reductionc C s t jddd�}| �|� d S )Nr� �C��order�r r� rr r� r r r �"test_identityless_reduction_corder s z,TestUfunc.test_identityless_reduction_corderc C s t jddd�}| �|� d S )Nr� �Frt rv r� r r r �"test_identityless_reduction_forder s z,TestUfunc.test_identityless_reduction_forderc C s$ t jddd��dd�}| �|� d S )N)r rg rH rs rt r r �r r� r& rr r� r r r �&test_identityless_reduction_otherorder s z0TestUfunc.test_identityless_reduction_otherorderc C s> t jddd��dd�}|dd �dd �dd �f }| �|� d S )N)rH r\ rg rs rt r r rz r� r r r �%test_identityless_reduction_noncontig! s z/TestUfunc.test_identityless_reduction_noncontigc C sP t jddd�}|dd � jdd�}d|_|dd �dd �dd �f }| �|� d S )N)i� Zi1r+ r r� )rH rg r\ )r r� rd r5 rr r� r r r �/test_identityless_reduction_noncontig_unaligned& s z9TestUfunc.test_identityless_reduction_noncontig_unalignedc C s t tjjg dd�d� t tjjg tjd�tj� t tjjg tj d�tj � t tjjdgdd�d� t tjjdgdd�d� t tjjdgdd�d� t tjjdgdd�d� tttjjg d d� tttjjg d d� t tjjg tj d�d� tj dgt d�}tjj|dd�}t |d� d S )Nr r� r\ rg r* r+ r� ) r r �maximumrG rq �infr r r Z_NoValuer_ r> �r r� rX r r r �test_initial_reduction- s z TestUfunc.test_initial_reductionr� )r r Nr� FTc C sh t �d��dd�}|�� }t �|�}t j|||d� t jj|||d�}|�|�}t ||� t ||� d S )Nr� rH ri �r� r� ) r r^ r� �copyr �positiver rG r� r r )r r� r� r� �a_copy�a_checkrX �checkr r r �test_reduction_with_whereJ s z#TestUfunc.test_reduction_with_wherer� r r )NTr� r� c C sl t �d��dd�}|�� }t �|jt j �}t j|||d� t jj ||||d�}|j ||d�}t ||� d S )Nr� rH ri )r� r� r� r� ) r r^ r� r� r- r5 r r� r~ rG r� r ) r r� r� r� r� r� r� rX r� r r r �%test_reduction_with_where_and_initial\ s z/TestUfunc.test_reduction_with_where_and_initialc C s4 t �d��dd�}dddg}ttt jj||d� d S )Nr� rH FTr� )r r^ r� r r r~ rG rn r r r �#test_reduction_where_initial_neededk s z-TestUfunc.test_reduction_where_initial_neededc C s� t �dddgdddgg�}t jj|dd�}t|dddg� t jj|d d�}t|ddg� t jj|d d�}t||� ttt jj|d d� d S ) Ng @r� r� r� g �?r r� r� r r rQ )r r_ �dividerG r r r r� r r r �*test_identityless_reduction_nonreorderablep s z4TestUfunc.test_identityless_reduction_nonreorderablec C sd dd� }dd� }dd� }t tjjd � ||tjjdd� ||tjjdd� ||tjjdd� ||tjjdd� |tjjg � tjjt�d �d d � ||tjjdd� ||tjjdd� ||tjjdd� ||tjjdd� tj�g � tjjt�d �d d � tjtjfD ]V}|jt�d �dd � |jt�d �dd � |jt�d�dd � |jt�d �dd � �qd S )Nc _ s | ||� d S r3 r �r/ �args�kwargsr r r �ok� s z+TestUfunc.test_reduce_zero_axis.<locals>.okc _ s t t| f|�|� d S r3 )r r r� r r r �err� s z,TestUfunc.test_reduce_zero_axis.<locals>.errc S sp | |t �||f�dd� | |t �||f�dd� | |t �|d |d |f�dd� | |t �|d ||d f�dd� | |t �||d |d f�dd� | |t �|d ||d f�dd� | |t �|d |d |d |d |d f�dd� | |t �d ||f�d d� | |t �d ||f�dd� | |t �|d |f�dd� | |t �d ||f�dd� | |t �d ||f�dd� d S ) Nr r� r r r r� rH )r r r r* rQ )r r� )Zexpect�funcr+ ro r r r �t� s" """" ��z*TestUfunc.test_reduce_zero_axis.<locals>.t� r ri r r� )r� r )r r� )r� r� )r r r~ ZidentityrG r� r rY )r r� r� r� Zufr r r �test_reduce_zero_axis~ s* z TestUfunc.test_reduce_zero_axisc C s� t jdddgtd�}ttt j|d�dddg� ttt j|d|d � d d � }tt||d� tt j|d|d d � t|dddg� d S )Nr r rH r+ g�������?g������ @g������@gffffff@r� c S s | |7 } d S r3 r )r� r� r r r � add_inplace� s z0TestUfunc.test_safe_casting.<locals>.add_inplacer� )r� r� rg )r r_ r r r r r r )r r� r� r r r �test_safe_casting� s �zTestUfunc.test_safe_castingc C s� t jdddgdd�}t jdddgdd�}t jdtjd�}t�|||�}t jdddgtjd�}t||� tt�� t�||� W 5 Q R X d S )Nr r r Zi8r+ rH rg ) r r_ r� rU �rationalrV r r r )r r� r� r� r# �targetr r r �test_ufunc_custom_out� s z TestUfunc.test_ufunc_custom_outc C s� t jddd��dd�}t jddd��dd�}t�|d d�d d�f |� t|t jdd ddgd dd d gd dd d gddd dggdd�� t �d�}t�|d� t|d� t�|ddg� t|d� d S )Nr� �lr+ rg r� rH rk r r r] � rk � r� r� r� r* )r r^ r� � opflag_testsZ inplace_addr r_ r� r r r �test_operand_flags� s �� z TestUfunc.test_operand_flagsc C sd dd l m m} tjdgdd�}tjdgdd�}|�||�}t|tjdgdd�� tt|j � d S )Nr r zu8,u8,u8r+ )r rg r[ ) Z numpy.core._struct_ufunc_tests�coreZ_struct_ufunc_testsr r_ Z add_tripletr r r! Z register_fail)r Z struct_ufuncr� r� r# r r r �test_struct_ufunc� s zTestUfunc.test_struct_ufuncc C s� t jt�dd�t�dd�t�dd�gtjd�}t jt�dd�t�dd�t�dd�gtjd�}t�||�}t jt�d�t�dd�t�dd�gtjd�}t||� d S )Nr r rH rg r+ )r r_ rU r� Ztest_add_rationalsr )r r� r� r# r r r r �test_custom_ufunc� s. �� �� ��zTestUfunc.test_custom_ufuncc C s6 t t��$ tjt�d�dtjtd fd� W 5 Q R X d S )Nr )r )r r r �multiplyrU r� r r r r r � test_custom_ufunc_forced_sig s �z&TestUfunc.test_custom_ufunc_forced_sigc sN G � fdd�dt �� t�d�� d� t� jdk� j� t� jdk� j� d S )Nc s@ e Zd ZdZdZdZdd� Zdd� Z� fdd�Z� fd d �Z d S ) z1TestUfunc.test_custom_array_like.<locals>.MyThingi� r c S s || _ d S r3 �r5 )r r5 r r r �__init__ s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__init__c S s | j d S )Nr r� r r r r �__len__ s z9TestUfunc.test_custom_array_like.<locals>.MyThing.__len__c sJ � j d7 _ t|t�s |f}t|�| jkr4td��� | jt|�d � �S )Nr Zboo)� getitem_count� isinstance�tupler� �ndim� IndexErrorr5 )r r� ��MyThingr r � __getitem__" s z=TestUfunc.test_custom_array_like.<locals>.MyThing.__getitem__c s � j d7 _ | S r6 )� rmul_countrJ r� r r �__rmul__+ s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__rmul__N) r% r& r' Z__array_priority__r� r� r� r� r� r� r r� r r r� s r� r\ r" r r )r> r r� r r� r� r r r� r �test_custom_array_like s z TestUfunc.test_custom_array_likec C s� t �d�}t j�|dddgd� t|dddddddd d d g � t �d�}t �d d d g�}t j�|dddg|� t|ddd ddddd d d g � t �d ��dd�}t �d d d gdddgdddgg�}t j�|td �dddgf|� t|dddgdddgdddgg� t �d��ddd�}t �d ddg�}t j�|td �td �dddgf|� t|ddd gdddgdddggd dd gd d d gd d!d"ggd#d$d%gd&d'd(gd)d*d+ggg� t �d ��dd�}t �d d d gdddgdddgg�}t j�|dddgtd �f|� t|dddgd,dd-gd.d/dgg� t �d��ddd�}t �d ddg�}t j�|td �dddgtd �f|� t|dddgd0dd1gd2d/dggd dd3gd4d d5gd6d7d8ggd#d9d:gd;d'd<gd=d>d?ggg� t �d ��dd�}t �d ddg�}t j�|ddddgf|� t|ddd gdddgdd d gg� t �d��ddd�}t �d ddg�}t j�|dddgdtd �f|� t|dddgdddgdd d ggd@ddAgd dBdCgd dDdEggdFdGdHgd&dIdJgd)dKdLggg� t �d��ddd�}t �d ddg�}t j�|td �td �td �f|� t|d ddMgdNdOdPgd2d/dggdQdRdSgdTdUdVgd6d7d8ggdFdGdHgdWdXdYgd=d>d?ggg� t �d�}t j�|dddg� t|ddddddZdd d d g � t �d�}t j�|d[d� t|d� t t t jj|dd� t t t jj|g d� t �d�}t j �|ddddgd\� t|t �ddd]d^dddd d d g �� t �d�}|d dk}t j �||ddddd g� t|dddddddd dd g � t jdd_d`�}t j �|dddg� t|ddddddadd d d g � t �d�}|d d �d f d d �dd�f }t j�|ddgd� t|t �d�� t �dddgt �db��� �}t �ddddgt �dc��� �}t j�||d� t|dd ddg� t jdddgtd`�}t tt jj|ddgd� t|t jdddgtd`�� t tt jjt �d�dg� d S )eNr* r r\ r r rg rH r[ r] r� r� r �� �i r r �� �f i� �� i_ i4 rj i� i� �� i� �� r� i� �� r� i� �� r� i� �� � i� �� r� i� �� i� i� �� �� �� i] �j r� �� if �s �� i= r� r� �� io r �� iF �� ic r� rk r� � �v �� i@ � � � r i. �g �� i1 �m �� i7 �p �� i: �y �� iC ���r r_ i> �. Zu4r+ l �� r� r/ r� )r r^ r Zatr r_ r� r0 r� r r� ra rN �invertr r Z newbyteorderr> r r Zmodf)r r� r� �indexZorig�valuesr r r �test_inplace_fancy_indexing3 s� " "$����� " "$����� " ����� "����� $ z%TestUfunc.test_inplace_fancy_indexingc C sB t jj}t jdtd�}t jd|jd�}|d }t||�|� t||dd�|� t||d�|� t||dd d�|� t||ddd�|� t||dd�|� t||dd �|� t||dd d d�|� t||dd |d�|� t||dd |�|� t||dd d �|� t||dd d d d �|� t||dd d d �|�d |j �� t||dd d d d�|� t||dd d d dd �|� t||dd d d dd �|� t||dd d d dd d�|� t||dd d d d�|� t||dd d d d�|� t||dd d d d�|� t||dd d d dd d�|� t t |� t t ||dd d d dd d� t t ||d� t t ||dd� t t ||dd d d� t t ||dd� t t ||dd� t t ||dd d� t t ||dd d� t t ||dd� t t ||dd d� t t ||dddd d� t t ||dd dd� t t ||dd� t t ||dd dd d� t t ||dd d d dd � t t ||dd d dd � d S ) N)r\ r r+ rX r\ r r� r� r� Fr Tr� 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� )r� r r r� r� )r� r r� r� ) r r rG r= r r r r� r5 r r )r r/ r� �o�rr r r �test_reduce_arguments� st " �� ��� � �z TestUfunc.test_reduce_argumentsc C st G dd� dt j�}t �d��dd�}|jdd��� }|�|�}t jdd d d gtd�}tt � |||d kk�d� d S ) Nc @ s e Zd Zdd� ZdS )z,TestUfunc.test_structured_equal.<locals>.MyAc _ s t ||�dd� |D �|�S )Nc s s | ]}|� tj�V qd S r3 )rd r rc )r? �inputr r r � <genexpr> s �zOTestUfunc.test_structured_equal.<locals>.MyA.__array_ufunc__.<locals>.<genexpr>)�getattr)r �ufunc�methodZinputsr� r r r �__array_ufunc__ s ��z<TestUfunc.test_structured_equal.<locals>.MyA.__array_ufunc__N)r% r& r' r� r r r r �MyA s r� g (@rg rH zf8,f8,f8r+ TFr ) r rc r^ r� rd r% r_ r: r r? )r r� r� ZraZmrar� r r r �test_structured_equal� s z TestUfunc.test_structured_equalc C sL t �d�}t �d�}t||k� t||k� t||k � t||k � d S )Nr� r� )r r_ r r� r r r �test_scalar_equal s zTestUfunc.test_scalar_equalc C s� t jt jt jt jt jt jt jt jt j t j t j t j t j t jt jt jt jt jt jt jt jt jt jt jt jt jt jt j t j t j g }t � d�}d}t � ddg�}|D ] }t t!|||� t t!|||� q�d S )N�1r r� r� )"r ra r rf r� r� r� Z floor_divideZ bitwise_andZ bitwise_orZ bitwise_xorZ left_shiftZ right_shiftZfmaxZfminZfmodZhypotZ logaddexpZ logaddexp2rF rE r; r~ rq �modZgreaterZ greater_equalZlessZ less_equalrN Z not_equalr_ r r )r Z binary_funcsr� r� r� r/ r r r � test_NotImplemented_not_returned sJ � z*TestUfunc.test_NotImplemented_not_returnedc C s� t jdt jd��ddd�}|dd�dd d�dd �f �dd d �}t jd t jd��dd�}|d d d �d d �f }|�� }t jj||�� d d�}t jj||d d�}t||� t|dd d �f |dd d �f � t|dd d �f |dd d �f � d S )Ni� r+ r] r� r� rg r[ r r� r\ r r r� r rH ) r r^ r� r� r r� r rG r )r r0 Zy_baser1 Z y_base_copyZr0Zr1r r r � test_reduce_noncontig_output) s & "z&TestUfunc.test_reduce_noncontig_outputc C s t dtjjk� d S )N� )r rm Zinner1d_no_docrJ r r r r �test_no_doc_string> s z TestUfunc.test_no_doc_stringc C s" t �ttjd �}|�d�s t�d S )Nz loop of ufunc does not support)rN Zraisesr r r9 �match�AssertionError)r �excr r r �test_invalid_argsB s zTestUfunc.test_invalid_args�natc C s, zt �|�rt�W n tk r& Y nX d S r3 )r r� r� r �r r� r r r �test_nat_is_not_finiteH s z TestUfunc.test_nat_is_not_finitec C s, zt �|�st�W n tk r& Y nX d S r3 )r Zisnanr� r r� r r r �test_nat_is_nanO s zTestUfunc.test_nat_is_nanc C s, zt �|�rt�W n tk r& Y nX d S r3 )r Zisinfr� r r� r r r �test_nat_is_not_infV s z TestUfunc.test_nat_is_not_infN)br% r& r' rY rZ re rf ro r rr ru rx rz 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; r< rH rO rU r[ r\ r] r^ re rh rl rm rp rr rw ry r{ r| r} r� rN rO rP r r_ r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Z datetime64Z timedelta64r� r� r� r r r r rQ � s� T J ! g,O G6 . �� �� A @ rQ r� c C s( g | ] }t tt|�tj�rtt|��qS r �r� r� r r� r> r r r r@ ^ s �r@ c C s� | j D ]�}d|ksd|kr q|�d�\}}dd� |D �}tjdd�� t�d� | |� }W 5 Q R X t|t�r�t|�}t|�t|�ks�t�t ||�D ] \}}|j t � |�ks�t�q�q|j t � |�kst�qd S ) z� Check all ufuncs that the correct type is returned. Avoid object and boolean types since many operations are not defined for for them. Choose the shape so even dot and matmul will succeed �O�?�->c S s g | ]}t �d |��qS )r" )r r= �r? r� r r r r@ m s z$test_ufunc_types.<locals>.<listcomp>Tr� r� N) �types�splitr� r� �filterwarningsr� r� r� r� ra r r ) r� �typ�inpr� r� rX Zoutsr� r� r r r �test_ufunc_types^ s r c C s( g | ] }t tt|�tj�rtt|��qS r r� r> r r r r@ y s �c C s" | j D �]}ttd�t|�@ �r"q|�d�\}}dd� |D �}dd� |D �}|D ]}tdd�|_qP|D ]}tdd�|_qftjdd �� t�d � | |� }| |� }W 5 Q R X t |�dkr�|f}|f}t ||�D ]N\} } | j } t � | t j��rt �| �j} d | } t||| | d � q�t| | � q�qd S )zs Check that contiguous and non-contiguous calls to ufuncs have the same results for values in range(9) zO?mMr� c S s g | ]}t �d |��qS )r[ �r r� r� r r r r@ � s z,test_ufunc_noncontiguous.<locals>.<listcomp>c S s" g | ]}t �d |�ddd� �qS )r� NrH r r� r r r r@ � s r r] Tr� r� r )r� r� N)r� rT �setr� rR Zflatr� r� r r� ra r r Z issubdtypeZfloatingr� Zepsr r )r� r r r� Zargs_cZargs_nr� Zres_cZres_nZc_arZn_arr� Zres_epsZtolr r r �test_ufunc_noncontiguousy s0 r )*Z __future__r r r r� r� rN Znumpyr Znumpy.core._umath_testsr� Z _umath_testsrm Znumpy.linalg._umath_linalgZlinalgZ _umath_linalgr Z numpy.core._operand_flag_testsZ_operand_flag_testsr� Znumpy.core._rational_testsrU Z numpy.testingr r r r r r r r Z numpy.compatr r> r r( rQ rO rP �dirr r r r r r �<module> s< ( a [