3 5^}(@s`dZddlmZmZmZddlZddlZddlmZm Z ddl j j Z ddl Zddl mZmZmZmZmZddl mZmZmZmZmZmZdd d d d d ddddddddgZddlmZdddddgZeeZd2dd Zd3d!dZd4d#d$Z d%dZ!d5d&dZ"d6d'd(Z#e#Z$d7d*d Z%e%Z&d8d+d Z'd9d,dZ(d:d-dZ)d;d.d Z*dd1dZ-dS)?zMiscellaneous 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 Th㈵>:0yE>c Cstt|t|}t|}t|}|jjdks:|jjdkrJtj||jStt|d|d|j t }tt|d|ddj 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. OF)copymaskr )r rrdtypecharnpequalravelrastyper less_equalumathabsolute) ab fill_valuertolZatolmd1d2xydr:]/home/brian/BIRCH/lib-linux-x86_64/python/lib64/python3.6/site-packages/numpy/ma/testutils.pyr+s $c Cstt|t|}t|}t|}|jjdks:|jjdkrJtj||jStt|d|d|j t }tt|d|ddj 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$F)r%r&r g$@) r rrr'r(r)r*r+rr,rZaroundabs) r0r1decimalr2r4r5r6r7r8r9r:r:r;rAs cCsJtt|t||x0tt|D] }t||||d||fq"WdS)z; Asserts the equality of two non-array sequences. z item=%r %sN)rlenrange)actualdesirederr_msgkr:r:r;_assert_equal_on_sequencesTs rFcCsht|j|jxT|jjD]H}tj||tj||}}|tk r|tk rttj||tj||qWdS)zI Asserts that two records are equal. Pretty crude for now. N)rr'namesoperatorgetitemr)r0r1fafZbfr:r:r;r_s cCst|trt|ts$ttt|tt|t||xH|jD]<\}}||krbtd||ft||||d||fqBWdSt|tt frt|tt frt ||ddSt|t pt|t st ||g|}||kst|dS|t ko|t k s |t kr(|t k r(t ||g|dd d}t|tj|d d d }tj|d d d }|j|j}}|jd kr|jd krt |j|jddSt|||S)z, Asserts that two items are equal. z %s not in %sz key=%r %sNr?)rDr7r8)headerrGFT)r%subokS)r7r8) isinstancedictAssertionErrorreprtyperr@itemslisttuplerFrr r ValueErrorr)arrayr'r(tolistr)rBrCrDrEimsgZ actual_dtypeZ desired_dtyper:r:r;rns:    cCs4t|trt|ts$ttt|tt|t||xD|jD]8\}}||kr^tt|t||||d||fqBWdSt|tt frt|tt frtt|t||x0t t|D] }t||||d||fqWdSt|t j st|t j rt |||St||g|}||ks0t|dS)z< Raises an assertion error if two items are equal. z key=%r %sNz item=%r %s)rOrPrQrRrS fail_if_equalr@rTrUrVrAr)rr r )rBrCrDrErZr[r:r:r;r\s&       r\cCs^t|tjst|tjr*t|||||dSt||g||d}tt|||dksZt|dS)z~ Asserts that two items are almost equal. The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal). )r>rDverbose)rDr^rN)rOr)rrr roundr=rQ)rBrCr>rDr^r[r:r:r;rs  c Cstt|t|}t|d|ddd}t|d|ddd}|tkrF|tk sV|tkrt|tk rtt||g|||dd}t|tjj||j ||j ||||dS)zn Asserts that comparison between two masked arrays is satisfied. The comparison is elementwise. F)r%r&Z keep_maskrMr7r8)rDr^rLrG)rDr^rL)r7r8) r rrrr rWr)testingrr) Z comparisonr7r8rDr^rLr2r4r[r:r:r;rs   cCsttj||||dddS)z@ Checks the elementwise equality of two masked arrays. zArrays are not equal)rDr^rLN)rrH__eq__)r7r8rDr^r:r:r;rs cCs dd}t|||||dddS)zT Raises an assertion error if two masked arrays are not equal elementwise. cSstjt|| S)N)r)Zalltruer)r7r8r:r:r;comparesz$fail_if_array_equal..comparezArrays are not equal)rDr^rLN)r)r7r8rDr^rbr:r:r;r s cs$fdd}t|||||dddS)z| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. cst||d dS)zr:r;rbsz*assert_array_approx_equal..comparezArrays are not almost equal)rDr^rLN)r)r7r8r>rDr^rbr:)r>r;rs  cs$fdd}t|||||dddS)z| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. cs t||S)zr:r;rbsz*assert_array_almost_equal..comparezArrays are not almost equal)rDr^rLN)r)r7r8r>rDr^rbr:)r>r;rs  cCsttj||||dddS)z7 Checks that x is smaller than y elementwise. zArrays are not less-ordered)rDr^rLN)rrH__lt__)r7r8rDr^r:r:r;rs cCs:|tkrt|tk|tkr(t|tkt|||ddS)z- Asserts the equality of two masks. )rDN)rrr)m1m2rDr:r:r;rs   )Tr"r#)r<T)r?)r?)r?)r]r?T)r?Tr?T)r?T)r?T)r<r?T)r<r?T)r?T)r?).__doc__ __future__rrrrHnumpyr)rrZnumpy.core.umathcorer.Z numpy.testingrrr r r r rrrrrZ __all__maskedZunittestr!Z__some__from_testing__all__rrrFrrr\rrrrrr rrrrr:r:r:r;sD       '