Edit File: test_subclassing.cpython-37.pyc
B ��Fd�6 � @ s d Z ddlmZmZmZ ddlZddlmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZ G dd� dej�Z e Z G d d � d e�Z G d d � d e e�Z e Z G d d� deej!�Z"e"Z#G dd� de$�Z%G dd� de �Z&G dd� de�Z'e(dk�re � dS )z�Tests suite for MaskedArray & subclassing. :author: Pierre Gerard-Marchant :contact: pierregm_at_uga_dot_edu :version: $Id: test_subclassing.py 3473 2007-10-29 15:18:13Z jarrod.millman $ � )�division�absolute_import�print_functionN)�TestCase�run_module_suite� assert_raises�dec)� assert_equal) �array�arange�masked� MaskedArray� masked_array�log�add�hypot�divide�asarray� asanyarray�nomaskc s@ e Zd Zi fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �SubArrayc C s t �|��| �}|�� |_|S )N)�npr �view�copy�info)�cls�arrr �x� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/tests/test_subclassing.py�__new__ s zSubArray.__new__c s< t ttt| �dd ��r&tt| ��|� t|di ��� | _d S )N�__array_finalize__r )�callable�getattr�superr r! r r )�self�obj)� __class__r r r! s zSubArray.__array_finalize__c s, t t| ��|�}|j�dd�d |jd< |S )N�addedr � )r$ r �__add__r �get)r% �other�result)r' r r r* $ s zSubArray.__add__c s, t t| ��|�}|j�dd�d |jd< |S )N�iaddedr r) )r$ r �__iadd__r r+ )r% r, r- )r' r r r/ ) s zSubArray.__iadd__)�__name__� __module__� __qualname__r r! r* r/ � __classcell__r r )r' r r s r c s" e Zd ZdZd� fdd� Z� ZS )�SubMaskedArrayz<Pure subclass of MaskedArray, keeping some info on subclass.Nc s" t t| �j| f|�}||jd< |S )Nr )r$ r4 r �_optinfo)r r �kwargsr& )r' r r r 4 s zSubMaskedArray.__new__)N)r0 r1 r2 �__doc__r r3 r r )r' r r4 2 s r4 c @ s, e Zd Zi efdd�Zdd� Zeed�ZdS )� MSubArrayc C s& t ||�}tj| ||d�}|j|_|S )N)�data�mask)r r r r )r r9 r r: Zsubarr�_datar r r r < s zMSubArray.__new__c C s | � t�}d|_|S )NF)r r � _sharedmask)r% �_viewr r r � _get_seriesB s zMSubArray._get_series)�fgetN)r0 r1 r2 r r r> �property�_seriesr r r r r8 : s r8 c @ s2 e Zd Zefdd�Zdd� Zdd� Zeed�ZdS ) �MMatrixc C s t �|�}tj| ||d�}|S )N)r9 r: )r �matrixr r )r r9 r: �matr; r r r r M s zMMatrix.__new__c C s t j�| |� t�| |� d S )N)r rC r! r )r% r& r r r r! R s zMMatrix.__array_finalize__c C s | � t�}d|_|S )NF)r r r< )r% r= r r r r> W s zMMatrix._get_series)r? N) r0 r1 r2 r r r! r> r@ rA r r r r rB K s rB c @ s< e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � ZeZ d S ) � CSAIteratorz� Flat iterator object that uses its own setter/getter (works around ndarray.flat not propagating subclass setters/getters see https://github.com/numpy/numpy/issues/4564) roughly following MaskedIterator c C s || _ |�tj�j| _d S )N)� _originalr r �ndarray�flat� _dataiter)r% �ar r r �__init__l s zCSAIterator.__init__c C s | S )Nr )r% r r r �__iter__p s zCSAIterator.__iter__c C s4 | j �|�}t|tj�s |�� }|�t| j��}|S )N) rI � __getitem__� isinstancer rG � __array__r �typerF )r% �indx�outr r r rM s s zCSAIterator.__getitem__c C s | j �|�| j|< d S )N)rF �_validate_inputrI )r% �index�valuer r r � __setitem__z s zCSAIterator.__setitem__c C s t | j��� �t| j��S )N)�nextrI rO r rP rF )r% r r r �__next__} s zCSAIterator.__next__N) r0 r1 r2 r7 rK rL rM rV rX rW r r r r rE e s rE c sh e Zd Zdd� Zdd� Zdd� Z� fdd�Z� fd d �Zed d � �Z e j d d � �Z d� fdd� Z � Z S )�ComplicatedSubArrayc C s d� | �t��S )Nzmyprefix {0} mypostfix)�formatr r )r% r r r �__str__� s zComplicatedSubArray.__str__c C s d� | jj| �S )Nz <{0} {1}>)rZ r' r0 )r% r r r �__repr__� s z ComplicatedSubArray.__repr__c C s t |t�std��|S )Nz!Can only set to MySubArray values)rN rY � ValueError)r% rU r r r rS � s z#ComplicatedSubArray._validate_inputc s t t| ��|| �|�� d S )N)r$ rY rV rS )r% �itemrU )r' r r rV � s z ComplicatedSubArray.__setitem__c s. t t| ��|�}t|tj�s*|�� �t�}|S )N)r$ rY rM rN r rG rO r )r% r^ rU )r' r r rM � s z ComplicatedSubArray.__getitem__c C s t | �S )N)rE )r% r r r rH � s zComplicatedSubArray.flatc C s | � � }||d d �< d S )N)�ravel)r% rU �yr r r rH � s Nc sD t t| ��||�}|d k r@|d tjkr@|j�dd�d |jd< |S )Nr Z multipliedr) )r$ rY �__array_wrap__r �multiplyr r+ )r% r& �context)r' r r ra � s z"ComplicatedSubArray.__array_wrap__)N) r0 r1 r2 r[ r\ rS rV rM r@ rH �setterra r3 r r )r' r rY � s rY c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) �TestSubclassingc C s2 t jddd�}t|dddddgd�}||f| _d S )N� �float)�dtyper r) )r: )r r �mmatrixr9 )r% r �mxr r r �setUp� s zTestSubclassing.setUpc C s^ t �d�}dddddg}t|�}t||d�}| �t|t�� t|j|� | �t|jt�� d S )Nrf r r) )r: ) r r r r � assertTruerN r r r; )r% r �m�xsubZxmsubr r r �test_data_subclassing� s z%TestSubclassing.test_data_subclassingc C s" | j \}}| �t|jtj�� d S )N)r9 rl rN r; r rC )r% r rj r r r � test_maskedarray_subclassing� s z,TestSubclassing.test_maskedarray_subclassingc C sN | j \}}tjdd��. | �tt|�t�� tt|�t�|�� W d Q R X d S )N�ignore)r )r9 r �errstaterl rN r ri r )r% r rj r r r � test_masked_unary_operations� s z,TestSubclassing.test_masked_unary_operationsc C s� | j \}}| �tt||�t�� | �tt||�t�� tt||�|| � | �tt||�jtj�� | �tt� ||�t�� | �tt ||�t�� | �tt ||�t�� d S )N) r9 rl rN r ri r r; r rC �outerr )r% r rj r r r � test_masked_binary_operations� s z-TestSubclassing.test_masked_binary_operationsc C sd | j \}}t|j �� |jd�}| �tt||�t�� | �tt||�t�� tt||�t||�� d S )N)r: ) r9 r rO r: rl rN r ri r )r% r rj Zxmxr r r � test_masked_binary_operations2� s z.TestSubclassing.test_masked_binary_operations2c C s� t td�dgdgd d�}tt|��}t|�}|d }| �t|t�� | �t|t� � | �t|j t �� t |j j i � |d }| �t|t�� | �t|t�� | �t|j t �� | �|j j d dk� |d7 }| �t|t�� | �t|t�� | �t|j t �� | �|j j d dk� |� dddddg� t |jdddddg� |j� dddddg� t |jdddddg� t|dd id �}t|�}| �t|d �� t |j |j � d S ) Nrf r r) � )r: r( r. �namer )r r )r r r �subarray� msubarrayrl rN r r8 r; r r r � _set_mask�_maskrA �hasattr)r% r �my�ym�zrn �mxsubr r r �test_attributepropagation� s4 z)TestSubclassing.test_attributepropagationc C s t �d�}dddddg}dd� t||�D �}t||d|id�}t|dd �}| �t|t� � | �t|t�� t|j |� t |�}| �t|t� � | �t|t�� t|j |� t|d d �}| �t|t�� t|j |j � t|j |j � t |�}| �t|t�� t|j |j � t|j |� d S ) Nrf r r) c S s g | ]\}}||f�qS r r )�.0�i�jr r r � <listcomp> s z=TestSubclassing.test_subclasspreservation.<locals>.<listcomp>rn )r: r F)�subokT) r r �zipr8 r rl rN r r r| r r r )r% r rm Zxinforn r� r r r �test_subclasspreservation s( z)TestSubclassing.test_subclasspreservationc C s� t �d�}t|�}t|dddddgd�}| �t|d t�� | �t|d t�� | �t|dd� t�� | �t|d t�� | �t|d jt�� | �|d tk� | �t|d jt�� | �t|dd� jt�� | �t|jd jt�� | �|jd tk� | �t|jdd� j t�� t t |j d|d � t t |j d|d � t t |j t dd�|dd� � |d |d< |dd� |dd�< t t |jj d|d � t t |jj t dd�|dd� � |d |jd< |dd� |jdd�< d S ) z,test that getter and setter go via baseclassrf TF)r: r) )r) .rw r )r .N)r r rY r rl rN r9 r rH �baser r] rV �slice)r% r �xcsub�mxcsubr r r �test_subclass_items s. z#TestSubclassing.test_subclass_itemsc C sr t �d�}t|�}t|�}| �t|d jt�� | �t|d jt�� | �t|d t�� | �t|d t�� d S )Nrf )r) .)r .r) r )r r rY r rl rN r9 )r% r r� Z mxcsub_nomaskr r r �test_subclass_nomask_itemsC s z*TestSubclassing.test_subclass_nomask_itemsc C sr t �d�}t|dddddgd�}| �t|��d�� t|�}t|dddddgd�}| �t|��d�tj��� dS )zOtest that repr uses the name of the subclass and 'array' for np.ndarrayrf TF)r: r z masked_{0}(data = [-- 1 -- 3 4]N) r r r rl �repr� startswithr rZ r0 )r% r rj rn r� r r r �test_subclass_reprN s z"TestSubclassing.test_subclass_reprc C s� t �d�}t|�}t|dddddgd�}| �t|�dk� t|�}tt|j dt j j j � t|dddddgd�}| �t|�dk� dS ) z7test str with subclass that has overridden str, setitemrf TF)r: z [-- 1 -- 3 4]r z myprefix [-- 1 -- 3 4] mypostfixN) r r r r rl �strrY r r] rV �ma�core�masked_print_option)r% r rn r� r� r� r r r �test_subclass_strY s z!TestSubclassing.test_subclass_strc C s� t dddddddgd�}t d dddddgd�}t�||�}| �d |jk� | �|jd dk� || }| �d |jk� | �|jd dk� d S ) N�testr) � � rw rf � )r9 r r )r4 r �subtractrl r5 )r% �arr1�arr2Zdiff1Zdiff2r r r �$test_pure_subclass_info_preservationg s z4TestSubclassing.test_pure_subclass_info_preservationN)r0 r1 r2 rk ro rp rs ru rv r� r� r� r� r� r� r� r r r r re � s !% re �__main__))r7 � __future__r r r �numpyr Z numpy.testingr r r r Znumpy.ma.testutilsr Z numpy.ma.corer r r r r r r r r r r r rG r ry r4 r8 rz rC rB ri �objectrE rY re r0 r r r r �<module> s$ 8 , G