Edit File: test_mixins.cpython-37.pyc
B ��Fd� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dl m Z m Z m Z m Z mZ ejjdk ZG dd� dejjj�Zdd� Zdd d �Zejejejejej ej ej ej ej ej!ej"ej#e$e%ej&ej'ej(ej)ej*gZ+G d d � d e �Z,e-d kr�e � dS )� )�division�absolute_import�print_functionN)�TestCase�run_module_suite�assert_� assert_equal� assert_raises� c @ s0 e Zd Zdd� ZejejfZdd� Z dd� Z dS )� ArrayLikec C s t �|�| _d S )N)�np�asarray�value)�selfr � r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_mixins.py�__init__ s zArrayLike.__init__c s� |� dd�}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�outr c 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 �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__7 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 | �tkr tdd� | D ��S t| �S d S )Nc s s | ]}t |�V qd S )N)r )r �rr r r r = s z"wrap_array_like.<locals>.<genexpr>)r r r )r# r r r �wrap_array_like; 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 r2 B s r2 c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� 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 )r2 r )r# r r r �checkj s z<TestNDArrayOperatorsMixin.test_array_like_add.<locals>.checkr )r r �array)r r5 r r r �test_array_like_addh 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 r6 r2 )r � array_liker6 r r r � test_inplacey s z&TestNDArrayOperatorsMixin.test_inplacec C s� G dd� dt �}td�}|� }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 r; 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? r8 )�objectr r r � TypeError)r r? r9 �opt_outr r r � test_opt_out� 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 rD r r8 )r r2 )r rD 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 r@ r rA )r r �objr r r � test_object� s z%TestNDArrayOperatorsMixin.test_objectc C sP t �ddddg�}t|�}x0tjtjttjgD ]}t||�t||��� q.W d S )N���r r8 � ) r r6 r �operator�neg�pos�abs�invertr2 )r r6 r9 �opr r r �test_unary_methods� s z,TestNDArrayOperatorsMixin.test_unary_methodsc C s\ t �ddddg�}t|�}x<tD ]4}t||d��}||d�}d�|�}t|||d� q W d S )NrI r r8 rJ zfailed for operator {})r. )r r6 r �_ALL_BINARY_OPERATORSr- �formatr2 )r r6 r9 rP r3 �actualr. r r r �test_forward_binary_methods� s z5TestNDArrayOperatorsMixin.test_forward_binary_methodsc C sF x@t D ]8}t|dd��}|dtd��}d�|�}t|||d� qW d S )NrJ r8 zfailed for operator {})r. )rR r- r rS r2 )r rP r3 rT r. r r r � test_reflected_binary_methods� s z7TestNDArrayOperatorsMixin.test_reflected_binary_methodsc C sP t t�ddddg��}ttj�|t�ddg��d k� t|t ddddg�� d S )Nr8 rJ r � r rI ���)r r r6 r �negativer r2 )r r6 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 r2 r6 )r Zmantissa�exponentr3 r r r �test_ufunc_two_outputs� s z0TestNDArrayOperatorsMixin.test_ufunc_two_outputsN) r% r' r( r7 r: rC rF rH rQ rU rV rZ r] r r r r r4 f s r4 �__main__)N).� __future__r r r r* rK �sys�numpyr Z numpy.testingr r r r r � version_info�major�PY2�lib�mixins�NDArrayOperatorsMixinr r- r2 �lt�le�eq�ne�gt�ge�add�sub�mul�truediv�floordiv�mod�divmod�pow�lshift�rshift�and_�xor�or_rR r4 r% r r r r �<module> s> ) s