File: //usr/local/lib/python3.10/dist-packages/numpy/lib/tests/__pycache__/test_ufunclike.cpython-310.pyc
o
���g� � @ 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 @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
TestUfunclikec C � t �t jt j t jdddg�}t �|jt�}t �g d��}t�|�}t ||� t�||�}t ||� t ||� |�
t j �}t
t�� t�|� W d � d S 1 sRw Y d S )N� � @� �)TFFFFF)�nx�array�inf�nan�zeros�shape�bool�ufl�isposinfr �astype�np�complex_r � TypeError��self�a�out�tgt�res� r �I/usr/local/lib/python3.10/dist-packages/numpy/lib/tests/test_ufunclike.py�
test_isposinf �
"�zTestUfunclike.test_isposinfc C r )Nr r
r )FTFFFF)r r
r r r r r r �isneginfr r r r r r r r r r �
test_isneginf r" 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 r5 r r r �__array_finalize__A s zHTestUfunclike.test_fix_with_subclass.<locals>.MyArray.__array_finalize__r3 )�__name__�
__module__� __qualname__r2 r9 r; r r7 r r r8 6 s
r8 r& r( �foo)r/ � ���)r .�bar)
r �ndarrayr
r r+ r r r4 r r/ )r r �m�f�m0d�f0dr r7 r �test_fix_with_subclass5 s
z$TestUfunclike.test_fix_with_subclassc 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 �True_r �typer+ �float64r
r )r �x�actual�expectedr r r r � test_scalarS s
zTestUfunclike.test_scalarN)r< r= r> r! r$ r, rH rO r r r r r s r )�numpyr �
numpy.core�corer �numpy.lib.ufunclike�lib� ufuncliker �
numpy.testingr r r r r r r r r r �<module> s