?????????? ????????? - ??????????????? - /usr/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
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????