File: //lib/python3/dist-packages/numpy/ma/tests/__pycache__/test_subclassing.cpython-310.pyc
o
6��ac1 � @ s� d Z ddlZddlmZmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ dd� ZG dd � d ej�ZeZG d
d � d e �ZG d d
� d
ee �ZeZG dd� d�Z G dd� de�Z G dd� d�Z 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 $
� N)�assert_�
assert_raises)� assert_equal) �array�arange�masked� MaskedArray� masked_array�log�add�hypot�divide�asarray�
asanyarray�nomaskc C s t | d t|�� |� d S �N)r �len)�a�b� r �A/usr/lib/python3/dist-packages/numpy/ma/tests/test_subclassing.py�assert_startswith s r c s@ e Zd Zi fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �SubArrayc C s t �|��| �}|�� |_|S r )�npr �view�copy�info)�cls�arrr �xr r r �__new__ s
zSubArray.__new__c s4 t tt� dd ��rt� �|� t|di ��� | _d S )N�__array_finalize__r )�callable�getattr�superr! r r )�self�obj�� __class__r r r! s zSubArray.__array_finalize__c �( t � �|�}|j�dd�d |jd< |S )N�addedr � )r$ �__add__r �get�r% �other�resultr' r r r, $ � zSubArray.__add__c r) )N�iaddedr r+ )r$ �__iadd__r r- r. r' r r r3 ) r1 zSubArray.__iadd__)�__name__�
__module__� __qualname__r r! r, r3 �
__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 � j| fi |��}||jd< |S )Nr )r$ r �_optinfo)r r �kwargsr& r' r r r 4 s
zSubMaskedArray.__new__r )r4 r5 r6 �__doc__r r7 r r r' r r8 2 s r8 c @ s& e Zd Zi efdd�Zedd� �ZdS )� MSubArrayc C s& t ||�}tj| ||d�}|j|_|S )N)�data�mask)r r r r )r r= r r> �subarr�_datar r r r < s
zMSubArray.__new__c C s | � t�}d|_|S )NF)r r � _sharedmask)r% �_viewr r r �_seriesB s
zMSubArray._seriesN)r4 r5 r6 r r �propertyrC r r r r r< : s r< c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd 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 r )� _originalr r �ndarray�flat� _dataiter)r% r r r r �__init__W s zCSAIterator.__init__c C s | S r r �r% r r r �__iter__[ s zCSAIterator.__iter__c C s4 | j �|�}t|tj�s|�� }|�t| j��}|S r ) rI � __getitem__�
isinstancer rG � __array__r �typerF )r% �indx�outr r r rM ^ s
zCSAIterator.__getitem__c C s | j �|�| j|< d S r )rF �_validate_inputrI )r% �index�valuer r r � __setitem__e s zCSAIterator.__setitem__c C s t | j��� �t| j��S r )�nextrI rO r rP rF rK r r r �__next__h s zCSAIterator.__next__N) r4 r5 r6 r; rJ rL rM rV rX r r r r rE P 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�� d�S )Nz myprefix z
mypostfix)r r rK r r r �__str__n s zComplicatedSubArray.__str__c C s d| j j� d| � d�S )N�<� �>)r( r4 rK r r r �__repr__q s z ComplicatedSubArray.__repr__c C s t |t�s td��|S )Nz!Can only set to MySubArray values)rN rY �
ValueError)r% rU r r r rS u s
z#ComplicatedSubArray._validate_inputc s t � �|| �|�� d S r )r$ rV rS �r% �itemrU r' r r rV z s z ComplicatedSubArray.__setitem__c s* t � �|�}t|tj�s|�� �t�}|S r )r$ rM rN r rG rO r rY r` r' r r rM s z ComplicatedSubArray.__getitem__c C s t | �S r )rE rK r r r rH � s zComplicatedSubArray.flatc C s | � � }||d d �< d S r )�ravel)r% rU �yr r r rH � s Nc s@ t � �||�}|d ur |d tju r |j�dd�d |jd< |S )Nr �
multipliedr+ )r$ �__array_wrap__r �multiplyr r- )r% r& �contextr' r r re � s z"ComplicatedSubArray.__array_wrap__r )
r4 r5 r6 rZ r^ rS rV rM rD rH �setterre r7 r r r' r rY l 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 s, t jddd�}t|g d�d�}||f| _d S )N� �float)�dtype)r r+ r r r �r> )r r � msubarrayr= �r% r �mxr r r �setup� s zTestSubclassing.setupc C sT t �d�}g d�}t|�}t||d�}tt|t�� t|j|� tt|jt�� d S )Nrj �r r r+ r r rm ) r r r r r rN r r r@ )r% r �m�xsub�xmsubr r r �test_data_subclassing� s
z%TestSubclassing.test_data_subclassingc C s | j \}}tt|jt�� d S r )r= r rN r@ �subarrayro r r r � test_maskedarray_subclassing� s
z,TestSubclassing.test_maskedarray_subclassingc C sb | j \}}tjdd�� ttt|�t�� tt|�t�|�� W d � d S 1 s*w Y d S )N�ignore)r
)r= r �errstater rN r
rn r ro r r r � test_masked_unary_operations� s
"�z,TestSubclassing.test_masked_unary_operationsc C s� | j \}}ttt||�t�� ttt||�t�� tt||�|| � ttt||�jt�� ttt�||�t�� ttt ||�t�� ttt ||�t�� d S r )
r= r rN r rn r r@ rw �outerr ro r r r � test_masked_binary_operations� s
z-TestSubclassing.test_masked_binary_operationsc C s` | j \}}t|j �� |jd�}ttt||�t�� ttt||�t�� tt||�t||�� d S )Nrm ) r= r rO r> r rN r
rn r )r% r rp �xmxr r r � test_masked_binary_operations2� s
z.TestSubclassing.test_masked_binary_operations2c C sz t td�dgdgd d�}tt|��}t|�}|d }tt|t�� tt|t� � tt|j t
�� t |j j i � |d }tt|t�� tt|t�� tt|j t
�� t|j j d dk� |d7 }tt|t�� tt|t�� tt|j t
�� t|j j d dk� |�
g d�� t |jg d�� |j�
g d �� t |jg d �� t|d
d id �}t|�}tt|d
�� t |j |j � d S )Nrj r r+ � rm r* r2 )r+ r r r r+ )r r r r r+ �namer )r r )r r r rw rn r rN r r< r@ r r r � _set_mask�_maskrC �hasattr)r% r �my�ym�zrt �mxsubr r r �test_attributepropagation� s4 z)TestSubclassing.test_attributepropagationc C s
t �d�}g d�}dd� t||�D �}t||d|id�}t|dd�}tt|t� � tt|t�� t|j |� t
|�}tt|t� � tt|t�� t|j |� t|d d�}tt|t�� t|j |j � t|j |j � t |�}tt|t�� t|j |j � t|j |� d S )
Nrj rr c S s g | ]\}}||f�qS r r )�.0�i�jr r r �
<listcomp>� s z=TestSubclassing.test_subclasspreservation.<locals>.<listcomp>rt )r> r F)�subokT)
r r �zipr< r r rN r r r� r r r )r% r rs �xinfort r� r r r �test_subclasspreservation� s(
z)TestSubclassing.test_subclasspreservationc C s� t �d�}t|�}t|g d�d�}tt|d t�� tt|d t�� tt|dd� t�� tt|d t�� tt|d jt�� t|d tu � tt|d jt�� tt|dd� jt�� tt|jd jt�� t|jd tu � tt|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 baseclassrj �TFTFFrm r+ �r+ .r� r �r .N)r r rY r r rN r= r rH �baser r_ rV �slice)r% r �xcsub�mxcsubr r r �test_subclass_items s.
z#TestSubclassing.test_subclass_itemsc C sj t �d�}t|�}t|�}tt|d jt�� tt|d jt�� tt|d t�� tt|d t�� d S )Nrj r� r� r+ r )r r rY r r rN r= )r% r r� �
mxcsub_nomaskr r r �test_subclass_nomask_items+ s
z*TestSubclassing.test_subclass_nomask_itemsc C s\ t �d�}t|g d�d�}tt|�d� t|�}t|g d�d�}tt|�dtj� d�� dS )zOtest that repr uses the name of the subclass
and 'array' for np.ndarrayrj r� rm r �masked_z(data=[--, 1, --, 3, 4]N)r r r r �reprr r4 )r% r rp rt r� r r r �test_subclass_repr6 s
�z"TestSubclassing.test_subclass_reprc C sp t �d�}t|�}t|g d�d�}tt|�d� t|�}tt|j dt j
j j � t|g d�d�}tt|�d� dS )z7test str with subclass that has overridden str, setitemrj r� rm z
[-- 1 -- 3 4]r z myprefix [-- 1 -- 3 4] mypostfixN)
r r r r r �strrY r r_ rV �ma�core�masked_print_option)r% r rt r� r� r� r r r �test_subclass_strA s
�z!TestSubclassing.test_subclass_strc C sv t dg d�d�}t g d�d�}t�||�}td|jv � t|jd dk� || }td|jv � t|jd dk� d S )N�test)r+ � � r� rj � )r= )r r+ r� r� r� rj r )r8 r �subtractr r9 )r% �arr1�arr2�diff1�diff2r r r �$test_pure_subclass_info_preservationO s z4TestSubclassing.test_pure_subclass_info_preservationN)r4 r5 r6 rq rv rx r{ r} r r� r� r� r� r� r� r� r r r r ri � s
!% ri ) r; �numpyr �
numpy.testingr r �numpy.ma.testutilsr �
numpy.ma.corer r r r r r
r r r
r r r r rG r rw r8 r< rn rE rY ri r r r r �<module> s 8 +