fE} d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZ d Z G d d ej ZeZ G d
d e Z G d d
ee Z e Z G d d Z G d de Z G d de Z G d d Z! G d d Z"d Z# G d d Z$dS )zTests 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) arrayarangemasked MaskedArray masked_arraylogaddhypotdivideasarray
asanyarraynomaskc P t | d t | | d S N)r len)abs s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_subclassing.pyassert_startswithr s& 7CFF7Q c < e Zd Zi fdZ fdZ fdZ fdZ xZS )SubArrayc t j | | }| |_ |S r )npr viewcopyinfo)clsarrr! xs r __new__zSubArray.__new__ s3 M# # #C ( (r c t | t |di | _ d S Nr! )super__array_finalize__getattrr r! )selfobj __class__s r r) zSubArray.__array_finalize__ s?
""3'''C,,1133 r c t | }|j dd dz |j d< |S )Naddedr )r( __add__r! getr+ otherresultr- s r r1 zSubArray.__add__$ s> '' %{w : :Q > G
r c t | }|j dd dz |j d< |S )Niaddedr r0 )r( __iadd__r! r2 r3 s r r8 zSubArray.__iadd__) s@ !!%(( & ! < >u E Eu r c t | j t | j S r )nextrW r_ r r` rT rZ s r __next__zCSAIterator.__next__h s6 DN##--//44T$.5I5IJJJr N) r9 r: r; rC rX r[ r] rg rj rP r r rR rR P sq 1 1 1 F F FK K K K Kr rR c | e Zd Zd Zd Zd Z fdZ fdZed Z e j
d Z d
fd Z xZ S ) ComplicatedSubArrayc > d| t dS )Nz myprefix z
mypostfix)r r rZ s r __str__zComplicatedSubArray.__str__n s :499X..::::r c ( d| j j d| dS )N< >)r- r9 rZ s r __repr__z ComplicatedSubArray.__repr__q s 44>*44T4444r c N t |t st d |S )Nz!Can only set to MySubArray values)r^ rl
ValueError)r+ rf s r rd z#ComplicatedSubArray._validate_inputu s* %!455 B @AA A r c r t || | d S r )r( rg rd r+ itemrf r- s r rg z ComplicatedSubArray.__setitem__z s3 D$"6"6u"="=>>>>>r c t | }t |t j s,| t }|S r )r( r] r^ r rU r_ r rl rw s r r] z ComplicatedSubArray.__getitem__ sR ##D))%,, @OO%%**+>??E r c t | S r )rR rZ s r rV zComplicatedSubArray.flat s 4 r c < | }||d d < d S r )ravel)r+ rf ys r rV zComplicatedSubArray.flat s JJLL!!!r Nc t || }|:|d t j u r&|j dd dz |j d< |S )Nr
multipliedr0 )r( __array_wrap__r multiplyr! r2 )r+ r, contextr- s r r z"ComplicatedSubArray.__array_wrap__ sX gg$$S'22 71:#<#<%(X\\,%B%BQ%FCH\ "
r r )
r9 r: r; rn rs rd rg r] rO rV setterr r< r= s @r rl rl l s ; ; ;5 5 5
? ? ? ? ?
! ! X!
[ [ r rl c 2 e Zd ZdZdZdZd Zd Zd Zd Z dS ) WrappedArrayz
Wrapping a MaskedArray rather than subclassing to test that
ufunc deferrals are commutative.
See: https://github.com/numpy/numpy/issues/15200)
_arrayattrs c " || _ || _ d S r r )r+ r r s r rX zWrappedArray.__init__ s
r c @ | j j d| j d| j dS )Nz(
z
))r- r9 r r rZ s r rs zWrappedArray.__repr__ s* .)LLdkLLTZLLLLr c 4 t j | j S r )r r r rZ s r r_ zWrappedArray.__array__ s z$+&&&r c l |dk r' fd|D } j ||i |fi j S t S )N__call__c L g | ] }t |j r|j n|!S rP )r^ r- r ).0argr+ s r
z0WrappedArray.__array_ufunc__.. s@ ) ) ) %/sDN$C$CLcjj ) ) )r )r- r NotImplemented)r+ ufuncmethodinputsrB s ` r __array_ufunc__z WrappedArray.__array_ufunc__ sg Z ) ) ) )!') ) )F!4>%%":6":":IIdjII I! !r N)
r9 r: r; rC __slots____array_priority__rX rs r_ r rP r r r r sj
$I M M M' ' '" " " " "r r c \ e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
ZdS )TestSubclassingc j t j dd }t |g d }||f| _ d S )N float)dtype)r r0 r r r rH )r r msubarrayrG r+ r$ mxs r setup_methodz TestSubclassing.setup_method s; Iaw ' ' '
q
/
/
/G r c 2 t j d }g d}t | }t || }t t |t t |j | t t |j t d S )Nr r r r0 r r r ) r r r r r r^ r
r rJ )r+ r$ mxsubxmsubs r test_data_subclassingz%TestSubclassing.test_data_subclassing s IaLL OO{{ T***
5+..///U[$'''
5;1122222r c h | j \ }}t t |j t d S r )rG r r^ rJ subarrayr s r test_maskedarray_subclassingz,TestSubclassing.test_maskedarray_subclassing s- )B
28X../////r c 2 | j \ }}t j d 5 t t t | t t t | t j | d d d d S # 1 swxY w Y d S Nignore)r )rG r errstater r^ r r r r s r test_masked_unary_operationsz,TestSubclassing.test_masked_unary_operations s )B
[
)
)
) , , Js2ww 22 3 3 3 Q + + + , , , , , , , , , , , , , , , , , ,s A B BBc | j \ }}t t t || t t t t || t t t || ||z t t t || j t t t t j || t t t t || t t t t || t d S r )
rG r r^ r
r r rJ r outerr r s r test_masked_binary_operationsz-TestSubclassing.test_masked_binary_operations s )B
3r2;; 22333
3r1::y11222SQZZA&&&
3r2;;,h77888
39R,,i88999
5R==)44555
5Q<<3344444r c | j \ }}t |j |j }t t t
|| t t t t
|| t t t
|| t
|| d S )Nr ) rG r r_ rH r r^ r r r )r+ r$ r xmxs r test_masked_binary_operations2z.TestSubclassing.test_masked_binary_operations2 s )B27,,..RW===
6"b>>955666
6"a==)44555VB^^VC%5%566666r c ~ t t d dgdgdz z }t t | }t | }|dz }t t
|t t t
|t t t
|j t t |j j i |dz }t t
|t t t
|t t t
|j t t |j j d dk |dz
}t t
|t t t
|t t t
|j t t |j j d dk |
g d t |j g d |j
g d t |j g d t |d
d i }t | }t t! |d
t |j |j d S )Nr r r0 r r/ r7 )r0 r r r r0 )r r r r r0 namer$ )r! r! )r r r r r r r^ r
rE rJ r r r! _set_mask_maskrN hasattr)r+ r$ myymzr mxsubs r test_attributepropagationz)TestSubclassing.test_attributepropagation s= &))1#qc!e) , , ,
(1++
&
&
q\\
T
1k**+++Jq),,,---
17H--...QW\2&&&
T
1k**+++
1i(()))
17H--... W%)***
a
2{++,,,
2y))***
28X..///
h'!+,,,
___%%%RX///
___---RX/// --- T""v&&'''UZ+++++r c t j d }g d}d t || D }t ||d|i }t |d }t t
|t t t
|t t |j | t | }t t
|t t t
|t t |j | t |d }t t
|t t |j |j t |j |j t | }t t
|t t |j |j t |j | d S ) Nr r c g | ] \ }}||f
S rP rP )r ijs r r z=TestSubclassing.test_subclasspreservation.. s 000FQ!Q000r r )rH r! F)subokT)
r r ziprE r r r^ r
r r r r! r )r+ r$ r xinfor r s r test_subclasspreservationz)TestSubclassing.test_subclasspreservation s IaLL OO00c!Qii000& 888 T///Jui000111
5+..///U[!$$$
Jui000111
5+..///U[!$$$ T...
5),,---UZ+++U[$*---4
5),,---UZ+++U[!$$$$$r c 2 t j d }t | }t |g d }t t |d t t t |d t t t |dd t t t |d t t t |d j t t |d t u t t |d j t t t |dd j t t t |j d j t t |j d t u t t |j dd j t t t |j d|d t t |j d|d t t |j t dd |dd |d |d< |dd |dd< t t |j j d|d t t |j j t dd |dd |d |j d< |dd |j dd< d S )
z,test that getter and setter go via baseclassr TFTFFr r0 r0 .r r r .N)r r rl r r r^ rG r rV baser ru rg slice)r+ r$ xcsubmxcsubs r test_subclass_itemsz#TestSubclassing.test_subclass_items" s IaLL#A&& e*K*K*KLLL
58%899:::
5<)<==>>>
51:':;;<<<
6!9&9::;;;
6%=-/BCCDDDq V#$$$
6%=-/BCCDDD
6!A#;+-@AABBB
6;q>.0CDDEEE A&()))
6;qs+02EFFGGG j%"3Q!===j&"4a1>>>j&"4eAqkk1QqS6JJJ!Hq AaCjqs j&+"91adCCCj&+"95A;;!A#OOO q A 1: AaCr c t j d }t | }t | }t t |d j t t t |d j t t t |d t t t |d t d S )Nr r r r0 r )r r rl r r r^ rG )r+ r$ r
mxcsub_nomasks r test_subclass_nomask_itemsz*TestSubclassing.test_subclass_nomask_itemsG s IaLL#A&&$U++
=/46IJJKKK
=/46IJJKKK
=+-@AABBB
=+-@AABBBBBr c ( t j d }t |g d }t t | d t | }t |g d }t t | dt
j d dS )zOtest that repr uses the name of the subclass
and 'array' for np.ndarrayr r r r masked_z(data=[--, 1, --, 3, 4]N)r r r r reprr r9 )r+ r$ r r r s r test_subclass_reprz"TestSubclassing.test_subclass_reprR s
IaLL
!"C"C"C
D
D
D$r((N333{{ T(I(I(IJJJ$u++ @h' @ @ @ B B B B Br c t j d }t | }t |g d }t t | d t
| }t t |j dt j
j j t |g d }t t | d dS )z7test str with subclass that has overridden str, setitemr r r z
[-- 1 -- 3 4]r z myprefix [-- 1 -- 3 4] mypostfixN)
r r r r r strrl r ru rg macoremasked_print_option)r+ r$ r r r r s r test_subclass_strz!TestSubclassing.test_subclass_str] s
IaLL{{ T(I(I(IJJJSZZ 111#A&&j%"3Qej4 6 6 6 e*K*K*KLLLS[["DEEEEEr c T t dg d }t g d }t j || }t d|j v t |j d dk ||z
}t d|j v t |j d dk d S )Ntest)r0 r r )rG )r r0 r r r r r! )r? r subtractr rA )r+ arr1arr2diff1diff2s r $test_pure_subclass_info_preservationz4TestSubclassing.test_pure_subclass_info_preservationk s f===999 ===111 D$''%.()))v&&0111t %.()))v&&011111r N)r9 r: r; r r r r r r r r r r r r r rP r r r r s
3 3 30 0 0
, , , 5 5 57 7 7 , , ,B% % %6#& #& #&J C C C B B B F F F
2
2
2
2
2r r c e Zd ZdZd Zd ZdS )ArrayNoInheritancez6Quantity-like class that does not inherit from ndarrayc " || _ || _ d S r ) magnitudeunits)r+ rG r s r rX zArrayNoInheritance.__init__z s
r c , t | j | S r )r* r )r+ attrs r __getattr__z ArrayNoInheritance.__getattr__~ s t~t,,,r N)r9 r: r; rC rX r rP r r r r x s8 @@ - - - - -r r c t j g dg d } t | d }t j | }t | j |j t | j |j g d| _ t | j |j t |j t j |d }t | j |j t | j |j g d| _ t g d|j t |j t j |d }t | j |j t | j g d t |j t |j d S )
N)r0 r r )TFTr meters)TFFT)r F) keep_mask) r r r r r rG rH r
sharedmask) data_maskeddata_masked_units new_arrays r test_array_no_inheritancer s %++iii.A.A.A+BBK*;AA -..I!9>222!9>222+++K!9>222 I !!! -D 99I!9>222!9>222***K%%%y~666 $ $%%% - ??I!9>222!#6#6#6777 $ $%%%%%r c & e Zd Zd Zd Zd Zd ZdS )TestClassWrappingc t j g dg d }t | }||f| _ d S )Nr0 r r FTFr )r r r r rG r+ r wms r r z TestClassWrapping.setup_method sA E yyy/C/C/C D D
!__G r c | j \ }}t j d 5 t t t j | t d d d d S # 1 swxY w Y d S r )rG r r r r^ r r r s r r z.TestClassWrapping.test_masked_unary_operations s )B
[
)
)
) : : Jrvbzz<88 9 9 9 : : : : : : : : : : : : : : : : : :s 5A""A&)A&c 0 | j \ }}t t t j || t
t t t j || t
t t t j || t
t
t j || ||z t t t j || t
t t t j || t
t t t j || t
t t t j || t
t
t j || |z t j || |z t j ||g }t t t j || t
t t t j || t
t
t j || t j || d S r )
rG r r^ r r
r r r r stack)r+ r r m2s r r z/TestClassWrapping.test_masked_binary_operations s )B
26"b>><88999
26!R==,77888
26"a==,77888RVAr]]AF+++
28Ar??L99:::
28B??L99:::
29R++\::;;;
29Q++\::;;;RYr1%%)29Q??R+?@@@
Xq!f
29R,,l;;<<<
29R,,l;;<<<RYr2&& "b(9(9:::::r c t }t t |j dd t j g dg d }t | }t t |j dd d S )Nnot_a_real_attrr0 r r r not_an_attrr )r r AttributeError __setattr__r r r r )r+ mixinr r s r test_mixins_have_slotsz(TestClassWrapping.test_mixins_have_slots sq %''ne&79JANNN E yyy/C/C/C D D
!__nbnmQGGGGGr N)r9 r: r; r r r r rP r r r r sU
: : : ; ; ;*H H H H Hr r )%rC numpyr numpy.lib.mixinsr
numpy.testingr r numpy.ma.testutilsr
numpy.ma.corer r r r
r r r
r r r r r r rU r r r? rE r rR rl r r r r r rP r r r s 2 2 2 2 2 2 0 0 0 0 0 0 0 0 + + + + + + rz 0 [ +
K K K K K K K K8( ( ( ( (( ( ( (V" " " " "( " " "8B2 B2 B2 B2 B2 B2 B2 B2J- - - - - - - - & & &@*H *H *H *H *H *H *H *H *H *Hr