Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_ufunclike.cpython-39.pyc
a
z[yc� � @ sT d dl Zd dlmZ d dlm mZ d 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 �t jt j t jdddg�}t �|jt�}t �g d��}t�|�}t ||� t�||�}t ||� t ||� |�
t j �}t
t�� t�|� W d � n1 s�0 Y d S )N� � @� �)TFFFFF)�nx�array�inf�nan�zeros�shape�bool�ufl�isposinfr �astype�np�complex_r � TypeError��self�a�outZtgt�res� r �D/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_ufunclike.py�
test_isposinf s
zTestUfunclike.test_isposinfc C s� t �t jt j t jdddg�}t �|jt�}t �g d��}t�|�}t ||� t�||�}t ||� t ||� |�
t j �}t
t�� t�|� W d � n1 s�0 Y d S )Nr r r
)FTFFFF)r r r
r r r r r �isneginfr r r r r r r r r r �
test_isneginf s
zTestUfunclike.test_isneginfc C s� t �g d�g d�g�}t �|jt�}t �g d�g d�g�}t�|�}t||� t�||�}t||� t||� tt�d�d� d S )N)� �?皙�����?g �?g������?)� �皙�����g ��g�������)r" r" r" r" )r$ r$ r$ r$ g ��Q� @� )r r r r �floatr �fixr r r r r �test_fix) s
zTestUfunclike.test_fixc s� G � fdd�dt j�� t �ddg�}� |dd�}t�|�}t|t �ddg�� tt|� �� t|j d� |d }d
|_ t�|�}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��| �}||_|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 r1 r r r �__array_finalize__A s zHTestUfunclike.test_fix_with_subclass.<locals>.MyArray.__array_finalize__)N)N)�__name__�
__module__� __qualname__r/ r5 r7 r r3 r r r4 6 s
r4 r# r% Zfoo)r, � ���)r .Zbar)
r Zndarrayr r r( r r r0 r r, )r r �m�fZm0dZf0dr r3 r �test_fix_with_subclass5 s
z$TestUfunclike.test_fix_with_subclassc C sX t ttjddgt�d�d� t ttjddgt�d�d� t ttjddgt�d�d� d S )Nr; � )�y)r �DeprecationWarningr r( r �emptyr r )r r r r �test_deprecatedS s z TestUfunclike.test_deprecatedc C s� t j}t �|�}t j}t||� tt|�t|�� d}t �|�}t �d�}t||� tt|�t|�� t �d�}t j||d�}t ||u � d S )Ng333333 �r
r )r )
r r
r ZTrue_r �typer( Zfloat64r r )r �x�actualZexpectedr r r r � test_scalarY s
zTestUfunclike.test_scalarN) r8 r9 r: r r! r) r? rD rH r r r r r s r )Znumpyr Z
numpy.coreZcorer Znumpy.lib.ufunclike�libZ ufuncliker Z
numpy.testingr r r r r r r r r r �<module> s
Mini Shell