Edit File: test_ufunc.cpython-37.pyc
B ��Fd�� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlm m Z d dl m m Z d dlmZmZmZ d dlmZmZmZmZmZmZmZmZm Z m Z G dd� de�Z G dd� de�Z e d kr�e� dS ) � )�division�absolute_import�print_functionN)�rational�test_add�test_add_rationals) �TestCase�run_module_suite�assert_� assert_equal� assert_raises�assert_array_equal�assert_almost_equal�assert_array_almost_equal�assert_no_warnings�assert_allclosec @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestUfuncKwargsc C s� t ttjdddd� t ttjddtjd� t ttjdddgd� t ttjddd d� t ttjddd d � t ttjddd d � t ttjddd d � t ttjdddgd� d S )N� � �safe)Zcastingx)Zdtypexi )Zextobjx)Zoutxzii->i)Zsigx)Z signaturexF)ZsubokxT)Zwherex)r � TypeError�np�add�int)�self� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_ufunc.py�test_kwarg_exact s z TestUfuncKwargs.test_kwarg_exactc C s t ttjddddd� d S )Nr r zii->i)�sig� signature)r � ValueErrorr r )r r r r �test_sig_signature s z"TestUfuncKwargs.test_sig_signaturec C s4 t ttjdddtjd� t ttjdddtjd� d S )Nr r zii->i)r �dtype)r r" )r � RuntimeErrorr r r )r r r r �test_sig_dtype s z TestUfuncKwargs.test_sig_dtypeN)�__name__� __module__� __qualname__r r! r$ r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z 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$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:dqdr� Z;dsS )t� TestUfuncc C s@ dd l }t|�|�tj��tjk� t|�|�t��tk� d S )Nr )�pickler �loads�dumpsr �sinr )r r) r r r � test_pickle( s zTestUfunc.test_picklec C s$ dd l }d}t|�|�tjk� d S )Nr sM cnumpy.core _ufunc_reconstruct p0 (S'numpy.core.umath' p1 S'cos' p2 tp3 Rp4 .)r) r r* r �cos)r r) Zastringr r r �test_pickle_withstring0 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 � � � ) r �arange�array�list�zip�ravelr r �reduceat)r �L�x�idxr r r �test_reduceat_shifting_sum6 s 0z$TestUfunc.test_reduceat_shifting_sumc C s� t j}dd� }d}d}d}t jdt jd�ddd � }t||�||d � d }t jdt jd�ddd � }t||�||d � d }t jdt jd�ddd � }t||�||d � d }t jdt jd�ddd � }t||�||d � d}t jdt jd�ddd � }t||�||d � d}t jdt j d�ddd � }t||�||d � d}t j dt jd�ddd � }t|||�||d � d}t j dt jd�ddd � }t|||�||d � d}t j dt jd�ddd � }t|||�||d � d}t j dt jd�ddd � }t|||�||d � d}t j dt jd�ddd � }t|||�||d � d}t j dt j d�ddd � }t|||�||d � G dd� dt �}d}t j dt j d�ddd � }t t � t �|�dk�|� d}t jdt j d�ddd � }x tt|��D ]}|� ||< �q�W t t � t �|�dk�|� d}t j dt j d�ddd � }t t � t �||�d k�|� d }t jdt j d�ddd � }x tt|��D ]}|� ||< �qjW t t � t �||��|� dS ) a� 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. Fixme, currently untested: PyUFunc_ff_f_As_dd_d PyUFunc_FF_F_As_DD_D PyUFunc_f_f_As_d_d PyUFunc_F_F_As_D_D PyUFunc_On_Om c S s | | S )Nr )r; �yr r r �<lambda>r � z.TestUfunc.test_generic_loops.<locals>.<lambda>r Z PyUFunc_f_f� )r" r Nr )�err_msgZ PyUFunc_d_dZ PyUFunc_g_gZ PyUFunc_F_FZ PyUFunc_D_DZ PyUFunc_G_GZ PyUFunc_ff_fZ PyUFunc_dd_dZ PyUFunc_gg_gZ PyUFunc_FF_FZ PyUFunc_DD_DZ PyUFunc_GG_Gc @ s e Zd Zdd� Zdd� ZdS )z)TestUfunc.test_generic_loops.<locals>.fooc S s t �d�S )Nr )r �bool_)r r r r � conjugate� s z3TestUfunc.test_generic_loops.<locals>.foo.conjugatec S s t �d�S )Nr )r rC )r �objr r r � logical_xor� s z5TestUfunc.test_generic_loops.<locals>.foo.logical_xorN)r% r& r' rD rF r r r r �foo� s rG Z PyUFunc_O_OZPyUFunc_O_O_methodTZ PyUFunc_OO_OZPyUFunc_OO_O_method)r �exp�zeros�singler �double� longdouble�csingle�cdouble� clongdouble�ones�objectr �all�abs�range�lenrD r rF ) r �foneZftwoZfone_valZftwo_val�msgr; rG �ir r r �test_generic_loops< sr 5 z TestUfunc.test_generic_loopsc 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_ufuncc C s* t t�ddd�d� t t�ddd�d� d}y t�ddd�}t |d |d� W n tk r` Y nX d }y t�ddd �}t |d |d� W n tk r� Y nX d }y t�ddd �}t |d |d� W n tk r� Y nX d }y t�ddd�}t |d |d� W n tk �r Y nX t t�ddd �d� d S )Nr r z (i),(i)->()z (),()->()r zcore_sig: extra parenthesisz ((i)),(i)->())rB z core_sig: parenthesis matchingz (i),)i(->()z core_sig: incomplete signaturez(i),->()z(i1,i2),(J_1)->(_kAB))r �umt�test_signaturer )r rW �retr r r r\ s6 zTestUfunc.test_signaturec C s t tjjd� d S )Nz (i),(i)->())r r[ �inner1dr )r r r r �test_get_signatureJ 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 )Ng �?r1 �f8)r" r g �?rX �unsafe)r �castingr zii->i)�i4)rc rc rc )r1 )�out)r rd rb )r r4 r r rI )r �a�br r r �test_forced_sigM s, zTestUfunc.test_forced_sigc C s� t �d�}t �d�}t �d�}�xfdD �]\}t �|�}|�|�}|�|�}�x6t�|| g|| g�D �]\}} |dkr�t|�t| � }nt|�t| � }t �|| �} t t � | �j d�} t | || d� |dkr�t | jjd k� nt | jj|jk� x*dD ]"} t �| �} ttt j|| | d � q�W x�d D ]�} t �| �} |dk�rRttt j|| | d � n�t|�t| � }t t � | �j d�} t t � | �jd �}t jd d�� t j|| | d �} W d Q R X t �| ��s�| dk�rȐq$t | || |d� t | jj| jk� �q$W x�dD ]�} t �| �} t|�t| � }t t � | �j d�} t t � | �jd �}t j|| | d �} t �| ��sX�q�t | || |d� t | jj| jk� �q�W qfW q&W t jdt jd �}t �||�} t | dk� t | jjd k� t �| |�} t | dk� t | jjd k� d S )NrA � g �?ZbhilqBHILQefdgFDG�FDGgV瞯�<)�rtol� bhilqBHILQ�float64)r" �efdggw�Mh�� �ignore)�invalid�e)rj �atolr g �?g )r r5 r" �astype� itertools�product�complex�float� true_divide�max�finfo� resolutionr r �namer r �tiny�errstate�isfiniterP rC )r re rf �tgtZtc�dt�aa�bbr; r> �resrj ZtcoutZdtoutrq r r r �test_true_dividef sd & zTestUfunc.test_true_dividec C s` t jdt jd�}t|d �� |jd dd� t jdt jd�}t|d �� |jd dd� d S )Ni� )r" g $@r � � )r rP �float32r �sum�sizerl )r re r r r �test_sum_stability� s z TestUfunc.test_sum_stabilityc C s� �x�t jt jt jt jt jfD �]b}x^dD ]V}|||d d �}t jd|d |d�}tt �|�|� tt �|d d d� �|� q(W t j 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�}||7 }t|d � q W d S )N) r r r r3 � � � � � � � i i� r r )r" ���i� g @o@r1 g �d@������)r g @) r r �float16r� rl rL r4 r r� rP )r r� �vr �dr r r �test_sum� s$ " zTestUfunc.test_sumc C s� �x�t jt jt jfD �]�}x�dD ]�}|||d d �|||d d d � }t j||d�}t �d|d �|_t �d|d � |_tt � |�|� tt � |d d d� �|� q W t 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� qW d S )N) r r r r3 r� r� r� r� r� r� r� i i� r r y �?)r" r� i� y @o@ @o@r1 y �d@ �d@r� r� )r y @ @) r � complex64� complex128rO �emptyr4 �real�imagr r� rP )r r� r� r r� r r r �test_sum_complex� s( , zTestUfunc.test_sum_complexc C sZ t �d��d�}tt�||�t j|| dd�� t �d�}tt�||�t �|| �� d S )Nr0 )r r1 r� )�axis)r r4 �reshaper r[ r^ r� )r re 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||� d S )N� broadcastr� )r r r )r r r r� )r� )rB z"extend & broadcast loop dimensions)r r r� )r� r )r� r r3 r )r r r r1 )r1 r r ) r r4 r� r r[ r^ r� r r r5 )r rW re rf r r r �test_broadcast� s $$ 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 castr0 �short)r" )r r1 r� )r� )rB ztype cast on one argumentg�������?)r r4 r� r r[ r^ r� r )r rW re rf r r r �test_type_cast 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 endianr0 z>i4)r" )r r1 r� )r� )rB z little endianz<i4r z>f8z<f8r` ) r r4 r� r r[ r^ r� r r" �absolute�negative)r rW re ZBaZLar r r � test_endian 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 r� )r r r z-make sure it references to the original array)r r r r r r )rB )r� � r r1 r� ) r r4 r� r r r[ r^ r� �T)r rW r; re rf �msg2r r r �test_incontiguous_array$ 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 argument� )r r1 r r� )r r r r )r r1 r )r" r� )r� )rB )rd z output argument with type cast�int16z(output argument with incontiguous layout)r r1 r� ).r )r r4 r� rI r[ r^ r r� )r rW re rf �cr r r �test_output_argument4 s. z TestUfunc.test_output_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 )Nr0 )r r1 rA r� rh � r� )r� �d �| )r r1 r� �� �� i, iD )r r4 r� r r[ �innerwtr� )r re rf �wr r r � test_innerwtO 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" r� )r� N)r r5 r r[ r� r� )r re rf r� r r r �test_innerwt_emptyY s z TestUfunc.test_innerwt_emptyc C s | � tj� | � tj� d S )N)� compare_matrix_multiply_resultsr �longrK )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 rA )rA r )r r )rA rA )r[ �matrix_multiplyr rP r rI )r r� r r r � test_matrix_multiply_umath_emptyd 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 }x�|D ]�} x�|D ]�} x�|D ]�} x�|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�W q�W q�W q�W t|d dd� d S )Nr r1 r� )r" zmatrix multiply on type %sc sx | dkrdgfS d}� | d �}xR|D ]J}xDt | �D ]8}|| d g }|| || d < | d ||< ||f7 }q4W q&W |S )Nr r r )rT )�nr] �base�permrX �new)� permute_nr r r� o s z<TestUfunc.compare_matrix_multiply_results.<locals>.permute_nc sX | dkr dS d}� | d �}x6|D ].}||t d �f f7 }||t dd�f f7 }q"W |S )Nr )r r r )�slice)r� r] r� �sl)�slice_nr r r� | s z:TestUfunc.compare_matrix_multiply_results.<locals>.slice_nc S s | |kp| dkp|dkS )Nr r )�s1�s2r r r � broadcastable� s z@TestUfunc.compare_matrix_multiply_results.<locals>.broadcastabler� Tr� r .r� )r� z %s %s)rB zreference check)r r5 �random�randr" r{ r� � transposer� �shaper r[ r� r� �newaxis�swapaxes�strr )r �tp�d1�d2rW r� Z permute_3Zslice_3�ref�p1�p2r� r� �a1�a2r )r� r� r r� j s4 $z)TestUfunc.compare_matrix_multiply_resultsc C s� t jdt jd��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" r� r1 r r r r� )r� )r r4 rv r� r� r� r" r[ �euclidean_pdist�sqrtr� �tri�boolr r r )r re rd rf r r r �test_euclidean_pdist� s * ( z TestUfunc.test_euclidean_pdistc 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 )Nr1 TF�test� )r" c S s g | ] }|pd �qS )Nr )�.0r; r r r � <listcomp>� s z1TestUfunc.test_object_logical.<locals>.<listcomp>c S s g | ] }|pd �qS )Tr )r� r; r r r r� � s r� c S s g | ] }|pd �qS )r� r )r� r; r r r r� � s �blahc S s g | ] }|pd �qS )r� r )r� r; r r r r� � s c S s g | ] }|od �qS )Nr )r� r; r r r r� � s c S s g | ] }|od �qS )Tr )r� r; r r r r� � s c S s g | ] }|od �qS )r� r )r� r; r r r r� � s c S s g | ] }|od �qS )r� r )r� r; r r r r� � s c S s g | ] }| �qS r r )r� r; r r r r� � s )r r5 rQ r � logical_or� logical_and� logical_not�reduce)r re r r r �test_object_logical� s* z TestUfunc.test_object_logicalc C s� 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� d S ) Nre rf r� )r" �abcTFr r r )r r ) r r5 rQ r r� rx �min�prod�anyrR )r re r r r �test_object_array_reduction� 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 )Nr� )r" c S s g | ] }d g�qS )r r )r� rX r r r r� � s zBTestUfunc.test_object_array_accumulate_inplace.<locals>.<listcomp>)rd c S s g | ]}d g| �qS )r r )r� rX r r r r� � s )r r1 r0 rA )r r� c S s g | ] }d g�qS )r r )r� rX r r r r� � s r r� )rd r� c S s g | ]}d g| �qS )r r )r� rX r r r r� � s )r rP rQ rT r � accumulater r5 )r �arrr r r �$test_object_array_accumulate_inplace� 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 ) Nr� )r" c S s g | ] }d g�qS )r r )r� rX r r r r� � s z@TestUfunc.test_object_array_reduceat_inplace.<locals>.<listcomp>c S s g | ] }d g�qS )r r )r� rX r r r r� � s )rd )r r� c S s g | ] }d g�qS )r r )r� rX r r r r� � s r c S s g | ] }d g�qS )r r )r� rX r r r r� � s r� )rd r� ) r r� rQ rT r r9 r4 r rP )r r� rd r r r �"test_object_array_reduceat_inplace� s z,TestUfunc.test_object_array_reduceat_inplacec C sN t jddgtd�}t jddggtd�}tt �|d�|� tt �d|�|� d S )Nr r )r" r1 r0 )r �matrixrQ r �multiply)r r� �desiredr r r �test_object_scalar_multiply� s z%TestUfunc.test_object_scalar_multiplyc C s| xvg t jg td�gD ]`}tt �|�d� tt �|�d� tt �|�d� tt �|�d� tt t j |� tt t j |� qW d S )N)r" r r FT) r r5 rQ r r� r� r� rR r r rx r� )r re 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 r5 r � AxisErrorrR r� )r re 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 ) Nr1 r )r� g @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� r� r� rR rx r� �powerr� r �typer� rC �ndarrayr5 �view)r r� re 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� )rd r i )r" r ) r rP �int64rl r r rI r� �subtract)r re rf 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 )Nr3 r r )rd �wherer r� r0 r1 � r� r2 TF� ) r r4 rP rI r r r� r� r )r re rf r� 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 )NrA g �?r r )rd r r )r rP r� rl r r )r re rf r� r r r � test_where_param_buffer_outputK 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" T)r )r r5 r� r� r r� rl )r re �mr r r �test_where_param_allocV 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� )r r )r r )r r r r )r r r )r r r )r r �minimumr� )r re r r r � check_identityless_reductiona sN $ $ $z&TestUfunc.check_identityless_reductionc C s t jddd�}| �|� d S )N)r r1 r� �C)�order)r r� r )r re r r r �"test_identityless_reduction_corder� s z,TestUfunc.test_identityless_reduction_corderc C s t jddd�}| �|� d S )N)r r1 r� �F)r )r r� r )r re r r r �"test_identityless_reduction_forder� s z,TestUfunc.test_identityless_reduction_forderc C s$ t jddd��dd�}| �|� d S )N)r r� r1 r )r r r )r r� r� r )r re 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)r1 r2 r� r )r r r )r r� r� r )r re 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� �i1)r" r r` )r1 r� r2 )r r� r� r� r )r re r r r �/test_identityless_reduction_noncontig_unaligned� s z9TestUfunc.test_identityless_reduction_noncontig_unalignedc 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 @g @g �?g �?g �?r )r� g @r r )r r )r r5 �divider� r r r )r re r� r r r �*test_identityless_reduction_nonreorderable� s z4TestUfunc.test_identityless_reduction_nonreorderablec C sh dd� }dd� }dd� }t tjjd � ||tjjdd� ||tjjdd� ||tjjdd� ||tjjdd� |tjjg � tjjt�d �d d � ||tjjdd� ||tjjdd� ||tjjdd� ||tjjdd� tj�g � tjjt�d �d d � xftjtjfD ]V}|jt�d �dd � |jt�d �dd � |jt�d�dd � |jt�d �dd � �q W d S )Nc _ s | ||� d S )Nr )�f�args�kwargsr r r �ok� s z+TestUfunc.test_reduce_zero_axis.<locals>.okc _ s t t| f|�|� d S )N)r r )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 )r r r1 )r r r rA )r r )r rI )Zexpect�funcr� r r r r �t� s """"z*TestUfunc.test_reduce_zero_axis.<locals>.t� r )r r r )r� )r r )r r )r r )r r �maximum�identityr� rI r r� )r r r r �ufr 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 r1 )r" g�������?g������ @g������@gffffff@)rd c S s | |7 } d S )Nr )re rf r r r � add_inplace s z0TestUfunc.test_safe_casting.<locals>.add_inplacera )rd rb r� )r r5 r r r r r r )r re r" r r r �test_safe_casting� s zTestUfunc.test_safe_castingc C sn t jdddgdd�}t jdddgdd�}t jdtd�}t|||�}t|t jdddgtd�� ttt||� d S )Nr r r �i8)r" r1 r� )r r5 r� r r r r r )r re rf r� �resultr 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� �l)r" r� r� r1 r� r r r3 � r � r� r� r� rA )r r4 r� � opflag_testsZ inplace_addr r5 )r re rf r r r �test_operand_flags s z TestUfunc.test_operand_flagsc C sX dd l m m} tjdgdd�}tjdgdd�}|�||�}t|tjdgdd�� d S )Nr )r r r1 zu8,u8,u8)r" )r r� r0 )Z numpy.core.struct_ufunc_test�coreZstruct_ufunc_testr r5 Z add_tripletr )r Z struct_ufuncre rf r% r r r �test_struct_ufunc% s zTestUfunc.test_struct_ufuncc C s� t jtdd�tdd�tdd�gtd�}t jtdd�tdd�tdd�gtd�}t||�}t jtd�tdd�tdd�gtd�}t||� d S )Nr r r1 r� )r" )r r5 r r r )r re rf r% �expectedr r r �test_custom_ufunc. s zTestUfunc.test_custom_ufuncc C s" t ttjtd�dttd fd� d S )Nr )r )r r r r� r r )r r r r � test_custom_ufunc_forced_sig9 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 )N)r� )r r� r r r �__init__F s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__init__c S s | j d S )Nr )r� )r r r r �__len__I 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�tuplerU �ndim� IndexErrorr� )r rX )�MyThingr r � __getitem__L s z=TestUfunc.test_custom_array_like.<locals>.MyThing.__getitem__c s � j d7 _ | S )Nr )� rmul_count)r �other)r8 r r �__rmul__U s z:TestUfunc.test_custom_array_like.<locals>.MyThing.__rmul__N) r% r& r' �__array_priority__r: r3 r1 r2 r9 r< r )r8 r r r8 @ s r8 r2 )r1 r1 r r )rQ r rl r r: r3 )r r )r8 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 jdddgt jd`�}| �tt jj|ddgd� t|t jdddgt jd`�� t tt jjt �d�dg� d S )eNrA r r2 r r r� r1 r0 r3 r� r� r� �� �i r� i, �� �f i� �� i_ i4 r i� i� �� i� �� r� i� �� r� i� �� r) i� �� � i� �� r� i� �� i� i� �� �� �� i] �j r( �� if �s �� i= r� rh �� io r� �� iF �� ic r� r r� � �v �� i@ � � � r� i. �g �� i1 �m �� i7 �p �� i: �y �� iC ���r g @i> �. �u4)r" l �� rX r re )r r4 r �atr r5 r� r� r� r r7 r� �equal�invertr r" � newbyteorderrQ � assertRaisesr r �modf)r re rf �index�orig�valuesr r r �test_inplace_fancy_indexing] s� " "$ " "$ " " $ z%TestUfunc.test_inplace_fancy_indexingc C s� 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 d�|� t||dd d d d�|� t t |� t t ||dd d d 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)r2 r )r" )r r2 r )r� rX )rd F)�keepdimsT)r )r� r" rd rq )r" rd rq )rd rq r ro )r� r" rq )r" rd )rd r" )rq r" rd )r� r" ro )ro )rq ro rd )r� r" rq rd ro )r� r" rd ro ) r r r� rP r r" r r� r� 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 )N)r� r r� )r� �inputr r r � <genexpr>& s zOTestUfunc.test_structured_equal.<locals>.MyA.__array_ufunc__.<locals>.<genexpr>)�getattr)r �ufunc�method�inputsr 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 (@r� r1 zf8,f8,f8)r" TFr ) r r� r4 r� r� �squeezer5 r� r rR )r r| re �raZmra�targetr r r �test_structured_equal! s z TestUfunc.test_structured_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 jg}t �d�}d}x|D ]}tt|||� qxW d S )N�1r ) r r� r r r� r rw � floor_divide� bitwise_and� bitwise_or� bitwise_xor� left_shift� right_shift�fmax�fmin�fmod�hypot� logaddexp� logaddexp2r� r� rF r r �modr5 r r )r Z binary_funcsre rf r r r r � test_NotImplemented_not_returned/ s 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" r3 r� r� r� r0 r r( r2 r r r� )rd r� r1 ) r r4 r� r� r� �copyr r� r )r r; �y_baser> � y_base_copyZr0�r1r r r � test_reduce_noncontig_outputD s & "z&TestUfunc.test_reduce_noncontig_outputc C s t dtjjk� d S )N� )r r[ �inner1d_no_doc�__doc__)r r r r �test_no_doc_stringY s z TestUfunc.test_no_doc_stringN)<r% r& r' r- r/ r= rY rZ r\ r_ rg 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/ r0 r> rp rt r� r� r� r� r r r r r( ' sv S/J 6 .A :r( �__main__)!� __future__r r r �warningsrs �numpyr Znumpy.core.umath_testsr, � umath_testsr[ Z numpy.core.operand_flag_testsZoperand_flag_testsr* �numpy.core.test_rationalr r r Z numpy.testingr r r r r r r r r r r r( r% r r r r �<module> s* 0 A