ó šÙÊKc@s7ddlZddlZddlmZddljjZddlmZej ddƒej Z dfd„ƒYZ e dkr3d Z d e Zd d f\ZZx‰ed d ƒD]xZdeZejeeƒjeed ƒZejeƒZdeddGHedeƒGHdeded fGHq´WndS(iÿÿÿÿN(tfloat_(t build_err_msgtalltignoret moduletestercBs‰eZd„Zdded„Zdd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „ZRS(cCs~||_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j |_ t |_ |j |_ |j |_ |j |_ |j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_y|j|_Wn tk rp|jj|_nXg|_dS(N(tmoduletallequaltarangetarrayt concatenatetcounttequaltfilledtgetmaskt getmaskarraytidtinnert make_masktmaskedt masked_arrayt masked_valuestmask_ortnomasktonestoutertrepeattresizetsortttaket transposetzerostMaskTypetumathtAttributeErrortcoret testnames(tselfR((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyt__init__s@                            tcCs|j|ƒ}|j|ƒ}|j|j|ƒ|j|ƒƒ} |j|j|d| ƒ|ƒ}|j|j|d| ƒ|ƒ}|jjdkr÷|jtƒ}t|t j ƒrÜ|j dkrÜd|t j |ƒtk rt||g|d|ddƒ} t| ƒ‚nXd S(s^Asserts that a comparison relation between two masked arrays is satisfied elementwise.tmasktOiis (shapes %s, %s mismatch)theadertnamestxtyidgY@s (mismatch %s%%)N(((R+R,(R+R,(R+R,(R RR RtdtypetchartastypeRt isinstancetnptndarraytsizetisnantshapeRtAssertionErrorRtbooltravelRttolistR tlent ValueError(R$t comparisonR+R,terr_msgR)t fill_valuetxftyftmtcondtmsgtvaltreducedtmatch((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pytassert_array_compare2s^$!!! ! 0       !   !cCs&|j|j||d|ddƒdS(s5Checks the elementwise equality of two masked arrays.R=R)sArrays are not equalN(RGR (R$R+R,R=((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pytassert_array_equalhsc Cs‚tjddddtddddddddg ƒ}d d d d d d d d d d d d g }|j|d |ƒ}|d d S( sTests creationgð?gÀg@g@g@g$Àg$@g@iiR'N(R1RtpiR(R$R+RAtxm((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_0ms7*c Cs?tjddddtddddddddg ƒ}tjdd 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 }|j|d|ƒ}|j|d|ƒ}tjdd ddgƒ}|j|dd d d d gƒ} tj|d|ƒ} |jdƒ||jd ƒjƒskt‚|j } |j t d„| ƒks•t‚|j |ƒt |ƒt d„|ƒksÆt‚xrddgD]d} | |_ | |_ | |_ | |_ | | _ |j |ƒt |ƒt d„|ƒksÓt‚qÓWdS(sTests creationgð?gÀg@g@g@g$Àg$@g@ggð¿gÀiiR'gà¿gà?gš™™™™™é?g@Œµx¯DcSs||S(N((R+R,((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyt…scSs||S(N((R+R,((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyRL†siiiicSs||S(N((R+R,((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyRLsN(ii(ii(R1RRIRtwheretset_fill_valueR tanyR6R5R3treduceR R:( R$R+R,ta10tm1tm2RJtymtztzmR?ts((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_1ts,73**!  !1     c Cs‹tjddddgƒ}|j|dddddgƒ}|j|dddddgƒ}|j|ƒ}t|ƒt|ƒ}}t|dƒt|dƒks«t‚|d|dksÅt‚d|d!!&      ++  c CsY|jdƒ}|j|d<|j|d ƒ}|j|j||gƒ|ƒsUt‚|j|j|ƒddddddddgƒsŽt‚|j|d ddƒ}|j|ddddddddgƒ|j|dddƒ}|j||ƒsþt‚|jd ddƒ}|j||ƒs+t‚|jddƒ}|j||ƒsUt‚dS( sTests resize/repeatiiiiitaxisiN(i(iiii(iiii( RRRRR R6R RRH(R$Rbty4ty5ty6ty7ty8((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_3Ås '9(cCsÝ|jdƒ}tjdƒ}|j|dd+|jdddƒ}|jdddƒ}|jtj|d ƒ|j|d ƒƒsˆt‚|jtj|ddƒ|j|ddƒƒs¾t‚|jtj|j |dƒ|j |dƒƒ|j||ƒƒst‚|jtj |j |dƒ|j |dƒƒ|j ||ƒƒsNt‚|j d dd ddgt ƒ}|j|d<|j|dddgƒ}|dd ks­t‚|ddksÃt‚|ddksÙt‚d S(s.Test of take, transpose, inner, outer productsiiiiiiiitabctdefN(iii(iii(iii(iii( RR1RtreshapeRRR6RRR RRR^(R$R+R,tt((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_4Ös"06--! cCsë|jdƒ}|jdƒ}|jdƒ}|j|d<|d7}|j||dƒs`t‚|d7}|j||dƒs†t‚|jdƒ}|jdƒ}|j|d<|d8}|j||dƒs×t‚|d8}|j||dƒsýt‚|jdƒd}|jdƒd}|j|d<|d9}|j||dƒsVt‚|d9}|j||dƒs|t‚|jdƒd}|jdƒd}|j|d<|d:}|j||ƒsÑt‚|d:}|j||ƒsót‚|jdƒd}|jdƒd}|j|d<|d:}|j||dƒsLt‚||jdƒ:}|j||jdƒƒ|jdƒjtƒ}|jdƒ}|j|d<|j|j ƒƒ}|d7}|j||dƒsçt‚dS(sTests inplace w/ scalari iigð?g@N(i ( RRRR6RHRR/RRtraw_data(R$R+R,RJtid1((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_5êsT                cCs|jddtƒ}|jdƒ}|jddtƒ}|j|d<|j}|jddtƒ}|j|d<||7}||7}|j|||ƒs¡t‚|j|||ƒs½t‚|j|j|j||jƒƒsçt‚|jddtƒ}|jddtƒ}|j|d<|j}|jddtƒ}|j|d<||8}||8}|j|||ƒsyt‚|j|||ƒs•t‚|j|j|j||jƒƒs¿t‚|jddtƒ}|jddtƒ}|j|d<|j}|jddtƒ}|j|d<||9}||9}|j|||ƒsQt‚|j|||ƒsmt‚|j|j|j||jƒƒs—t‚|jddtƒ}|jddtƒ}|j|d<|j}|jddtƒ}|j|d<||:}||:}dS(sTests inplace w/ arrayi R-iiÿÿÿÿN(RRRR'RR6R(R$R+R,RJRAta((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_6sT     *     *     *    cCsC|jdddtdgddddgdgdƒ|jdddtdgddddgdgdƒf}xÈdd d d d gD]±}yt|j|ƒ}Wn tk rÈtt|ƒ}nXt|j|ƒ}||j }||Œ}||Œ}|j|j dƒ|j dƒ|ƒ|j|j |j ƒqŠWd S(s Tests ufuncgð?iiÿÿÿÿiR'iitsqrttlogtlog10texpt conjugateN( RRItgetattrR R!t fromnumericRtninRHR t_mask(R$tdtftuftmftargsturtmr((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_7Qs7=    %c Cs|jddddgdddddgƒ}|jd|j|ddƒƒ|jd|j|d ddddgƒƒ|j|d ddddgd dƒ\}}|jd|ƒ|d ksÆt‚|j|(|j|ddƒ|jksôt‚|jddddgdddddgƒ}|jd d ƒ}|j|dd…df<|j|j|ddƒddgƒ|j|ddƒd|jks™t‚|jddg|j|ddƒƒ|j|ddd dƒ\}}|j|ddgƒddddddg}ddddddgddddddgg}|jd ƒ}|j|j|ddƒdƒ|j|j|ddd |ƒdƒ|j|jd ƒd|jd ƒgƒ}|j|j|dƒtj j tjd ƒƒddƒ|j|j|ddƒtjd ƒddƒ|j|j|ddƒ|j|ddƒ|j|ddƒdgƒ|j|j|dd |ƒddƒ|j|j|ddd |ƒddddd dgƒ|j|j|ddƒ|j|ddƒ|j|ddƒdgƒ|j d ƒ}ddddddg} ddddddgddddddgg} |j d ƒ} ddddddg} |j|j|j ||ƒddƒdƒ|j|j|j || ƒddƒdƒ|j|j|j || ƒddƒdƒ|j|j|j|j || ƒddƒƒdƒ|j || ƒ} |j|j| dƒddƒ|j|j| ddƒddddd dgƒ|j|j| ddƒddgƒ|j|j| ddd |ƒddddd dgƒdS(Nggð?g@g@R'iiRjtweightstreturnedg@iig@g(@g4@g@g$@gÀX@g@g@(RRHtaverageR6RRsRtNoneR1taddRPRRRR (R$totttresulttwtstw1tw2R+R,RRRSRhtm4tm5RU((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_99usT-+- $-%(%!0%+90G&7G0+++4 1%cCsG|jdƒ}tjdƒ}|j|dd+|jdddƒ}dS(Niiiiii(RR1RRs(R$R+R,((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyttest_A£s(t__name__t __module__R%tTrueRGRHRKRXRiRpRuRxRzR‹R˜R™(((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyR s # 5    4   5 2 $ .t__main__sNfrom __main__ import moduletester import numpy tester = moduletester(module) simport np.ma.core as module i iistester.test_%i()s#%ii2t.smoduletester.test_%i.__doc__score_current : %.3f - %.3fi(ttimeittnumpyR1Rtnp.core.fromnumericR"Rtnp.testing.utilsRtseterrRIRRšt setup_baset setup_curtnrepeattnlooptrangetitfunctTimerRtcurRteval(((s=/usr/lib/python2.7/site-packages/numpy/ma/timer_comparison.pyts&   ÿŸ   "