σ ΨΖχXc@`sdZddlmZmZmZddlZddlZddlmZm Z ddl j j Z ddl mZmZmZmZmZmZmZmZddljjZddl mZmZmZmZmZmZdd d d d d ddddddddgZddddddgZ ee Z!e"ddd„Z#de"d „Z$d!d"„Z%d#„Z&d!d$„Z'd!d%„Z(e(Z)d&d!e"d'„Z*e*Z+d!e"d!e"d(„Z,d!e"d)„Z-d!e"d*„Z.dd!e"d+„Z/dd!e"d,„Z0d!e"d-„Z1d!d.„Z2dS(/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_functionN(tndarraytfloat_(tTestCasetassert_tassert_allclosetassert_array_almost_equal_nulpt assert_raisest build_err_msgtrun_module_suitetsuppress_warningsi(tmask_ortgetmaskt masked_arraytnomasktmaskedtfilledtalmosttapproxtassert_almost_equaltassert_array_almost_equaltassert_array_approx_equaltassert_array_comparetassert_array_equaltassert_array_lesst assert_closet assert_equaltassert_equal_recordstassert_mask_equaltassert_not_equaltfail_if_array_equalRRRRR R gρ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(s† Returns true if all components of a and b are equal 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(R RRtdtypetchartnptequaltravelRtFalsetastypeRt less_equaltumathtabsolute( tatbt fill_valuetrtoltatoltmtd1td2txtytd((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR*s   $**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. R!R"R#ig$@(R RRR$R%R&R'R(RR)R*Rtaroundtabs( R.R/tdecimalR0R3R4R5R6R7R8((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR@s  $***tcC`s_tt|ƒt|ƒ|ƒx<tt|ƒƒD](}t||||d||fƒq/WdS(s; Asserts the equality of two non-array sequences. s item=%r %sN(Rtlentrange(tactualtdesiredterr_msgtk((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyt_assert_equal_on_sequencesSs&cC`s“t|j|jƒxy|jjD]k}tj||ƒtj||ƒ}}|tk r |tk r ttj||ƒtj||ƒƒq q WdS(sI Asserts that two records are equal. Pretty crude for now. N(RR$tnamestoperatortgetitemR(R.R/tftaftbf((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR^s %,c C`sdt|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 ƒsnt ||g|ƒ}t ƒ2}|jtdƒ||ksdt|ƒ‚nWdQXdS|tkr†|tk sž|tkrΞ|tk rΞt ||g|ddddƒ}t|ƒ‚ntj|d td tƒ}tj|d td tƒ}|j|j}}|jd krT|jd krTt |jƒ|jƒddƒSt|||ƒS(s, Asserts that two items are equal. s %s not in %ss key=%r %sNRAR<s.*NAT ==theaderRDR6R7R"tsuboktS(R6R7(t isinstancetdicttAssertionErrortreprttypeRR=titemstlistttupleRCRR R tfiltert FutureWarningRt ValueErrorR&tarrayR)tTrueR$R%ttolistR( R?R@RARBtitmsgtsupt actual_dtypet desired_dtype((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRms> &*     cC`sͺt|tƒrΉt|tƒs9ttt|ƒƒƒ‚ntt|ƒt|ƒ|ƒx]|jƒD]O\}}||krtt|ƒƒ‚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(s< Raises an assertion error if two items are equal. s key=%r %sNs item=%r %s(RMRNRORPRQt fail_if_equalR=RRRSRTR>R&RR R (R?R@RARBR[R\((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR`–s& &*&$ ic C`s“t|tjƒs$t|tjƒrCt||d|d|d|ƒSt||gd|d|ƒ}tt||ƒ|ƒdkst|ƒ‚ndS(s~ Asserts that two items are almost equal. The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal). R;RAtverboseiN(RMR&RRR troundR:RO(R?R@R;RARaR\((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR³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( sn Asserts that comparison between two masked arrays is satisfied. The comparison is elementwise. R"R#t keep_maskRKRARaRJRDR6R7(R6R7( R RRR)RR RWtutilsRR( t comparisonR6R7RARaRJR0R3R\((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRΖs $$  c C`s)ttj||d|d|ddƒdS(s@ Checks the elementwise equality of two masked arrays. RARaRJsArrays are not equalN(RREt__eq__(R6R7RARa((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRίs c C`s/d„}t|||d|d|ddƒdS(sT Raises an assertion error if two masked arrays are not equal elementwise. cS`stjt||ƒƒ S(N(R&talltrueR(R6R7((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pytcompareξsRARaRJsArrays are not equalN(R(R6R7RARaRh((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyR ιs c `s5‡fd†}t|||d|d|ddƒdS(s| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. c`st||ddˆ ƒS(s<Returns the result of the loose comparison between x and y).R1g$@(R(R6R7(R;(s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRhϋsRARaRJsArrays are not almost equalN(R(R6R7R;RARaRh((R;s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRτsc `s5‡fd†}t|||d|d|ddƒdS(s| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. c`st||ˆƒS(s<Returns the result of the loose comparison between x and y).(R(R6R7(R;(s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRh sRARaRJsArrays are not almost equalN(R(R6R7R;RARaRh((R;s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRsc C`s)ttj||d|d|ddƒdS(s7 Checks that x is smaller than y elementwise. RARaRJsArrays are not less-orderedN(RREt__lt__(R6R7RARa((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRs cC`sU|tkrt|tkƒn|tkr>t|tkƒnt||d|ƒdS(s- Asserts the equality of two masks. RAN(RRR(tm1tm2RA((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pyRs   (3t__doc__t __future__RRRREtnumpyR&RRtnumpy.core.umathtcoreR,t numpy.testingRRRRR R R R tnumpy.testing.utilsttestingRdR RRRRRt __all__maskedt__some__from_testingt__all__RYRRRCRRR`RRRRRR RRRR(((s1/tmp/pip-build-X4mzal/numpy/numpy/ma/testutils.pytsB  :.        )