OwlCyberSecurity - MANAGER
Edit File: test_umath.cpython-38.pyc
U �p�]ض � @ s: d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl m Z d dl m m Z d dlmZ d dlZd dlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! dd� Z"G dd � d e#�Z$G d d � d e#�Z%G d d � d e#�Z&G dd� de#�Z'G dd� de#�Z(G dd� de#�Z)dd� Z*dd� Z+G dd� de#�Z,G dd� de#�Z-G d d � d e#�Z.G d d � d e#�Z/G d d!� d!e#�Z0G d"d#� d#e#�Z1G d$d%� d%e$�Z2G d&d'� d'e#�Z3G d(d)� d)e#�Z4G d*d+� d+e#�Z5G d,d-� d-e#�Z6G d.d/� d/e$�Z7G d0d1� d1e#�Z8G d2d3� d3e#�Z9G d4d5� d5e#�Z:d6d7� Z;d8d9� Z<G d:d;� d;e#�Z=d<d=� Z>d>d?� Z?d@dA� Z@dBdC� ZAdDdE� ZBG dFdG� dGe#�ZCG dHdI� dIe#�ZDG dJdK� dKe$�ZEG dLdM� dMe$�ZFG dNdO� dOe$�ZGG dPdQ� dQe$�ZHG dRdS� dSe#�ZIG dTdU� dUe#�ZJG dVdW� dWe#�ZKG dXdY� dYe#�ZLG dZd[� d[e#�ZMG d\d]� d]e#�ZNG d^d_� d_e#�ZOG d`da� dae#�ZPG dbdc� dce#�ZQG ddde� dee#�ZRG dfdg� dge#�ZSG dhdi� die#�ZTG djdk� dke#�ZUG dldm� dme#�ZVG dndo� doe#�ZWG dpdq� dqe#�ZXG drds� dse#�ZYG dtdu� due#�ZZdvdwdxe[fdydz�Z\d{d|� Z]d}d~� Z^dd�� Z_d�d�� Z`ejajbe�cejd�e�ceje�kd�d��ejajfe�g� �hd��d�d��d�d�� ��Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zmejajbe�cejd�e�ceje�kd�d��ejajfe�g� �hd��d�d��d�d�� ��Znd�d�� Zod�d�� Zpd�d�� Zqd�d�� Zrd�d�� Zsd�d�� Ztd�d�� Zud�d�� Zveja�wd�e�xd��e�yd dvg�e�yd dvgd�d�gg�g�d�d�� �ZzdS )�� )�division�absolute_import�print_functionN)�Fraction)� _umath_tests) �assert_� assert_equal� assert_raises�assert_raises_regex�assert_array_equal�assert_almost_equal�assert_array_almost_equal�assert_array_max_ulp�assert_allclose�assert_no_warnings�suppress_warnings�_gen_alignment_data� assert_array_almost_equal_nulpc C s t �� dkpt �� �d�S )z/ True if we are running on a Power PC platform.ZpowerpcZppc)�platformZ processor�machine� startswith� r r �=/usr/lib/python3/dist-packages/numpy/core/tests/test_umath.py� on_powerpc s �r c @ s e Zd Zdd� Zdd� ZdS )�_FilterInvalidsc C s t jdd�| _d S )N�ignore��invalid��npZseterrZolderr��selfr r r �setup s z_FilterInvalids.setupc C s t jf | j� d S �Nr r r r r �teardown s z_FilterInvalids.teardownN)�__name__� __module__� __qualname__r" r$ r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestConstantsc C s t tjdd� d S )Ng-DT�! @�V瞯�<)r �ncu�pir r r r �test_pi# s zTestConstants.test_pic C s t tjdd� d S )NgiW� �@r) )r r* �er r r r �test_e& s zTestConstants.test_ec C s t tjdd� d S )Ng�o��x�?r) )r r* Z euler_gammar r r r �test_euler_gamma) s z TestConstants.test_euler_gammaN)r% r& r' r, r. r/ r r r r r( " s r( c @ s e Zd Zdd� Zdd� ZdS )�TestOutc C s� dD �]�}t �d�}t �d�}t j|d||d�}t||k� t j|d||d�}t||k� t j|d|f|d�}t||k� t �d�}t �d�}t jdt jd�}t j||d |d�\}} t||k� t j|d ||d�\}} t| |k� t j||||d�\}} t||k� t| |k� t j||d f|d�\}} t||k� t j|d |f|d�\}} t| |k� t j|||f|d�\}} t||k� t| |k� tjd d ��F} t� d d t � t j|||d�\}} t||k� t| d j t k� W 5 Q R X t t t j|d|||d� t t t j|d|||d� t t t j|dd ||d� t t t j|d||f|d� t t t j|dd|d� t tt j|dg |d� t tt j|dg |d� t tt j|dg f|d� d|j_t t t j|d||d� t t t j|d||d� t t t j|d|f|d� qd S )N�TF� �?r � ��subok��outr5 �������@��dtypeT��record�always� r F)r �array�empty�addr �int32�frexp�warnings�catch_warnings�filterwarnings�DeprecationWarning�categoryr � ValueError� TypeError�flagsZ writeable) r! r5 �a�o�r�d�o1�o2�r1�r2�wr r r �test_out_subok. sV zTestOut.test_out_subokc C s� G dd� dt j�}dD �]�}|dg�}t j|d|d�}|rJtt||�� ntt|�t jk� t j|dd |d�}|r�tt||�� ntt|�t jk� t j|dd |d�}|r�tt||�� ntt|�t jk� t j|dd|d�}|r�tt||�� ntt|�t jk� |d g�}t �d �}t jd t jd �}t j|||d�\}} |�rRtt| |�� ntt| �t jk� t j||d |d�\}} |�r�tt| |�� ntt| �t jk� t j|d ||d�\}} |�r�tt||�� ntt|�t jk� t j||d f|d�\}} |�rtt| |�� ntt| �t jk� t j|d |f|d�\}} |�rNtt||�� ntt|�t jk� t j d d ��b} t � ddt � t j|||d�\}} |�r�tt| |�� ntt| �t jk� t| d j t k� W 5 Q R X qd S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z.TestOut.test_out_wrap_subok.<locals>.ArrayWrap� c S s t �|��| ��� S r# �r �asarray�view�copy)�cls�arrr r r �__new__e s z6TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__new__c S s |� t| ��S r# �rY �type�r! r\ �contextr r r �__array_wrap__h s z=TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__array_wrap__N�r% r& r' �__array_priority__r] rb r r r r � ArrayWrapb s re r1 r2 r3 r4 r6 r# r8 �� r9 Tr; r= r> r )r �ndarrayrA r � isinstancer_ r@ rB rC rD rE rF rG rH ) r! re r5 rL rN rO rP rQ rR rS rT r r r �test_out_wrap_suboka sb zTestOut.test_out_wrap_subokN)r% r& r' rU rj r r r r r0 - s 3r0 c @ s e Zd Zdd� Zdd� ZdS )�TestComparisonsc C s� t jt �dddg�d gtd�}ttt j||� G dd� dt�}t �|� g�}ttt j||� t jt jgtd�}tt �||�dg� d S )Nrg r3 � r9 c @ s e Zd Zdd� ZdS )zGTestComparisons.test_ignore_object_identity_in_equal.<locals>.FunkyTypec S s t d��d S �NzI won't compare�rJ �r! �otherr r r �__eq__� s zNTestComparisons.test_ignore_object_identity_in_equal.<locals>.FunkyType.__eq__N)r% r& r' rq r r r r � FunkyType� s rr F) r r? �objectr rI ZequalrJ �nanr �r! rL rr r r r �$test_ignore_object_identity_in_equal� s z4TestComparisons.test_ignore_object_identity_in_equalc C s� t jt �dddg�d gtd�}ttt j||� G dd� dt�}t �|� g�}ttt j||� t jt jgtd�}tt �||�dg� d S )Nrg r3 rl r9 c @ s e Zd Zdd� ZdS )zKTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyTypec S s t d��d S rm rn ro r r r �__ne__� s zRTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyType.__ne__N)r% r& r' rw r r r r rr � s rr T) r r? rs r rI � not_equalrJ rt r ru r r r �(test_ignore_object_identity_in_not_equal� s z8TestComparisons.test_ignore_object_identity_in_not_equalN)r% r& r' rv ry r r r r rk � s rk c @ s e Zd Zdd� ZdS )�TestAddc C s< t jddt jfdt jfgd�}d|d< t|d �� d� d S )Nr3 rL �br9 ���r )r �zerosrB �float64r �sum�r! rL r r r �test_reduce_alignment� s z TestAdd.test_reduce_alignmentN)r% r& r' r� r r r r rz � s rz c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestDivisionc C s� t �ddddddddd g �}d d krFt|d d d d ddddddg � n t|d dddddddddg � t|d dddddddddg � t|d dddddddddg � d S )N� rV �Z �d ������i�������i����r2 g�������?皙�����?g�������?rg g��������皙������g�������r| g333333�r ����_ �P )r r? r �r! �xr r r �test_division_int� s � z TestDivision.test_division_intc C sn d}t jdddgt jd�}t|d | ||d� d}t jd d gt jd�}|d | }t|| d d g|d� d S ) Nz%Complex division implementation check� �? �?y �? �?� �? @r9 r3 ��err_msgz)Complex division overflow/underflow check��G0J�V��ys?� )rg )r r? � complex128r �r! �msgr� �yr r r �test_division_complex� s z"TestDivision.test_division_complexc C s� t jddd��� t jdgt jd�}d| }tt �|�d � tt jt j�| }tt �|�d � tt jt j�| }tt �|�d � tt jt j�| }tt �|�d � d| }tt � |�d � W 5 Q R X d S )Nr �r �divide� r9 � �?r ) r �errstater? r� r �isinf�complex�infrt �isnan)r! r� r� r r r �test_zero_division_complex� s z'TestDivision.test_zero_division_complexc C s� d}t jddddgt jd�}t jddddgt jd�}tt �|d |�||d � d }t jd d gt jd�}t �|d |�}t|d dg|d � d S )Nz+Complex floor division implementation checky�������? �?y�������� �?y�������? �?y�������? @r9 r� � �r3 r� z/Complex floor division overflow/underflow checkr� r� r )r r? r� r � floor_divider� r r r �test_floor_division_complex s z(TestDivision.test_floor_division_complexc C s8 t �d�}tt �|d �d� tt �| d �d� d S )NrV rg r )r r} r �signbitr� r r r � test_floor_division_signed_zero s z,TestDivision.test_floor_division_signed_zeroN)r% r& r' r� r� r� r� r� r r r r r� � s r� c C s t �| |�t �| |�fS r# )r r� � remainder�r� r� r r r �floor_divide_and_remainder s r� c C s | t jd krdS dS d S )N�UnsignedIntegerrf �rg r| )r � typecodes)�dtr r r �_signs s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestRemainderc C s t jd t jd }tt jfD ]�}t�||�D ]�\}}t�t|�t|��D ]�\}}d}||j||||f }t j|d |d�} t j|d |d�} || | �\} } t | | | | |d� |dkr�t | | k o�d kn |� qJt | | ko�d kn |� qJq.q d S ) N� AllInteger�Float�*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�G r9 � r� r| r ) r r� r� �divmod� itertools�productr� r% r? r r � r! r� �opZdt1Zdt2Zsg1Zsg2Zfmtr� rL r{ �div�remr r r �test_remainder_basic( s z"TestRemainder.test_remainder_basicc C s( t tdd��}t tdd��}|dg | }|| }t t�||��}t dd� |D ��}tj|td�j\}}tj|td�j\} } t� | dk|dk |dk A @ d | �} t� | dk|dk @ d | �} t tj fD ]`} tj d D ]P} d | j | f } |�| �}|�| �}| ||�\}}t|| | d � t|| | d � q�q�d S ) Ni����r rg � c s s | ]}t |� V qd S r# )r� )�.0�tr r r � <genexpr>@ s z;TestRemainder.test_float_remainder_exact.<locals>.<genexpr>r9 r� � �r� zop: %s, dtype: %sr� )�list�ranger� r� r r? �int�T�float�wherer� r� r� r% �astyper )r! ZnlstZplstZdividendZdivisor�arg�tgtrL r{ ZtgtdivZtgtremr� r� r� ZfaZfbr� r� r r r �test_float_remainder_exact8 s$ " z(TestRemainder.test_float_remainder_exactc C s� t jd }tt jfD ]�}t�||�D ]�\}}t�dd�D ]�\}}d}||j||||f }t j|d d |d�} t j|d |d�} || | �\} } t| | | | |d� |dkr�t | | k o�d kn |� q8t | | ko�d kn |� q8q$qd S ) Nr� r� r� �N g+i�)+p>r9 r� r| r ) r r� r� r� r� r� r% r? r r r� r r r � test_float_remainder_roundoffR s z+TestRemainder.test_float_remainder_roundoffc C sR t jd D ]l}t jd|d�}t �t jd|d�| �}t �||�}t||kd| � t �| | �}t|| kd| � q t� ��}|�td� t jd D ]�}t jd|d�}t jd|d�}t jt j |d�}t jt j |d�} t �||�}tt � |�d||f � t �|| �}tt � |�d||f � t �||�}tt � |�d||f � q�W 5 Q R X d S )Nr� r� r9 r� zdt: %sz&invalid value encountered in remainderzdt: %s, rem: %s) r r� r? � nextafterr� r r �filter�RuntimeWarningr� rt r� ) r! r� r{ rL r� �supZfoneZfzerZfinfZfnanr r r �!test_float_remainder_corner_casesd s( z/TestRemainder.test_float_remainder_corner_casesN)r% r& r' r� r� r� r� r r r r r� & s r� c @ s e Zd Zdd� Zdd� ZdS )�TestCbrtc C s t t�t�d�d �d� d S )N� �rl )r r �cbrt�float32r r r r �test_cbrt_scalar� s zTestCbrt.test_cbrt_scalarc C st t �dddt jt j g�}tt �|d �|� tt �t �t j��� tt �t j�t j� tt �t j �t j � d S )Nr� � @� �rl ) r r? r� r r� r r� rt r r� r r r � test_cbrt� s zTestCbrt.test_cbrtN)r% r& r' r� r� r r r r r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestPowerc C sP t �dddg�}t|d dddg� t|d |� t|d dddg� |�� }|dC }t|dddg� t|d dd d g� t|d dt�d�t�d �g� tt jd dd�D ]F\}}}dd� |D �}t|d ||d� t j||d� t|||d� q�tt j d dd�D ]H\}}}dd� |D �}t|d ||d� t j||d� t|||d� �qd S )Nr� r� � @r rg r3 � @� "@r| r2 gUUUUUU�?rl �unary� �r: r_ Zmax_sizec S s g | ]}t �|��qS r �r* �sqrt�r� �ir r r � <listcomp>� s z.TestPower.test_power_float.<locals>.<listcomp>r� �r7 � c S s g | ]}t �|��qS r r� r� r r r r� � s ) r r? r rZ r r* r� r r� r~ )r! r� r� r7 �inpr� �expr r r �test_power_float� s2 ��zTestPower.test_power_floatc s� t �dddg�}t|d dddg� t|d |� t|d dd d g� t|d d d dg� t|d dddg� t|d dddg� t|d dddg� t|d d d d g� t|d t�d�t�d�t�d�g� d|d d � t|d � � fd!d"�d#D �� d$d%� }tdt j�tdt j�fD ]f}t j|gt jd&�}t j d'd(��< ||d |� ||d || � ||d || | � W 5 Q R X �q d S ))Nr� y @ @y @ @r r� rg r3 y � @y � (@y � 8@rl y &� �y G� "@y @]� F@� y � 8�y �]� ^�y x�� u�r| y�������?������ٿy;�;��?�؉�؉Ϳy� ��Q��?{�G�zĿr� y� ��Q���{�G�zĿy�n_Y�K��&�q-��yǺ���a2U0*������y�I +�������Mb�?y:бb�p��ԗ�|�p�yF�̱��~��&�|�g�r2 � c s g | ] }|� �qS r r r� �Znormr r r� � s z0TestPower.test_power_complex.<locals>.<listcomp>)y ��� ~�@y 0�vA ��Ay � ���A ��]�Ac S s t | j|j� t | j|j� d S r# )r �real�imagr� r r r �assert_complex_equal� s z:TestPower.test_power_complex.<locals>.assert_complex_equalr9 r r ) r r? r r r* r� r� r� �complex_r� )r! r� r� �zr r� r �test_power_complex� s2 �� � z TestPower.test_power_complexc C s� t �dg�}t �dg�}t �tt jt j�g�}dd� }dD ]}|t �||�|� q:|t �|d�|� t jdd��H |t �|d �|� dD ]}|t �|| �|� q�|t �|d �|� W 5 Q R X d S ) Ny y �? c S s6 t �| �t �|� } }t| j|j� t| j|j� d S r# )r rX r r� r� r� r r r r� � s z7TestPower.test_power_zero.<locals>.assert_complex_equal) g ��Q� �?r2 rg � �?r3 rl r� r� gffffff@r r r � �?y �����?)r r? r� rt �powerr� )r! �zero�one�cnanr� �pr r r �test_power_zero� s zTestPower.test_power_zeroc C s� t �dddgt j�}|d }t|d j|jk� t|dddg� tt �||� � t|dddg� |t �dggg� }t|jd� d S ) Nrg r3 rl r� g9b-> @r� � )rg rg rl ) r r? �int16r r: r Zmay_share_memoryr �shape)r! r� �resr r r �test_fast_power� s zTestPower.test_fast_powerc C s. t �ddgd�}t �||�}t|ddg� d S )N� Zi8l ��2�X�0)r r? r� r �r! rL r{ r r r �test_integer_power� s z TestPower.test_integer_powerc C sz t jd }|D ],}t jdd|d�}tt �|d�t �|�� qt jd }|D ]*}t jd|d�}tt �|d�t �|�� qJd S )N�Integerr� rV r9 r r� �r r� �aranger r� � ones_like�r! �dtypesr� r\ r r r �-test_integer_power_with_integer_zero_exponent� s z7TestPower.test_integer_power_with_integer_zero_exponentc C s> t jd }|D ]*}t jd|d�}tt �d|�t �|�� qd S )Nr� rV r9 rg r r r r r �test_integer_power_of_1 s z!TestPower.test_integer_power_of_1c C s@ t jd }|D ],}t jdd|d�}tt �d|�t �|�� qd S )Nr� rg rV r9 r )r r� r r r� � zeros_liker r r r �test_integer_power_of_zero s z$TestPower.test_integer_power_of_zeroc C s� t jd }|D ]�}t jddddg|d�}t jddddg|d�}t jd|d�}t jd|d�}ttt j||� ttt j||� ttt j||� ttt j||� qd S ) Nr r rg r3 rl r9 r� r| )r r� r? r rI r� )r! r r� rL r{ r� Zminusoner r r � test_integer_to_negative_power s z(TestPower.test_integer_to_negative_powerN) r% r& r' r� r� r r r r r r r r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestFloat_powerc C s` d}d}t ||�D ]H\}}d||f }tjd|d�}t�||�}t|jjt�|�jk|� qd S )Nz?bhilBHILefdgFDGZddddddddddddgDDGzdtin: %s, dtout: %srg r9 )�zipr �onesZ float_powerr r: �name)r! Zarg_typeZres_typeZdtinZdtoutr� r� r r r r �test_type_conversion s z$TestFloat_power.test_type_conversionN)r% r& r' r r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestLog2c C sn ddddddddd d d g }d ddd dddddddg }dD ]0}t j||d�}t j||d�}tt �|�|� q8d S �Nrg r3 r� � � � �@ r� � � � r rl r� � r� r rV ��frO �gr9 )r r? r �log2�r! r� r� r� �xf�yfr r r �test_log2_values% s zTestLog2.test_log2_valuesc C s8 t dd�D ](}t�d| �}t|t|�d| d� q d S )Nrg �A r� zat exponent %dr� )r� r r' r r� )r! r� �vr r r �test_log2_ints- s zTestLog2.test_log2_intsc C s� t t�d�d� t t�tj�tj� tt�t�tj��� tjdd���}t� ddt � tt�t�d��� tt�t�tj ��� t t�d�tj � t|d j t k� t|d j t k� t|d j t k� W 5 Q R X d S ) Nr� r� Tr; r= r> r� r rg r3 ) r r r' r� r r� rt rD rE rF r� rH )r! rT r r r �test_log2_special4 s zTestLog2.test_log2_specialN)r% r& r' r+ r. r/ r r r r r $ s r c @ s e Zd Zdd� ZdS )�TestExp2c C sn ddddddddd d d g }d ddd dddddddg }dD ]0}t j||d�}t j||d�}tt �|�|� q8d S r )r r? r �exp2r( r r r �test_exp2_valuesD s zTestExp2.test_exp2_valuesN)r% r& r' r2 r r r r r0 C s r0 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestLogAddExp2c C s� dddddg}dddddg}dddddg}t ddd gdd d g�D ]Z\}}t�tj||d ��}t�tj||d ��}t�tj||d ��}tt�||�||d � q@d S � Nrg r3 rl r� r� r# r% rO r& r r9 ��decimal)r r r' r? r � logaddexp2� r! r� r� r� r� �dec_r) r* Zzfr r r �test_logaddexp2_valuesO s z%TestLogAddExp2.test_logaddexp2_valuesc C sn ddddg}ddddg}ddddg}dD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� q(d S �N�@B i���iC i���r$ r9 )r r? r r7 �r! r� r� r� r� �logxf�logyf�logzfr r r �test_logaddexp2_rangeY s z$TestLogAddExp2.test_logaddexp2_rangec C s� t j}|| || |d| dg}||| | d|d| g}|||| ||ddg}t jdd��L dD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� qbW 5 Q R X d S �Nrg �raiser r$ r9 )r r� r� r? r r7 � r! r� r� r� r� r� r> r? r@ r r r �test_infc s zTestLogAddExp2.test_infc C s� t t�t�tjtj��� t t�t�tjtj��� t t�t�tjd��� t t�t�dtj��� t t�t�tjtj��� d S �Nr )r r r� r7 rt r� r r r r �test_nano s zTestLogAddExp2.test_nanN)r% r& r' r: rA rE rG r r r r r3 M s r3 c @ s e Zd Zdd� ZdS )�TestLogc C sv ddddddddd d d g }d ddd dddddddg }dD ]8}d}t j||d�}t j||d�| }tt �|�|� q8d S �Nrg r3 r� r r r r r� r r! r" r rl r� r# r� r rV r$ g�9��B.�?r9 )r r? r �log�r! r� r� r� Zlog2_r) r* r r r �test_log_valuesx s zTestLog.test_log_valuesN)r% r& r' rL r r r r rH w s rH c @ s e Zd Zdd� ZdS )�TestExpc C sv ddddddddd d d g }d ddd dddddddg }dD ]8}d}t j||d�}t j||d�| }tt �|�|� q8d S rI )r r? r r� rK r r r �test_exp_values� s zTestExp.test_exp_valuesN)r% r& r' rN r r r r rM � s rM c @ s e Zd Zdd� Zdd� ZdS )�TestSpecialFloatsc C s� t jt jt jdg}t jt j t jt j g}dD ]0}t j||d�}t j||d�}tt �|�|� q.t jdd��. ttt jt � d�� ttt jt � d�� W 5 Q R X d S )Nr� r$ r9 rC �Zover� Y@g =�`�X�C) r rt r� r? r r� r� r �FloatingPointErrorr� r( r r r rN � s z!TestSpecialFloats.test_exp_valuesc C s� t jdd��v t jt jt jt jt j t jg}t jt j t jt j ddg}dD ]0}t j||d�}t j||d�}tt �|�|� qLW 5 Q R X t jdd�� ttt jt � d�� W 5 Q R X t jdd ��2 ttt jt � t j �� ttt jt � d�� W 5 Q R X d S ) Nr )�allr� r� r$ r9 rC �r� r ) r r� rt r� r? r rJ r rR r� r( r r r rL � s z!TestSpecialFloats.test_log_valuesN)r% r& r' rN rL r r r r rO � s rO c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestExpLogFloat32c C sR t j�d� t �t jjdddd��}t �|�}tt �|�t �t �|��dd� d S )N�* r� �fffffV@r< �ZlowZhigh�sizeg������@�Zmaxulp)r �random�seedr� �uniformr~ r r� �r! �x_f32Zx_f64r r r �test_exp_float32� s z"TestExpLogFloat32.test_exp_float32c C sR t j�d� t �t jjdddd��}t �|�}tt �|�t �t �|��dd� d S )NrV r� �� r< rX g333333@rZ )r r[ r\ r� r] r~ r rJ r^ r r r �test_log_float32� s z"TestExpLogFloat32.test_log_float32c C s� t j�d� t jjdddd�}t jjdddd�}|D ]�}t �t jjdd|d��}t �|�}t �|�}|D ]T}tt �|d d |� �|d d |� d d � tt �|d d |� �|d d |� d d � qhq4d S ) NrV r� r� rX rg i� �{�G�z�?rW r3 )Znulp) r r[ r\ Zrandintr� r] r� rJ r )r! �stridesZsizesZiir_ Zexp_trueZlog_trueZjjr r r � test_strided_exp_log_float32� s (z.TestExpLogFloat32.test_strided_exp_log_float32N)r% r& r' r` rb re r r r r rU � s rU c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestLogAddExpc C s� dddddg}dddddg}dddddg}t ddd gdd d g�D ]Z\}}t�tj||d ��}t�tj||d ��}t�tj||d ��}tt�||�||d � q@d S r4 )r r rJ r? r � logaddexpr8 r r r �test_logaddexp_values� s z#TestLogAddExp.test_logaddexp_valuesc C sn ddddg}ddddg}ddddg}dD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� q(d S r; )r r? r rg r= r r r �test_logaddexp_range� s z"TestLogAddExp.test_logaddexp_rangec C s� t j}|| || |d| dg}||| | d|d| g}|||| ||ddg}t jdd��L dD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� qbW 5 Q R X d S rB )r r� r� r? r rg rD r r r rE � s zTestLogAddExp.test_infc C s� t t�t�tjtj��� t t�t�tjtj��� t t�t�tjd��� t t�t�dtj��� t t�t�tjtj��� d S rF )r r r� rg rt r� r r r r rG � s zTestLogAddExp.test_nanc C s, t tjjtj � t tj�g �tj � d S r# )r r rg �identityr� �reducer r r r � test_reduce� s zTestLogAddExp.test_reduceN)r% r& r' rh ri rE rG rl r r r r rf � s rf c @ s e Zd Zdd� Zdd� ZdS )� TestLog1pc C s0 t t�d�t�d�� t t�d�t�d�� d S )N皙�����?g333333�?���ư>g zo �?)r r* �log1prJ r r r r � test_log1p� s zTestLog1p.test_log1pc C s� t jddd��j tt�t j�t j� tt�t j�t j� tt�d�t j � tt�d�t j� tt�t j �t j� W 5 Q R X d S )Nr r� r� g �)r r� r r* rp rt r� r r r r � test_special� s zTestLog1p.test_specialN)r% r& r' rq rr r r r r rm � s rm c @ s e Zd Zdd� Zdd� ZdS )� TestExpm1c C s8 t t�d�t�d�d � t t�d�t�d�d � d S )Nrn rg ro )r r* �expm1r� r r r r � test_expm1� s zTestExpm1.test_expm1c C s` t t�tj�tj� t t�d�d� t t�d�d� t t�tj�tj� t t�tj �d� d S )Nr� r� r� )r r* rt r r� r r r r rr s zTestExpm1.test_specialN)r% r& r' ru rr r r r r rs � s rs c @ s e Zd Zdd� Zdd� ZdS )� TestHypotc C s. t t�dd�t�d�� t t�dd�d� d S �Nrg r3 r )r r* �hypotr� r r r r � test_simple s zTestHypot.test_simplec C s\ t tj�ddg�d� t tj�dddg�d� t tj�dddg�d� ttj�g �d � d S ) Nr� r� � @r r� g (@� 4@g 9@r� )r r* rx rk r r r r r rl s zTestHypot.test_reduceN)r% r& r' ry rl r r r r rv s rv c C sF t jdd��0 tt �t�| |��d| |t�| |�f � W 5 Q R X d S )Nr r z hypot(%s, %s) is %s, not nan)r r� r r� r* rx r� r r r �assert_hypot_isnan s �r| c C sF t jdd��0 tt �t�| |��d| |t�| |�f � W 5 Q R X d S )Nr r z hypot(%s, %s) is %s, not inf)r r� r r� r* rx r� r r r �assert_hypot_isinf s �r} c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestHypotSpecialValuesc C s t tjtj� t tjd� d S �Nrg )r| r rt r r r r �test_nan_outputs# s z'TestHypotSpecialValues.test_nan_outputsc C sR t tjtj� t tjtj� t tjd� t dtj� t tjtj� t tjd� d S )Nr g 7@)r} r rt r� r r r r �test_nan_outputs2' s z(TestHypotSpecialValues.test_nan_outputs2c C s t tjtjd� d S rF )r r* rx r r� r r r r � test_no_fpe/ s z"TestHypotSpecialValues.test_no_fpeN)r% r& r' r� r� r� r r r r r~ " s r~ c C s. t t�t�| |��d| |t�| |�f � d S )Nz arctan(%s, %s) is %s, not nan)r r r� r* �arctan2r� r r r �assert_arctan2_isnan3 s r� c C s> t t�t�| |��o"t�| |�dkd| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not +inf�r r r� r* r� r� r r r �assert_arctan2_ispinf7 s r� c C s> t t�t�| |��o"t�| |�dk d| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not -infr� r� r r r �assert_arctan2_isninf; s r� c C s@ t t�| |�dko$t�t�| |�� d| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not +0�r r* r� r r� r� r r r �assert_arctan2_ispzero? s r� c C s> t t�| |�dko"t�t�| |��d| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not -0r� r� r r r �assert_arctan2_isnzeroC s r� c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) �TestArctan2SpecialValuesc C sL t t�dd�dtj � t t�dd�dtj � t t�dd�dtj � d S )Nrg � �?r| � п� �?)r r* r� r r+ r r r r � test_one_oneH s z%TestArctan2SpecialValues.test_one_onec C s6 t t�tjtj�tj� t t�tjtj�tj � d S r# )r r* r� r �PZERO�NZEROr+ r r r r �test_zero_nzeroN s z(TestArctan2SpecialValues.test_zero_nzeroc C s t tjtj� ttjtj� d S r# �r� r r� r� r� r r r r �test_zero_pzeroS s z(TestArctan2SpecialValues.test_zero_pzeroc C s2 t t�tjd�tj� t t�tjd�tj � d S )Nr| �r r* r� r r� r+ r� r r r r �test_zero_negativeX s z+TestArctan2SpecialValues.test_zero_negativec C s t tjd� ttjd� d S r r� r r r r �test_zero_positive] s z+TestArctan2SpecialValues.test_zero_positivec C s8 t t�dtj�dtj � t t�dtj�dtj � d S )Nrg r2 r� r r r r �test_positive_zerob s z+TestArctan2SpecialValues.test_positive_zeroc C s8 t t�dtj�dtj � t t�dtj�dtj � d S )Nr| � �r� r r r r �test_negative_zerog s z+TestArctan2SpecialValues.test_negative_zeroc C s2 t t�dtj�tj� t t�dtj�tj � d S �Nrg r| )r r* r� r ZNINFr+ r r r r � test_any_ninfl s z&TestArctan2SpecialValues.test_any_ninfc C s t dtj� tdtj� d S r� )r� r r� r� r r r r � test_any_pinfq s z&TestArctan2SpecialValues.test_any_pinfc C s: t t�tjd�dtj � t t�tj d�dtj � d S )Nrg r2 r� �r r* r� r r� r+ r r r r � test_inf_anyv s z%TestArctan2SpecialValues.test_inf_anyc C sB t t�tjtj �dtj � t t�tj tj �dtj � d S )Nr� g �r� r r r r � test_inf_ninf{ s z&TestArctan2SpecialValues.test_inf_ninfc C s> t t�tjtj�dtj � t t�tj tj�dtj � d S )Nr� r� r� r r r r � test_inf_pinf� s z&TestArctan2SpecialValues.test_inf_pinfc C s. t tjtj� t tjtj� t tjtj� d S r# )r� r rt r� r r r r � test_nan_any� s z%TestArctan2SpecialValues.test_nan_anyN)r% r& r' r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� G s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestLdexpc C sp t t�t�dtj�t�d|��d� t t�t�dtj�t�d|��d� t t�t�dtj�t�d|��d� d S )Nr� rl � 0@)r r* �ldexpr r? r� r~ � longdouble)r! �tpr r r � _check_ldexp� s �� �� ��zTestLdexp._check_ldexpc C sN t t�dd�d� | �tj� | �tj� | �tj� | �d� | �d� d S )Nr� rl r� r� �l)r r* r� r� r �int8r rB r r r r � test_ldexp� s zTestLdexp.test_ldexpc C sf t jdd��P t �t �d��j}t �t �d��j}tt�d|�t j � tt�d|�d� W 5 Q R X d S )Nr rP r� r� r ) r r� �iinfor: �max�minr r* r� r� )r! ZimaxZiminr r r �test_ldexp_overflow� s z TestLdexp.test_ldexp_overflowN)r% r& r' r� r� r� r r r r r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestMaximumc C s� t jd }t jd }t �d�}|d d d� }t jj}|D ]4}|�|�}|�|�}t||�d� t||�d� q8|D ]t}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�t j� t||�t j� qrd S )N�AllFloatr� r� r| rV r3 )r r� r �maximumrk r� r rt � r! ZdfltZdintZseq1Zseq2�funcr� Ztmp1Ztmp2r r r rl � s& zTestMaximum.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S �Nrg � @y �? @)r r r� rk r r r r �test_reduce_complex� s z TestMaximum.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �|||g�}tt �||�|� d S rF )r rt r? r r� �r! rt �arg1�arg2r7 r r r �test_float_nans� s zTestMaximum.test_float_nansc C s^ t d�D ]P}t�td�t�}d}t�td�t�}tt�||�dk� tt�||�dk� qd S �Nrg rt r� )r� r r? r� rs r r� �r! r� r� r� r� r r r �test_object_nans� s z TestMaximum.test_object_nansc C sz t j}t|d�td|�t||�fD ]R}t jd||gtd�}t j|d|gtd�}t j|||gtd�}tt �||�|� q"d S �Nr r9 )r rt r� r? r r� �r! rt r� r� r� r7 r r r �test_complex_nans� s z TestMaximum.test_complex_nansc C s, t jdtd�}|d }tt �||�|� d S �Nr� r9 rg )r r rs r r� �r! r� r� r r r �test_object_array� s z TestMaximum.test_object_arrayN� r% r& r' rl r� r� r� r� r� r r r r r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestMinimumc C s� t jd }t jd }t �d�}|d d d� }t jj}|D ]4}|�|�}|�|�}t||�d� t||�d� q8|D ]t}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�t j� t||�t j� qrd S )Nr� r� r� r| r r3 )r r� r �minimumrk r� r rt r� r r r rl � s& zTestMinimum.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S r� )r r r� rk r r r r r� � s z TestMinimum.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �|||g�}tt �||�|� d S rF )r rt r? r r� r� r r r r� � s zTestMinimum.test_float_nansc C s^ t d�D ]P}t�td�t�}d}t�td�t�}tt�||�dk� tt�||�dk� qd S r� )r� r r? r� rs r r� r� r r r r� s z TestMinimum.test_object_nansc C sz t j}t|d�td|�t||�fD ]R}t jd||gtd�}t j|d|gtd�}t j|||gtd�}tt �||�|� q"d S r� )r rt r� r? r r� r� r r r r� s z TestMinimum.test_complex_nansc C s, t jdtd�}|d }tt �||�|� d S r� )r r rs r r� r� r r r r� s z TestMinimum.test_object_arrayNr� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFmaxc C s� t jd }t jd }t �d�}|d d d� }t jj}|D ]4}|�|�}|�|�}t||�d� t||�d� q8|D ]p}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�d� t||�d� qrd S )Nr� r� r� r| rV r3 r )r r� r �fmaxrk r� r rt r� r r r rl s& zTestFmax.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S r� )r r r� rk r r r r r� 2 s z TestFmax.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �dd|g�}tt �||�|� d S rF )r rt r? r r� r� r r r r� 6 s zTestFmax.test_float_nansc C sz t j}t|d�td|�t||�fD ]R}t jd||gtd�}t j|d|gtd�}t jdd|gtd�}tt �||�|� q"d S r� )r rt r� r? r r� r� r r r r� = s zTestFmax.test_complex_nansN�r% r& r' rl r� r� r� r r r r r� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFminc C s� t jd }t jd }t �d�}|d d d� }t jj}|D ]4}|�|�}|�|�}t||�d� t||�d� q8|D ]p}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�d� t||�d� qrd S )Nr� r� r� r| r r3 rg )r r� r �fminrk r� r rt r� r r r rl G s& zTestFmin.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S r� )r r r� rk r r r r r� \ s z TestFmin.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �dd|g�}tt �||�|� d S rF )r rt r? r r� r� r r r r� ` s zTestFmin.test_float_nansc C sz t j}t|d�td|�t||�fD ]R}t jd||gtd�}t j|d|gtd�}t jdd|gtd�}tt �||�|� q"d S r� )r rt r� r? r r� r� r r r r� g s zTestFmin.test_complex_nansNr� r r r r r� F s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestBoolc C s@ t jdt jd�}ttt j|� ttt j|� ttt j||� d S )Nrg r9 )r r Zbool_r rJ �negative�positive�subtractr� r r r �test_exceptionsq s zTestBool.test_exceptionsc C s� ddddg}ddddg}t jd t jd d }tt j|�D ]�}t j||d�}t j||d�}d d d d g}t jt jfD ]}t|||��t �|� qtd d d d g}t j t j fD ]}t|||��t �|� q�d d d d g}t j t j fD ]}t|||��t �|� q�q<d S ) Nr rl r3 r� r� r� �?r9 FT)r r� �mapr: rX � logical_orr� r r� �bool� logical_andr� � logical_xorrx ) r! Zinput1Zinput2r� r: r� r� r7 r� r r r �test_truth_table_logicalw s& �� z!TestBool.test_truth_table_logicalc C sv ddddg}ddddg}ddddg}t t�||�|� ddddg}t t�||�|� ddddg}t t�||�|� d S )NFT)r r � bitwise_or� bitwise_and� bitwise_xor)r! r� r� r7 r r r �test_truth_table_bitwise� s z!TestBool.test_truth_table_bitwisec C s� t �ddddgt�}t �ddddgt�}t �ddddgt�}t �g t�}||||g}|D ]}tt j�|�t|�� qX|D ]}tt j�|�t|�� qx|D ]"}tt j �|�|� � d dk� q�d S )Nr rg r3 ) r r? r� r r� rk rS r� �anyr� r )r! ZnoneZsomeZeveryr@ Zarrsr\ r r r rl � s zTestBool.test_reduceN)r% r& r' r� r� r� rl r r r r r� p s r� c @ s: e Zd Zdd� dD �Zdd� Zdd� Zdd � Zd d � Zd S ) �TestBitwiseUFuncsc C s g | ]}t �|��qS r )r r: )r� �cr r r r� � s z TestBitwiseUFuncs.<listcomp>z ?bBhHiIlLqQOc C sl | j D �]^}tjdg|d�}tjdg|d�}d|j }tt�|�||d� tt�|�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� tt�||�||d� qd S )Nr r9 r| � dt = '%s'r� ) � bitwise_typesr r? �charr � bitwise_notr� r� r� �r! r� r} r r� r r r � test_values� s$ z TestBitwiseUFuncs.test_valuesc C s� | j D ]�}tjdg|d�}tjdg|d�}d|j }tt�|�j|k|� tt�||�j|k|� tt�||�j|k|� tt� ||�j|k|� qd S )Nr r9 r| r� ) r� r r? r� r r� r: r� r� r� r� r r r � test_types� s z TestBitwiseUFuncs.test_typesc C s: t tjjdkd� t tjjdkd� t tjjdkd� d S )Nr r� r� r| r� )r r r� rj r� r� r r r r � test_identity� s z TestBitwiseUFuncs.test_identityc C sn t jt jt jf}| jD ]b}t jdg|d�}t jdg|d�}|D ]8}d||f }t|�|�||d� t|�|�||d� q>q| jd d� D ]b}t jg |d�}|D ]J}d||f }t j|j|d�}|�|�} t| ||d� t | j |j k|� q�q�|D ]:}d|f }t jg t d�}|j}|�|�} t| ||d� q�|D ]8}d|f }t jdgt d�} t t |�| ��t k|� �q0d S )Nr r9 r| zdt: '%s', f: '%s'r� zdt: '%s'T)r r� r� r� r� r? r rk rj r r: rs r_ r� ) r! Z binary_funcsr� r} r r% r� r@ r� r Zbtyper r r �test_reduction� s4 z TestBitwiseUFuncs.test_reductionN)r% r& r' r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestIntc C sd t jdt jd�}t jdtd�}|�� }d|d d d�< |d d d� }tt j||d�d� t||� d S )NrV r9 � Fr3 r� )r r r r� rZ r Z logical_not)r! r� rM r� �osr r r �test_logical_not s zTestInt.test_logical_notN)r% r& r' r� r r r r r� s r� c @ s e Zd Zdd� ZdS )�TestFloatingPointc C s t tjd� d S r )r r* ZFLOATING_POINT_SUPPORTr r r r �test_floating_point s z%TestFloatingPoint.test_floating_pointN)r% r& r' r� r r r r r� s r� c @ s e Zd Zdd� ZdS )� TestDegreesc C s, t t�tj�d� t t�dtj �d� d S )N� �f@r� � �V�)r r* Zdegreesr r+ r r r r � test_degrees s zTestDegrees.test_degreesN)r% r& r' r� r r r r r� s r� c @ s e Zd Zdd� ZdS )� TestRadiansc C s, t t�d�tj� t t�d�dtj � d S )Nr� r� r� )r r* Zradiansr r+ r r r r � test_radians s zTestRadians.test_radiansN)r% r& r' r� r r r r r� s r� c @ s e Zd Zdd� ZdS )� TestHeavsidec C s� t �ddddgdt jt jt j gg�}t �ddddgdt jddgg�}|�� }d|d < t�|d�}t||� t�|d�}t||� |�t j �}t�|t � d��}t||�t j �� t�|t � d��}t||�t j �� d S ) Ng >�r� r� rn g @r2 r� rg �r r3 ) r r? rt r� rZ r* Z heavisider r� r� )r! r� Z expectedhalfZ expected1�hr r r �test_heaviside s & zTestHeavside.test_heavisideN)r% r& r' r� r r r r r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestSignc C s� t �t jt j t jdddg�}t �|j�}t �ddt jdddg�}t jdd��: t�|�}t ||� t�||�}t ||� t ||� W 5 Q R X d S )Nr� r� r� r� r� r r ) r r? r� rt r} r r� r* �signr )r! rL r7 r� r r r r � test_sign6 s zTestSign.test_signc C s8 t �dddg�}t �|�t��}t �|�}t||� d S )Nr� r r� )r r? r� r� rs r )r! �foorL r{ r r r �test_sign_dtype_objectB s z TestSign.test_sign_dtype_objectc C s dd� }t t|� d S )Nc S s" t �t jg�} t �| �t��}d S r# )r r? rt r� r� rs )r� rL r r r rG M s z5TestSign.test_sign_dtype_nan_object.<locals>.test_nan)r rJ )r! rG r r r �test_sign_dtype_nan_objectK s z#TestSign.test_sign_dtype_nan_objectN)r% r& r' r� r� r� r r r r r� 5 s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestMinMaxc s� t jdft jdffD ]�\}}t|d|d�D ]�\}� �t� j�D ]�}t j� j|d�� d d �< t j� |< � �fdd�}t� �>}|� t d� t t � � � � �|d � t t � � �� �|d � W 5 Q R X d � |< t� � � d �d � d � |< t� �� d �d � q>q*qd S ) Nr r� r� r� r9 c s d� �f S )Nz%r %sr r �r� r� r r �<lambda>^ � z0TestMinMax.test_minmax_blocked.<locals>.<lambda>z#invalid value encountered in reduce)r� g _�Br� g _��)r r� r~ r r� rY r rt r r� r� r r� r� r� r )r! r� �szr7 r� Zemsgr� r r r �test_minmax_blockedU s$ � � z TestMinMax.test_minmax_blockedc C sH t jdt jd�dd� �t j�}t|�� |d � t|�� |d � d S )N� r9 r� ���r )r r} r� rY r~ r r� r� �r! rO r r r �test_lower_alignj s zTestMinMax.test_lower_alignc C sX dD ]N}t jt jt jfD ]8}t �t jt jg| |d��D ]}tt �|�t j� q8qqd S )N)r3 r� r r r r9 ) r r� Zfloat16� complex64Zdiagflatr? rt r r� )r! �nr� rN r r r �test_reduce_reorderq s z TestMinMax.test_reduce_reorderc C s t �t jd�}t|t j� d S r )r r� rt r r� r r r �test_minimize_no_warnsz s z!TestMinMax.test_minimize_no_warnsN)r% r& r' r r r r r r r r r� T s r� c @ s e Zd Zdd� Zdd� ZdS )�TestAbsoluteNegativec C st t jdft jdffD �]X\}}t|d|d�D �]>\}}}dd� |D �}t j||d� t|||d� t|d k�� � d d� |D �}t j||d� t|||d� t j t j t j fD ]�}t |j �D ]�}t j |j |d �} | |d d �< |||< |t j kr�| n|| |< tt �|�| |d� t j||d� t|| |d� t| d | |d� d | } t j||d� t|| |d� q�q�q,qd S ) Nr� r� r� r� c S s g | ]}t �|��qS r )r* �absoluter� r r r r� � s z=TestAbsoluteNegative.test_abs_neg_blocked.<locals>.<listcomp>r� r� r c S s g | ] }d | �qS )r| r r� r r r r� � s r9 r| )r r� r~ r r r r rS r� rt r� r� rY r r �abs) r! r� r r7 r� r� r� r- r� rO r r r �test_abs_neg_blocked� s0 �z)TestAbsoluteNegative.test_abs_neg_blockedc C s� t jdt jd�dd� �t j�}tt �|�|� tt �|�| � t j||d� t jt �|�|d� t j||d� t jt �|�|d� d S )Nr r9 r� r r� ) r r} r� rY r~ r r r� r r r r r r � s z%TestAbsoluteNegative.test_lower_alignN)r% r& r' r r r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� TestPositivec C sD t tttg}|D ].}tjd|d�}t�|�}t||t|�d� qd S )Nr� r9 r� ) r� r� r� rs r r r� r �str)r! Z valid_dtypesr: r� �resultr r r � test_valid� s zTestPositive.test_validc C s� t t�� t�d� W 5 Q R X t t�� t�t�d�� W 5 Q R X t t�� t�tjdgtd�� W 5 Q R X t t�� t�tjdgtd�� W 5 Q R X d S )NTz 2000-01-01r� r9 �bar)r rJ r r� Z datetime64r? r rs r r r r � test_invalid� s zTestPositive.test_invalidN)r% r& r' r r r r r r r � s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0�TestSpecialMethodsc s� G � fdd�dt �� � � }t�||�}t|jt�d�� |j\}}}t|tjk� tt |�d� t|d |� t|d |� t|d� d S )Nc s e Zd Zdd� Z� fdd�ZdS )z/TestSpecialMethods.test_wrap.<locals>.with_wrapc S s t �d�S r �r r} r r r r � __array__� s z9TestSpecialMethods.test_wrap.<locals>.with_wrap.__array__c s � � }||_ ||_|S r# )r\ ra �r! r\ ra rN �� with_wrapr r rb � s z>TestSpecialMethods.test_wrap.<locals>.with_wrap.__array_wrap__N�r% r& r' r rb r r r r r � s r rg r3 r ) rs r* r� r r\ r r} ra r �len)r! rL r� r� �argsr� r r r � test_wrap� s z TestSpecialMethods.test_wrapc s� G dd� dt j�� � fdd�}|dd� dd� � |dd� d d� � |d d� d d� � |d d� d d� � |dd� dd� � |dd� dd� � |dd� dd� � d S )Nc @ s@ e Zd ZdZdZdd� Zdd� Zdd� Zedd � �Z d d � Z dS ) zKTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrapNc S s t �d��| �S )Nr )r r@ rY �r[ r r r r] � s zSTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__new__c S s |d | _ |S r )� _wrap_args�r! �objra r r r rb � s zZTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__array_wrap__c S s |d | _ |S r )� _prepare_argsr# r r r �__array_prepare__� s z]TestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__array_prepare__c S s | j | jfS r# )r% r" r r r r r � s zPTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.argsc S s dS )NrL r r r r r �__repr__� s zTTestSpecialMethods.test_wrap_and_prepare_out.<locals>.StoreArrayPrepareWrap.__repr__) r% r& r' r" r% r] rb r& �propertyr r' r r r r �StoreArrayPrepareWrap� s r) c s� � � }| |� |j \}}||�}zt||� t||� W nJ tk r� } z,td�dd�|�d�|�d�|�g���W 5 d }~X Y nX d S )N� z"Bad arguments passed in ufunc callz expected: {}z __array_prepare__ got: {}z __array_wrap__ got: {})r r �AssertionError�join�format)Zf_callZ f_expectedrL r rT Zexpectedr- �r) r r �do_test� s �z=TestSpecialMethods.test_wrap_and_prepare_out.<locals>.do_testc S s t �| d�S rF �r rA �rL r r r r � r z>TestSpecialMethods.test_wrap_and_prepare_out.<locals>.<lambda>c S s | dfS rF r r1 r r r r � r c S s t �| dd �S rF r0 r1 r r r r � r c S s | dfS rF r r1 r r r r � r c S s t j| dd d�S �Nr r� r0 r1 r r r r � r c S s | dfS rF r r1 r r r r � r c S s t j| ddd�S )Nr r# r� r0 r1 r r r r � r c S s | dfS rF r r1 r r r r � r c S s t �dd| �S rF r0 r1 r r r r r c S s dd| fS rF r r1 r r r r r c S s t jdd| d�S r2 r0 r1 r r r r r c S s dd| fS rF r r1 r r r r r c S s t jdd| fd�S r2 r0 r1 r r r r r c S s dd| fS rF r r1 r r r r r )r rh )r! r/ r r. r �test_wrap_and_prepare_out� s z,TestSpecialMethods.test_wrap_and_prepare_outc C sF G dd� dt j�}|� }t�|d�}tt||�� t|t �d�� d S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z=TestSpecialMethods.test_wrap_with_iterable.<locals>.with_wraprV c S s t �d��| ��� S r rW r! r r r r] s zETestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__new__c S s |� t| ��S r# r^ r` r r r rb s zLTestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__array_wrap__Nrc r r r r r s r )rg r3 rl )r rh r* �multiplyr ri r r? �r! r rL r� r r r �test_wrap_with_iterable s z*TestSpecialMethods.test_wrap_with_iterablec C sH G dd� dt j�}|� }t �d�| }tt||�� t|t �d�� d S )Nc @ s e Zd ZdZdd� ZdS )z7TestSpecialMethods.test_priority_with_scalar.<locals>.ArV c S s t �dd��| ��� S )Nr� r~ rW r! r r r r] s z?TestSpecialMethods.test_priority_with_scalar.<locals>.A.__new__N)r% r& r' rd r] r r r r �A s r7 rg )r rh r~ r ri r r? )r! r7 rL r� r r r �test_priority_with_scalar s z,TestSpecialMethods.test_priority_with_scalarc s< G � fdd�dt �� � � }t�||�}t|jt�d�� d S )Nc s e Zd Zdd� Z� fdd�ZdS )z3TestSpecialMethods.test_old_wrap.<locals>.with_wrapc S s t �d�S r r r r r r r ( s z=TestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array__c s � � }||_ |S r# )r\ )r! r\ rN r r r rb + s zBTestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array_wrap__Nr r r r r r ' s r rg )rs r* r� r r\ r r} �r! rL r� r r r � test_old_wrap% s z TestSpecialMethods.test_old_wrapc C s� G dd� dt �}G dd� d|�}G dd� d|�}t�d�}|� }|� }|� }tj}tt|||��tjk� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� tt|||��|k� ttt�|�|k�� ttt�|�|k�� ttt�|�|k�� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z+TestSpecialMethods.test_priority.<locals>.Ac S s t �d�S r r r r r r r 7 s z5TestSpecialMethods.test_priority.<locals>.A.__array__c S s t | �� }||_||_|S r# )r_ r\ ra r r r r rb : s z:TestSpecialMethods.test_priority.<locals>.A.__array_wrap__Nr r r r r r7 6 s r7 c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Br{ N�r% r& r' rd r r r r �B@ s r< c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Cg D@Nr; r r r r �CC s r= rg ) rs r r} r* r� r r_ rh r� ) r! r7 r< r= r� rL r{ r� r% r r r � test_priority4 s2 z TestSpecialMethods.test_priorityc C s: G dd� dt �}|� }tttj||� tttjj|� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z/TestSpecialMethods.test_failing_wrap.<locals>.Ac S s t �d�S �Nr3 r r r r r r b s z9TestSpecialMethods.test_failing_wrap.<locals>.A.__array__c S s t �d S r# �� RuntimeErrorr` r r r rb e s z>TestSpecialMethods.test_failing_wrap.<locals>.A.__array_wrap__Nr r r r r r7 a s r7 )rs r rA r* r� rk �r! r7 rL r r r �test_failing_wrap_ s z$TestSpecialMethods.test_failing_wrapc sx t �dg�� G � fdd�dt j�}G dd� dt j�}t �d��|�}t �d��|�}td�D ]}tttj d||� q\d S )Nr� c s e Zd Z� fdd�ZdS )z4TestSpecialMethods.test_failing_out_wrap.<locals>.Okc s � S r# r �r! r$ �Z singletonr r rb q s zCTestSpecialMethods.test_failing_out_wrap.<locals>.Ok.__array_wrap__N�r% r& r' rb r rE r r �Okp s rG c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_failing_out_wrap.<locals>.Badc S s t �d S r# r@ rD r r r rb u s zDTestSpecialMethods.test_failing_out_wrap.<locals>.Bad.__array_wrap__NrF r r r r �Badt s rH rg rV ) r r? rh r@ rY r� r rA r* rC )r! rG rH �okZbadr� r rE r �test_failing_out_wrapl s z(TestSpecialMethods.test_failing_out_wrapc C s, G dd� dt �}|� }tt�||�d � d S )Nc @ s e Zd Zdd� Zddd�ZdS )z,TestSpecialMethods.test_none_wrap.<locals>.Ac S s t �d�S r r r r r r r � s z6TestSpecialMethods.test_none_wrap.<locals>.A.__array__Nc S s d S r# r r` r r r rb � s z;TestSpecialMethods.test_none_wrap.<locals>.A.__array_wrap__)Nr r r r r r7 � s r7 )rs r r* r� rB r r r �test_none_wrap s z!TestSpecialMethods.test_none_wrapc C sF G dd� dt �}|� }t�||�}t|t�d�� tt|�tj� d S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z:TestSpecialMethods.test_default_prepare.<locals>.with_wraprV c S s t �d�S r r r r r r r � s zDTestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array__c S s |S r# r r` r r r rb � s zITestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array_wrap__N)r% r& r' rd r rb r r r r r � s r rg )rs r* r� r r r} r_ rh r5 r r r �test_default_prepare� s z'TestSpecialMethods.test_default_preparec sV G � fdd�dt j�� t �d�j� d�}t �||�}t|t �d�� tt|�� � d S )Nc s e Zd ZdZ� fdd�ZdS )z5TestSpecialMethods.test_prepare.<locals>.with_preparerV c s t �|�j� d�S �N�r_ �r r? rY r` �� with_preparer r r& � s zGTestSpecialMethods.test_prepare.<locals>.with_prepare.__array_prepare__N�r% r& r' rd r& r rP r r rQ � s rQ rg rN r3 )r rh r? rY rA r r_ r9 r rP r � test_prepare� s z TestSpecialMethods.test_preparec sn G � fdd�dt j�� t �dg�j� d�}t �|||�}tt �||� � t|t �dg�� tt|�� � d S )Nc s e Zd ZdZ� fdd�ZdS )z9TestSpecialMethods.test_prepare_out.<locals>.with_preparerV c s t �|�j� d�S rM rO r` rP r r r& � s zKTestSpecialMethods.test_prepare_out.<locals>.with_prepare.__array_prepare__NrR r rP r r rQ � s rQ rg rN r3 ) r rh r? rY rA r Z shares_memoryr r_ r9 r rP r �test_prepare_out� s z#TestSpecialMethods.test_prepare_outc C s* G dd� dt �}|� }tttj||� d S )Nc @ s e Zd Zdd� Zddd�ZdS )z2TestSpecialMethods.test_failing_prepare.<locals>.Ac S s t �d�S r r r r r r r � s z<TestSpecialMethods.test_failing_prepare.<locals>.A.__array__Nc S s t �d S r# r@ r` r r r r& � s zDTestSpecialMethods.test_failing_prepare.<locals>.A.__array_prepare__)N)r% r& r' r r& r r r r r7 � s r7 )rs r rA r* r� rB r r r �test_failing_prepare� s z'TestSpecialMethods.test_failing_preparec C s� G dd� dt �}G dd� dt �}G dd� dt �}|� }t�t�d�|� t|jtjk� t|jd d� t|jd |k� t|j dk� tt�||� �d� tt�||� �d� d S ) Nc @ s e Zd Zddd�ZdS )z5TestSpecialMethods.test_array_with_context.<locals>.ANc S s& |\}}}|| _ || _|| _t�d�S r )r� r r� r r} )r! r: ra r� r r� r r r r � s z?TestSpecialMethods.test_array_with_context.<locals>.A.__array__)NN�r% r& r' r r r r r r7 � s r7 c @ s e Zd Zddd�ZdS )z5TestSpecialMethods.test_array_with_context.<locals>.BNc S s t �d|�S r r )r! r: r r r r � s z?TestSpecialMethods.test_array_with_context.<locals>.B.__array__)NrV r r r r r< � s r< c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_array_with_context.<locals>.Cc S s t �d�S r r r r r r r � s z?TestSpecialMethods.test_array_with_context.<locals>.C.__array__NrV r r r r r= � s r= rg r ) rs r* r� r r} r r� r r r� )r! r7 r< r= rL r r r �test_array_with_context� s z*TestSpecialMethods.test_array_with_contextc C s� G dd� dt �}G dd� dtj�}|� }t�dg��|�}t�||�}tj|||d�}t|d |� t|d |� t|d tj� t|d tj� t|d d � t|d d � t|d ||f� t|d ||f� t|d i � t|d d |fi� d S ) Nc @ s e Zd Zdd� ZdS )z1TestSpecialMethods.test_ufunc_override.<locals>.Ac _ s | ||||fS r# r �r! r� �method�inputs�kwargsr r r �__array_ufunc__� s zATestSpecialMethods.test_ufunc_override.<locals>.A.__array_ufunc__N�r% r& r' r\ r r r r r7 � s r7 c @ s e Zd ZdZdS )z9TestSpecialMethods.test_ufunc_override.<locals>.MyNDArrayr� Nr; r r r r � MyNDArray� s r^ rg r� r r3 �__call__rl r� r7 )rs r rh r? rY r4 r )r! r7 r^ rL r{ �res0�res1r r r �test_ufunc_override� s z&TestSpecialMethods.test_ufunc_overridec C s� dd� }dd� }t �|dd�}t �|dd�}G dd � d t�}G d d � d |�}G d d � d t�}G dd� dt�}G dd� d|�} |� } |� } |� } |� } t �| | �}t|d � t �| | �}t|d � t �| | �}t|d � t| jd� t �| | �}t| jd� |� } | � }ttt j| |� t| jd� t|jd� d | _|_ttt j|| � t| jd� t|jd� d| _ttt j| | � t| jd� d| _ttt jd| � t| jd� t|| dd�d � t|d| d�d � t|dd| �d � t|| | d�d � t|| d| �d � t|| d| �d � t|| d| �d � t|| | d�d � d| _t|| | d�d � t| jd� d| _t|d| | �d � t| jd� d| _t|| | | �d � t| jd� d|_t|| | |�d � t|jd� t|dd| �d � tt|dd| � tt||d| � tt||dd� t|| ddd�d � t|d| dd�d � t|dd| d�d � t|ddd| �d � t|| | dd�d � t|d| d| �d � t|| d| d�d � t|| dd| �d � t|| dd| �d � |� } | � }tt|ddd| � t| jd� d| _tt|dd|| � t|jd� t| jd� |� }d | _|_tt|d| ||� t|jd� t| jd� t|jd� d | _ |_|_tt||| || � t|jd� t| jd� t|jd� d S )Nc S s | | | S r# r )rL r{ r� r r r �tres_mul� s z<TestSpecialMethods.test_ufunc_override_mro.<locals>.tres_mulc S s | | | | S r# r )rL r{ r� rO r r r � quatro_mul s z>TestSpecialMethods.test_ufunc_override_mro.<locals>.quatro_mulrl rg r� c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Ac _ s dS )Nr7 r rX r r r r\ s zETestSpecialMethods.test_ufunc_override_mro.<locals>.A.__array_ufunc__Nr] r r r r r7 s r7 c @ s e Zd Zdd� ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.ASubc _ s dS )N�ASubr rX r r r r\ s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.ASub.__array_ufunc__Nr] r r r r re s re c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Bc _ s dS )Nr< r rX r r r r\ s zETestSpecialMethods.test_ufunc_override_mro.<locals>.B.__array_ufunc__Nr] r r r r r< s r< c @ s e Zd Zdd� Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Cc S s d| _ d S rF )�countr r r r �__init__ s z>TestSpecialMethods.test_ufunc_override_mro.<locals>.C.__init__c _ s | j d7 _ tS r �rf �NotImplementedrX r r r r\ s zETestSpecialMethods.test_ufunc_override_mro.<locals>.C.__array_ufunc__N)r% r& r' rg r\ r r r r r= s r= c @ s e Zd Zdd� ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.CSubc _ s | j d7 _ tS r rh rX r r r r\ s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.CSub.__array_ufunc__Nr] r r r r �CSub s rj r3 r r# )r Z frompyfuncrs r4 r rf r rJ )r! rc rd Zthree_mul_ufuncZfour_mul_ufuncr7 re r< r= rj rL Za_subr{ r� r Zc_sub�c2r r r �test_ufunc_override_mro� s� z*TestSpecialMethods.test_ufunc_override_mroc C sH G dd� dt �}|� }tjjd|ddd�}t|d |� t|d tj� t|d d � t|d d|f� t|d ddd�� tttj|� tttj||||� tttj||d d d � tttj||dddgd� tj� |dddd�}t|d |� t|d tj� t|d d� t|d |f� t|d ddddd�� tjj |ddddddd�}t|d |� t|d tj� t|d d� t|d |f� t|d ddddddd�� tj� |dd d d�}t|d dd dd�� tjj |d dd d �}t|d dd d �� tjj |d d d d �}t|d d d d!�� tj� |dd d ddd �}t|d dd ddd d"�� tj� |dd d dtj d �}t|d dd dd d#�� tj� |dd d dd d �}t|d dd dd d d"�� tt tjj |d$d%� tt tjj |d&d%� tttjj |ddd'� tj� |ddd�}t|d |� t|d tj� t|d d(� t|d |f� t|d dddd)�� tjj |dddd*�}t|d |� t|d tj� t|d d(� t|d |f� t|d dddd)�� tj� |dd d �}t|d dd d!�� tjj |d dd+d,�}t|d dd+d!�� tjj |d d d d �}t|d d d d!�� tt tjj |d$d%� tt tjj |d&d%� tttjj |ddd'� tj� |d dgddd�}t|d |� t|d tj� t|d d-� t|d |d dgf� t|d dddd)�� tjj |d dgdddd*�}t|d |� t|d tj� t|d d-� t|d |d dgf� t|d dddd)�� tj� |d dgdd d �}t|d dd d!�� tjj |d dgd d d.d/�}t|d d d.d!�� tjj |d dgd d d d%�}t|d d d d!�� tt tjj |d dgd$d%� tt tjj |d dgd&d%� tttjj |d dgddd'� tj�|d�}t|d |� t|d tj� t|d d0� t|d |df� t|d i � tttjj|� tttjj||||� tttjj||d d d � tj�|d dgd1�}t|d |� t|d tj� t|d d2� t|d |d dgd1f� tttjj|� tttjj||||� d S )3Nc @ s e Zd Zdd� ZdS )z9TestSpecialMethods.test_ufunc_override_methods.<locals>.Ac _ s | ||||fS r# r �r! �ufuncrY rZ r[ r r r r\ � s zITestSpecialMethods.test_ufunc_override_methods.<locals>.A.__array_ufunc__Nr] r r r r r7 � s r7 rg r rV )r� Zanswerr r3 r_ rl r� rL )ZsigZ signature)�axisZaxesZaxis0Zdtype0�out0Zkeep0rk )rp )r: r7 �keepdimsro Zinit0Zwhere0)ro r: r7 rq �initialr� )r: r7 rq ro rr r� F)ro r: rq T)r7 ro rq )ro rq r# )r7 r: )ro r: )ro r: rq rr r� )ro r: rq r� r r� )rp �out1�ro � accumulate)r: r7 ro )ro r: r7 Zdtype1)r7 ro r: �reduceatr� )ro r7 r: �outerZb0�at)rs r r4 r_ r r rJ � ncu_tests�inner1drk Z_NoValuerI ru rv rw rx )r! r7 rL r r r r �test_ufunc_override_methods� s( � � � � �� � � � � � � �� �� �z.TestSpecialMethods.test_ufunc_override_methodsc C s� G dd� dt �}G dd� dt �}|� }|� }t�||d�}tj||dd�}t�d|d�}tjd|dd�}t�|d d�} tj|d dd�} t|d d d� t|d d d� t|d d d� t|d d d� t| d d d� t| d d d� t�|d d�} t�|d d�} t| d d d � t| d d d� t| d d d � t| d d d� tt�|d �i k� ttj|d d�i k� ttj|dd�i k� tt�|d �i k� tt�|d d �i k� ttj|dd�i k� tj dd��<} t� ddt � ttj|d d�i k� t| d j t k� W 5 Q R X t ttj||ddd� t ttj||dd� t ttj||dd� t ttj|dd� t ttj|ddd� t ttj|ddd� t ttj|d d� t ttj|d d� d S ) Nc @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_out.<locals>.Ac _ s |S r# r rm r r r r\ . s zETestSpecialMethods.test_ufunc_override_out.<locals>.A.__array_ufunc__Nr] r r r r r7 - s r7 c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_out.<locals>.Bc _ s |S r# r rm r r r r\ 2 s zETestSpecialMethods.test_ufunc_override_out.<locals>.B.__array_ufunc__Nr] r r r r r< 1 s r< Zout_argr� r3 rl r� r� r7 r rp rs rg r# )NNTr; r= r> r� �two�r� r| r )r| �threer~ )r� r| r~ )r� )rs r r4 r �modfrC r �sinrD rE rF rG rH r rJ rI )r! r7 r< rL r{ r` ra Zres2Zres3Zres4Zres5Zres6Zres7rT r r r �test_ufunc_override_out+ sP z*TestSpecialMethods.test_ufunc_override_outc C sJ G dd� dt �}|� }tttjd|d� tttj|� tttjd|� d S )Nc @ s e Zd Zdd� ZdS )z;TestSpecialMethods.test_ufunc_override_exception.<locals>.Ac _ s t d��d S )NZoops)rI )r! rL r[ r r r r\ g s zKTestSpecialMethods.test_ufunc_override_exception.<locals>.A.__array_ufunc__Nr] r r r r r7 f s r7 rg r� r� )rs r rI r r� r� rB r r r � test_ufunc_override_exceptiond s z0TestSpecialMethods.test_ufunc_override_exceptionc C st G dd� dt �}d}ttt�|��� t�|� � W 5 Q R X d}ttt�|��� tj|� t � dd� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )zATestSpecialMethods.test_ufunc_override_not_implemented.<locals>.Ac _ s t S r# )ri �r! r r[ r r r r\ r s zQTestSpecialMethods.test_ufunc_override_not_implemented.<locals>.A.__array_ufunc__Nr] r r r r r7 q s r7 zjoperand type(s) all returned NotImplemented from __array_ufunc__(<ufunc 'negative'>, '__call__', <*>): 'A'z�operand type(s) all returned NotImplemented from __array_ufunc__(<ufunc 'add'>, '__call__', <*>, <object *>, out=(1,)): 'A', 'object', 'int'rg r� )rs r rJ �fnmatch� translater r� rA )r! r7 r� r r r �#test_ufunc_override_not_implementedo s z6TestSpecialMethods.test_ufunc_override_not_implementedc C s� G dd� dt �}|� }d}tt|�� t�|d� W 5 Q R X tt|�� t�d|� W 5 Q R X tt|�� t�|� W 5 Q R X G dd� dt �}|� }tt�|�|k� tt|�� t�||� W 5 Q R X tt|�� tj|d|d� W 5 Q R X d S )Nc @ s e Zd ZdZdS )z?TestSpecialMethods.test_ufunc_override_disabled.<locals>.OptOutNr] r r r r �OptOut� s r� z(operand 'OptOut' does not support ufuncsrg c @ s e Zd Zdd� ZdS )zDTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArrayc _ s | S r# r r� r r r r\ � s zTTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArray.__array_ufunc__Nr] r r r r � GreedyArray� s r� r� )rs r rJ r rA r� r )r! r� Zopt_outr� r� Zgreedyr r r � test_ufunc_override_disabled� s z/TestSpecialMethods.test_ufunc_override_disabledc C s( G dd� dt �}tj}|� }|||�}t|d |� t|d |� t|d d� t|d ||f� t|d i � |dd|d �}t|d |� t|d |� t|d d� t|d d � t|d d |fi� tt||d d � tt|||d d d � tt|||d d � tt|||dd � tt|||dd � d S )Nc @ s e Zd Zdd� ZdS )z2TestSpecialMethods.test_gufunc_override.<locals>.Ac _ s | ||||fS r# r rm r r r r\ � s zBTestSpecialMethods.test_gufunc_override.<locals>.A.__array_ufunc__Nr] r r r r r7 � s r7 r rg r3 r_ rl r� r� )rg rg r7 r| r� r} r )rs ry rz r r rJ rI )r! r7 rz rL r r r r �test_gufunc_override� s( z'TestSpecialMethods.test_gufunc_overridec s\ G � fdd�dt j�� G � fdd�dt�}t �d�}t �d��� �}t �|�}t �|�}tt �||k�� t|j ddgi� t j||fd�}tt �||k�� t|j d dgi� t||k� t �d��� �}t j||d�}tt �||k�� t|j dgdgd �� t �d��� �}t � |�\}}t|j ddgi� t j |d |fd�\}}t||k� t|j d d gi� t �d��� �}t �d��� �}t j |||fd�\}} t||k� t| |k� t|j dgdd gd �� t �d��� �}t �d��� �}t j |||d�} t| |k� t| j dd gdgd �� t �d�}|� }t|� t j d ||�t k� t|� t j d ||�t k� ttt j ||� |�� �}t|� t j d ||�t k� t|� t j d ||�d k� tt � ||�d k� t �d ddgd ddgg�}|�� �}|�� } |�� }t| |� t| j ddgi� |�� } |�� }t| |� t| j ddgi� t �d��� �}|j|d�} t| |� t| |k� t| j dgdgd �� |jdd�}t �|��� �}|jd|d�} t| |� t| |k� t| j dgdgd �� t j j|d d�}t j j|d d�} t| |� t| j ddgi� t �| �}t j �|d d |�} t| |� t| |k� t| j dgdgd �� t j j|dd�}t j j|dd�} t| |� t| j ddgi� t �| �}t j �|dd |�} t| |� t| |k� t| j dgdgd �� ddd g} t j j|| d d�}t j j|| d d�} t| |� t| j ddgi� t �| �}t j �|| d d |�} t| |� t| |k� t| j dgdgd �� t �d ddgd ddgg�}|�� }|�� �� �}t j �|dd gddgfd� t j �|dd gddgfd� t||� t|j ddgi� t �d��� �}|�� �� �}t j �|dd gddgf|� t||� t|j dddgi� d S )Nc s e Zd Z�� fdd�Z� ZS )z<TestSpecialMethods.test_ufunc_override_with_super.<locals>.Ac s� g }g }t |�D ]:\}}t|� �r@|�|� |�|�tj�� q|�|� q|�dd �} g } | r�g } t | �D ]:\} } t| � �r�| �| � | �| �tj�� ql| �| � qlt| �|d< n d|j } i }|r�||d< | r�| |d< t � | �j ||f|�|�}|t k�rt S |dk�r,t|d � ��r(||d _ d S |jdk�r>|f}t� fdd �t || �D ��}|�rzt|d � ��rz||d _ t|�dk�r�|d S |S ) Nr7 r# rZ �outputsrx r rg c 3 s. | ]&\}}|d kr"t �|��� �n|V qd S r# )r rX rY )r� r �output�r7 r r r� � s ��z_TestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__.<locals>.<genexpr>)� enumerateri �appendrY r rh �pop�tuple�nout�superr\ ri �infor r )r! rn rY rZ r[ r Zin_nor� �input_r� Zout_noZout_args�jr� r� Zresults)r7 � __class__r r r\ � sT �� � zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__)r% r& r' r\ � __classcell__r r� )r� r r7 � s r7 c s e Zd Z� fdd�ZdS )z<TestSpecialMethods.test_ufunc_override_with_super.<locals>.Bc s" t � fdd�|D ��rdS tS d S )Nc 3 s | ]}t |� �V qd S r# )ri )r� r� r� r r r� � s z_TestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__.<locals>.<genexpr>�A!)r� ri rm r� r r r\ � s zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__Nr] r r� r r r< � s r< rz rZ r r� r� )rZ r� rg r_ r� r3 rl rt )ro r7 r� )r rh rs r rY r� r rS r r� r rA r\ ri r rJ r? r� r� r rk ru rv rZ rx ) r! r< rO rL r{ �checkZb1Zb2Zc1rk r� �indicesr r� r � test_ufunc_override_with_super� s� 4 z1TestSpecialMethods.test_ufunc_override_with_superN)r% r& r' r r3 r6 r8 r: r> rC rJ rK rL rS rT rU rW rb rl r{ r� r� r� r� r� r� r r r r r � s2 3+ *9 r c @ s e Zd Zdd� ZdS )� TestChoosec C s@ t �ddg�}t �ddg�}tt �||df�t �ddg�� d S )NTrg )r r? r Zchoose)r! r� rL r r r � test_mixedk s zTestChoose.test_mixedN)r% r& r' r� r r r r r� j s r� c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )�TestRationalFunctionsc C s | � tj� | � tj� d S r# )�_test_lcm_innerr r �uint16r r r r �test_lcmr s z TestRationalFunctions.test_lcmc C s | � tj� d S r# )r� r �object_r r r r �test_lcm_objectv s z%TestRationalFunctions.test_lcm_objectc C s | � tj� | �tj� d S r# )�_test_gcd_innerr r r� r� r r r r �test_gcdy s z TestRationalFunctions.test_gcdc C s | � tj� d S r# )r� r r� r r r r �test_gcd_object} s z%TestRationalFunctions.test_gcd_objectc C s� t jddg|d�}t jddg|d�}tt �||�ddg� t|t j�s�t jddddg|d�}t jddd d g|d�}tt �||�dgd � t jd ddg|d�}tt j�d ddg�d� t �d ��|�}d}tt �||�d dddddg� d S )N� �x r9 r� �� �< iX �����r� rl r# r ) r r? r �lcm� issubclass�unsignedintegerrk r r� �r! r: rL r{ r r r r� � s z%TestRationalFunctions._test_lcm_innerc C s� t jddg|d�}t jddg|d�}tt �||�ddg� t|t j�s�t jddddg|d�}t jddd d g|d�}tt �||�dgd � t jd d d g|d�}tt j�|�d � t �d��|�}d}tt �||�dddddd g� d S )Nr� r� r9 r� r� r� �( r� r� r � �# r� r# rg r3 ) r r? r �gcdr� r� rk r r� r� r r r r� � s z%TestRationalFunctions._test_gcd_innerc C sB t �t �t j�jd �}d| }d| }tt �||�d| � d S )Nr� r3 r� rV )r rB r� r� r r� )r! ZbigrL r{ r r r �test_lcm_overflow� s z'TestRationalFunctions.test_lcm_overflowc C s^ t jt jfD ]L}|t �|�j�}|d }tt �||d �|� tt �|| d �|� q d S )Nr� rl )r rB �int64r� r� r r� )r! r: rL �qr r r �test_gcd_overflow� s z'TestRationalFunctions.test_gcd_overflowc C s| ddl m} t�ddddg�|d� }t�ddddg�|d� }tt�||�d|d�g � tt�||�d|d �g � d S ) Nr )�Decimalrg r| z0.20z0.12r� z0.04z0.60)r6 r� r r? r r� r� )r! r� rL r{ r r r � test_decimal� s z"TestRationalFunctions.test_decimalc C s$ t ttjdd� t ttjdd� d S )Ng333333�?g�������?)r rJ r r� r� r r r r � test_float� s z TestRationalFunctions.test_floatc C s� t t�dd ��� dd � t�dd d �}t�dd d dg�}t t�||�dd dg� t t�||�dd d d dd d g� t t�dd d d �d � d S ) Nr3 r� r� �� i-1 l U9 l ` i�� rl rg )r r r? �itemr� r� r r r r �test_builtin_long� s *z'TestRationalFunctions.test_builtin_longN)r% r& r' r� r� r� r� r� r� r� r� r� r� r� r r r r r� q s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestRoundingFunctionsc C s` G dd� d�}t �|� |� g�}tt �|�ddg� tt �|�ddg� tt �|�ddg� dS )z3 test direct implementation of these magic methods c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z3TestRoundingFunctions.test_object_direct.<locals>.Cc S s dS r r r r r r � __floor__� s z=TestRoundingFunctions.test_object_direct.<locals>.C.__floor__c S s dS r? r r r r r �__ceil__� s z<TestRoundingFunctions.test_object_direct.<locals>.C.__ceil__c S s dS )Nrl r r r r r � __trunc__� s z=TestRoundingFunctions.test_object_direct.<locals>.C.__trunc__N)r% r& r' r� r� r� r r r r r= � s r= rg r3 rl N)r r? r �floor�ceil�trunc�r! r= r\ r r r �test_object_direct� s z(TestRoundingFunctions.test_object_directc C sl G dd� d�}t �|� |� g�}tt �|�ddg� tt �|�ddg� t�t�� t �|� W 5 Q R X dS )z$ test implementations via __float__ c @ s e Zd Zdd� ZdS )z5TestRoundingFunctions.test_object_indirect.<locals>.Cc S s dS )Nr� r r r r r � __float__� s z?TestRoundingFunctions.test_object_indirect.<locals>.C.__float__N)r% r& r' r� r r r r r= � s r= r� r� N) r r? r r� r� �pytestZraisesrJ r� r� r r r �test_object_indirect� s z*TestRoundingFunctions.test_object_indirectc C s> t dd�}tt�|�d� tt�|�d� tt�|�d� d S )Nr rl r� r| )r r r r� r� r� )r! r% r r r � test_fraction� s z#TestRoundingFunctions.test_fractionN)r% r& r' r� r� r� r r r r r� � s r� c @ s� e Zd Zejejejejejej ej ej ej ej ejejejejejejgZdd� Zdd� Zdd� Zdd� Zd d � Zej� d ej ej ej g�d d � �Z dS )�TestComplexFunctionsc C s` | j D ]T}|tjkrd}nd}||�}|t|��}t|j|d| d� t|jdd| d� qd S )Nr� r2 z real part %sr� r� z imag part %s)�funcsr �arccoshr� r r� r� )r! r% r� �frZfzr r r �test_it� s z TestComplexFunctions.test_itc C sf d}| j D ]V}|t�|��}|t�|��}|t�|��}t||dd| d� t||dd| d� q d S )Nr� r# z fch-fcd %s)r6 r� r z fch-fcl %s)r� r ZcsingleZcdoubleZ clongdoubler )r! r� r% ZfcfZfcdZfclr r r �test_precisions_consistent s z/TestComplexFunctions.test_precisions_consistentc C s� t tjddddd� t tjddddd� t tjddddd� t tjddddd� t tjddddd� t tjddgddgddd� t tjddgddgddd� t tj d d gddgddd� t tj d d gddgddd� t tj dd gddgddd� t tj ddgddgddd� t tjd d gddgdd� t tjd d gddgdd� t tj ddgddgdd� t tj ddd gdddgdd� t tj d d dgdddgdd� t tj d d d gdddgdd� d S ) Nr� r� rg r| T� ��r� r3 � �r� r2 r ) �_check_branch_cutr rJ r' �log10rp r� �arcsin�arccos�arctan�arcsinhr� �arctanhr r r r �test_branch_cuts s" z%TestComplexFunctions.test_branch_cutsc C s t tjdddddtj� t tjdddddtj� t tjdddddtj� t tjdddddtj� t tjdddddtj� t tjddgddgdddtj� t tj ddgddgdddtj� t tj d d gddgdddtj� t tj d d gddgdddtj� t tj dd gddgdddtj� t tj ddgddgdddtj� t tjd d gddgddd tj� t tj d d gddgddd tj� t tj ddgddgddd tj� t tj ddd gdddgddd tj� t tj d d dgdddgddd tj� t tj d d d gdddgddd tj� d S )Nr� r� rg r| Tr� r� r3 r� r� r2 Fr )r� r rJ r r' r� rp r� r� r� r� r� r� r� r r r r �test_branch_cuts_complex64+ s" $$z/TestComplexFunctions.test_branch_cuts_complex64c C s� dd l }ddddg}dddd d d d �}d t�t�j }| jD ]�}|j�d�d }|�||�}zt ||�}W n t k r� Y q<Y nX |D ]@} t|t� | ���} || �} t t | | �|k d|| | | f � q�q<d S )Nr y � �y � �?y �? �r� ZasinZacosZatanZasinhZacoshZatanh)r� r� r� r� r� r� r� �.r| z%s %s: %s; cmath: %s)�cmathr �finfor� �epsr� r% �split�get�getattr�AttributeErrorr� r r ) r! r� ZpointsZname_map�atolr� �fnameZcnameZcfuncr rL r{ r r r �test_against_cmathD s& � z'TestComplexFunctions.test_against_cmathr: c s� t �� �}� d�jj�|j�� �fdd�}t �ddd�}t jddd d d �}� t jkrd||d � � n||d � � ||d� d � t jdg� d�}d}t �dt � |�| �}t t � |dk �� d}t �dt � |�| �}t t � |dk �� d}t �dt � |�| �}t t � |dk �� d}t �dt �|�| �}t t � |dk �� d � �fdd� }t j t j t jt j t j fD ]8} dd� dD �} || | d� || | d � || | d � �qtd S ) z1Check loss of precision in complex arc* functionsr� c s~ | � ��} | � � �}t�t�| �t�|�j d �}tt�||k �t�|�| t�|� |�� df� d| � � �}t�t�| �t� |�j d �}tt�||k �t�|�| t�|� |�� df� | � � �}t�t� | �t� |�j d �}tt�||k �t�|�| t�|� |�� df� d| � � �}t�t� | �t� |�j d �}tt�||k �t�|�| t�|� |�� df� d S )Nrg r� r� r� r� r� ) r� r r r� r� r rS Zargmaxr� r� r� r� r� )r� Zrtolr� rO )r: � real_dtyper r r� a s* (� (� (� (�z:TestComplexFunctions.test_loss_of_precision.<locals>.checkr� g5^�I �r� gˡE���r rV F)Zendpointg I@������� @r� ���MbP?y�h㈵��>�h㈵��>r9 y�g݈���>+j鈵��>rg r) y�i戵��>Sh���>y+j鈵��>�g݈���>ySh���>�i戵��>c s� t j|� d�}|t|�| � d }|t|�| � d }tt �||k�||f� t| |�| |� �d� k }tt �|�| || f� d S )Nr9 r3 )r rX r r rS )r� Zz0rO ZzpZzmZgood)r: r� r r r� � s c S s2 g | ]*}d D ] }|dks |dkr |d| �q qS )�g����MbP�r r� r r� r )r� ZrpZipr r r r� � s �z?TestComplexFunctions.test_loss_of_precision.<locals>.<listcomp>r� r� r� N)rg )r r� r� r: r� Zlogspace� longcomplexr? r r� r rS r� r� r� ) r! r: r� r� Zx_seriesZx_basicr� r rO r� Zptsr )r: r� r� r �test_loss_of_precisionW s: z+TestComplexFunctions.test_loss_of_precisionN)!r% r& r' r r� r� r� r� r� r� r� ZcosZtanr� r1 rJ r� r� r' rp r� r� r� r� r� r� r� �mark� parametrizer r� r� r� r r r r r� � s( � r� c @ s e Zd Zdd� Zdd� ZdS )�TestAttributesc C sV t j}t|jd� t|jdk� td|jk� t|jd� t|jd� t|j d� d S )NrA � zii->ir3 rg r ) r* rA r r% r Zntypes�typesZninr� rj )r! rA r r r �test_attributes� s z TestAttributes.test_attributesc C s( t tjj�d�� t tjj�d�� d S )Nz&add(x1, x2, /, out=None, *, where=Truez<frexp(x[, out1, out2], / [, out=(None, None)], *, where=True)r r* rA �__doc__r rC r r r r �test_doc� s � �zTestAttributes.test_docN)r% r& r' r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� TestSubclassc C s, G dd� dt j�}|d�}t|| |� d S )Nc @ s e Zd Zdd� ZdS )z-TestSubclass.test_subclass_op.<locals>.simplec S s t jj| |td�}|�d� |S )Nr9 r )r rh r] rs Zfill)Zsubtyper r! r r r r] � s z5TestSubclass.test_subclass_op.<locals>.simple.__new__N)r% r& r' r] r r r r �simple� s r� )rl r� )r rh r )r! r� rL r r r �test_subclass_op� s z TestSubclass.test_subclass_opN)r% r& r' r� r r r r r� � s r� rg r| Fc C s� t �|��|�}t �|��|�}t �|�jdkrLt �|�jd }t �d�}nt �|�jd }d}| |�} | ||| t �|� t �|� �} | ||| t �|� t �|� �} t t � t �| j | j �|k �| | f� t t � t �| j | j �|k �| | f� t t � t �| j | j | �|k �| | f� t t � t �| j | j | �|k �| | f� |�r�|j dk|j dk@ } |j dk|j dk@ } t � | ��r|| }t j|_ | |�} t t � t �| | j | j | �|k �| | | f� t t � t �| | j | j | �|k �| | | f� t � | ��r�|| }t j|_ | |�} t t � t �| | j | j | �|k �| | | f� t t � t �| | j | j | �|k �| | | f� dS )aR Check for a branch cut in a function. Assert that `x0` lies on a branch cut of function `f` and `f` is continuous from the direction `dx`. Parameters ---------- f : func Function to check x0 : array-like Point on branch cut dx : array-like Direction to check continuity in re_sign, im_sign : {1, -1} Change of sign of the real or imaginary part expected sig_zero_ok : bool Whether to check if the branch cut respects signed zero (if applicable) dtype : dtype Dtype to check (should be complex) �FrQ rc g @�@g-C ��6?r N)r Z atleast_1dr� r: r� r� r� r� r r rS r� r� r� r� )r% Zx0ZdxZre_signZim_signZ sig_zero_okr: Zscaler� Zy0ZypZymZjrZjir� r r r r� � s: $$&&** 22 2r� c C s� t t�dd�dk� tjdd��6 t dt�dd� dk � t dt�dd� dk� W 5 Q R X t t�t�tjd��� t t�t�tjd�� � d S )Nrg r| r rT r )r r Zcopysignr� r� rt r r r r � test_copysign s "r� c C s� | d�}| d�}| d�}t �| �j}tt �||�| |k� tt �||�| dk � tt �t �t j|��� tt �t �|t j��� tt �||�|k� d S rw )r r� r� r r� r� rt )r� r� r| r� r� r r r �_test_nextafter s r� c C s t tj�S r# )r� r r~ r r r r �test_nextafter s r� c C s t tj�S r# )r� r r� r r r r �test_nextafterf# s r� z long double is same as double)�reasonZppc64zIBM double double)Z conditionr� c C s t tj�S r# )r� r r� r r r r �test_nextafterl' s r c C s� t �tjd d�D ]j\} }t�| �j}td|t�| d�| |�� k oN|k n � tt�| d�| |��| d� |d � qd S )Nr� r� r� r r� ) r� r� r Zsctypesr� �tinyr r� r )r� � directionr r r r �test_nextafter_0/ s 0r c C s� | d�}t �| �j}| t j�}| t j�}t jdd��l tt �|�|k� tt �t �|��� tt �t �|��� tt �t �| ��� tt �| d��dk� W 5 Q R X d S )Nrg r r gꌠ9Y>)Fr ) r r� r� rt r� r� r �spacingr� )r� r� r� rt r� r r r � _test_spacing5 s r c C s t tj�S r# )r r r~ r r r r � test_spacingA s r c C s t tj�S r# )r r r� r r r r � test_spacingfD s r c C s t tj�S r# )r r r� r r r r � test_spacinglH s r c C sn t jddddgt jddddgi} tt jt jgd �D ]6\}}t jd d d d g|d�}tt �|�| | |d� q2d S )Ng �;g �<g @=g �=g5( p=g�� �>g�K8 ?g P?)rV r� ��h㈵��>rg ra i) r9 r5 )r r~ r� r r? r r )�refr� r9 r� r r r �test_spacing_gfortranO s � �� r c C sR t jt jfD ]@} dD ]6}| |�}| |d �}tt �||�| t �|�k� qq d S )N)rg r ra rg )r r� r~ r r� r )r� Z_fr% �f1r r r �test_nextafter_vs_spacingm s r c C s t t�tj�dk� dS )z Check np.nan is a positive nan.r N)r r r� rt r r r r � test_pos_nanu s r c C s t �ddt jfdt jfg�} t jdg| d�}d|d< d|d< d|d< d d d d g}g }|d }|d d� D ]&}|�t j�|d ||� �� |}qh|�t j�|d |d� �� t �|�}t j� |d |�}t ||� t � d� t j� |d |�}t � t j � t ||� dS )z;Test bug in reduceat when structured arrays are not copied.)r ZS11�time�valuer� r9 ZSimpler rV r r� r r� rg Nr ) r r: r� r� r@ r� rA rk r? rv r Z setbufsizeZUFUNC_BUFSIZE_DEFAULT)ZdbrL ZindxZh2Zval1Zval2Zh1r r r � test_reduceaty s& r c C s� t �g d�} t �g d�}t j�|| �}t|j|j� t|jd� t �d�}t jj|g dd�}t|j|j� t|jd� t jj|g dd�}t|j|j� t|jd � d S ) z&Reduceat should work with empty arraysZi4Zf8)r )r� r3 r rt r� rg )r� r N)r r? rA rv r r: r r )r� r� r r r r �test_reduceat_empty� s r c C sH t tjd�t dtj�t tjtj�g} t dd�t dd�t dd�t dd�t dd�t dd�t dd�g}tjdd��� | | D ]�}t�|g�}| | D ]�}t�|g�}t�|�r�t�|�r�q�t||k dd||f d� t||kdd ||f d� t||kdd ||f d� t||kdd ||f d� t||kdd ||f d� q�qvW 5 Q R X d S ) Nr rg r| r r Fz%r < %rr� z%r > %rz%r <= %rz%r >= %rz%r == %r)r� r rt r� r? Zisfiniter )ZnansZfinsr� r� r r r � test_complex_nan_comparisons� s$ $ � r c C s* d} t | tt| ��� t | t�| �� d S )Nl *�n�� )r r� r� r Zrint)�valr r r �test_rint_big_int� s r c C s0 t � � tjdddd�} t�| � W 5 Q R X d S )Nr r� s �)r r: �buffer)r r rh r� r1 r r r � test_signaling_nan_exceptions� s r r\ r3 r� c C s@ G dd� dt j�}t j�| �|�| �|��}|jjdks<t�d S )Nc @ s e Zd ZdS )z)test_outer_subclass_preserve.<locals>.fooN)r% r& r' r r r r r� � s r� )r rh r4 rw rY r� r% r+ )r\ r� Zactualr r r � test_outer_subclass_preserve� s r ){Z __future__r r r r rD r� r� r� Z fractionsr Znumpy.core.umathZcoreZumathr* Z numpy.corer ry Znumpyr Z numpy.testingr r r r r r r r r r r r r r rs r r( r0 rk rz r� r� r� r� r� r� r r r0 r3 rH rM rO rU rf rm rs 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� Zskipifr� Zdoubler� Zxfailr r r r r r r r r r r r r r r r r� r Zmatrixr r r r r �<module> s� < & 8[ * - E ::**?T +* 5c$ 8 � = �� �� �