3
���h� � @ sL d dl Zd dljZd dljjZd dl m
Z
m Z m Z m
Z
mZ G dd� d�ZdS )� N)�assert_� assert_equal�assert_array_equal� assert_warns�
assert_raisesc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
TestUfunclikec C s� t jt jt j t jdddg�}t j|jt�}t jddddddg�}tj|�}t ||� tj||�}t ||� t ||� |j
t j �}t
t�� tj|� W d Q R X d S )Ng g @TFg �)�nx�array�inf�nan�zeros�shape�bool�ufl�isposinfr �astype�np�complex_r � TypeError)�self�a�out�tgt�res� r �?/tmp/pip-build-5_djhm0z/numpy/numpy/lib/tests/test_ufunclike.py�
test_isposinf s
zTestUfunclike.test_isposinfc C s� t jt jt j t jdddg�}t j|jt�}t jddddddg�}tj|�}t ||� tj||�}t ||� t ||� |j
t j �}t
t�� tj|� W d Q R X d S )Ng g @FTg �)r r r
r r r
r r �isneginfr r r r r r )r r r r r r r r �
test_isneginf s
zTestUfunclike.test_isneginfc C s� t jddddgddd d
gg�}t j|jt�}t jddddgd d d
dgg�}tj|�}t||� tj||�}t||� t||� ttjd�d� d S )Ng �?g�������?g �?g������?g ��Q� @� g �g�������g ��g�������g �g �g �g �)r r r r
�floatr �fixr )r r r r r r r r �test_fix) s
zTestUfunclike.test_fixc s� G � fdd�dt j�� t jdd
g�}� |dd�}tj|�}t|t jdd g�� tt|� �� t|j d� |d }d |_ tj|�}tt|� �� t|j d � d S )
Nc s, e Zd Zddd�Zd � fdd� Zdd� ZdS )
z5TestUfunclike.test_fix_with_subclass.<locals>.MyArrayNc S s t j|dd�j| �}||_|S )NT)�copy)r r �view�metadata)�cls�datar% r r r r �__new__7 s z=TestUfunclike.test_fix_with_subclass.<locals>.MyArray.__new__c s t |� �r| j|_|S )N)�
isinstancer% )r �obj�context)�MyArrayr r �__array_wrap__< s
zDTestUfunclike.test_fix_with_subclass.<locals>.MyArray.__array_wrap__c S s t |dd �| _| S )Nr% )�getattrr% )r r* r r r �__array_finalize__A s zHTestUfunclike.test_fix_with_subclass.<locals>.MyArray.__array_finalize__)N)N)�__name__�
__module__� __qualname__r( r- r/ r )r, r r r, 6 s
r, g�������?Zfoo)r% � r .�barg����������)r .)
r Zndarrayr r r! r r r) r r% )r r �m�fZm0dZf0dr )r, r �test_fix_with_subclass5 s
z$TestUfunclike.test_fix_with_subclassc C sX t ttjddgtjd�d� t ttjddgtjd�d� t ttjddgtjd�d� d S )Nr3 � )�y)r �DeprecationWarningr r! r �emptyr r )r r r r �test_deprecatedS s z TestUfunclike.test_deprecatedc C s� t j}t j|�}t j}t||� tt|�t|�� d}t j|�}t jd�}t||� tt|�t|�� t jd�}t j||d�}t ||k� d S )Ng333333 @g @g )r g333333 �g �)
r r
r ZTrue_r �typer! Zfloat64r r )r �x�actual�expectedr r r r � test_scalarY s
zTestUfunclike.test_scalarN) r0 r1 r2 r r r" r8 r= rB r r r r r s r )Znumpyr Z
numpy.core�corer Znumpy.lib.ufunclike�libZ ufuncliker Z
numpy.testingr r r r r r r r r r �<module> s
|