B Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_ej`Z`dfdd ZaGd d d eZbGd d d eZcGdddeZdddZeefdkredS))divisionabsolute_importprint_function)reduceN)TestCaserun_module_suiteassert_suppress_warnings)assert_array_equal)KMaskType MaskedArrayabsoluteaddallallcloseallequalalltruearangearccosarcsinarctanarctan2arrayaveragechoose concatenate conjugatecoscoshcountdivideequalexpfilledgetmaskgreater greater_equalinner isMaskedArrayless less_equalloglog10 make_maskmasked masked_array masked_equalmasked_greatermasked_greater_equal masked_inside masked_lessmasked_less_equalmasked_not_equalmasked_outsidemasked_print_option masked_values masked_wheremaximumminimummultiplynomasknonzero not_equalonesouterproductputravelrepeatresizeshapesinsinhsometruesortsqrtsubtractsumtaketantanh transposewherezeroscCs,t||}|s(td|t|t|f|S)NzNot eq:%s %s ----%s)rprintstr)vwmsgresultr]M/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/tests/test_old_ma.pyeqs r_c@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7S)8TestMac Cstddddtddddddddg }tdd ddd d d dddd dg }d}d d d d d d d d d d d d g }d d d d d d d d d d d d g }t||d}t||d}tdd ddg}t|d d d d gd} t|d|} |j} |d||||||||| | | f |_dS)Ng?gg@g@g@g$g$@g@gggr)maskgg?g?g@xD)nprpirTrHZset_fill_valued) selfxya10m1m2xmymzzmxfsr]r]r^setUp%s&"   z TestMa.setUpc Cs|j\ }}}}}}}}} } } |t||t||t|| ||j| ||j|j||jtdd| |t |t |tdd||t || |t t |d| |t ||dS)NcSs||S)Nr])rgrhr]r]r^<z)TestMa.test_testBasic1d..cSs||S)Nr])rgrhr]r]r^rs=rtg@xD) re assertFalser( assertTrue assertEqualrHdtypesizerrlenr_r#) rfrgrhrirjrkrlrmrnrorprqr]r]r^test_testBasic1d4s"zTestMa.test_testBasic1dc CsxdD]}|j\ }}}}}}}} } } }||_||_||_||_|| _|t||t||t||||j|||jtdd||t|t |tdd||t || |t t |d| |t ||| qWdS)N)))cSs||S)Nr])rgrhr]r]r^rsPrtz)TestMa.test_testBasic2d..cSs||S)Nr])rgrhr]r]r^rsRrtg@xD) rerHrur(rvrwryrrrzr_r#rr) rfrqrgrhrirjrkrlrmrnrorpr]r]r^test_testBasic2dBs$  zTestMa.test_testBasic2dc Cs |j\ }}}}}}}}} } } tddgddgg} t| ddgddgg} |t| | | | |t| | | | |t| | | | xdD]} || }|| }|| }|| }| | } |t| | |t|||||t|||||t||||tjddd|t||||WdQRX|t|||||t|||||t||||tjddd|t||||WdQRX|t|||||t|||||t|||||t|||||t|d|d|tt|dt|d|t|||||tt ||t |||tt ||t |||tt ||t ||tjddd$|tt ||t ||WdQRXqWdS) Nrarrr|)) )r|r})rr~ignore)r invalidg@) rerr/rvr_reshapercerrstateabsrrNr=r )rfrgrhrirjrkrlrmrnrorprqa2da2dmr]r]r^test_testArithmeticXsF     "" zTestMa.test_testArithmeticcCsBtdg}tdg}|t||t|t||tdS)Nra)rcrrv isinstancer )rfZnamar]r]r^test_testMixedArithmetic~s  zTestMa.test_testMixedArithmeticc Cs*|j\ }}}}}}}}} } } |tt|t||tt|t||tt|t||tt|t||tt|t||tt |t |tj ddd`|tt t |t ||tt t |t ||ttt |t|WdQRX|tt|t||tt|t| |tt|t| |tt|t| |tt||t|||tt|t||tt||t|||tt||t|||tt||t|||tt||t|||tt||t|||tt||t|||tt|t||tt||ft||f|tt||ft||f|tt||ft||f|tt|||ft|||fdS)Nr)r r)rervr_rcrrrIrJrQrRrrMrr+r,r"rrrrr r!r@r)r%r*r&rr) rfrgrhrirjrkrlrmrnrorprqr]r]r^test_testUfuncs1s8("""zTestMa.test_testUfuncs1cCstddddgddddgd}|t|jjtjk|dt||dtd|tdtddgd| d }|t|jjtjkt t t|dtj |t|jjtjk|tdt|t t t|dtk|tdd gt|ddS) Ngg?g@g@rar)rbr})rrr)rrvrrxtypercZintprwr_rrrndarrayr$r>)rfottr]r]r^test_xtestCounts zTestMa.test_xtestCountc Csf|j\ }}}}}}}}} } } t|} t|} |tt| t| |tt| t | dS)N) rercrErvr_maxr;rminr<)rfrgrhrirjrkrlrmrnrorprqZxrZxmrr]r]r^test_testMinMaxs  zTestMa.test_testMinMaxc Cs|j\ }}}}}}}}} } } |ttj|t||ttj|t||tdttddd|tdttddd|ttj|ddt|dd|ttjt |dddt|dd|tt|dt|d|ttj |ddt |dd|tt |dt |d|ttj t |dddt |ddt | dkr|tt ||fdt ||fd|ttj|dt|d|tt|dt|d|tt |dt |ddS)Nr|r)axisra) rervr_rcrr accumulaterOrr#rCrzr) rfrgrhrirjrkrlrmrnrorprqr]r]r^test_testAddSumProds$"(""zTestMa.test_testAddSumProdcCs0tddddg}t|ddddgd}t|ddddgd}t|}t|t|ttt|t|ddtt|dt|dkt|d|dkt|dtktt|d|dtt|dd|ddtt|dd|ddtt|dd|ddd |d<d |d<tt||d |dd<d |dd<tt||t|d<tt||t|dd<tt||||dd<t|d<tt t |tddddgt ddddgddddg|dd<tt t |tddddgt ddddgddddg|dd<tt t |tddddgtt |tddddgt dd }t |d }tt||tt tdddddgt|jttd |jtdd ddgt}tdd ddgt}|d}|d}|t|t|t|t|||t|ddjdkdS)Nrarr|r}r)rb) fill_value cg?g@Zhello)r)rcrrXreprrr_rLrr.rr$r/rr9r rbrobjectrwrH)rfx1Zx2Zx3x4s1s2r]r]r^ test_testCIsZ    ""   zTestMa.test_testCIc Cst}|tjjjddddddg}t|}t|}|||kt|dd}|||k td}t ||d}||j |k |t ||j ||j |kt |dd}||j |j kt ||dd} || j |k|| dt kd | d<|| dt k || j |k |t | j dt |d |d} |t| j|d jktd } t | d<t| d } |tt| | g| |tt| ddddddddgt| d dd} |t| ddddddddgt| ddd}|t| |WdQRXdS)NzGsetting an item on a masked array which has a shared mask will not copyrra)copyr)rb)rbrrrg?r|))rrrr)rr})r filterrcrcoreMaskedArrayFutureWarningr-rvrr_datarrbr.r#rxrGr_rr$rF)rfsupnmrkm3rZy1Zy1aZy2Zy3rZy4Zy5Zy6r]r]r^test_testCopySizesF      $ zTestMa.test_testCopySizec Cs8t&}|tjjjdtd}dddddg}t|}t||d}| |dt k| |dt kdd g|ddg<| |j |k | |dt k| |dt k | t |ddd d d gt||d}| ddd gd d d g| t |d d d ddg| |dt k| |dt kWdQRXdS)NzGsetting an item on a masked array which has a shared mask will not copyrrra)rbr}r| (rd)r rrcrrrrr-rrvr.rbr_rD)rfrrerrrgr]r]r^ test_testPut0s(   zTestMa.test_testPutc Csp|j\ }}}}}}}}} } } ddddddddddddg } t| d} t|| | ttt|| dd| kdS)Nrar)r)rercr?rDrrrP)rfrgrhrirjrkrlrmrnrorprqrir]r]r^test_testMaPutIs  zTestMa.test_testMaPutc Cstd}|dd}d|jd<t|ddk|d|}tt|j|tt|jd|tt|t|jd||d |jd <td}t|d <tt |d t tk|d k}tt t |ttd ktt t |tt|j kt ||t}t|j |j kt|d tkt|dtkt|dtkt|d tk t|dtk tt||t |t|}t|j |j kt|d tkt|dtk t|dtk t|d tkt|dtkt||}t|j |j kt|d tkt|dtk t|dtk t|d tkt|dtktt||tdddddg}tdddd d g}t|d<t ||| }tt|ddd ddgt|d <t ||| }tt|ddd ddgt|d tkt|dtk t|dtktttt|d|t|dtttt|d|t|dtttt|d|t|dtttt|d|t|dtttt|d|t|dtttt|d|t|dtttt|d|t|dtttttddd d ddddgtttttddd dddd dgttttttddd d d d gddd j ddddd gttttttdd dd d d gddd j ddd d dgttttttddd d d d gddj dd dd d gttttdddddgdd d d d gddj dd dd dgtttddd d d gddd ddgddd ddgt!dt"j#d}t$|j t%}t||}tt||t&|| |f}tt|ddd ddgt|d tkt|dtk t|dtktd}t|d<tdd}t|d<tdddd d d gdd d d d d gd}|'d}t |||}t |||} tt|| tt(| t)ktt| d dddd d!gt |td}tt|ddddddgt |dt}tt|ddddddgdS)"Nr|rr)rary$@reg.r}rrrg?g@g@g@g@rarg)rbr)rrr)rxr~r2)*rrZflatrr_realimagrr.rXrrTrHrxr:rr%r1r&r2r)r4r*r5r@r6r!r0r3listranger7rbrArcfloat32rUr rr#r$r>) rfrgrncZatestZbtestZctestrhcmror]r]r^test_testOddFeaturesPs              ((    $     zTestMa.test_testOddFeaturescCstttdddgdddgdddgtttdddgdddgdddgtd}tdd}t|d<t|d<ttt||tt||||ttt||tt||||tt |dktt |dkdS)Nrarr}r|rrr) rr_r<r;rr.rTr)r%r)rfrgrhr]r]r^test_testMinMax2s&& ""zTestMa.test_testMinMax2cCs&td}td}t|dd<|ddd}|ddd}ttt|dt|dttt|ddt|ddtttt |d t |d t||ttt t |d t |d t ||t d dd ddgt }t|d<t|d ddg}t|d d kt|ddkt|ddkdS) Nrr~rr}r|)rrrararabcdef) rrcr.rrr_rSrPr'r#rBrr)rfrgrhtr]r]r^ test_testTakeTransposeInnerOuters"   z'TestMa.test_testTakeTransposeInnerOutercCstd}td}td}t|d<|d7}tt||d|d7}tt||dtd}td}t|d<|d8}tt||d|d8}tt||dtdd}tdd}t|d<|d9}tt||d|d9}tt||dtdd}td}t|d<|d}tt|||d}tt||tdd}tdd}t|d<|d}tt||d|td}tt|tdtdtj}td}t|d<|d7}tt||ddS)Nrrrag?g@)r)rr.rr_rAZastypercr)rfrhrgrlr]r]r^test_testInplacesR      zTestMa.test_testInplacecCsPddl}td}t|ddd<|dd}||}||}tt||dS)Nrrr|rrr})picklerr.rdumpsloadsrr_)rfrrgrqrhr]r]r^test_testPickles   zTestMa.test_testPicklecCsNtd}t|d<|ttdk||dtk|t|ddddS)Nr~raz--r)rr.rvrXrwr#)rfZxxr]r]r^test_testMaskeds zTestMa.test_testMaskedc Csbtddddgddddgd}|tdt|dd|tdt|ddddgd t|ddddgdd \}}|td|||d kt|dd<|t|ddtktddddgddddgd}|d d }t|dddf<|tt|ddddg|t|dddtk|tddgt|ddt|ddd \}}|t|ddgdS)Ngg?g@g@rar)rb)r)weights)rreturnedg@r)rr)rrvr_rr.r)rfrr\Zwtsr]r]r^test_testAverage1s    zTestMa.test_testAverage1c Csdddddddg}ddddddgddddddgg}td}|tt|ddd|tt|d|ddttddtdg}|tt|dtjtddd |tt|ddtddd|tt|ddt|ddt|dddg|tt|d|d d |tt|d|dd d ddddg|tt|ddt|ddt|dddgtd}ddddddg}ddddddgddddddgg}t d}ddddddg} |ttt ||ddd|ttt ||ddd|tt ||ddt k| tt || ddd | t tt ||dddt ||} |tt| dd |tt| ddd d ddddg|tt| ddddg|tt| d|dd d ddddgtd} tdd} t| | g| | ggddd\} }| t| t|| | j|jtt ddddgdd\}}| t|t|tt ddd\}}| t|t|tt dt ddd\}}|t|t|ktddgddggt}t |ddgddgg}t|dd}|t|ddgt|dd}|t|d dgt|dd}|t|dt|dd}|t|ddgdS)Nrrar~)rg@)rrg@g@g(@)rg @gg?g@g$@gX@g@g@r})rr)rrr})rrr)r)rrrr|g?g@g?)rrvrrrrcrrrUrAr/r.rwrrHfloatr_)rfZw1Zw2rgrhrjrkrZm4Zm5rnabZr1Zr2rrZa2daZa2dmar]r]r^test_testAverage2-sl &         zTestMa.test_testAverage2cCs|dttd|dttd|dttdggg|dttdgg|tttddg|tttddg|tttddgddgddS)Nrag?r)rb)rwintrrZ assertRaises TypeError ValueErrorbool)rfr]r]r^test_testToPythongszTestMa.test_testToPythonc Cstddd}tjdd|dtdjWdQRX|d|j|| j|| j|t||j|t||j||j|j jktddd}|||j k| t |t t dS)Nrra)rbr)r ) rrcrrvrbr;r<r#rxrrwrXr8)rfrlrgr]r]r^test_testScalarArithmeticps  z TestMa.test_testScalarArithmeticc Cstdddg}|t||j|t||j|t||j|t||j|t|ddddd|jddddd|t| dddg|j dddg|t| |j |t| |j tddgddgg}|t| |j |t| |j |t|ddg|jddg|t||jdS)Nrar}rrr|)rrvr_anyrrZargmaxZargminrcompressZconjrdiagonalrOrPrS)rfrrr]r]r^test_testArrayMethodss*&zTestMa.test_testArrayMethodscCs tdddg}||jddS)Nrar}r)rrwndim)rfrr]r]r^test_testArrayAttributesszTestMa.test_testArrayAttributescCs|ddttjDdS)NcSs&g|]}|ttkr|ds|qS)_)dirr startswith).0rr]r]r^ s z'TestMa.test_testAPI..)rurrcr)rfr]r]r^ test_testAPIszTestMa.test_testAPIcCs`tdddg}tdddgdddgd}||djd||djd||djddS)Nrar}rr)rbr])rrwrH)rfrrr]r]r^test_testSingleElementSubscripts z&TestMa.test_testSingleElementSubscriptN)__name__ __module__ __qualname__rrr{rrrrrrrrrrrrrrrrrrrrrrrrrr]r]r]r^r`#s6& 2-e 2 : r`c@s4eZdZddZddZddZddZd d Zd S) TestUfuncscCsZtdddtdgdddgdgddtdddtdgdddgdgddf|_dS)Ng?rrrrar~)rb)rrdre)rfr]r]r^rrs(zTestUfuncs.setUpc Csdddddddddd d d d g }xd D]}ytt|}Wntk rTtt|}YnXttj|}|jd|j}t>||krtj dd|dkrtj dd||}||}WdQRX| t | d| d|| t |j|jq$WdS)NrMarctanhrrarccoshr+r,r true_divide floor_divide remainderfmod)*rMr+r,r"rrIrrQrrrrJrrRZarcsinhrrr ZfabsnegativeZfloorZceilZ logical_notrrNr=r rrrrZhypotrr!r@r*r&r)r%Z logical_andZ logical_orZ logical_xorr)r)rr+r,)r r)getattrumathAttributeError fromnumericrcrreZninrZseterrrvr_r#eqmaskrb)rfZf_invalid_ignorefZufZmfargsZurZmrr]r]r^test_testUfuncRegressions&       z#TestUfuncs.test_testUfuncRegressioncCsb|jd}|t|dd|t|dd|t|ddddd|t|ddddS)Nr)rr})rerurrvrKrwrOrC)rfrr]r]r^ test_reduces  zTestUfuncs.test_reducecCstdddd}t|dk|}|||||d||d|dk||ddddd gk||ddj||ddjdS) Nra r}r|rrr~rr) rrr:rwrrrvrrb)rfrZamaskr]r]r^ test_minmaxs zTestUfuncs.test_minmaxcCsDx>dD]6}tddddgddddgd}|tt|dgqWdS)Nz?bhilqpBHILQPfdgFDGOrarr)rb)rrvr_r?)rfrrgr]r]r^ test_nonzeros zTestUfuncs.test_nonzeroN)rrrrrrrrrr]r]r]r^rs % rc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)TestArrayMethodsc&Cs tddddddddd d d d d ddddddddddddddddddd d!d"d#d$g$}|d%d%}|d&d'd'd&}td(d)d(d)d(d(d)d(d)d)d(d)d(d(d(d)d(d)d(d(d(d)d)d)d)d(d(d)d(d(d(d(d)d(d)d(g$}t||d*}t|||jd*}t|||jd*}|||||||f|_dS)+Ng @gGz.@g-!@g!@g&1?gʡE@g\( @gQ@g{G#@gPn@gh|?!@g!rh@gjt@gJ +@gzGa@g!rh@gʡE@g-@g)\(@g(\B#@g +@guV @g~j@g"~@gK7@g ףp=J#@gGz@gMb@g}?5^#@g/$@g!rhm@g-#@gJ +@gMb@gʡ@g-?r~r}rrra)datarb)rcrrrHre)rfrgXXXrmxmXmXXr]r]r^rrs$          zTestArrayMethods.setUpc Csj|j\}}}}}}}|}||||t||t|j|dddS)Nr)r) rerrwZtrace compressedrOrvr_rb) rfrgrrrrrrZmXdiagr]r]r^ test_traces  zTestArrayMethods.test_tracec Csn|j\}}}}}}}|dd}|t|j|j|t|j|dd|t|j|jdddS)Nrr)reZcliprvr_rbr) rfrgrrrrrrZclippedr]r]r^ test_clips  zTestArrayMethods.test_clipc Cs|j\}}}}}}}|j\}}|||t|tj} t|tj} x,t|D] } |dd| f| | <q\Wx$t|D]} || | | <qW| t |d| | t |d| dS)Nrra) rerHrwZptpr rcrUZfloat_rrvr_) rfrgrrrrrrrZrowsZcolskr]r]r^test_ptp s  zTestArrayMethods.test_ptpc Cs^|j\}}}}}}}|dd}|t|d|dddf|dd} || jddS)Nrrarr)rrr}r})reZswapaxesrvr_rwrH) rfrgrrrrrrZ mXswappedZ mXXswappedr]r]r^ test_swapaxess    zTestArrayMethods.test_swapaxesc Csh|j\}}}}}}}|d}|t|j|dd|d}|t|j|dddS)Nrra)reZcumprodrvr_rr#) rfrgrrrrrrmXcpr]r]r^ test_cumprods   zTestArrayMethods.test_cumprodc Csh|j\}}}}}}}|d}|t|j|dd|d}|t|j|dddS)Nrra)reZcumsumrvr_rr#) rfrgrrrrrrrr]r]r^ test_cumsum$s   zTestArrayMethods.test_cumsumc Cs:|j\}}}}}}}|t|jdd||t|jdd||t|jddj|jddj|t|j|j|jdd|jdd}} xtdD]z} |t| | || |t|| |dd| f|tt || |dd| fqWdS)N)rr}rrar~) rervr_varr ZstdrHrrcrM) rfrgrrrrrrZmXvar0ZmXvar1r r]r]r^ test_varstd+s  $ (zTestArrayMethods.test_varstdN) rrrrrr r r rrrrr]r]r]r^rs rcCs,|tkr|tkS|tkr |tkS||kS)N)r>r)rjrkr]r]r^r9s r__main__)gZ __future__rrr functoolsrZnumpyrcZnumpy.core.umathrrZnumpy.core.fromnumericrZ numpy.testingrrrr Znumpy.ma.testutilsr Znumpy.mar r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrdr_r`rrrrr]r]r]r^s*  5 AW