File: //usr/local/lib/python3.10/dist-packages/numpy/lib/tests/__pycache__/test_mixins.cpython-310.pyc
o
���gv � @ s� d dl Z d dlZd dlZd dlmZmZmZ G dd� dejj j
�Z dd� Z d dd�Z
ejejejejejejejejejejejejeeej ej ej ej ej gZ!G d d
� d
�Z"dS ) � N)�assert_� assert_equal�
assert_raisesc @ s0 e Zd Zdd� ZejejfZdd� Z dd� Z
dS )� ArrayLikec C s t �|�| _d S �N)�np�asarray�value)�selfr � r �F/usr/local/lib/python3.10/dist-packages/numpy/lib/tests/test_mixins.py�__init__ s zArrayLike.__init__c s� |� dd�}|| D ]}t|� jtf �st S q
tdd� |D ��}|r0tdd� |D ��|d< t||�|i |��}t|�tu rKt� fdd�|D ��S |dkrQd S t� �|�S )N�outr c s �$ � | ]
}t |t�r |jn|V qd S r ��
isinstancer r ��.0�xr r r � <genexpr> s � �z,ArrayLike.__array_ufunc__.<locals>.<genexpr>c s r r r r r r r r ! s
� �
�c 3 s � | ] }t � �|�V qd S r )�typer �r
r r r ( s � �at)�getr �_HANDLED_TYPESr �NotImplemented�tuple�getattrr )r
�ufunc�method�inputs�kwargsr r �resultr r r �__array_ufunc__ s$ �� � zArrayLike.__array_ufunc__c C s dt | �j| jf S )Nz%s(%r))r �__name__r r r r r �__repr__0 s zArrayLike.__repr__N) r$ �
__module__� __qualname__r
r �ndarray�numbers�Numberr r# r% r r r r r s
r c C s& t | �tu rtdd� | D ��S t| �S )Nc s s � | ]}t |�V qd S r )r )r �rr r r r 6 s � z"wrap_array_like.<locals>.<genexpr>)r r r �r" r r r �wrap_array_like4 s r- c C s� t t| �t|�|d� t| t�r-t t| �t|�|d� t| |�D ]
\}}t|||� q d S t | j|j|d� t t| jdd �t|jdd �|d� d S )N��err_msg�dtype) r r r r �len�zip� _assert_equal_type_and_valuer r )r" �expectedr/ � result_item�
expected_itemr r r r3 ; s
�
�r3 c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TestNDArrayOperatorsMixinc C s� dd� }|t d�d � |dt d� � |t d�t�d� � |t�d�t d� � |t t�d��d � |dt t�d�� � |t t�d��t�d� � |t�d�t t�d�� � d S )Nc S s t | td�� d S �Nr )r3 r r, r r r �checkb s z<TestNDArrayOperatorsMixin.test_array_like_add.<locals>.checkr )r r �array)r
r9 r r r �test_array_like_add` s z-TestNDArrayOperatorsMixin.test_array_like_addc C s` t t�dg��}|d7 }t|t t�dg��� t�dg�}|t d�7 }t|t t�dg��� d S )Nr � )r r r: r3 )r
�
array_liker: r r r � test_inplaceq s z&TestNDArrayOperatorsMixin.test_inplacec C s� G dd� d�}t d�}|� }t|| |u � t|| |u � tt�� ||7 }W d � n1 s1w Y tt�� || W d � n1 sIw Y tt��
|| W d � d S 1 sbw Y d S )Nc @ s$ e Zd ZdZdZdd� Zdd� ZdS )z6TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOutz(Object that opts out of __array_ufunc__.Nc S � | S r r �r
�otherr r r �__add__� � z>TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__add__c S r? r r r@ r r r �__radd__� rC z?TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__radd__)r$ r& r' �__doc__r# rB rD r r r r �OptOut| s
rF r< )r r r � TypeError)r
rF r= �opt_outr r r � test_opt_outz s
�
�
"�z&TestNDArrayOperatorsMixin.test_opt_outc C s@ G dd� dt �}t d�}|d�}t|| |� t|| |� d S )Nc @ s e Zd ZdZdS )z=TestNDArrayOperatorsMixin.test_subclass.<locals>.SubArrayLikez&Should take precedence over ArrayLike.N)r$ r& r' rE r r r r � SubArrayLike� s rJ r r< )r r3 )r
rJ r �yr r r �
test_subclass� s
z'TestNDArrayOperatorsMixin.test_subclassc C s� t d�}t� }tt�� || W d � n1 sw Y tt�� || W d � n1 s2w Y tt��
||7 }W d � d S 1 sKw Y d S r8 )r �objectr rG )r
r �objr r r � test_object� s
�
�
"�z%TestNDArrayOperatorsMixin.test_objectc C sH t �g d��}t|�}tjtjttjfD ]
}t||�t||��� qd S )N����r r< � ) r r: r �operator�neg�pos�abs�invertr3 )r
r: r= �opr r r �test_unary_methods� s ��z,TestNDArrayOperatorsMixin.test_unary_methodsc C sT t �g d��}t|�}tD ]}t||d��}||d�}d�|�}t|||d� q
d S )NrP r< �failed for operator {}r. )r r: r �_ALL_BINARY_OPERATORSr- �formatr3 )r
r: r= rX r4 �actualr/ r r r �test_forward_binary_methods� s
�z5TestNDArrayOperatorsMixin.test_forward_binary_methodsc C sB t D ] }t|dd��}|dtd��}d�|�}t|||d� qd S )NrR r< rZ r. )r[ r- r r\ r3 )r
rX r4 r] r/ r r r � test_reflected_binary_methods� s
�z7TestNDArrayOperatorsMixin.test_reflected_binary_methodsc C sd t jddgt jd�}t|�}tt �d��}t|t �||�� t|t�||�� t|t�||�� d S )Nr< rR )r0 � )r r: �float64r r3 �matmulrS )r
r: r= r4 r r r � test_matmul� s � �z%TestNDArrayOperatorsMixin.test_matmulc C sH t t�g d���}ttj�|t�ddg��d u � t|t g d��� d S )N)r< rR � � r r< )rQ ���rd re )r r r: r �negativer r3 )r
r: r r r �
test_ufunc_at� s z'TestNDArrayOperatorsMixin.test_ufunc_atc C sP t �d�\}}t|�t|�f}tt �td��|� tt �tt �d���|� d S )Ng �?)r �frexpr r3 r: )r
�mantissa�exponentr4 r r r �test_ufunc_two_outputs� s ��z0TestNDArrayOperatorsMixin.test_ufunc_two_outputsN)r$ r& r' r; r> rI rL rO rY r^ r_ rc rh rl r r r r r7 ^ s
r7 r )#r) rS �numpyr �
numpy.testingr r r �lib�mixins�NDArrayOperatorsMixinr r- r3 �lt�le�eq�ne�gt�ge�add�sub�mul�truediv�floordiv�mod�divmod�pow�lshift�rshift�and_�xor�or_r[ r7 r r r r �<module> s8 )
�