Submit
Path:
~
/
/
opt
/
alt
/
python37
/
lib64
/
python3.7
/
site-packages
/
numpy
/
core
/
tests
/
__pycache__
/
File Content:
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� � Zsafe)Zcastingx)Zdtypexi )Zextobjx)Zoutxzii->i)Zsigx)Z signaturexF)ZsubokxT)Zwherex)r � TypeError�np�add�int)�self� r �N/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 Zsinr )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 Zcos)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�zipZravelr 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 )r7 �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 r? )r �objr r r � logical_xor� s z5TestUfunc.test_generic_loops.<locals>.foo.logical_xorN)r$ r% r& r@ rB r r r r �foo� s rC Z PyUFunc_O_OZPyUFunc_O_O_methodTZ PyUFunc_OO_OZPyUFunc_OO_O_method)r Zexp�zerosZsingler �double� longdoubleZcsingleZcdouble� clongdouble�ones�objectr �all�abs�range�lenr@ r rB ) r ZfoneZftwoZfone_valZftwo_val�msgr7 rC �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)->())r> z core_sig: parenthesis matchingz (i),)i(->()z core_sig: incomplete signaturez(i),->()z(i1,i2),(J_1)->(_kAB))r �umt�test_signaturer )r rN �retr r r rS s6 zTestUfunc.test_signaturec C s t tjjd� d S )Nz (i),(i)->())r rR �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 �?r. �f8)r! r g �?rO �unsafe)r �castingr zii->i)�i4)rZ rZ rZ )r. )�out)r r[ rY )r r1 r r rD )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 )Nr= � g �?ZbhilqBHILQefdgFDGZFDGgV瞯�<)�rtolZ bhilqBHILQ�float64)r! Zefdggw�Mh�� �ignore)�invalid�e)r` �atolr g �?g )r r2 r! Zastype� itertools�product�complex�float� true_divide�maxZfinfoZ resolutionr r �namer r ZtinyZerrstateZisfiniterH r? )r r\ r] �tgtZtc�dtZaaZbbr7 r: �resr` ZtcoutZdtoutre 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 rH �float32r �sum�sizera )r r\ 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 r0 � � � � � � � i i� r r )r! ���i� g @o@r. g �d@������)r g @) r r Zfloat16rs ra rF r1 r rt rH )r rn �vrm �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 r0 rw rx ry rz r{ r| r} i i� r r y �?)r! r~ i� y @o@ @o@r. y �d@ �d@r r� )r y @ @) r Z complex64Z complex128rG �emptyr1 �real�imagr rt rH )r rn r� rm 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 )Nr- )r r. r~ )�axis)r r1 �reshaper rR rU rt )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||� d S )NZ broadcastrq )r r r )r r r r~ )r� )r> z"extend & broadcast loop dimensions)r r rw )rq r )rq r r0 r )r r r r. )r. r r ) r r1 r� r rR rU rt r r r2 )r rN r\ r] 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 castr- Zshort)r! )r r. r~ )r� )r> ztype cast on one argumentg�������?)r r1 r� r rR rU rt r )r rN r\ r] 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 endianr- z>i4)r! )r r. r~ )r� )r> z little endianz<i4r z>f8z<f8rW ) r r1 r� r rR rU rt r r! Zabsolute�negative)r rN r\ 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 )r> )r� � r r. rq ) r r1 r� r r rR rU rt �T)r rN r7 r\ r] Zmsg2r 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 r. r rq )r r r r )r r. r )r! r~ )r� )r> )r[ z output argument with type cast�int16z(output argument with incontiguous layout)r r. rq ).r )r r1 r� rD rR rU r rt )r rN r\ r] �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 )Nr- )r r. r= rz r_ � r~ )r� �d �| )r r. rq �� �� i, iD )r r1 r� r rR �innerwtrt )r r\ r] �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 operandsrW )r! r~ )r� N)r r2 r rR r� rt )r r\ r] r� r r r �test_innerwt_emptyY s z TestUfunc.test_innerwt_emptyc C s | � tj� | � tj� d S )N)� compare_matrix_multiply_resultsr ZlongrE )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 )r r )r= r= )rR �matrix_multiplyr rH r rD )r ro 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 r. rq )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 )rL )�nrT �baseZpermrO �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� rT r� Zsl)�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)r> zreference check)r r2 ZrandomZrandr! rl r� � transposer� �shaper rR r� rt Znewaxis�swapaxes�strr )r �tpZd1Zd2rN r� Z permute_3Zslice_3�refZp1Zp2r� r� Za1Za2r )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! rq r. r r r r~ )r� )r r1 ri r� r� r� r! rR Zeuclidean_pdist�sqrtrt Ztri�boolr r r )r r\ r[ r] 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 )Nr. TFZtest� )r! c S s g | ] }|pd �qS )Nr )�.0r7 r r r � <listcomp>� s z1TestUfunc.test_object_logical.<locals>.<listcomp>c S s g | ] }|pd �qS )Tr )r� r7 r r r r� � s r� c S s g | ] }|pd �qS )r� r )r� r7 r r r r� � s �blahc S s g | ] }|pd �qS )r� r )r� r7 r r r r� � s c S s g | ] }|od �qS )Nr )r� r7 r r r r� � s c S s g | ] }|od �qS )Tr )r� r7 r r r r� � s c S s g | ] }|od �qS )r� r )r� r7 r r r r� � s c S s g | ] }|od �qS )r� r )r� r7 r r r r� � s c S s g | ] }| �qS r r )r� r7 r r r r� � s )r r2 rI r � logical_or� logical_andZ logical_not�reduce)r r\ 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 ) Nr\ r] r� )r! �abcTFr r r )r r ) r r2 rI r rt rk �min�prod�anyrJ )r r\ 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 )Nrq )r! c S s g | ] }d g�qS )r r )r� rO r r r r� � s zBTestUfunc.test_object_array_accumulate_inplace.<locals>.<listcomp>)r[ c S s g | ]}d g| �qS )r r )r� rO r r r r� � s )r r. r- r= )r rq c S s g | ] }d g�qS )r r )r� rO r r r r� � s r r~ )r[ r� c S s g | ]}d g| �qS )r r )r� rO r r r r� � s )r rH rI rL r � accumulater r2 )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 ) Nrq )r! c S s g | ] }d g�qS )r r )r� rO r r r r� � s z@TestUfunc.test_object_array_reduceat_inplace.<locals>.<listcomp>c S s g | ] }d g�qS )r r )r� rO r r r r� � s )r[ )r rq c S s g | ] }d g�qS )r r )r� rO r r r r� � s r c S s g | ] }d g�qS )r r )r� rO r r r r� � s r~ )r[ r� ) r r� rI rL r r5 r1 r rH )r r� r[ 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! r. r- )r ZmatrixrI r �multiply)r r� Zdesiredr 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 r2 rI r rt r� r� rJ r r rk r� )r 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 r2 r Z AxisErrorrJ r� )r 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 ) Nr. 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 rt r� r� rJ rk r� �powerr� r �typers r? �ndarrayr2 �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 rH �int64ra r r rD rs �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 )Nr0 r r )r[ �wherer rq r- r. � rz r/ TF� ) r r1 rH rD r r r� r� r� )r r\ r] 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 )Nr= g �?r r )r[ r� r )r rH r� ra r r )r r\ r] 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 r2 r� r� r r� ra )r r\ �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 r\ r r r � check_identityless_reductiona sN $ $ $z&TestUfunc.check_identityless_reductionc C s t jddd�}| �|� d S )N)r r. rq �C)�order)r r� r� )r r\ r r r �"test_identityless_reduction_corder� s z,TestUfunc.test_identityless_reduction_corderc C s t jddd�}| �|� d S )N)r r. rq �F)r� )r r� r� )r r\ r r r �"test_identityless_reduction_forder� s z,TestUfunc.test_identityless_reduction_forderc C s$ t jddd��dd�}| �|� d S )N)r rq r. r� )r� r r )r r� r� r� )r 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)r. r/ rq r� )r� r r )r r� r� r� )r 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� Zi1)r! r rW )r. rq r/ )r r� r� r� r� )r r\ 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 r2 �divider� r r r )r r\ ro 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 r. )r r r r= )r r )r rD )�expect�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 �maximumZidentityr� rD r r� )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 r. )r! g�������?g������ @g������@gffffff@)r[ c S s | |7 } d S )Nr )r\ r] r r r � add_inplace s z0TestUfunc.test_safe_casting.<locals>.add_inplacerX )r[ rY rq )r r2 r r r r r r )r r\ 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 Zi8)r! r. rq )r r2 r� r r r r r )r r\ r] 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 )Nrz �l)r! rq rx r. r~ r r r0 � r� � r� rr ry r= )r r1 r� � opflag_testsZ inplace_addr r2 )r r\ r] 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 r. zu8,u8,u8)r! )r rq r- )Z numpy.core.struct_ufunc_test�coreZstruct_ufunc_testr r2 Z add_tripletr )r Z struct_ufuncr\ r] 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 r. rq )r! )r r2 r r r )r r\ r] r� Zexpectedr 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�tuplerM �ndim� IndexErrorr� )r rO )�MyThingr r � __getitem__L s z=TestUfunc.test_custom_array_like.<locals>.MyThing.__getitem__c s � j d7 _ | S )Nr )� rmul_count)r �other)r r r �__rmul__U 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 )rI r ra 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 jdddgt jd`�}| �tt jj|ddgd� t|t jdddgt jd`�� t tt jjt �d�dg� d S )eNr= r r/ r r rq r. r- r0 rw rx r� �� �i r� i, �� �f i� �� i_ i4 r� i� i� �� i� �� r� i� �� ry i� �� r� i� �� � i� �� r� i� �� i� i� �� �� �� i] �j r� �� if �s �� i= r{ r_ �� io r� �� iF �� ic rr r� rz � �v �� i@ � � � r� i. �g �� i1 �m �� i7 �p �� i: �y �� iC ���r g @i> �. Zu4)r! l �� rO r� r\ )r r1 r Zatr r2 r� r� r� r r r� Zequal�invertr r! Z newbyteorderrI Z assertRaisesr r Zmodf)r r\ r] �indexZorig�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)r/ r )r! )r r/ r )r� rO )r[ F)�keepdimsT)r )r� r! r[ r7 )r! r[ r7 )r[ r7 r rc )r� r! r7 )r! r[ )r[ r! )r7 r! r[ )r� r! rc )rc )r7 rc r[ )r� r! r7 r[ rc )r� r! r[ rc ) r r r� rH 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 Zufunc�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 (@rq r. zf8,f8,f8)r! TFr ) r r� r1 r� r� Zsqueezer2 r� r rJ )r r@ r\ ZraZmra�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� rj Z floor_divideZ bitwise_andZ bitwise_orZ bitwise_xorZ left_shiftZ right_shiftZfmaxZfminZfmodZhypotZ logaddexpZ logaddexp2r� r� rB r� r� �modr2 r r )r Z binary_funcsr\ r] 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! r0 rr rw rq r- r r� r/ r r rz )r[ r� r. ) r r1 r� r� r� �copyr r� r )r r7 Zy_baser: Z y_base_copyZr0Zr1r r r � test_reduce_noncontig_outputD s & "z&TestUfunc.test_reduce_noncontig_outputc C s t dtjjk� d S )N� )r rR Zinner1d_no_doc�__doc__)r r r r �test_no_doc_stringY s z TestUfunc.test_no_doc_stringN)<r$ r% r& r+ r, r9 rP rQ rS rV r^ rp 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� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r6 r: rB rE rG rJ r r r r r' ' sv S/J 6 .A :r' �__main__)!Z __future__r r r �warningsrf Znumpyr Znumpy.core.umath_testsr� Z umath_testsrR Z numpy.core.operand_flag_testsZoperand_flag_testsr� Znumpy.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
Submit
FILE
FOLDER
Name
Size
Permission
Action
test_abc.cpython-37.opt-1.pyc
2107 bytes
0644
test_abc.cpython-37.pyc
2107 bytes
0644
test_api.cpython-37.opt-1.pyc
11505 bytes
0644
test_api.cpython-37.pyc
11505 bytes
0644
test_arrayprint.cpython-37.opt-1.pyc
10986 bytes
0644
test_arrayprint.cpython-37.pyc
10986 bytes
0644
test_datetime.cpython-37.opt-1.pyc
49163 bytes
0644
test_datetime.cpython-37.pyc
49163 bytes
0644
test_defchararray.cpython-37.opt-1.pyc
29206 bytes
0644
test_defchararray.cpython-37.pyc
29206 bytes
0644
test_deprecations.cpython-37.opt-1.pyc
20438 bytes
0644
test_deprecations.cpython-37.pyc
20438 bytes
0644
test_dtype.cpython-37.opt-1.pyc
21522 bytes
0644
test_dtype.cpython-37.pyc
21647 bytes
0644
test_einsum.cpython-37.opt-1.pyc
26207 bytes
0644
test_einsum.cpython-37.pyc
26207 bytes
0644
test_errstate.cpython-37.opt-1.pyc
1914 bytes
0644
test_errstate.cpython-37.pyc
1914 bytes
0644
test_extint128.cpython-37.opt-1.pyc
6702 bytes
0644
test_extint128.cpython-37.pyc
6702 bytes
0644
test_function_base.cpython-37.opt-1.pyc
12187 bytes
0644
test_function_base.cpython-37.pyc
12337 bytes
0644
test_getlimits.cpython-37.opt-1.pyc
5695 bytes
0644
test_getlimits.cpython-37.pyc
5695 bytes
0644
test_half.cpython-37.opt-1.pyc
13558 bytes
0644
test_half.cpython-37.pyc
13558 bytes
0644
test_indexerrors.cpython-37.opt-1.pyc
7545 bytes
0644
test_indexerrors.cpython-37.pyc
7545 bytes
0644
test_indexing.cpython-37.opt-1.pyc
41938 bytes
0644
test_indexing.cpython-37.pyc
41938 bytes
0644
test_item_selection.cpython-37.opt-1.pyc
3473 bytes
0644
test_item_selection.cpython-37.pyc
3473 bytes
0644
test_longdouble.cpython-37.opt-1.pyc
7461 bytes
0644
test_longdouble.cpython-37.pyc
7461 bytes
0644
test_machar.cpython-37.opt-1.pyc
1389 bytes
0644
test_machar.cpython-37.pyc
1389 bytes
0644
test_mem_overlap.cpython-37.opt-1.pyc
26795 bytes
0644
test_mem_overlap.cpython-37.pyc
26833 bytes
0644
test_memmap.cpython-37.opt-1.pyc
7512 bytes
0644
test_memmap.cpython-37.pyc
7589 bytes
0644
test_multiarray.cpython-37.opt-1.pyc
227544 bytes
0644
test_multiarray.cpython-37.pyc
227633 bytes
0644
test_nditer.cpython-37.opt-1.pyc
69947 bytes
0644
test_nditer.cpython-37.pyc
69947 bytes
0644
test_numeric.cpython-37.opt-1.pyc
91401 bytes
0644
test_numeric.cpython-37.pyc
91401 bytes
0644
test_numerictypes.cpython-37.opt-1.pyc
13866 bytes
0644
test_numerictypes.cpython-37.pyc
13866 bytes
0644
test_print.cpython-37.opt-1.pyc
7470 bytes
0644
test_print.cpython-37.pyc
7470 bytes
0644
test_records.cpython-37.opt-1.pyc
15572 bytes
0644
test_records.cpython-37.pyc
15572 bytes
0644
test_regression.cpython-37.opt-1.pyc
87286 bytes
0644
test_regression.cpython-37.pyc
87286 bytes
0644
test_scalarinherit.cpython-37.opt-1.pyc
1902 bytes
0644
test_scalarinherit.cpython-37.pyc
1902 bytes
0644
test_scalarmath.cpython-37.opt-1.pyc
21079 bytes
0644
test_scalarmath.cpython-37.pyc
21079 bytes
0644
test_scalarprint.cpython-37.opt-1.pyc
1257 bytes
0644
test_scalarprint.cpython-37.pyc
1257 bytes
0644
test_shape_base.cpython-37.opt-1.pyc
17242 bytes
0644
test_shape_base.cpython-37.pyc
17242 bytes
0644
test_ufunc.cpython-37.opt-1.pyc
45351 bytes
0644
test_ufunc.cpython-37.pyc
45351 bytes
0644
test_umath.cpython-37.opt-1.pyc
94853 bytes
0644
test_umath.cpython-37.pyc
94853 bytes
0644
test_umath_complex.cpython-37.opt-1.pyc
14807 bytes
0644
test_umath_complex.cpython-37.pyc
14807 bytes
0644
test_unicode.cpython-37.opt-1.pyc
13276 bytes
0644
test_unicode.cpython-37.pyc
13276 bytes
0644
N4ST4R_ID | Naxtarrr
