Current File : //lib64/python3.9/site-packages/numpy/ma/tests/__pycache__/test_subclassing.cpython-39.pyc
a
z[yc�@ � @ s d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZ dd� ZG d d
� d
ej�ZeZG d d � d e�ZG d
d� dee�Z e Z G dd� d�Z G dd� de�Z G dd� de�Z G dd� d�Z!G dd� d�Z"dd� 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)�NDArrayOperatorsMixin)�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 �E/usr/lib64/python3.9/site-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)�clsZarrr �xr r r �__new__ s
zSubArray.__new__c s" t � �|� t|di ��� | _d S �Nr )�super�__array_finalize__�getattrr r )�self�obj�� __class__r r r# s zSubArray.__array_finalize__c s( t � �|�}|j�dd�d |jd< |S )N�addedr � )r" �__add__r �get�r% �other�resultr' r r r+ $ s zSubArray.__add__c s( t � �|�}|j�dd�d |jd< |S )N�iaddedr r* )r" �__iadd__r r, r- r' r r r1 ) s zSubArray.__iadd__)�__name__�
__module__� __qualname__r r# r+ r1 �
__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 r! )r" r �_optinfo)r r �kwargsr& r' r r r 4 s
zSubMaskedArray.__new__)N)r2 r3 r4 �__doc__r r5 r r r' r r6 2 s r6 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< Zsubarr�_datar r r r < s
zMSubArray.__new__c C s | � t�}d|_|S )NF)r r Z _sharedmask)r% Z_viewr r r �_seriesB s
zMSubArray._seriesN)r2 r3 r4 r r �propertyr> 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 ) rD � __getitem__�
isinstancer rB � __array__r �typerA )r% Zindx�outr r r rH ^ s
zCSAIterator.__getitem__c C s | j �|�| j|< d S r )rA �_validate_inputrD )r% �index�valuer r r � __setitem__e s zCSAIterator.__setitem__c C s t | j��� �t| j��S r )�nextrD rJ r rK rA rF r r r �__next__h s zCSAIterator.__next__N) r2 r3 r4 r9 rE rG rH rP rR r r r r r@ P s r@ 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 rF r r r �__str__n s zComplicatedSubArray.__str__c C s d| j j� d| � d�S )N�<� �>)r( r2 rF r r r �__repr__q s z ComplicatedSubArray.__repr__c C s t |t�std��|S )Nz!Can only set to MySubArray values)rI rS �
ValueError)r% rO r r r rM u s
z#ComplicatedSubArray._validate_inputc s t � �|| �|�� d S r )r" rP rM �r% �itemrO r' r r rP z s z ComplicatedSubArray.__setitem__c s* t � �|�}t|tj�s&|�� �t�}|S r )r" rH rI r rB rJ r rS rZ r' r r rH s z ComplicatedSubArray.__getitem__c C s t | �S r )r@ rF r r r rC � s zComplicatedSubArray.flatc C s | � � }||d d �< d S r )Zravel)r% rO �yr r r rC � s Nc s@ t � �||�}|d ur<|d tju r<|j�dd�d |jd< |S )Nr Z
multipliedr* )r" �__array_wrap__r Zmultiplyr r, )r% r& �contextr' r r r] � s z"ComplicatedSubArray.__array_wrap__)N)
r2 r3 r4 rT rX rM rP rH r? rC �setterr] r5 r r r' r rS l s
rS c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) � WrappedArrayz�
Wrapping a MaskedArray rather than subclassing to test that
ufunc deferrals are commutative.
See: https://github.com/numpy/numpy/issues/15200)
� c K s || _ || _d S r )�_array�attrs)r% r rc r r r rE � s zWrappedArray.__init__c C s | j j� d| j� d| j� d�S )Nz(
�
z
))r( r2 rb rc rF r r r rX � s zWrappedArray.__repr__c C s t �| j�S r )r r rb rF r r r rJ � s zWrappedArray.__array__c s@ |dkr8� fdd�|D �}� j ||i |��fi � j��S tS d S )N�__call__c s" g | ]}t |� j�r|jn|�qS r )rI r( rb )�.0�argrF r r �
<listcomp>� s �z0WrappedArray.__array_ufunc__.<locals>.<listcomp>)r( rc �NotImplemented)r% Zufunc�methodZinputsr8 r rF r �__array_ufunc__� s
� z WrappedArray.__array_ufunc__N) r2 r3 r4 r9 Z__array_priority__rE rX rJ rk r r r r r` � s r` 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)Zdtype)r r* r r r �r< )r r � msubarrayr; �r% r �mxr r r � setup_method� s z TestSubclassing.setup_methodc C sT t �d�}g d�}t|�}t||d�}tt|t�� t|j|� tt|jt�� d S )Nrm �r r r* r r ro ) r r r r
r rI r r r= )r% r �m�xsubZxmsubr r r �test_data_subclassing� s
z%TestSubclassing.test_data_subclassingc C s | j \}}tt|jt�� d S r )r; r rI r= �subarrayrq r r r � test_maskedarray_subclassing� s
z,TestSubclassing.test_maskedarray_subclassingc C s` | j \}}tjdd��6 ttt|�t�� tt|�t�|�� W d � n1 sR0 Y d S �N�ignore)r )r; r �errstater rI r rp r rq 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 rI r rp r r= rx Zouterr
rq 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 )Nro ) r; r
rJ r< r rI r rp r )r% r rr Zxmxr 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 )Nrm r r* � ro r) r0 )r* r r r r* )r r r r r* �namer )r r )r r r
rx rp r rI r r: r= r r r Z _set_mask�_maskr> �hasattr)r% r ZmyZym�zrv �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 )
Nrm rt c S s g | ]\}}||f�qS r r )rf �i�jr r r rh
� z=TestSubclassing.test_subclasspreservation.<locals>.<listcomp>rv )r< r F)ZsubokT)
r r �zipr: r
r rI r r r� r r r )r% r ru Zxinforv 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 baseclassrm �TFTFFro r* �r* .r� r �r .N)r r rS r
r rI r; r rC �baser rY rP �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 )Nrm r� r� r* r )r r rS r
r rI r; )r% r r� Z
mxcsub_nomaskr r r �test_subclass_nomask_itemsF 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.ndarrayrm r� ro r
Zmasked_z(data=[--, 1, --, 3, 4]N)r r r
r �reprr r2 )r% r rr rv r� r r r �test_subclass_reprQ 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, setitemrm r� ro z
[-- 1 -- 3 4]r z myprefix [-- 1 -- 3 4] mypostfixN)
r r r r
r �strrS r rY rP �maZcoreZmasked_print_option)r% r rv r� r� r� r r r �test_subclass_str\ 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� rm � )r; )r r* r� r� r� rm r )r6 r �subtractr r7 )r% Zarr1Zarr2Zdiff1Zdiff2r r r �$test_pure_subclass_info_preservationj s z4TestSubclassing.test_pure_subclass_info_preservationN)r2 r3 r4 rs rw ry r} r~ r r� r� r� r� r� r� r� r r r r rl � s
!% rl c @ s e Zd ZdZdd� Zdd� ZdS )�ArrayNoInheritancez6Quantity-like class that does not inherit from ndarrayc C s || _ || _d S r )� magnitude�units)r% r; r� r r r rE y s zArrayNoInheritance.__init__c C s t | j|�S r )r$ r� )r% �attrr r r � __getattr__} s z ArrayNoInheritance.__getattr__N)r2 r3 r4 r9 rE r� r r r r r� w s r� c C s t jjg d�g d�d�} t| d�}t j�|�}t| j|j� t| j|j� g d�| _t| j|j� t|j� t jj|dd�}t| j|j� t| j|j� g d�| _tg d�|j� t|j � t jj|dd �}t| j|j� t| jg d�� t|j � t|j � d S )
N)r* r� r� )TFTro Zmeters)TFFT)r F)Z keep_mask) r r� r r� r r; r< r Z
sharedmask)Z data_maskedZdata_masked_unitsZ new_arrayr r r �test_array_no_inheritance� s&
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestClassWrappingc C s. t jjg d�g d�d�}t|�}||f| _d S )N)r* r� rm )FTFro )r r� r
r` r; �r% ru �wmr r r rs � s z TestClassWrapping.setup_methodc C sN | j \}}tjdd��$ ttt�|�t�� W d � n1 s@0 Y d S rz )r; r r| r rI r r` r� r r r r} � s
z.TestClassWrapping.test_masked_unary_operationsc C s4 | j \}}ttt�||�t�� ttt�||�t�� ttt�||�t�� tt�||�|| � ttt�||�t�� ttt�||�t�� ttt�||�t�� ttt�||�t�� tt�||�| t�||�| � t� ||g�}ttt�||�t�� ttt�||�t�� tt�||�t�||�� d S r )
r; r rI r r r` r r
r �stack)r% ru r� Zm2r r r r~ � s
"z/TestClassWrapping.test_masked_binary_operationsN)r2 r3 r4 rs r} r~ r r r r r� � s r� )%r9 Znumpyr Znumpy.lib.mixinsr Z
numpy.testingr r Znumpy.ma.testutilsr Z
numpy.ma.corer r r r r
r r r
r r r r r rB r rx r6 r: rp r@ rS r` rl r� r� r� r r r r �<module> s& 8 + F
Mini Shell