�
�9Zc@`
s~
dZd
dl
mZmZmZ
d
dlZd
dlmZm	Z	m
Z

d
dlmZ
d
dl
mZmZmZmZmZmZmZmZmZmZmZmZ
dejf
d��YZeZd	ef
d
��YZdeefd��YZeZd
eej fd��YZ!e!Z"de#f
d��YZ$def
d��YZ%def
d��YZ&e'dkrz
e	�
ndS(s�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 $

i(tdivisiontabsolute_importtprint_functionN(tTestCasetrun_module_suitet
assert_raises(
tassert_equal(tarraytarangetmaskedtMaskedArraytmasked_arraytlogtaddthypottdividetasarrayt
asanyarraytnomasktSubArraycB`
s/eZ
id�
Zd
�Zd�Zd�ZRS(cC`
s+tj
|
�
j|�
}|j�|_|S(
N(tnpRtviewtcopytinfo(tclstarrRt
x((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__new__s


cC`
sYtt
tt|�d
d��
r:tt|�j|
�

nt
|
di�j�|_dS(Nt__array_finalize__R(tcallabletgetattrtsuperRtNoneRRR(tselftobj((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRs





cC`
s<tt
|�j|
�
}|jjd
d�d|jd
<|S(Ntaddedii
(RRt__add__Rtget(R!tothertresult((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR$$s

 
cC`
s<tt
|�j|
�
}|jjd
d�d|jd
<|S(Ntiaddedii
(RRt__iadd__RR%(R!R&R'((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR))s

 
(t__name__t
__module__RRR$R)(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRs		tSubMaskedArraycB`
seZ
dZdd
�
ZRS(s<Pure subclass of MaskedArray, keeping some info on subclass.cK`
s,tt
|�j||�
}|
|jd
<|S(NR(RR,Rt_optinfo(RRtkwargsR"((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR4s



N(R*R+t__doc__R R(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR,2s

t	MSubArraycB`
s/eZ
ied�Zd
�Zede�
ZRS(cC`
s:t|
|�}t
j|d
|d|�
}|j|_|S(Ntdatatmask(RR
RR(RR1RR2tsubarrt_data((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR<s



c
C`
s|jt
�
}
t|
_|
S(
N(RR
tFalset_sharedmask(R!t_view((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt_get_seriesBs

	
tfget(R*R+RRR8tpropertyt_series(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR0:s	tMMatrixcB`
s5eZ
ed�
Zd
�Zd�Zede�
ZRS(cC`
s.tj
|
�
}tj|d
|d|�
}|S(NR1R2(RtmatrixR
R(RR1R2tmatR4((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRMs


cC`
s'tj
j||
�
tj||
�
dS(
N(RR=RR
(R!R"((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRRs


c
C`
s|jt
�
}
t|
_|
S(
N(RR
R5R6(R!R7((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR8Ws

	
R9(R*R+RRRR8R:R;(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR<Ks		tCSAIteratorc
B`
sAeZ
dZd
�Zd�Zd�Zd�Zd�ZeZRS(s�
    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
    cC`
s%|
|_|
j
tj�
j|_dS(
N(t	_originalRRtndarraytflatt	_dataiter(R!t
a((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__init__ls
	
c


C`
s|S(
N((
R!((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__iter__ps
cC`
sO|jj
|
�
}t|tj�s3|j�}n|jt|j�
�
}|S(
N(	RCt__getitem__t
isinstanceRRAt	__array__RttypeR@(R!tindxtout((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRGss





cC`
s|jj
|�
|j|
<dS(
N(R@t_validate_inputRC(R!tindextvalue((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__setitem__zs
c

C`
s%t|j
�
j�jt|j�
�
S(
N(tnextRCRIRRJR@(
R!((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__next__}s
(	R*R+R/RERFRGRPRRRQ(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR?es
					tComplicatedSubArraycB`
sbeZ
d�Zd
�Zd�Zd�Zd�Zed��
Zej	d��
Zdd�
ZRS(	c

C`
sd
j|j
t�
�
S(Nsmyprefix {0} mypostfix(tformatRR(
R!((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__str__�s
c

C`
sd
j|j
j|�S(Ns	<{0} {1}>(RTt	__class__R*(
R!((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt__repr__�scC`
s"t|
t
�std
�
�
n|
S(Ns!Can only set to MySubArray values(RHRSt
ValueError(R!RO((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRM�s


cC`
s&tt
|�j|
|j|�
�
dS(
N(RRSRPRM(R!titemRO((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRP�s
cC`
sFtt
|�j|
�
}t|tj�sB|j�jt
�
}n|S(
N(RRSRGRHRRARIR(R!RYRO((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRG�s


c

C`
s
t|�
S(
N(
R?(
R!((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRB�scC`
s|j�}|
|(dS(
N(
travel(R!ROt
y((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRB�s
cC`
satt
|�j|
|�}
|dk	r]|d
tjkr]|
jjdd
�d|
jd<n|
S(Nit
multipliedi
(RRSt__array_wrap__R RtmultiplyRR%(R!R"tcontext((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyR]�s


#N(R*R+RURWRMRPRGR:RBtsetterR R](((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRS�s					tTestSubclassingc
B`
steZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
RS(c
C`
sCtj
d
�
}
t|
ddddddg�

}|
|f|_dS(NiR2ii
(RRtmmatrixR1(R!Rtmx((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pytsetUp�s

!
c
C`
s�tj
d
�
}
dddddg}t|
�
}t|d|�

}|jt|t��

t|j|�
|jt|jt��

dS(Niii
R2(	RRRRt
assertTrueRHR
RR4(R!Rt
mtxsubtxmsub((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_data_subclassing�s





c
C`
s/|j\}
}|j
t|jtj��

dS(
N(R1ReRHR4RR=(R!RRc((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_maskedarray_subclassing�s
c
C`
sd|j\}
}t
jd
d�
�=
|jtt|�
t��

tt|
�
t
j|
�
�
WdQXdS(NRtignore(R1RterrstateReRHRRbR(R!RRc((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_masked_unary_operations�s


c
C`
s�|j\}
}|j
tt||�t��

|j
tt||
�t��

tt||
�||
�
|j
tt||�jtj��

|j
ttj	||�t��

|j
tt
||�t��

|j
tt
||
�t��

dS(
N(R1ReRHR
RbRR4RR=touterR(R!RRc((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_masked_binary_operations�s

%
"

c
C`
s�|j\}
}t
|jj�d
|j�

}|jtt||�t��

|jtt||
�t��

tt||�t||��
dS(NR2(	R1RRIR2ReRHRRbR(R!RRctxmx((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_masked_binary_operations2�s




c
C`
sJtt
d
�
ddg
dg
d�

}
tt|
�
�
}t|
�
}|d}|jt|t��

|jt|t��

|jt|j	t
��

t|j	ji�
|d}|jt|t��

|jt|t��

|jt|j	t
��

|j|j	jddk�

|d7}|jt|t��

|jt|t��

|jt|j	t
��

|j|j	jddk�

|j
dddddg�

t|jdddddg�
|jj
dddddg�

t|jdddddg�
t|
di
d	d
6�

}t|�
}|jt|d��

t|j|j�
dS(NiR2ii
iR#R(RRtname(RRRtsubarrayt	msubarrayReRHR
R0R4RRRt	_set_maskt_maskR;thasattr(R!Rtmytymt
zRgtmxsub((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_attributepropagation�s4
&






















c
C`
s�
tj
d
�
}
dddddg}gt|
|�D]\}}||f^q4}t|
d|di
|d6�
}t|dt�

}|jt|t��

|jt|t��

t	|j
|�
t|�
}|jt|t��

|jt|t��

t	|j
|�
t|dt�

}|jt|t��

t	|j
|j
�
t	|j
|j
�
t|�
}|jt|t��

t	|j
|j
�
t	|j
|�
dS(Niii
R2RRgtsubok(RRtzipR0RR5ReRHR
RRvRtTrueRR(R!RRft
it
jtxinfoRgR{((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_subclasspreservation
s(

.












c
C`
stj
d
�
}
t|
�
}t|dtttttg�

}|jt|dt��

|jt|dd!t��

|jt|dt��

|j|dtk�

|jt|dd!j	t��

|jt|j
dj	t��

|j|dtk�

|jt|j
dd!jt��

tt
|jd|
d�
tt
|jd|
d�
tt
|jtdd�|
dd!�
|d|d<|dd!|dd+tt
|j
jd|
d�
tt
|j
jtdd�|
dd!�
|d|j
d<|dd!|j
dd+dS(s,test that getter and setter go via baseclassiR2i
iiN(RRRSRRR5ReRHR	R1RBtbaseRRXRPtslice(R!Rtxcsubtmxcsub((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_subclass_items
s(

!


  

#
&


)

c
C`
s�tj
d
�
}
t|
dtttttg�

}|jt|�
jd�
�

t|
�
}t|dtttttg�

}|jt|�
jdj	tj
�
�
�

dS(sOtest that repr uses the name of the subclass
        and 'array' for np.ndarrayiR2Rsmasked_{0}(data = [-- 1 -- 3 4]N(RRRRR5Retreprt
startswithRRTR*(R!RRcRgR{((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_subclass_repr>
s
!


!

c
C`
s�tj
d
�
}
t|
�
}t|dtttttg�

}|jt|�
dk�

t|
�
}t	t
|jdtjj
j�
t|dtttttg�

}|jt|�
dk�

dS(s7test str with subclass that has overridden str, setitemiR2s
[-- 1 -- 3 4]is myprefix [-- 1 -- 3 4] mypostfixN(RRRRRR5RetstrRSRRXRPtmatcoretmasked_print_option(R!RRgR{R�R�((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyttest_subclass_strI
s

!



!
c
	C`
s�td
dddddddg�

}
tdd	dddddg�
}t
j|
|�}|jd
|jk�

|j|jd
d
k�

|
|}|jd
|jk�

|j|jd
d
k�

dS(NttestR1i
iiiiiiR(R,RtsubtractReR-(R!tarr1tarr2tdiff1tdiff2((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt$test_pure_subclass_info_preservationW
s$
!






(R*R+RdRiRjRmRoRqR|R�R�R�R�R�(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyRa�s		
			
		!		 		t__main__((R/t
__future__RR
RtnumpyRt
numpy.testingRRRtnumpy.ma.testutilsRt
numpy.ma.coreRRR	R
RRR
RRRRRRARRsR,R0RtR=R<RbtobjectR?RSRaR*(((sR/opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_subclassing.pyt<module>s"



R,�