Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_mixins.cpython-39.opt-1.pyc
a
z[ycv � @ 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)�npZasarray�value)�selfr � r
�A/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_mixins.py�__init__ s zArrayLike.__init__c s� |� dd�}|| D ] }t|� jtf �st S qtdd� |D ��}|r`tdd� |D ��|d< t||�|i |��}t|�tu r�t� fdd�|D ��S |dkr�d S t� �|�S d S )N�outr
c s 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 s" | ]}t |t�r|jn|V qd S r r r r
r
r r ! s �c 3 s | ]}t � �|�V qd S r )�typer �r r
r r ( � �at)�getr �_HANDLED_TYPESr �NotImplemented�tuple�getattrr )r Zufunc�methodZinputs�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 Zndarray�numbers�Numberr r r" r
r
r
r r s r c C s* t | �tu r tdd� | D ��S t| �S d S )Nc s s | ]}t |�V qd S r )r )r �rr
r
r r 6 r 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�rXt t| �t|�|d� t| |�D ]\}}t|||� q@n4t | 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+ Z result_itemZ
expected_itemr
r
r r/ ; s
�r/ 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 )r/ r r( r
r
r �checkb s z<TestNDArrayOperatorsMixin.test_array_like_add.<locals>.checkr )r r �array)r r3 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 r4 r/ )r �
array_liker4 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 sb0 Y tt�� || W d � n1 s�0 Y tt�� || W d � n1 s�0 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 | S r r
�r �otherr
r
r �__add__� s z>TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__add__c S s | S r r
r9 r
r
r �__radd__� s z?TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__radd__)r! r# r$ �__doc__r r; r< r
r
r
r �OptOut| s r> r6 )r r r � TypeError)r r> r7 Zopt_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$ r= r
r
r
r � SubArrayLike� s rA r r6 )r r/ )r rA r �yr
r
r �
test_subclass� s
z'TestNDArrayOperatorsMixin.test_subclassc C s� t d�}t� }tt�� || W d � n1 s40 Y tt�� || W d � n1 sd0 Y tt�� ||7 }W d � n1 s�0 Y d S r2 )r �objectr r? )r r �objr
r
r � test_object� s
&
&
z%TestNDArrayOperatorsMixin.test_objectc C sH t �g d��}t|�}tjtjttjfD ]}t||�t||��� q(d S )N����r r6 � ) r r4 r �operator�neg�pos�abs�invertr/ )r r4 r7 �opr
r
r �test_unary_methods� s �z,TestNDArrayOperatorsMixin.test_unary_methodsc C sT t �g d��}t|�}tD ]4}t||d��}||d�}d�|�}t|||d� qd S )NrG r6 �failed for operator {}r* )r r4 r �_ALL_BINARY_OPERATORSr) �formatr/ )r r4 r7 rO r0 �actualr+ r
r
r �test_forward_binary_methods� s
z5TestNDArrayOperatorsMixin.test_forward_binary_methodsc C sB t D ]8}t|dd��}|dtd��}d�|�}t|||d� qd S )NrI r6 rQ r* )rR r) r rS r/ )r rO r0 rT 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 )Nr6 rI )r, � )r r4 Zfloat64r r/ �matmulrJ )r r4 r7 r0 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)r6 rI � � r r6 )rH ���rZ r[ )r r r4 r �negativer r/ )r r4 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 Zfrexpr r/ r4 )r ZmantissaZexponentr0 r
r
r �test_ufunc_two_outputs� s ��z0TestNDArrayOperatorsMixin.test_ufunc_two_outputsN)r! r# r$ r5 r8 r@ rC rF rP rU rV rY r^ r_ r
r
r
r r1 ^ s
r1 )N)#r% rJ Znumpyr Z
numpy.testingr r r �libZmixinsZNDArrayOperatorsMixinr r) r/ �lt�le�eq�ne�gt�ge�add�sub�mul�truediv�floordiv�mod�divmod�pow�lshift�rshift�and_�xor�or_rR r1 r
r
r
r �<module> s6 )
�
Mini Shell