Edit File: test_umath.cpython-37.pyc
B ��Fd� � @ sp 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(dd� Z)dd� Z*G dd� de�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/ee#�Z7d0d1� Z8d2d3� Z9G d4d5� d5e�Z:d6d7� Z;d8d9� Z<d:d;� Z=d<d=� Z>d>d?� Z?G d@dA� dAe�Z@G dBdC� dCe�ZAG dDdE� dEe$�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�ZRdfdg� ZSG dhdi� die#�ZTG djdk� dke�ZUG dldm� dme�ZVdndodpejWfdqdr�ZXdsdt� ZYdudv� ZZdwdx� Z[dydz� Z\e �]ejd{kd|�d}d~� �Z^dd�� Z_d�d�� Z`d�d�� Zad�d�� Zbe �]ejd{kd|�d�d�� �Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�� Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkeld�k�rle� dS )�� )�division�absolute_import�print_functionN)�_gen_alignment_data)� umath_tests) �TestCase�run_module_suite�assert_� assert_equal� assert_raises�assert_raises_regex�assert_array_equal�assert_almost_equal�assert_array_almost_equal�dec�assert_allclose�assert_no_warnings�suppress_warningsc C s t �� dkpt �� �d�S )z/ True if we are running on a Power PC platform.Zpowerpc�ppc)�platform� processor�machine� startswith� r r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_umath.py� on_powerpc s r c @ s e Zd Zdd� Zdd� ZdS )�_FilterInvalidsc C s t jdd�| _d S )N�ignore)�invalid)�np�seterr�olderr)�selfr r r �setUp s z_FilterInvalids.setUpc C s t jf | j� d S )N)r r r! )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�! @gV瞯�<)r �ncu�pi)r" r r r �test_pi$ s zTestConstants.test_pic C s t tjdd� d S )NgiW� �@gV瞯�<)r r) �e)r" r r r �test_e' s zTestConstants.test_ec C s t tjdd� d S )Ng�o��x�?gV瞯�<)r r) � euler_gamma)r" 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� �x�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 d 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� qW d S )N)TFg �?r � )�subok)�outr2 g������@)�dtypeT)�record�always� r F)r �array�empty�addr �int32�frexp�warnings�catch_warnings�filterwarnings�DeprecationWarning�categoryr � ValueError� TypeError�flags� writeable) r" r2 �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�}�x�dD �]�}|dg�}t j|d|d�}|rNtt||�� 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�\}} |�rVtt| |�� 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�\}} |�rRtt||�� ntt|�t jk� t j d d ��b} t � ddt � t j|||d�\}} |�r�tt| |�� ntt| �t jk� t| d j t k� W d Q R X qW d S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z.TestOut.test_out_wrap_subok.<locals>.ArrayWrap� c S s t �|��| ��� S )N)r �asarray�view�copy)�cls�arrr r r �__new__f s z6TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__new__c S s |� t| ��S )N)rR �type)r" rU �contextr r r �__array_wrap__i s z=TestOut.test_out_wrap_subok.<locals>.ArrayWrap.__array_wrap__N)r% r& r' �__array_priority__rV rY r r r r � ArrayWrapc s r[ )TFg �?r1 )r2 )r3 r2 )Ng������@)� )r4 T)r5 r6 r7 r )r �ndarrayr: r � isinstancerW r9 r; r< r= r> r? r@ rA ) r" r[ r2 rF rH rI rJ rK rL rM rN r r r �test_out_wrap_subokb sb zTestOut.test_out_wrap_subokN)r% r& r' rO r_ 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 )Nr\ r1 � )r4 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)rC )r" �otherr r r �__eq__� s zNTestComparisons.test_ignore_object_identity_in_equal.<locals>.FunkyType.__eq__N)r% r& r' rc r r r r � FunkyType� s rd F) r r8 �objectr rB �equalrC �nanr )r" rF rd 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 )Nr\ r1 ra )r4 c @ s e Zd Zdd� ZdS )zKTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyTypec S s t d��d S )NzI won't compare)rC )r" rb r r r �__ne__� s zRTestComparisons.test_ignore_object_identity_in_not_equal.<locals>.FunkyType.__ne__N)r% r& r' ri r r r r rd � s rd T) r r8 re r rB � not_equalrC rg r )r" rF rd r r r �(test_ignore_object_identity_in_not_equal� s z8TestComparisons.test_ignore_object_identity_in_not_equalN)r% r& r' rh rk r r r r r` � s r` c @ s, e Zd Zdd� Zdd� Zdd� Zdd� 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� rP �Z �d ���i����i����i����i����g �?g�������?g�������?g�������?r\ g��������g��������g����������g333333�r ����_ �P )r r8 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 checky �? �?y �? �?y �? @)r4 r1 )�err_msgz)Complex division overflow/underflow checkg���G0J�Vg�ys?� )r\ )r r8 � complex128r )r" �msgru �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 d Q R X d S )Nr )r �divideg )r4 g �?r ) r �errstater8 rx r �isinf�complex�infrg �isnan)r" ru rz 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�������? @)r4 g g �r1 )rw z/Complex floor division overflow/underflow checkg���G0J�Vg�ys?� )r )r r8 rx r � floor_divide)r" ry ru rz r r r �test_floor_division_complex� s z(TestDivision.test_floor_division_complexN)r% r& r' rv r{ r� r� r r r r rl � s rl c C s t �| |�t �| |�fS )N)r r� � remainder)ru rz r r r �floor_divide_and_remainder s r� c C s | t jd krdS dS d S )N�UnsignedInteger)r\ )r\ rq )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 }x�tt jgD ]�}x�t�||�D ]�\}}x�t�t|�t|��D ]�\}}d}||j||||f }t j|d |d�} t j|d |d�} || | �\} } t | | | | |d� |dkr�t | | k o�d kn |� qPt | | ko�d kn |� qPW q2W q W d S ) N� AllInteger�Floatz*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�G )r4 � )rw rq r ) r r� r� �divmod� itertools�productr� r% r8 r r ) r" r� �op�dt1�dt2�sg1�sg2�fmtry rF �b�div�remr r r �test_remainder_basic s z"TestRemainder.test_remainder_basicc C s0 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 | �} xrt tj gD ]d} x^tj d D ]P} d | j | f } |�| �}|�| �}| ||�\}}t|| | d � t|| | d � q�W q�W d S ) Ni����r r\ � c s s | ]}t |� V qd S )N)r� )�.0�tr r r � <genexpr>0 s z;TestRemainder.test_float_remainder_exact.<locals>.<genexpr>)r4 g g �r� zop: %s, dtype: %s)rw )�list�ranger� r� r r8 �int�T�float�wherer� r� r� r% �astyper )r" �nlst�plst�dividend�divisor�arg�tgtrF r� �tgtdiv�tgtremr� r� ry �fa�fbr� r� r r r �test_float_remainder_exact( s$ " z(TestRemainder.test_float_remainder_exactc C s t jd }x�tt jgD ]�}x�t�||�D ]�\}}x�t�dd�D ]�\}}d}||j||||f }t j|d d |d�} t j|d |d�} || | �\} } t| | | | |d� |dkr�t | | k o�d kn |� q>t | | ko�d kn |� q>W q(W qW d S ) Nr� )r\ rq z*op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s�N g+i�)+p>)r4 )rw rq r ) r r� r� r� r� r� r% r8 r r ) r" r� r� r� r� r� r� r� ry rF r� r� r� r r r � test_float_remainder_roundoffB s z+TestRemainder.test_float_remainder_roundoffc C sZ xzt jd D ]l}t jd|d�}t �t jd|d�| �}t �||�}t||kd| � t �| | �}t|| kd| � q W t� ��}|�td� x�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 W d Q R X d S )Nr� g �?)r4 g zdt: %sz&invalid value encountered in remainderzdt: %s, rem: %s) r r� r8 � nextafterr� r r �filter�RuntimeWarningr� rg r� ) r" r� r� rF r� �sup�fone�fzer�finf�fnanr r r �!test_float_remainder_corner_casesT 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 )Ng �ra )r r �cbrt�float32)r" r r r �test_cbrt_scalarr 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 )Ng �?g @g �ra ) r r8 r� r r� r r� rg r )r" ru r r r � test_cbrtu s zTestCbrt.test_cbrtN)r% r& r' r� r� r r r r r� q s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� TestPowerc C sX 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� xZtt jd dd�D ]F\}}}dd� |D �}t|d ||d� t j||d� t|||d� q�W x\tt j d dd�D ]H\}}}dd� |D �}t|d ||d� t j||d� t|||d� �qW d S )Ng �?g @g @r r\ r1 g @g "@rq g �?gUUUUUU�?ra �unary� )r4 rW �max_sizec S s g | ]}t �|��qS r )r) �sqrt)r� �ir r r � <listcomp>� s z.TestPower.test_power_float.<locals>.<listcomp>)rw )r3 � c S s g | ]}t �|��qS r )r) r� )r� r� r r r r� � s ) r r8 r rS r r) r� r r� �float64)r" ru rz r3 �inpry �expr r r �test_power_float s. 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%� }x�tdt j�tdt j�gD ]f}t j|gt jd&�}t j d'd(��< ||d |� ||d || � ||d || | � W d Q R X �q"W d S ))Ny �? @y @ @y @ @r g �?r\ r1 y � @y � (@y � 8@ra y &� �y G� "@y @]� F@� y � 8�y �]� ^�y x�� u�rq y�������?������ٿy;�;��?�؉�؉Ϳy� ��Q��?{�G�zĿrr y� ��Q���{�G�zĿy�n_Y�K��&�q-��yǺ���a2U0*������y�I +�������Mb�?y:бb�p��ԗ�|�p�yF�̱��~��&�|�g�g �?� c s g | ] }|� �qS r r )r� r� )�normr 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 )N)r �real�imag)ru rz r r r �assert_complex_equal� s z:TestPower.test_power_complex.<locals>.assert_complex_equal)r4 r )r ) r r8 r r r) r� r r� �complex_r} )r" ru r� �zr )r� r �test_power_complex� s, z TestPower.test_power_complexc C s� t �dg�}t �dg�}t �tt jt j�g�}dd� }x dD ]}|t �||�|� q<W |t �|d�|� t jdd��L |t �|d �|� x dD ]}|t �|| �|� q�W |t �|d �|� W d Q R X d S ) Ny y �? c S s6 t �| �t �|� } }t| j|j� t| j|j� d S )N)r rQ r r� r� )ru rz r r r r� � s z7TestPower.test_power_zero.<locals>.assert_complex_equal) g ��Q� �?g �?r\ g �?r1 ra r� rm gffffff@r r )r y �?y �����?)r r8 r rg �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 ) Nr\ r1 ra g @g9b-> @r� � )r\ r\ ra ) r r8 �int16r r4 r �may_share_memoryr �shape)r" ru �resr r r �test_fast_power� s zTestPower.test_fast_powerc C s. t �ddgd�}t �||�}t|ddg� d S )N� �i8l ��2�X�0)r r8 r� r )r" rF r� r r r �test_integer_power� s z TestPower.test_integer_powerc C s� t jd }x4|D ],}t jdd|d�}tt �|d�t �|�� qW t jd }x2|D ]*}t jd|d�}tt �|d�t �|�� qPW d S )N�Integeri����rP )r4 r r� )r r� �aranger r� � ones_like)r" �dtypesr� rU r r r �-test_integer_power_with_integer_zero_exponent� s z7TestPower.test_integer_power_with_integer_zero_exponentc C sB t jd }x2|D ]*}t jd|d�}tt �d|�t �|�� qW d S )Nr� rP )r4 r\ )r r� r� r r� r� )r" r� r� rU r r r �test_integer_power_of_1� s z!TestPower.test_integer_power_of_1c C sD t jd }x4|D ],}t jdd|d�}tt �d|�t �|�� qW d S )Nr� r\ rP )r4 r )r r� r� r r� � zeros_like)r" r� r� rU r r r �test_integer_power_of_zero� s z$TestPower.test_integer_power_of_zeroc C s� t jd }x�|D ]�}t jddddg|d�}t jddddg|d�}t jd|d�}t jd|d�}ttt j||� ttt j||� ttt j||� ttt j||� qW d S ) Nr� r r\ r1 ra )r4 r� rq )r r� r8 r rB r� )r" r� r� rF 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 sd d}d}xVt ||�D ]H\}}d||f }tjd|d�}t�||�}t|jjt�|�jk|� qW d S )Nz?bhilBHILefdgFDGZddddddddddddgDDGzdtin: %s, dtout: %sr\ )r4 )�zipr �ones� float_powerr r4 �name)r" Zarg_type�res_typeZdtin�dtoutry 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 sr ddddddddd d d g }d ddd dddddddg }x8dD ]0}t j||d�}t j||d�}tt �|�|� q:W d S )Nr\ r1 r� � � � �@ r� � i i r ra rm � r� r� rP )�frI �g)r4 )r r8 r �log2)r" ru rz r� �xf�yfr r r �test_log2_values s zTestLog2.test_log2_valuesc C s< x6t dd�D ](}t�d| �}t|t|�d| d� q W d S )Nr\ �A g @zat exponent %d)rw )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 d Q R X d S ) Ng �?g T)r5 r6 r7 g �r r\ r1 ) r r r r� r r� rg r= r> r? r� rA )r" rN r r r �test_log2_special$ 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 sr ddddddddd d d g }d ddd dddddddg }x8dD ]0}t j||d�}t j||d�}tt �|�|� q:W d S )Nr\ r1 r� r� r� r� r� r� r� i i r ra rm r� r� r� rP )r rI r )r4 )r r8 r �exp2)r" ru rz r� r r r r r �test_exp2_values4 s zTestExp2.test_exp2_valuesN)r% r& r' r r r r r r 3 s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestLogAddExp2c C s� dddddg}dddddg}dddddg}xtt ddd gdd d g�D ]Z\}}t�tj||d ��}t�tj||d ��}t�tj||d ��}tt�||�||d � qBW d S ) Nr\ r1 ra r� rm r� r rI r r� )r4 )�decimal)r� r r r8 r � logaddexp2) r" ru rz r� r� �dec_r r �zfr r r �test_logaddexp2_values? s z%TestLogAddExp2.test_logaddexp2_valuesc C sr ddddg}ddddg}ddddg}xHdD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� q*W d S )Ni@B i���iC i���)r rI r )r4 )r r8 r r )r" ru rz r� r� �logxf�logyf�logzfr r r �test_logaddexp2_rangeI s z$TestLogAddExp2.test_logaddexp2_rangec C s� t j}|| || |d| dg}||| | d|d| g}|||| ||ddg}t jdd��P xHdD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� qdW W d Q R X d S )Nr\ �raise)r )r rI r )r4 )r r� r} r8 r r ) r" r� ru rz r� r� r r r r r r �test_infS 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� r rg r� )r" r r r �test_nan_ s zTestLogAddExp2.test_nanN)r% r& r' r r r r r r r r r = s r c @ s e Zd Zdd� ZdS )�TestLogc C sz ddddddddd d d g }d ddd dddddddg }x@dD ]8}d}t j||d�}t j||d�| }tt �|�|� q:W d S )Nr\ r1 r� r� r� r� r� r� r� i i r ra rm r� r� r� rP )r rI r g�9��B.�?)r4 )r r8 r �log)r" ru rz r� �log2_r r r r r �test_log_valuesh s zTestLog.test_log_valuesN)r% r& r' r r r r r r g s r c @ s e Zd Zdd� ZdS )�TestExpc C sz ddddddddd d d g }d ddd dddddddg }x@dD ]8}d}t j||d�}t j||d�| }tt �|�|� q:W d S )Nr\ r1 r� r� r� r� r� r� r� i i r ra rm r� r� r� rP )r rI r g�9��B.�?)r4 )r r8 r r� )r" ru rz r� r r r r r r �test_exp_valuess s zTestExp.test_exp_valuesN)r% r& r' r r r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestLogAddExpc C s� dddddg}dddddg}dddddg}xtt ddd gdd d g�D ]Z\}}t�tj||d ��}t�tj||d ��}t�tj||d ��}tt�||�||d � qBW d S ) Nr\ r1 ra r� rm r� r rI r r� )r4 )r )r� r r r8 r � logaddexp) r" ru rz r� r� r r r r r r r �test_logaddexp_values~ s z#TestLogAddExp.test_logaddexp_valuesc C sr ddddg}ddddg}ddddg}xHdD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� q*W d S )Ni@B i���iC i���)r rI r )r4 )r r8 r r! )r" ru rz r� r� r r 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��P xHdD ]@}t j||d�}t j||d�}t j||d�}tt �||�|� qdW W d Q R X d S )Nr\ r )r )r rI r )r4 )r r� r} r8 r r! ) r" r� ru rz r� r� r r r r r r r � 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 )Nr )r r r� r! rg r� )r" r r r r � s zTestLogAddExp.test_nanN)r% r& r' r" r# r r r r r r r } s r c @ s e Zd Zdd� Zdd� ZdS )� TestLog1pc C s0 t t�d�t�d�� t t�d�t�d�� d S )Ng�������?g333333�?g���ư>g zo �?)r r) �log1pr )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 d Q R X d S )Nr )r r| g �g �)r r} r r) r% rg r� )r" r r r � test_special� s zTestLog1p.test_specialN)r% r& r' r& r' r r r r r$ � s r$ c @ s e Zd Zdd� Zdd� ZdS )� TestExpm1c C s8 t t�d�t�d�d � t t�d�t�d�d � d S )Ng�������?r\ g���ư>)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 )Ng g �g �)r r) r) r r� )r" r r r r' � s zTestExpm1.test_specialN)r% r& r' r* r' r r r r r( � s r( c @ s e Zd Zdd� Zdd� ZdS )� TestHypotc C s. t t�dd�t�d�� t t�dd�d� d S )Nr\ r1 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 ) Ng @g @g @r g "@g (@g 4@g 9@g )r r) r, �reducer )r" r r r � test_reduce� s zTestHypot.test_reduceN)r% r& r' r- r/ r r r r r+ � s r+ c C sF t jdd��0 tt �t�| |��d| |t�| |�f � W d Q R X d S )Nr )r z hypot(%s, %s) is %s, not nan)r r} r r� r) r, )ru rz r r r �assert_hypot_isnan� s r0 c C sF t jdd��0 tt �t�| |��d| |t�| |�f � W d Q R X d S )Nr )r z hypot(%s, %s) is %s, not inf)r r} r r~ r) r, )ru rz r r r �assert_hypot_isinf� s r1 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestHypotSpecialValuesc C s t tjtj� t tjd� d S )Nr\ )r0 r rg )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@)r1 r rg r� )r" r r r �test_nan_outputs2� s z(TestHypotSpecialValues.test_nan_outputs2c C s t tjtjd� d S )Nr )r r) r, r r� )r" r r r � test_no_fpe� s z"TestHypotSpecialValues.test_no_fpeN)r% r& r' r3 r4 r5 r r r r r2 � s r2 c C s. t t�t�| |��d| |t�| |�f � d S )Nz arctan(%s, %s) is %s, not nan)r r r� r) �arctan2)ru rz r r r �assert_arctan2_isnan� s r7 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) r6 )ru rz r r r �assert_arctan2_ispinf� s r8 c C s> t t�t�| |��o"t�| |�dk d| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not -inf)r r r~ r) r6 )ru rz r r r �assert_arctan2_isninf� s r9 c C s@ t t�| |�dko$t�t�| |�� d| |t�| |�f � d S )Nr z arctan(%s, %s) is %s, not +0)r r) r6 r �signbit)ru rz 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 -0)r r) r6 r r: )ru rz r r r �assert_arctan2_isnzero� 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 )Nr\ g �?rq g пg �?)r r) r6 r r* )r" r r r � test_one_one� s z%TestArctan2SpecialValues.test_one_onec C s6 t t�tjtj�tj� t t�tjtj�tj � d S )N)r r) r6 r �PZERO�NZEROr* )r" r r r �test_zero_nzero s z(TestArctan2SpecialValues.test_zero_nzeroc C s t tjtj� ttjtj� d S )N)r; r r? r< r@ )r" r r r �test_zero_pzero s z(TestArctan2SpecialValues.test_zero_pzeroc C s2 t t�tjd�tj� t t�tjd�tj � d S )Nrq )r r) r6 r r? r* r@ )r" r r r �test_zero_negative s z+TestArctan2SpecialValues.test_zero_negativec C s t tjd� ttjd� d S )Nr\ )r; r r? 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 )Nr\ g �?)r r) r6 r r? r* r@ )r" r r r �test_positive_zero s z+TestArctan2SpecialValues.test_positive_zeroc C s8 t t�dtj�dtj � t t�dtj�dtj � d S )Nrq g �)r r) r6 r r? r* r@ )r" r r r �test_negative_zero s z+TestArctan2SpecialValues.test_negative_zeroc C s2 t t�dtj�tj� t t�dtj�tj � d S )Nr\ rq )r r) r6 r �NINFr* )r" r r r � test_any_ninf# s z&TestArctan2SpecialValues.test_any_ninfc C s t dtj� tdtj� d S )Nr\ rq )r; r r� r< )r" r r r � test_any_pinf( s z&TestArctan2SpecialValues.test_any_pinfc C s: t t�tjd�dtj � t t�tj d�dtj � d S )Nr\ g �?g �)r r) r6 r r� r* )r" r r r � test_inf_any- s z%TestArctan2SpecialValues.test_inf_anyc C sB t t�tjtj �dtj � t t�tj tj �dtj � d S )Ng �?g �)r r) r6 r r� r* )r" r r r � test_inf_ninf2 s z&TestArctan2SpecialValues.test_inf_ninfc C s> t t�tjtj�dtj � t t�tj tj�dtj � d S )Ng �?g п)r r) r6 r r� r* )r" r r r � test_inf_pinf7 s z&TestArctan2SpecialValues.test_inf_pinfc C s. t tjtj� t tjtj� t tjtj� d S )N)r7 r rg r� )r" r r r � test_nan_any< s z%TestArctan2SpecialValues.test_nan_anyN)r% r& r' r> rA rB rC rD rE rF rH rI rJ rK rL rM r r r r r= � 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 )Ng @ra g 0@)r r) �ldexpr r8 r� r� � longdouble)r" �tpr r r � _check_ldexpD s zTestLdexp._check_ldexpc C sN t t�dd�d� | �tj� | �tj� | �tj� | �d� | �d� d S )Ng @ra g 0@r� �l)r r) rO rR r �int8r� r; )r" r r r � test_ldexpL 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 d Q R X d S )Nr )�overrS g @r ) r r} �iinfor4 �max�minr r) rO r� )r" �imax�iminr r r �test_ldexp_overflowV s z TestLdexp.test_ldexp_overflowN)r% r& r' rR rU r\ r r r r rN C s rN 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}x<|D ]4}|�|�}|�|�}t||�d� t||�d� q:W x||D ]t}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�t j� t||�t j� qxW d S )N�AllFloatr� r� rq rP r1 )r r� r� �maximumr. r� r rg ) r" �dflt�dint�seq1�seq2�funcr� �tmp1�tmp2r r r r/ ` s& zTestMaximum.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S )Nr\ y @y �? @)r r r_ r. )r" r r r �test_reduce_complexu s z TestMaximum.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �|||g�}tt �||�|� d S )Nr )r rg r8 r r_ )r" rg �arg1�arg2r3 r r r �test_float_nansy s zTestMaximum.test_float_nansc C sf x`t d�D ]T}t�td�tj�}d}t�td�tj�}tt�||�dk� tt�||�dk� q W d S )Nr\ rg g �?)r� r r8 r� re r r_ )r" r� ru rz r� r r r �test_object_nans� s z TestMaximum.test_object_nansc C s� t j}xxt|d�td|�t||�gD ]X}t jd||gt jd�}t j|d|gt jd�}t j|||gt jd�}tt �||�|� q$W d S )Nr )r4 )r rg r r8 r r_ )r" rg r� rh ri r3 r r r �test_complex_nans� s "z TestMaximum.test_complex_nansc C s. t jdt jd�}|d }tt �||�|� d S )Nrm )r4 r\ )r r� re r r_ )r" rh ri r r r �test_object_array� s z TestMaximum.test_object_arrayN) r% r& r' r/ rg rj rk rl rm 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}x<|D ]4}|�|�}|�|�}t||�d� t||�d� q:W x||D ]t}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�t j� t||�t j� qxW d S )Nr^ r� r� rq r r1 )r r� r� �minimumr. r� r rg ) r" r` ra rb rc rd r� re rf r r r r/ � s& zTestMinimum.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S )Nr\ y @y �? @)r r ro r. )r" r r r rg � s z TestMinimum.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �|||g�}tt �||�|� d S )Nr )r rg r8 r ro )r" rg rh ri r3 r r r rj � s zTestMinimum.test_float_nansc C sf x`t d�D ]T}t�td�tj�}d}t�td�tj�}tt�||�dk� tt�||�dk� q W d S )Nr\ rg g �?)r� r r8 r� re r ro )r" r� ru rz r� r r r rk � s z TestMinimum.test_object_nansc C s� t j}xxt|d�td|�t||�gD ]X}t jd||gt jd�}t j|d|gt jd�}t j|||gt jd�}tt �||�|� q$W d S )Nr )r4 )r rg r r8 r ro )r" rg r� rh ri r3 r r r rl � s "z TestMinimum.test_complex_nansc C s. t jdt jd�}|d }tt �||�|� d S )Nrm )r4 r\ )r r� re r ro )r" rh ri r r r rm � s z TestMinimum.test_object_arrayN) r% r& r' r/ rg rj rk rl rm r r r r rn � s rn c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFmaxc C s� t jd }t jd }t �d�}|d d d� }t jj}x<|D ]4}|�|�}|�|�}t||�d� t||�d� q:W xx|D ]p}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�d� t||�d� qxW d S )Nr^ r� r� rq rP r1 r� )r r� r� �fmaxr. r� r rg ) r" r` ra rb rc rd r� re rf r r r r/ � s& zTestFmax.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S )Nr\ y @y �? @)r r rq r. )r" r r r rg � s z TestFmax.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �dd|g�}tt �||�|� d S )Nr )r rg r8 r rq )r" rg rh ri r3 r r r rj � s zTestFmax.test_float_nansc C s� t j}xxt|d�td|�t||�gD ]X}t jd||gt jd�}t j|d|gt jd�}t jdd|gt jd�}tt �||�|� q$W d S )Nr )r4 )r rg r r8 r rq )r" rg r� rh ri r3 r r r rl � s "zTestFmax.test_complex_nansN)r% r& r' r/ rg rj rl r r r r rp � s rp c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestFminc C s� t jd }t jd }t �d�}|d d d� }t jj}x<|D ]4}|�|�}|�|�}t||�d� t||�d� q:W xx|D ]p}|�|�}|�|�}t||�d� t||�d� t j|d d d�< t j|d d d�< t||�d� t||�d� qxW d S )Nr^ r� r� rq r r1 r\ )r r� r� �fminr. r� r rg ) r" r` ra rb rc rd r� re rf r r r r/ � s& zTestFmin.test_reducec C s0 t tj�ddg�d� t tj�ddg�d� d S )Nr\ y @y �? @)r r rs r. )r" r r r rg s z TestFmin.test_reduce_complexc C sL t j}t �d||g�}t �|d|g�}t �dd|g�}tt �||�|� d S )Nr )r rg r8 r rs )r" rg rh ri r3 r r r rj s zTestFmin.test_float_nansc C s� t j}xxt|d�td|�t||�gD ]X}t jd||gt jd�}t j|d|gt jd�}t jdd|gt jd�}tt �||�|� q$W d S )Nr )r4 )r rg r r8 r rs )r" rg r� rh ri r3 r r r rl s "zTestFmin.test_complex_nansN)r% r& r' r/ rg rj rl r r r r rr � s rr c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestBoolc C s0 t jdt jd�}ttt j|� ttt j|� d S )Nr\ )r4 )r r� �bool_r rC �negative�positive)r" rF r r r �test_exceptions( s zTestBool.test_exceptionsc C s ddddg}ddddg}t jd t jd d }x�tt j|�D ]�}t j||d�}t j||d�}d d d d g}x*t jt jfD ]}t|||��t �|� qxW d d d d g}x*t j t j fD ]}t|||��t �|� q�W d d d d g}x*t j t j fD ]}t|||��t �|� q�W q>W d S ) Nr ra r1 r� r^ r� �?)r4 FT)r r� �mapr4 rQ � logical_orr_ r r� �bool� logical_andro � logical_xorrj ) r" Zinput1Zinput2r� r4 rh ri r3 rd r r r �test_truth_table_logical- 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" rh ri r3 r r r �test_truth_table_bitwiseF s z!TestBool.test_truth_table_bitwisec C s� t �ddddgt�}t �ddddgt�}t �ddddgt�}t �g t�}||||g}x"|D ]}tt j�|�t|�� qZW x"|D ]}tt j�|�t|�� q~W x*|D ]"}tt j �|�|� � d dk� q�W d S )Nr r\ r1 ) r r8 r| r r} r. �allr{ �anyr~ �sum)r" �noneZsomeZeveryr9 �arrsrU r r r r/ S s zTestBool.test_reduceN)r% r& r' rx r r� r/ r r r r rt ' s rt c @ s: e Zd Zdd� dD �Zdd� Zdd� Zdd � Zd d � Zd S ) �TestBitwiseUFuncsc C s g | ]}t �|��qS r )r r4 )r� �cr r r r� g s z TestBitwiseUFuncs.<listcomp>z ?bBhHiIlLqQOc C sr �xj| 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� q W d S )Nr )r4 rq z dt = '%s')rw ) � bitwise_typesr r8 �charr � bitwise_notr� r� r� )r" r� �zerosr� ry r r r � test_valuesi s$ z TestBitwiseUFuncs.test_valuesc C s� x�| j D ]�}tjdg|d�}tjdg|d�}d|j }tt�|�j|k|� tt�||�j|k|� tt�||�j|k|� tt� ||�j|k|� qW d S )Nr )r4 rq z dt = '%s') r� r r8 r� r r� r4 r� r� r� )r" r� r� r� ry r r r � test_types� s z TestBitwiseUFuncs.test_typesc C s: t tjjdkd� t tjjdkd� t tjjdkd� d S )Nr r� r� rq r� )r r r� �identityr� r� )r" r r r � test_identity� s z TestBitwiseUFuncs.test_identityc C s� t jt jt jf}xp| jD ]f}t jdg|d�}t jdg|d�}x@|D ]8}d||f }t|�|�||d� t|�|�||d� qBW qW xx| jd d� D ]f}t jg |d�}xR|D ]J}d||f }t j|j|d�}|�|�} t| ||d� t | j |j k|� q�W q�W xD|D ]<}d|f }t jg t d�}|j}|�|�} t| ||d� �qW x@|D ]8}d|f }t jdgt d�} t t |�| ��t k|� �qHW d S )Nr )r4 rq zdt: '%s', f: '%s')rw zdt: '%s'T)r r� r� r� r� r8 r r. r� r r4 re rW r| ) r" � binary_funcsr� r� r� r ry r9 r� r� �btyper r r �test_reduction� s4 z TestBitwiseUFuncs.test_reductionN)r% r& r' r� r� r� r� r� r r r r r� e s r� c @ s e Zd Zdd� ZdS )�TestIntc C sf t jdt jd�}t jdt jd�}|�� }d|d d d�< |d d d� }tt j||d�d� t||� d S )NrP )r4 � Fr1 )r3 )r r� r� r| rS r � logical_not)r" ru rG 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 )Nr\ )r r) �FLOATING_POINT_SUPPORT)r" 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 )Ng �f@g �g �V�)r r) �degreesr 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 )Ng �f@g �V�g �)r r) �radiansr 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 >�g��������g g�������?g @g �?g �?r\ )r r1 ) r r8 rg r� rS r) � heavisider r� r� )r" ru 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 d Q R X d S )Ng g @g �g �?g �r )r ) r r8 r� rg r� r� r} r) �signr )r" rF r3 r� r� r r r � test_sign� s zTestSign.test_signc C s: t �dddg�}t �|�t j��}t �|�}t||� d S )Ng��������r g�������?)r r8 r� r� re r )r" �foorF r� r r r �test_sign_dtype_object� s z TestSign.test_sign_dtype_objectc C s dd� }t t|� d S )Nc S s$ t �t jg�} t �| �t j��}d S )N)r r8 rg r� r� re )r� rF r r r r s z5TestSign.test_sign_dtype_nan_object.<locals>.test_nan)r rC )r" r r r r �test_sign_dtype_nan_object s z#TestSign.test_sign_dtype_nan_objectN)r% r& r' r� r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestMinMaxc s� x�t jdft jdfgD ]�\}}x�t|d|d�D ]�\}� �x�t� j�D ]�}t j� j|d�� d d �< t j� |< � �fdd�}tt � � � � �|d� tt � � � � �|d� d � |< t � � � d �d � d � |< t � � � d �d � qDW q.W qW d S ) Nr� r� r� )r4 rW r� )r4 c s d� �f S )Nz%r %sr r )r� ry r r �<lambda> � z0TestMinMax.test_minmax_blocked.<locals>.<lambda>)ry g _�B)rw g _��) r r� r� r r� �sizer� rg r r� rX rY r )r" r� �szr3 r� Zemsgr )r� ry r �test_minmax_blocked s z TestMinMax.test_minmax_blockedc C sH t jdt jd�dd� �t j�}t|�� |d � t|�� |d � d S )N� )r4 r� ���r )r r� rT rR r� r rX rY )r" rI r r r �test_lower_align s zTestMinMax.test_lower_alignN)r% r& r' r� r� r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )�TestAbsoluteNegativec C s� �x�t jdft jdfgD �]j\}}�x^t|d|d�D �]J\}}}dd� |D �}t j||d� t|||d� | �|d k�� � d d� |D �}t j||d� t|||d� x�t j t j t j gD ]�}x�t |j �D ]�}t j |j |d �} | |d d �< |||< |t j k�r| n|| |< tt �|�| |d� t j||d� t|| |d� t| d | |d� d | } t j||d� t|| |d� q�W q�W q4W qW d S ) Nr� rm r� )r4 rW r� c S s g | ]}t �|��qS r )r) �absolute)r� r� r r r r� + s z=TestAbsoluteNegative.test_abs_neg_blocked.<locals>.<listcomp>)r3 )rw r c S s g | ] }d | �qS )rq r )r� r� r r r r� 0 s )r4 rq )r r� r� r r� r � assertTruer� rv rg r� r� r� r� r �abs) r" r� r� r3 r� ry r� r r� rI r r r �test_abs_neg_blocked& s. " 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� )r4 r� r� )r3 ) r r� rT rR r� r r� rv r� )r" rI r r r r� C 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 sH t tttg}x6|D ].}tjd|d�}t�|�}t||t|�d� qW d S )Nrm )r4 )rw ) r� r� r re r r� rw r �str)r" Z valid_dtypesr4 ru �resultr r r � test_validP s zTestPositive.test_validc C s� t t�� t�d� W d Q R X t t�� t�t�d�� W d Q R X t t�� t�tjdgtd�� W d Q R X t t�� t�tjdgtd�� W d Q R X d S )NTz 2000-01-01r� )r4 �bar)r rC r rw � datetime64r8 r� re )r" r r r � test_invalidW s zTestPositive.test_invalidN)r% r& r' r� r� r r r r r� O 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+S ),�TestSpecialMethodsc s� G � fdd�dt �� � � }t�||�}t|jt�d�� |j\}}}| �|tjk� | � t |�d� t|d |� t|d |� | � |d� d S )Nc s e Zd Zdd� Z� fdd�ZdS )z/TestSpecialMethods.test_wrap.<locals>.with_wrapc S s t �d�S )Nr\ )r r� )r" r r r � __array__f s z9TestSpecialMethods.test_wrap.<locals>.with_wrap.__array__c s � � }||_ ||_|S )N)rU rX )r" rU rX rH )� with_wrapr r rY i s z>TestSpecialMethods.test_wrap.<locals>.with_wrap.__array_wrap__N)r% r& r' r� rY r )r� r r r� e s r� r\ r1 r ) re r) ro r rU r r� rX r� � assertEqual�len)r" rF ru rd �argsr� r )r� r � test_wrapc s z TestSpecialMethods.test_wrapc C sH G dd� dt j�}|� }t�|d�}| �t||�� t|t �d�� d S )Nc @ s e Zd ZdZdd� Zdd� ZdS )z=TestSpecialMethods.test_wrap_with_iterable.<locals>.with_wraprP c S s t �d��| ��� S )Nr\ )r rQ rR rS )rT r r r rV s zETestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__new__c S s |� t| ��S )N)rR rW )r" rU rX r r r rY � s zLTestSpecialMethods.test_wrap_with_iterable.<locals>.with_wrap.__array_wrap__N)r% r& r' rZ rV rY r r r r r� | s r� )r\ r1 ra )r r] r) �multiplyr� r^ r r8 )r" r� rF ru r r r �test_wrap_with_iterabley s z*TestSpecialMethods.test_wrap_with_iterablec C sJ G dd� dt j�}|� }t �d�| }| �t||�� t|t �d�� d S )Nc @ s e Zd ZdZdd� ZdS )z7TestSpecialMethods.test_priority_with_scalar.<locals>.ArP c S s t �dd��| ��� S )Ng �?r� )r rQ rR rS )rT r r r rV � s z?TestSpecialMethods.test_priority_with_scalar.<locals>.A.__new__N)r% r& r' rZ rV r r r r �A� s r� r\ )r r] r� r� r^ r r8 )r" r� rF ru 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 )Nr\ )r r� )r" r r r r� � s z=TestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array__c s � � }||_ |S )N)rU )r" rU rH )r� r r rY � s zBTestSpecialMethods.test_old_wrap.<locals>.with_wrap.__array_wrap__N)r% r& r' r� rY r )r� r r r� � s r� r\ )re r) ro r rU r r� )r" rF ru 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}| �t|||��tjk� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �t|||��|k� | �tt�|�|k�� | �tt�|�|k�� | �tt�|�|k�� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z+TestSpecialMethods.test_priority.<locals>.Ac S s t �d�S )Nr\ )r r� )r" r r r r� � s z5TestSpecialMethods.test_priority.<locals>.A.__array__c S s t | �� }||_||_|S )N)rW rU rX )r" rU rX rH r r r rY � s z:TestSpecialMethods.test_priority.<locals>.A.__array_wrap__N)r% r& r' r� rY r r r r r� � s r� c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Bg 4@N)r% r& r' rZ r r r r �B� s r� c @ s e Zd ZdZdS )z+TestSpecialMethods.test_priority.<locals>.Cg D@N)r% r& r' rZ r r r r �C� s r� r\ ) re r r� r) ro r� rW r] r� ) r" r� r� r� ru rF r� r� r r r r � test_priority� s2 z TestSpecialMethods.test_priorityc C s, G dd� dt �}|� }| �ttj||� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z/TestSpecialMethods.test_failing_wrap.<locals>.Ac S s t �d�S )Nr\ )r r� )r" r r r r� � s z9TestSpecialMethods.test_failing_wrap.<locals>.A.__array__c S s t �d S )N)� RuntimeError)r" rU rX r r r rY � s z>TestSpecialMethods.test_failing_wrap.<locals>.A.__array_wrap__N)r% r& r' r� rY r r r r r� � s r� )re � assertRaisesr� r) r_ )r" r� rF r r r �test_failing_wrap� s z$TestSpecialMethods.test_failing_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 )Nr\ )r r� )r" r r r r� � s z6TestSpecialMethods.test_none_wrap.<locals>.A.__array__Nc S s d S )Nr )r" rU rX r r r rY � s z;TestSpecialMethods.test_none_wrap.<locals>.A.__array_wrap__)N)r% r& r' r� rY r r r r r� � s r� )re r r) r_ )r" r� rF 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_wraprP c S s t �d�S )Nr\ )r r� )r" r r r r� � s zDTestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array__c S s |S )Nr )r" rU rX r r r rY � s zITestSpecialMethods.test_default_prepare.<locals>.with_wrap.__array_wrap__N)r% r& r' rZ r� rY r r r r r� � s r� r\ )re r) ro r r r� rW r] )r" r� rF ru 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_preparerP c s t �|�j� d�S )N)rW )r r8 rR )r" rU rX )� with_preparer r �__array_prepare__ s zGTestSpecialMethods.test_prepare.<locals>.with_prepare.__array_prepare__N)r% r& r' rZ r� r )r� r r r� � s r� r\ )rW r1 )r r] r8 rR r: r rW )r" rF ru r )r� 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_preparerP c s t �|�j� d�S )N)rW )r r8 rR )r" rU rX )r� r r r� s zKTestSpecialMethods.test_prepare_out.<locals>.with_prepare.__array_prepare__N)r% r& r' rZ r� r )r� r r r� s r� r\ )rW r1 ) r r] r8 rR r: r � shares_memoryr rW )r" rF ru r )r� r �test_prepare_out s z#TestSpecialMethods.test_prepare_outc C s, G dd� dt �}|� }| �ttj||� d S )Nc @ s e Zd Zdd� Zddd�ZdS )z2TestSpecialMethods.test_failing_prepare.<locals>.Ac S s t �d�S )Nr\ )r r� )r" r r r r� s z<TestSpecialMethods.test_failing_prepare.<locals>.A.__array__Nc S s t �d S )N)r� )r" rU rX r r r r� s zDTestSpecialMethods.test_failing_prepare.<locals>.A.__array_prepare__)N)r% r& r' r� r� r r r r r� s r� )re r� r� r) r_ )r" r� rF 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�|� | �|jtjk� t|jd d� | �|jd |k� | �|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 )Nr\ )rd r� r� r r� )r" r4 rX rd 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 r� ' s r� c @ s e Zd Zddd�ZdS )z5TestSpecialMethods.test_array_with_context.<locals>.BNc S s t �d|�S )Nr\ )r r� )r" r4 r r r r� 0 s z?TestSpecialMethods.test_array_with_context.<locals>.B.__array__)N)r% r& r' r� 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 )Nr\ )r r� )r" r r r r� 4 s z?TestSpecialMethods.test_array_with_context.<locals>.C.__array__N)r% r& r' r� r r r r r� 3 s r� r\ r ) re r) r_ r r� r� rd r r� r� )r" r� r� r� rF r r r �test_array_with_context% s z*TestSpecialMethods.test_array_with_contextc C s� G dd� dt �}|� }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 )Nr )r" rd �method�inputs�kwargsr r r �__array_ufunc__C s zATestSpecialMethods.test_ufunc_override.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� B s r� r\ )r3 r r1 �__call__ra r� r3 )re r �matrixr� r )r" r� rF 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 � ttt j| |� ttt j|| � ttt 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 � t|| | d�d � t|d| | �d � t|| | | �d � t|| | |�d � 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| � tt|dd|| � tt|d| || � d S )Nc S s | | | S )Nr )rF r� r� r r r �tres_mulZ s z<TestSpecialMethods.test_ufunc_override_mro.<locals>.tres_mulc S s | | | | S )Nr )rF r� r� rI r r r � quatro_mul] s z>TestSpecialMethods.test_ufunc_override_mro.<locals>.quatro_mulra r\ r� c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Ac _ s dS )Nr� r )r" rd r� r� r� r r r r� e s zETestSpecialMethods.test_ufunc_override_mro.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� d s r� c @ s e Zd Zdd� ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.ASubc _ s dS )N�ASubr )r" rd r� r� r� r r r r� i s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.ASub.__array_ufunc__N)r% r& r' r� r r r r r� h s r� c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Bc _ s dS )Nr� r )r" rd r� r� r� r r r r� m s zETestSpecialMethods.test_ufunc_override_mro.<locals>.B.__array_ufunc__N)r% r& r' r� r r r r r� l s r� c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_mro.<locals>.Cc _ s t S )N)�NotImplemented)r" rd r� r� r� r r r r� q s zETestSpecialMethods.test_ufunc_override_mro.<locals>.C.__array_ufunc__N)r% r& r' r� r r r r r� p s r� c @ s e Zd Zdd� ZdS )z8TestSpecialMethods.test_ufunc_override_mro.<locals>.CSubc _ s t S )N)r� )r" rd r� r� r� r r r r� u s zHTestSpecialMethods.test_ufunc_override_mro.<locals>.CSub.__array_ufunc__N)r% r& r' r� r r r r �CSubt s r� r1 r� )r � frompyfuncre r� r r rC )r" r� r� Zthree_mul_ufuncZfour_mul_ufuncr� r� r� r� r� rF Za_subr� r� Zc_subr� r r r �test_ufunc_override_mroW sf z*TestSpecialMethods.test_ufunc_override_mroc C sx 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 � tj�|dddd�}t|d |� t|d tj� t|d d� t|d |f� t|d ddddd�� tjj|ddddd�}t|d |� t|d tj� t|d d� t|d |f� t|d ddddd�� tj�|dd d d�}t|d dd dd�� tjj|d ddd�}t|d ddd�� tjj|d dd d �}t|d d d d �� tttjj|d d � tttjj|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 dd d �}t|d d d d �� tttjj |d d � tttjj |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 dd �}t|d d d d �� tttjj|d dgd d � tttjj|d dgd d � tttjj|d dgddd!� tj� |d�}t|d |� t|d tj� t|d d*� t|d |df� t|d i � tttjj |� tttjj ||||� tj� |d dgd+�}t|d |� t|d tj� t|d d,� t|d |d dgd+f� tttjj |� tttjj ||||� d S )-Nc @ s e Zd Zdd� ZdS )z9TestSpecialMethods.test_ufunc_override_methods.<locals>.Ac _ s | ||||fS )Nr )r" �ufuncr� r� r� r r r r� � s zITestSpecialMethods.test_ufunc_override_methods.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� � s r� r\ r� �* )r� �answerr r1 r� ra r� rF )�sig� signatureZaxis0Zdtype0�out0Zkeep0r. )r� )r4 r3 �keepdims�axis)r� r4 r3 r� F)r� r4 r� T)r3 r� r� )r� r� )N)r3 r4 )r� r4 r )r3 )r� �out1)r� � accumulate)r4 r3 r� )r� r4 r3 �dtype1)r3 r� r4 �reduceatr� )r� r3 r4 �outer�b0�at) re r r� r� r r rC r. rB r� r� r� r� )r" r� rF 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 d 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 )Nr )r" r� r� r� r� r r r r� M s zETestSpecialMethods.test_ufunc_override_out.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� L s r� c @ s e Zd Zdd� ZdS )z5TestSpecialMethods.test_ufunc_override_out.<locals>.Bc _ s |S )Nr )r" r� r� r� r� r r r r� Q s zETestSpecialMethods.test_ufunc_override_out.<locals>.B.__array_ufunc__N)r% r& r' r� r r r r r� P s r� Zout_arg)r3 r1 ra r� rm r3 r r� r� r\ )N)NNT)r5 r6 r7 r� �two)r� r� r )r� �threer� )r� r� r� )r� )re r r� r �modfr< r �sinr= r> r? r@ rA r rC rB )r" r� r� rF r� r� r� �res2Zres3Zres4Zres5Zres6Zres7rN r r r �test_ufunc_override_outJ 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 )N�oops)rB )r" rF r� r r r r� � s zKTestSpecialMethods.test_ufunc_override_exception.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� � s r� r\ )r3 g �?)re r rB r rv r| )r" r� rF r r r � test_ufunc_override_exception� s z0TestSpecialMethods.test_ufunc_override_exceptionc C st G dd� dt �}d}ttt�|��� t�|� � W d Q R X d}ttt�|��� tj|� t � dd� W d Q R X d S )Nc @ s e Zd Zdd� ZdS )zATestSpecialMethods.test_ufunc_override_not_implemented.<locals>.Ac _ s t S )N)r� )r" r� r� r r r r� � s zQTestSpecialMethods.test_ufunc_override_not_implemented.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� � s r� 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'r\ )r3 )re r rC �fnmatch� translater rv r: )r" r� ry r r r �#test_ufunc_override_not_implemented� s z6TestSpecialMethods.test_ufunc_override_not_implementedc C s� G dd� dt �}|� }d}tt|�� t�|d� W d Q R X tt|�� t�d|� W d Q R X tt|�� t�|� W d Q R X G dd� dt �}|� }tt�|�|k� tt|�� t�||� W d Q R X tt|�� tj|d|d� W d Q R X d S )Nc @ s e Zd ZdZdS )z?TestSpecialMethods.test_ufunc_override_disabled.<locals>.OptOutN)r% r& r' r� r r r r �OptOut� s r z(operand 'OptOut' does not support ufuncsr\ c @ s e Zd Zdd� ZdS )zDTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArrayc _ s | S )Nr )r" r� r� r r r r� � s zTTestSpecialMethods.test_ufunc_override_disabled.<locals>.GreedyArray.__array_ufunc__N)r% r& r' r� r r r r � GreedyArray� s r )r3 )re r rC r r: rv r )r" r Zopt_outry r �greedyr 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 )Nr )r" r� r� r� r� r r r r� � s zBTestSpecialMethods.test_gufunc_override.<locals>.A.__array_ufunc__N)r% r& r' r� r r r r r� � s r� r r\ r1 r� ra r� )r3 )r\ r\ r3 r� r� )r� r� r )re � ncu_tests�inner1dr r rC rB )r" r� r rF 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 }xFt |�D ]:\}}t|� �rB|�|� |�|�tj�� q|�|� qW |�dd �} g } | r�g } xFt | �D ]:\} } t| � �r�| �| � | �| �tj�� qr| �| � qrW t| �|d< n d|j } i }|r�||d< | r�| |d< t � | �j ||f|�|�}|t k�r t S |dk�r4t|d � ��r0||d _ d S |jdk�rF|f}t� fdd �t || �D ��}|�r�t|d � ��r�||d _ t|�dk�r�|d S |S ) Nr3 )Nr� �outputsr� r r\ c 3 s. | ]&\}}|d kr"t �|��� �n|V qd S )N)r rQ rR )r� r� �output)r� r r r� s z_TestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__.<locals>.<genexpr>)� enumerater^ �appendrR r r] �pop�tuple�nout�superr� r� �infor� r� )r" r� r� r� r� r� Zin_nor� �input_r Zout_no�out_args�jr r �results)r� � __class__r r r� � sL zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.A.__array_ufunc__)r% r& r' r� � __classcell__r )r� )r r r� � s r� 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 )N)r^ )r� r )r� r r r� s z_TestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__.<locals>.<genexpr>zA!)r� r� )r" r� r� r� r� )r� r r r� s zLTestSpecialMethods.test_ufunc_override_with_super.<locals>.B.__array_ufunc__N)r% r& r' r� r )r� r r r� s r� g @r� r )r3 r )r� r r\ r� zA!r1 ra )r� )r� r3 g �?)r r] re r� rR r r r� r r r r: r� r� r rC r8 r� rX r� r. r� r� rS r� ) r" r� rI rF r� �check�b1�b2�c1�c2r� �indicesr )r� r � test_ufunc_override_with_super� s� 4 z1TestSpecialMethods.test_ufunc_override_with_superN)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� b s, + [ 9 r� c @ s e Zd Zdd� ZdS )� TestChoosec C s@ t �ddg�}t �ddg�}tt �||df�t �ddg�� d S )NTr\ )r r8 r �choose)r" r� rF r r r � test_mixed� s zTestChoose.test_mixedN)r% r& r' r( r r r r r& � s r& c C s& t �t j�} t �t �| j| j �� S )N)r �finfo� longcomplex�isfinite�log10�tiny�eps)r r r r �is_longdouble_finfo_bogus� s r/ c @ s� e Zd Zejejejejejej ej ej ej ej ejejejejejejgZdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Ze � e � d�dd� �Z dS )�TestComplexFunctionsc C sf x`| j D ]V}|tjkr d}nd}||�}|t�|��}t|j|d| d� t|jdd| d� qW d S )Ng �?g �?z real part %s)rw g z imag part %s)�funcsr �arccoshr r r� r� )r" r ru �frZfzr r r �test_it� s z TestComplexFunctions.test_itc C sj d}x`| j D ]V}|t�|��}|t�|��}|t�|��}t||dd| d� t||dd| d� q W d S )Ny �? �?r� z fch-fcd %s)r rw r� z fch-fcl %s)r1 r �csingle�cdouble� clongdoubler )r" r� r ZfcfZfcdZfclr r r �test_precisions_consistent� s z/TestComplexFunctions.test_precisions_consistentc c s� t tjdddddfV t tjdddddfV t tjdddddfV t tjdddddfV t tjdddddfV t tjddgddgdddfV t tjddgddgdddfV t tj d d gddgdddfV t tj d d gddgdddfV t tj dd gddgdddfV t tj ddgddgdddfV t tjd d gddgddfV t tjd d gddgddfV t tj ddgddgddfV t tj ddd gdddgddfV t tj d d dgdddgddfV t tj d d d gdddgddfV d S ) Ng �y �?r\ rq Tg ��rr r1 y �y @g �?r ) �_check_branch_cutr r r r, r% r� �arcsin�arccos�arctan�arcsinhr2 �arctanh)r" r r r �test_branch_cuts� s" z%TestComplexFunctions.test_branch_cutsc c s* t tjdddddtjfV t tjdddddtjfV t tjdddddtjfV t tjdddddtjfV t tjdddddtjfV t tjddgddgdddtjfV t tj ddgddgdddtjfV t tj d d gddgdddtjfV t tj d d gddgdddtjfV t tj dd gddgdddtjfV t tj ddgddgdddtjfV t tjd d gddgddd tjfV t tj d d gddgddd tjfV t tj ddgddgddd tjfV t tj ddd gdddgddd tjfV t tj d d dgdddgddd tjfV t tj d d d gdddgddd tjfV d S )Ng �y �?r\ rq Tg ��rr r1 y �y @g �?Fr )r9 r r � complex64r r, r% r� r: r; r< r= r2 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�tj�j }x�| jD ]�}|j�d�d }|�||�}yt ||�}W n t k r� w@Y nX xH|D ]@} t|t� | ���} || �} t t | | �|k d|| | | f � q�W q@W d S )Nr y � �y � �?y �? �y �? �?�asin�acos�atan�asinh�acosh�atanh)r: r; r< r= r2 r> r� �.rq z%s %s: %s; cmath: %s)�cmathr r) r r. r1 r% �split�get�getattr�AttributeErrorr� r r� ) r" rI ZpointsZname_map�atolrd �fname�cnameZcfuncr� rF r� r r r �test_against_cmath� s z'TestComplexFunctions.test_against_cmathc 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� }xTt j t j t jt j t j fD ]8} dd� dD �} || | d� || | d � || | d � �qvW d S ) z1Check loss of precision in complex arc* functionsg 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 )Nr\ r= y �?r: r> r< ) r� r r� r= r� r r� �argmaxrX r: r� r> r< )ru �rtolr� rI )r4 � real_dtyper r r � s" ( ( ( (z;TestComplexFunctions.check_loss_of_precision.<locals>.checki��g5^�I ��� gˡE���r rP F)�endpoint�2 g������ @r1 g����MbP?y�h㈵��>�h㈵��>)r4 y�g݈���>+j鈵��>r\ gV瞯�<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 )N)r4 r1 )r rQ r� r r� )rd Zz0rI ZzpZzm�good)r4 r. r r r 8 s c S s2 g | ]*}d D ] }|dks |dkr |d| �q qS ))g����MbP�r g����MbP?r y �?r )r� �rp�ipr r r r� C s z@TestComplexFunctions.check_loss_of_precision.<locals>.<listcomp>)g����MbP�r g����MbP?y �?y �? �?N)r\ )r r) r� r4 r. �logspacer* r8 r� r> r r� r= r< r: ) r" r4 r r Zx_seriesZx_basicr� r� rI rd Zptsr )r4 r. rT r �check_loss_of_precision� s: z,TestComplexFunctions.check_loss_of_precisionc c s& x t jt jgD ]}| j|fV qW d S )N)r r@ r� r\ )r" r4 r r r �test_loss_of_precisionI s z+TestComplexFunctions.test_loss_of_precisionzBogus long double finfoc C s | � tj� d S )N)r\ r r* )r" r r r �"test_loss_of_precision_longcomplexM s z7TestComplexFunctions.test_loss_of_precision_longcomplexN) r% r& r' r r: r; r< r= r2 r> r �cos�tanr� r r r� r, r r% r1 r4 r8 r? rA rQ r\ r] r �knownfailureifr/ r^ r r r r r0 � s Ur0 c @ s e Zd Zdd� Zdd� ZdS )�TestAttributesc C sZ t j}t|jd� | �|jdk� | �d|jk� t|jd� t|jd� t|j d� d S )Nr: � zii->ir1 r\ r ) r) r: r r% r� �ntypes�types�ninr r� )r" r: r r r �test_attributesS 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) r: �__doc__r r< )r" r r r �test_doc\ s zTestAttributes.test_docN)r% r& r' rg ri r r r r rb R s rb 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 )N)r4 r )r r] rV re �fill)�subtyper� r" r r r rV j s z5TestSubclass.test_subclass_op.<locals>.simple.__new__N)r% r& r' rV r r r r �simplei s rm )ra r� )r r] r )r" rm rF r r r �test_subclass_opg s z TestSubclass.test_subclass_opN)r% r& r' rn r r r r rj e s rj r\ rq 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) �Fg Y@g{�G�z�?g @�@g-C ��6?r N)r � atleast_1dr� r4 r� r) r. r� r� r r� r� r� r� r@ )r �x0�dxZre_signZim_signZ sig_zero_okr4 �scalerN �y0Zyp�ymZjr�jiru r r r r9 r s: $$&&** 22 2r9 c C s� t t�dd�dk� tjdd��6 t dt�dd� dk � t dt�dd� dk� W d Q R X t t�t�tjd��� t t�t�tjd�� � d S )Nr\ rq r )r| r )r r �copysignr} r: rg r r r r � test_copysign� s "rx c C s� | d�}| d�}| d�}t �| �j}tt �||�| |k� tt �||�| dk � tt �t �t j|��� tt �t �|t j��� tt �||�|k� d S )Nr\ r1 r )r r) r. r r� r� rg )r� r� r� r� r. r r r �_test_nextafter� s ry c C s t tj�S )N)ry r r� r r r r �test_nextafter� s rz c C s t tj�S )N)ry r r� r r r r �test_nextafterf� s r{ �win32z0Long double support buggy on win32, ticket 1664.c C s t tj�S )N)ry r rP r r r r �test_nextafterl� s r} c C s� x�t �tjd d�D ]j\} }t�| �j}td|t�| d�| |�� k oP|k n � tt�| d�| |��| d� |d � qW d S )Nr� )r\ rq g r g������ @) r� r� r �sctypesr) r- r 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 d Q R X d S )Nr\ r )r gꌠ9Y>)Fr ) r r) r. rg r� r} r �spacingr� )r� r� r. rg r� r r r � _test_spacing� s r� c C s t tj�S )N)r� r r� r r r r � test_spacing� s r� c C s t tj�S )N)r� r r� r r r r � test_spacingf� s r� c C s t tj�S )N)r� r rP r r r r � test_spacingl� s r� c C sr t jddddgt jddddgi} xLtt jt jgd �D ]6\}}t jd d d d g|d�}tt �|�| | |d� q4W d S )Ng �;g �<g @=g �=g5( p=g�� �>g�K8 ?g P?)rP r� g�h㈵��>r\ i� i) )r4 )r )r r� r� r� r8 r r� )�refr� r ru r r r �test_spacing_gfortran� s r� c C sZ xTt jt jgD ]D} x>dD ]6}| |�}| |d �}tt �||�| t �|�k� qW qW d S )N)r\ g�h㈵��>i� r\ )r r� r� r r� r� )r� �_fr �f1r r r �test_nextafter_vs_spacing s r� c C s t t�tj�dk� dS )z Check np.nan is a positive nan.r N)r r r: rg r r r r � test_pos_nan 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 }x6|d d� D ]&}|�t j�|d ||� �� |}qjW |�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� �S11�time�valuero )r4 ZSimpler� rP r r� r� � r\ Nr� ) r r4 �int64r� r9 r r: r. r8 r� r � setbufsize�UFUNC_BUFSIZE_DEFAULT)�dbrF �indxZh2�val1�val2Zh1r r r � test_reduceat 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 arrays�i4�f8)r )rm r1 r )r� )r r1 r\ )rm r N)r r8 r: r� r r4 r� r� )r$ ru r� r r r �test_reduceat_empty0 s r� c C sP t tjd�t dtj�t tjtj�g} t dd�t dd�t dd�t dd�t dd�t dd�t dd�g}tjdd��� x�| | D ]�}t�|g�}x�| | 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�W qxW W d Q R X d S ) Nr r\ rq r )r Fz%r < %r)rw z%r > %rz%r <= %rz%r >= %rz%r == %r)r r rg r} r8 r+ r )ZnansZfinsru rz 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 �rint)�valr r r �test_rint_big_intU s r� c C s0 t � � tjdddd�} t�| � W d Q R X d S )Nr r� s �)r� r4 �buffer)r r r] r� )rF r r r � test_signaling_nan_exceptions_ s r� �__main__)m� __future__r r r �sysr r= r r� Znumpy.testing.utilsr Znumpy.core.umath�core�umathr) � numpy.corer r �numpyr Z numpy.testingr r r r r r r r r r r r r r re r r( r0 r` rl r� r� r� r� r� r� r� r r r r r r$ r( r+ r0 r1 r2 r7 r8 r9 r; r< r= rN r] rn rp rr rt r� r� r� r� r� r� r� r� r� r� r� r& r/ r0 rb rj r r9 rx ry rz r{ ra r} r� r� r� r� r� r� r� r� r� r� r� r� r� r% r r r r �<module> s� < &2[ * ) E ::**>U * - > <