;>"^}(@s/dZddlmZmZmZddlZddlZddlmZm Z ddl j j Z ddl Zddl mZmZmZmZmZddl mZmZmZmZmZmZdd d d d d ddddddddgZddlmZdddddgZeeZddddd Zd dd!dZd"d#d$Z d%dZ!d"d&dZ"d"d'd(Z#e#Z$d)d"dd*d Z%e%Z&d"dd"dd+d Z'd"dd,dZ(d"dd-dZ)d d"dd.d Z*d d"dd/d Z+d"dd0dZ,d"d1dZ-dS)2zMiscellaneous 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 $ )divisionabsolute_importprint_functionN)ndarrayfloat_)assert_assert_allcloseassert_array_almost_equal_nulp assert_raises build_err_msg)mask_orgetmask masked_arraynomaskmaskedfilledalmostapproxassert_almost_equalassert_array_almost_equalassert_array_approx_equalassert_array_compareassert_array_equalassert_array_less assert_close assert_equalassert_equal_recordsassert_mask_equalassert_not_equalfail_if_array_equal)TestCaser!rrr r Tgh㈵>g:0yE>c Cstt|t|}t|}t|}|jjdksW|jjdkrmtj||jStt|ddd||j t }tt|ddd|dj t } tj t j || ||t j | } | jS)a 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. OcopyFmaskr )r rrdtypecharnpequalravelrastyper less_equalumathabsolute) ab fill_valuertolZatolmd1d2xydr8D/home/birch/.local/lib/python3.5/site-packages/numpy/ma/testutils.pyr+s   $**0c Cstt|t|}t|}t|}|jjdksW|jjdkrmtj||jStt|ddd||j t }tt|ddd|dj t }tj tj |||d| k} | jS)z 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#Fr$r g$@) r rrr%r&r'r(r)rr*rZaroundabs) r.r/decimalr0r2r3r4r5r6r7r8r8r9rAs  $***cCs_tt|t||x<tt|D](}t||||d||fq/WdS)z; Asserts the equality of two non-array sequences. z item=%r %sN)rlenrange)actualdesirederr_msgkr8r8r9_assert_equal_on_sequencesTs&rDcCst|j|jxv|jjD]h}tj||tj||}}|tk r |tk r ttj||tj||q WdS)zI Asserts that two records are equal. Pretty crude for now. N)rr%namesoperatorgetitemr)r.r/fafZbfr8r8r9r_s %)cCs6t|trt|ts6ttt|tt|t||x^|jD]P\}}||krtd||ft||||d||fq_WdSt|tt frt|tt frt ||ddSt|t pt|t sCt ||g|}||ks?t|dS|t kr[|t k ss|t kr|t k rt ||g|dddd}t|tj|d d d d }tj|d d d d }|j|j}}|jdkr&|jdkr&t |j|jddSt|||S)z, Asserts that two items are equal. z %s not in %sz key=%r %sNrBr=headerrEr5r6r#FsubokTS)r5r6) isinstancedictAssertionErrorreprtyperr>itemslisttuplerDrr r ValueErrorr'arrayr%r&tolistr)r@rArBrCimsgZ actual_dtypeZ desired_dtyper8r8r9rns: &*      cCst|trt|ts6ttt|tt|t||xZ|jD]L\}}||krtt|t||||d||fq_WdSt|tt fr<t|tt fr<tt|t||x<t t|D](}t||||d||fq WdSt|t j s`t|t j rpt |||St||g|}||kst|dS)z< Raises an assertion error if two items are equal. z key=%r %sNz item=%r %s)rMrNrOrPrQ fail_if_equalr>rRrSrTr?r'rr r )r@rArBrCrXrYr8r8r9rZs& &*&$ rZc Cst|tjs$t|tjrCt||d|d|d|St||gd|d|}tt|||dkst|dS)z~ Asserts that two items are almost equal. The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal). r<rBverboserN)rMr'rrr roundr;rO)r@rAr<rBr\rYr8r8r9rs$ c Cstt|t|}t|ddd|dddd}t|ddd|dddd}|tkr{|tk s|tkr|tk rt||gd|d|d|d d }t|tjj||j ||j |d|d|d|S) zn Asserts that comparison between two masked arrays is satisfied. The comparison is elementwise. r#Fr$Z keep_maskrKrBr\rJrEr5r6)r5r6) r rrrr rUr'testingrr) Z comparisonr5r6rBr\rJr0r2rYr8r8r9rs $$   c Cs)ttj||d|d|dddS)z@ Checks the elementwise equality of two masked arrays. rBr\rJzArrays are not equalN)rrF__eq__)r5r6rBr\r8r8r9rs c Cs2dd}t|||d|d|dddS)zT Raises an assertion error if two masked arrays are not equal elementwise. cSstjt|| S)N)r'Zalltruer)r5r6r8r8r9comparesz$fail_if_array_equal..comparerBr\rJzArrays are not equalN)r)r5r6rBr\r`r8r8r9r s c s8fdd}t|||d|d|dddS)z| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. cst||dd S)z.comparerBr\rJzArrays are not almost equalN)r)r5r6r<rBr\r`r8)r<r9rsc s8fdd}t|||d|d|dddS)z| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. cst||S)z.comparerBr\rJzArrays are not almost equalN)r)r5r6r<rBr\r`r8)r<r9rsc Cs)ttj||d|d|dddS)z7 Checks that x is smaller than y elementwise. rBr\rJzArrays are not less-orderedN)rrF__lt__)r5r6rBr\r8r8r9rs cCsO|tkrt|tk|tkr8t|tkt||d|dS)z- Asserts the equality of two masks. rBN)rrr)m1m2rBr8r8r9rs   ).__doc__ __future__rrrrFnumpyr'rrZnumpy.core.umathcorer,Z numpy.testingrrr r r r rrrrrZ __all__maskedunittestr!Z__some__from_testing__all__rrrDrrrZrrrrrr rrrrr8r8r8r9sD   (.        '