3
���hv � @ 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 j|�| _d S )N)�npZasarray�value)�selfr � r �</tmp/pip-build-5_djhm0z/numpy/numpy/lib/tests/test_mixins.py�__init__ s zArrayLike.__init__c s� |j df �}x&|| D ]}t|� jtf �stS qW tdd� |D ��}|r`tdd� |D ��|d< t||�||�}t|�tkr�t� fdd�|D ��S |dkr�d S t� �|�S d S )N�outc s s" | ]}t |t�r|jn|V qd S )N)�
isinstancer r )�.0�xr r r
� <genexpr> s z,ArrayLike.__array_ufunc__.<locals>.<genexpr>c s s" | ]}t |t�r|jn|V qd S )N)r
r r )r r r r r
r " s c 3 s | ]}t � �|�V qd S )N)�type)r r )r r r
r ( s �at)�getr
�_HANDLED_TYPESr �NotImplemented�tuple�getattrr )r Zufunc�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 Zndarray�numbers�Numberr r r r r r r
r s r c C s* t | �tkr tdd� | D ��S t| �S d S )Nc s s | ]}t |�V qd S )N)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� xXt| |�D ]\}}t|||� qBW 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�tjd� � |tjd�t d� � |t tjd��d � |dt tjd�� � |t tjd��tjd� � |tjd�t tjd�� � 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 r, r r r
�test_array_like_add` s z-TestNDArrayOperatorsMixin.test_array_like_addc C s` t tjdg��}|d7 }t|t tjdg��� tjdg�}|t d�7 }t|t tjdg��� d S )Nr � )r r r- r) )r �
array_liker- r r r
� test_inplaceq s z&TestNDArrayOperatorsMixin.test_inplacec C s� G dd� d�}t d�}|� }t|| |k� t|| |k� tt�� ||7 }W d Q R X tt�� || W d Q R X tt�� || W d Q R X 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 )Nr )r �otherr r r
�__add__� s z>TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__add__c S s | S )Nr )r r2 r r r
�__radd__� s z?TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__radd__)r r r �__doc__r r3 r4 r r r r
�OptOut| s r6 r/ )r r r � TypeError)r r6 r0 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 r5 r r r r
� SubArrayLike� s r9 r r/ )r r) )r r9 r �yr r r
�
test_subclass� s
z'TestNDArrayOperatorsMixin.test_subclassc C sf t d�}t� }tt�� || W d Q R X tt�� || W d Q R X tt�� ||7 }W d Q R X d S )Nr )r �objectr r7 )r r �objr r r
� test_object� s
z%TestNDArrayOperatorsMixin.test_objectc C sP t jddddg�}t|�}x0tjtjttjgD ]}t||�t||��� q.W d S )Nr/ r � ���) r r- r �operator�neg�pos�abs�invertr) )r r- r0 �opr r r
�test_unary_methods� s z,TestNDArrayOperatorsMixin.test_unary_methodsc C s\ t jddddg�}t|�}x<tD ]4}t||d��}||d�}dj|�}t|||d� q W d S )Nr/ r r? zfailed for operator {})r% r@ )r r- r �_ALL_BINARY_OPERATORSr$ �formatr) )r r- r0 rF r* �actualr% r r r
�test_forward_binary_methods� s
z5TestNDArrayOperatorsMixin.test_forward_binary_methodsc C sF x@t D ]8}t|dd��}|dtd��}dj|�}t|||d� qW d S )Nr? r/ zfailed for operator {})r% )rH r$ r rI r) )r rF r* rJ r% r r r
� test_reflected_binary_methods� s
z7TestNDArrayOperatorsMixin.test_reflected_binary_methodsc C sd t jddgt jd�}t|�}tt jd��}t|t j||�� t|tj||�� t|tj||�� d S )Nr/ r? )r& � )r r- Zfloat64r r) �matmulrA )r r- r0 r* r r r
� test_matmul� s z%TestNDArrayOperatorsMixin.test_matmulc C sP t tjddddg��}ttjj|tjddg��d k� t|t ddddg�� d S )Nr/ r? � � r r@ ���)r r r- r �negativer r) )r r- r r r
�
test_ufunc_at� s z'TestNDArrayOperatorsMixin.test_ufunc_atc C sP t jd�\}}t|�t|�f}tt jtd��|� tt jtt jd���|� d S ) Nr? rP ���g �?rU g �?rU g �?)r �frexpr r) r- )r Zmantissa�exponentr* r r r
�test_ufunc_two_outputs� s z0TestNDArrayOperatorsMixin.test_ufunc_two_outputsN)r r r r. r1 r8 r; r> rG rK rL rO rT rX r r r r
r+ ^ s
r+ )N)#r! rA 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_rH r+ r r r r
�<module> s4 )
|