ó ¢áSc@`s²dZddlmZmZmZdZdZdZdZddl Z ddl Z ddl m Z m Z ddljjZdd lTddljjZd d lmZmZmZmZmZmZmZmZed d d„Zded„Zdd„Z d„Z!dd„Z"dd„Z#e#Z$dded„Z%e%Z&deded„Z'ded„Z(ded„Z)dded„Z*dded„Z+ded„Z,dd„Z-dS(sÌMiscellaneous functions for testing masked arrays and subclasses :author: Pierre Gerard-Marchant :contact: pierregm_at_uga_dot_edu :version: $Id: testutils.py 3529 2007-11-13 08:01:14Z jarrod.millman $ i(tdivisiontabsolute_importtprint_functions5Pierre GF Gerard-Marchant ($Author: jarrod.millman $)s1.0s$Revision: 3529 $s5$Date: 2007-11-13 10:01:14 +0200 (Tue, 13 Nov 2007) $N(tndarraytfloat_(t*i(tmask_ortgetmaskt masked_arraytnomasktmaskedtfilledtequaltlessgñh㈵øä>g:Œ0âŽyE>c C`sûtt|ƒt|ƒƒ}t|ƒ}t|ƒ}|jjdksW|jjdkrmtj||ƒjƒStt|dt d|ƒ|ƒj t ƒ}tt|dt d|ƒdƒj t ƒ} tj t j|| ƒ||t j| ƒƒ} | jƒS(ssReturns true if all components of a and b are equal subject to given tolerances. If fill_value is True, masked values considered equal. Otherwise, masked values are considered unequal. The relative error rtol should be positive and << 1.0 The absolute error atol comes into play for those elements of b that are very small or zero; it says how small a must be also. tOtcopytmaski(RRR tdtypetchartnpR travelRtFalsetastypeRt less_equaltumathtabsolute( tatbt fill_valuetrtoltatoltmtd1td2txtytd((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytapproxs   $**0ic C`sõtt|ƒt|ƒƒ}t|ƒ}t|ƒ}|jjdksW|jjdkrmtj||ƒjƒStt|dt d|ƒ|ƒj t ƒ}tt|dt d|ƒdƒj t ƒ}tj tj ||ƒ|ƒd| k} | jƒS(s¤Returns True if a and b are equal up to decimal places. If fill_value is True, masked values considered equal. Otherwise, masked values are considered unequal. RRRig$@(RRR RRRR RRRRRtaroundtabs( RRtdecimalRRR R!R"R#R$((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytalmost0s  $***tcC`s_tt|ƒt|ƒ|ƒx<tt|ƒƒD](}t||||d||fƒq/WdS(s0Asserts the equality of two non-array sequences.s item=%r %sN(t assert_equaltlentrange(tactualtdesiredterr_msgtk((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyt_assert_equal_on_sequencesAs&cC`s“t|j|jƒxy|jjD]k}tj||ƒtj||ƒ}}|tk r |tk r ttj||ƒtj||ƒƒq q WdS(s9Asserts that two records are equal. Pretty crude for now.N(R+RtnamestoperatortgetitemR (RRtftaftbf((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytassert_equal_recordsHs %,cC`sBt|tƒr½t|tƒs9ttt|ƒƒƒ‚ntt|ƒt|ƒ|ƒxa|jƒD]S\}}||kr“td||fƒ‚nt||||d||fƒqbWdSt|tt fƒrút|tt fƒrút ||ddƒSt|t ƒpt|t ƒsLt ||g|ƒ}||ksHt|ƒ‚ndS|t krd|t k s||t kr¬|t k r¬t ||g|dddd ƒ}t|ƒ‚ntj|d td tƒ}tj|d td tƒ}|j|j}}|jd kr2|jd kr2t |jƒ|jƒddƒSt|||ƒS(s!Asserts that two items are equal.s %s not in %ss key=%r %sNR0R*theaderR3R"R#RtsuboktS(R"R#(t isinstancetdicttAssertionErrortreprttypeR+R,titemstlistttupleR2Rt build_err_msgR t ValueErrorRtarrayRtTrueRRttolisttassert_array_equal(R.R/R0R1titmsgt actual_dtypet desired_dtype((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyR+Rs: &*     cC`sªt|tƒr¹t|tƒs9ttt|ƒƒƒ‚ntt|ƒt|ƒ|ƒx]|jƒD]O\}}||krtt|ƒƒ‚nt||||d||fƒqbWdSt|tt fƒrBt|tt fƒrBtt|ƒt|ƒ|ƒx<t t|ƒƒD](}t||||d||fƒqWdSt|t j ƒsft|t j ƒrvt |||ƒSt||g|ƒ}||ks¦t|ƒ‚ndS(s6Raises an assertion error if two items are equal. s key=%r %sNs item=%r %s(R=R>R?R@RAt fail_if_equalR,RBRCRDR-RRtfail_if_array_equalRE(R.R/R0R1RKRL((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyRO~s& &*&$ ic C`s“t|tjƒs$t|tjƒrCt||d|d|d|ƒSt||gd|d|ƒ}tt||ƒ|ƒdkst|ƒ‚ndS(svAsserts that two items are almost equal. The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal) R(R0tverboseiN(R=RRtassert_array_almost_equalREtroundR'R?(R.R/R(R0RQRL((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytassert_almost_equal˜s$ c C`stt|ƒt|ƒƒ}t|dtd|dtdtƒ}t|dtd|dtdtƒ}|tkr{|tk s“|tkrÌ|tk rÌt||gd|d|d|dd ƒ}t|ƒ‚ntj||j |ƒ|j |ƒd|d|d|ƒS( sZAsserts that a comparison relation between two masked arrays is satisfied elementwise.RRt keep_maskR;R0RQR:R3R"R#(R"R#( RRRRR RERFtutilstassert_array_compareR ( t comparisonR"R#R0RQR:RRRL((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyRW¨s$$  c C`s)ttj||d|d|ddƒdS(s5Checks the elementwise equality of two masked arrays.R0RQR:sArrays are not equalN(RWR4t__eq__(R"R#R0RQ((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyRJÀs c C`s/d„}t|||d|d|ddƒdS(sKRaises an assertion error if two masked arrays are not equal (elementwise).cS`stjt||ƒƒ S(N(RtalltrueR%(R"R#((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytcompareÉsR0RQR:sArrays are not equalN(RW(R"R#R0RQR[((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyRPÇs c `s5‡fd†}t|||d|d|ddƒdS(s[Checks the elementwise equality of two masked arrays, up to a given number of decimals.c`st||ddˆ ƒS(s<Returns the result of the loose comparison between x and y).Rg$@(R%(R"R#(R((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyR[ÒsR0RQR:sArrays are not almost equalN(RW(R"R#R(R0RQR[((R(sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytassert_array_approx_equalÏsc `s5‡fd†}t|||d|d|ddƒdS(s[Checks the elementwise equality of two masked arrays, up to a given number of decimals.c`st||ˆƒS(s<Returns the result of the loose comparison between x and y).(R)(R"R#(R((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyR[ÜsR0RQR:sArrays are not almost equalN(RW(R"R#R(R0RQR[((R(sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyRRÙsc C`s)ttj||d|d|ddƒdS(s,Checks that x is smaller than y elementwise.R0RQR:sArrays are not less-orderedN(RWR4t__lt__(R"R#R0RQ((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytassert_array_lessãs cC`sU|tkrt|tkƒn|tkr>t|tkƒnt||d|ƒdS(s"Asserts the equality of two masks.R0N(R tassert_RJ(tm1tm2R0((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pytassert_mask_equalês   (.t__doc__t __future__RRRt __author__t __version__t __revision__t__date__R4tnumpyRRRtnumpy.core.umathtcoreRt numpy.testingtnumpy.testing.utilsttestingRVRRRR R R R R RHR%R)R2R9R+ROtassert_not_equalRTt assert_closeRWRJRPR\RRR^Rb(((sL/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/numpy/ma/testutils.pyts:   :  ,