U
|
P ±d|B ã @ s d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZ dd ZG d d
|
d
|
ejZeZG d d d eZG d d deeZ e Z G dd dZ G dd deZ G dd deZ G dd dZ!G dd dZ"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 úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/tests/test_subclassing.pyÚassert_startswith s r c s@ e Zd Zi fddZ fddZ fddZ fddZ 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| f|}||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 efddZedd 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|tjs | ¡ }| 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 fddZ fd d
|
Zed d Z e j
|
d d Z d fdd Z Z S )ÚComplicatedSubArrayc C s d| t¡ dS )Nz myprefix z
|
mypostfix)r r rF r r r Ú__str__n s zComplicatedSubArray.__str__c C s d| j j d| dS )Nú<ú ú>)r( r2 rF r r r Ú__repr__q s z ComplicatedSubArray.__repr__c C s t |tstd|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|tjs&| ¡ 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 k r<|d tjkr<|j dd¡d |jd< |S )Nr Z
|
multipliedr* )r" Ú__array_wrap__r Úmultiplyr 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 rd r r r rE s zWrappedArray.__init__c C s | j j d| j d| j dS )Nz(
|
Ú
|
z
|
))r( r2 rc rd rF r r r rX £ s zWrappedArray.__repr__c C s t | j¡S r )r r rc rF r r r rJ ¦ s zWrappedArray.__array__c s8 |dkr0 fdd|D } j |||f jS tS d S )NÚ__call__c s" g | ]}t | jr|jn|qS r )rI r( rc )Ú.0ÚargrF r r Ú
|
<listcomp>« s ÿz0WrappedArray.__array_ufunc__.<locals>.<listcomp>)r( rd ÚNotImplemented)r% ZufuncÚmethodÚinputsr8 r rF r Ú__array_ufunc__© s
|
ÿz WrappedArray.__array_ufunc__N) r2 r3 r4 r9 Z__array_priority__rE rX rJ rm r r r r ra s ra 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)Zdtyper r* ©r< )r r Ú msubarrayr; ©r% r Úmxr r r Ú setup_methodµ s z TestSubclassing.setup_methodc C sZ t d¡}dddddg}t|}t||d}tt|t t|j| tt|jt d S )Nro r r* rq ) 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= Úsubarrayrs r r r Ú test_maskedarray_subclassingÄ s
|
z,TestSubclassing.test_maskedarray_subclassingc C sL | j \}}tjdd, ttt|t tt|t |¡ W 5 Q R X d S ©NÚignore)r )r; r Úerrstater rI r rr r rs 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 rr r r= ry Úouterr rs 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 )Nrq ) r; r
|
rJ r< r rI r rr r )r% r rt Zxmxr r r Ú test_masked_binary_operations2Ý s
|
|
z.TestSubclassing.test_masked_binary_operations2c C s t tddgdgd 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 | dddddg¡ t |jdddddg |j dddddg¡ t |jdddddg t|dd id
|
}t|}tt|d t |j |j d S ) Nro r r* é rq r) r0 Únamer )r r )r r r
|
ry rr r rI r r: r= r r r Z _set_maskÚ_maskr> Úhasattr)r% r ZmyZymÚzrw Úmxsubr r r Útest_attributepropagationå s4 z)TestSubclassing.test_attributepropagationc C s t d¡}dddddg}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 ) Nro r r* c S s g | ]\}}||fqS r r )rg ÚiÚjr r r ri
|
s z=TestSubclassing.test_subclasspreservation.<locals>.<listcomp>rw )r< r F)ZsubokT) r r Úzipr: r
|
r rI r r r r r r )r% r rv Zxinforw r r r r Útest_subclasspreservation s(
|
z)TestSubclassing.test_subclasspreservationc C sÊ t d¡}t|}t|dddddgd}tt|d t tt|d t tt|dd
t tt|d t tt|d jt t|d tk tt|d jt tt|dd
jt tt|jd jt t|jd tk 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 baseclassro TFrq 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 )Nro 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 sh t d¡}t|dddddgd}tt|d t|}t|dddddgd}tt|dtj d dS ) zOtest that repr uses the name of the subclass
|
and 'array' for np.ndarrayro TFrq r
|
Zmasked_z(data=[--, 1, --, 3, 4]N)r r r
|
r Úreprr r2 )r% r rt rw r r r r Útest_subclass_reprQ s
|
ÿz"TestSubclassing.test_subclass_reprc C s| t d¡}t|}t|dddddgd}tt|d t|}tt|j dt j
|
j j t|dddddgd}tt|d dS ) z7test str with subclass that has overridden str, setitemro TFrq z [-- 1 -- 3 4]r z myprefix [-- 1 -- 3 4] mypostfixN) r r r r
|
r ÚstrrS r rY rP ÚmaÚcoreZmasked_print_option)r% r rw r r r r r r Útest_subclass_str\ s
|
|
ÿz!TestSubclassing.test_subclass_strc C s t dddddddgd}t d dddddgd}t ||¡}td
|
|jk t|jd
|
dk || }td
|
|jk t|jd
|
dk d S ) NÚtestr* é é r ro é )r; r r )r6 r Úsubtractr r7 )r% Zarr1Zarr2Zdiff1Zdiff2r r r Ú$test_pure_subclass_info_preservationj s z4TestSubclassing.test_pure_subclass_info_preservationN)r2 r3 r4 ru rx rz r~ r r r r r r r r r r r r r rn ² s
|
!% rn 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 jjdddgdddgd} t| d}t j |¡}t| j|j t| j|j dddg| _t| j|j t|j t jj|dd}t| j|j t| j|j dddg| _tdddg|j t|j t jj|dd }t| j|j t| jdddg t|j t|j d S )
|
Nr* r r TFrq Zmeters)r )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 s2 t jjdddgdddgd}t|}||f| _d S )Nr* r ro FTrq )r r r
|
ra r; ©r% rv Úwmr r r ru ¤ s z TestClassWrapping.setup_methodc C s: | j \}}tjdd ttt |¡t W 5 Q R X d S r{ )r; r r} r rI r ra 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 ra r r r Ústack)r% rv r© Úm2r r r r ¯ s
|
"z/TestClassWrapping.test_masked_binary_operationsN)r2 r3 r4 ru r~ r r r r r r§ ¡ s r§ )%r9 Únumpyr 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 ry r6 r: rr r@ rS ra rn r¡ r¦ r§ r r r r Ú<module> s& 8 + F
|
|
