Ñò \ÐKc@s‚dZdZdZdZdZddkZddklZl Z ddk Tddk Tddk Td efd „ƒYZ d efd „ƒYZd efd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd „ƒYZed!jo e ƒndS("såTests suite for MaskedArray. Adapted from the original test_ma by Pierre Gerard-Marchant :author: Pierre Gerard-Marchant :contact: pierregm_at_uga_dot_edu :version: $Id: test_extras.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(tTestCasetrun_module_suite(t*t TestGenericcBs,eZd„Zd„Zd„Zd„ZRS(cCs4tddtƒ}tddgdddgdtƒ}t||ƒtihddgd6ddgd 6ƒ}tdd|ƒ}tddgdddgd|ƒ}t||ƒtdd|ƒ}tddgddggdddgddggd|ƒ}t||ƒtiddddgfgƒ}td d|ƒ}tdd!fdd"fgddd#fdd$fgd|ƒ}t||ƒtd%d|ƒ}tdd&fdd'fgddd(fdd)fgd|ƒ}t||ƒtd*d|ƒ}tdd+fggddd,fggd|ƒ}t||ƒd S(-sTests masked_allitdtypeitmasktatbtnamestftformatsitbatbbN(i(i(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(RR (R R (R R (i(ii(ii(ii(ii(i(ii(ii(ii(ii(ii(ii(ii(t masked_alltfloattarrayt assert_equaltnpR(tselfttesttcontroltdt((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_masked_alls0$ )$   !! ! 0cCsYtddgdtƒ}t|ƒ}tddgdddgdtƒ}t||ƒtihddgd6ddgd 6ƒ}tddgdddgd|ƒ}t|ƒ}tddgdddgd|ƒ}t||ƒtiddddgfgƒ}tddfddfgdddfddfgd|ƒ}t|ƒ}t||ƒdS(sTests masked_alliiRRRRRR R ii R R N(ii(ii(ii(ii(i i (i i (ii(ii(RR (sbaR (sbbR (ii(ii(ii(ii(RRtmasked_all_likeRRR(RtbaseRRR((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_masked_all_like6s $ )$ $ !! cCsxttidƒƒ}t|ddddddg0(3*) ) 0 ""("(+c Cstdƒ}tdƒd}t||g||ggddddƒ\}}tt|ƒt|ƒƒt|i|iƒttdƒdddddgddƒ\}}tt|ƒt|ƒƒttdƒddƒ\}}tt|ƒt|ƒƒttdƒdtdƒddƒ\}}tt|ƒt|ƒƒtddgdd ggtƒ}t|ttgt tggƒ}t|ddƒ} t| d d gƒt|ddƒ} t| d d gƒt|ddƒ} t| d d ƒt|ddƒ} t| ddgƒdS(sYet more tests of average!iiR%iR'iiR&igà?g@gð?g@gø?g@N(iii(iii(iii(iii( RR*RtshapeR6RRRR)R(R2( RRRtr1R8tr2R9ta2dta2dmta2data2dma((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_testAverage3”s* 00*!!cCsfttddgƒƒ}t|dƒttddddgdttttgƒƒ}t|dƒdS(s"Test average on integers with maskiigø?iiRN(R*RRR)R((RR((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_onintegers_with_mask¬s 0(R"R#t__doc__R0RBRJRK(((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyR$]s   " tTestConcatenatorcBs eZdZd„Zd„ZRS(s@ Tests for mr_, the equivalent of r_ for masked arrays. cCsðttd tddddddgƒƒtdƒ}dddddg}t|d|ƒ}t|dd|f}|it|tƒpt|tiƒƒt|ddddddddddddg ƒt|i t|dd|fƒd S( sTests mr_ on 1D arrays.iiiiiiiRN(iiiiii( tassert_array_equaltmr_RR6RR+t isinstancet MaskedArraytcoreR(RRtmtdtc((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_1d¹s) )1cCs”tddƒ}tddƒ}titddƒdƒ}titddƒdƒ}t|d|ƒ}t|d|ƒ}td||f}|i|idjƒt|dd…dd…f|ƒt|dd…dd…f|ƒt|iti d||fƒt||f}|i|idjƒt|dd…dd…f|ƒt|dd…dd…f|ƒt|iti ||fƒdS( sTests mr_ on 2D arrays.iiRt1i N(ii (i i( trandRtround_RROR+RCRNRtr_(Rta_1ta_2tm_1tm_2tb_1tb_2RT((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyttest_2dÄs ## ##(R"R#RLRVRa(((s@/usr/lib64/python2.6/site-packages/numpy/ma/tests/test_extras.pyRM´s t TestNotMaskedcBs eZdZd„Zd„ZRS(s9 Tests notmasked_edges and notmasked_contiguous. c CsJttidƒiddƒddddddgdddddgdddddgdddddgdddddggƒ}t|d ƒ}t|ddgƒt|dƒ}t|dd dgƒt|dddgƒt|dƒ}t|dddgƒt|dddgƒt|id ƒ}t|dd gƒt|idƒ}t|dddgƒt|dddgƒt|id ƒ}t|dddgƒt|dddgƒt|d s.    HW%As (A;