Ñò \ÐKc@sÚdZdZdZdZdZddkZddkTddkTddk Tdei fd „ƒYZ e Z d e e fd „ƒYZeZd e eifd „ƒYZeZdefd„ƒYZedjo 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 $ s5Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s1.0s$Revision: 3473 $s5$Date: 2007-10-29 17:18:13 +0200 (Mon, 29 Oct 2007) $iÿÿÿÿN(t*tSubArraycBs,eZdZhd„Zd„Zd„ZRS(s_Defines a generic np.ndarray subclass, that stores some metadata in the dictionary `info`.cCs%ti|ƒi|ƒ}||_|S(N(tnpt asanyarraytviewtinfo(tclstarrRtx((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyt__new__s cCst|dhƒ|_dS(NR(tgetattrR(tselftobj((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyt__array_finalize__scCsCtii||ƒ}|iih|iiddƒdd6ƒ|S(Ntaddedii(Rtndarrayt__add__Rtupdatetpop(R tothertresult((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyRs*(t__name__t __module__t__doc__R R R(((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyRs  t MSubArraycBs8eZhed„Zd„Zd„ZedeƒZRS(cCs:t||ƒ}ti|d|d|ƒ}|i|_|S(Ntdatatmask(Rt MaskedArrayR R(RRRRtsubarrt_data((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR $s cCs$ti||ƒti||ƒdS(N(RR R(R R ((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR )scCs|itƒ}t|_|S(N(RRtFalset _sharedmask(R t_view((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyt _get_series-s tfget(RRtnomaskR R R!tpropertyt_series(((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR#s  tMMatrixcBs5eZed„Zd„Zd„ZedeƒZRS(cCs.ti|ƒ}ti|d|d|ƒ}|S(NRR(RtmatrixRR (RRRtmatR((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR 6scCs'tii||ƒti||ƒdS(N(RR'R R(R R ((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR :scCs|itƒ}t|_|S(N(RRRR(R R ((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR!>s R"(RRR#R R R!R$R%(((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR&5s   tTestSubclassingcBsVeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS( s,Test suite for masked subclasses of ndarray.cCsCtidƒ}t|ddddddgƒ}||f|_dS(NiRii(RtarangetmmatrixR(R Rtmx((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pytsetUpIs!cCs…tidƒ}dddddg}t|ƒ}t|d|ƒ}|it|tƒƒt|i|ƒ|it|itƒƒdS(s#Tests whether the subclass is kept.iiiRN( RR*Rt masked_arrayt failUnlesst isinstanceRt assert_equalR(R Rtmtxsubtxmsub((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_data_subclassingNs cCs/|i\}}|it|itiƒƒdS(sTests subclassing MaskedArrayN(RR/R0RRR'(R RR,((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_maskedarray_subclassingXscCsK|i\}}|itt|ƒtƒƒtt|ƒti|ƒƒdS(sTests masked_unary_operationN(RR/R0tlogR+R1R(R RR,((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_masked_unary_operations]scCsð|i\}}|itt||ƒtƒƒ|itt||ƒtƒƒtt||ƒ||ƒ|itt||ƒitiƒƒ|itti ||ƒtƒƒ|itt ||ƒtƒƒ|itt ||ƒtƒƒdS(sTests masked_binary_operationN( RR/R0taddR+R1RRR'touterthypot(R RR,((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_masked_binary_operationscs%"cCsŽ|i\}}t|iiƒd|iƒ}|itt||ƒtƒƒ|itt||ƒtƒƒtt||ƒt||ƒƒdS(s&Tests domained_masked_binary_operationRN( RR.t __array__RR/R0tdivideR+R1(R RR,txmx((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR<ps cCsÞttdƒddgdgdƒ}tt|ƒƒ}t|ƒ}|d}|it|tƒƒ|it|tƒ ƒ|it|i t ƒƒt |i i hƒ|d}|it|tƒƒ|it|tƒƒ|it|i t ƒƒ|i|i i ddjƒ|i dddddgƒt |idddddgƒ|ii dddddgƒt |idddddgƒt|dhdd 6ƒ}t|ƒ}|it|dƒƒt |i |i ƒdS( NiRiiiRRRtname(tarrayR*R.tsubarrayt msubarrayR/R0RRRRR1Rt _set_maskt_maskR%thasattr(R RtmytymtzR3tmxsub((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_attributepropagationxs*&    c Cs¨tidƒ}dddddg}g}t||ƒD]\}}|||fq8~}t|d|dh|d6ƒ}t|dtƒ}|it|tƒ ƒ|it|tƒƒt |i |ƒt |ƒ}|it|tƒ ƒ|it|tƒƒt |i |ƒt|dt ƒ}|it|tƒƒt |i |i ƒt |i |i ƒt|ƒ}|it|tƒƒt |i |i ƒt |i |ƒdS( s=Checks that masked_array(...,subok=True) preserves the class.iiiRRR3tsubokN(RR*tzipRR.RR/R0RR1REtasarraytTrueRR( R RR2t_[1]titjtxinfoR3RJ((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyttest_subclasspreservation“s(6  ( RRRR-R5R6R8R<RKRT(((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyR)Fs     t__main__(Rt __author__t __version__t __revision__t__date__tnumpyRt numpy.testingtnumpy.ma.testutilst numpy.ma.coreRRRBRRRCR'R&R+tTestCaseR)Rtrun_module_suite(((sE/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_subclassing.pyts"    j