o k)h (@s4dZddlZddlZddlmZddlmmZddlZddlm Z m Z m Z m Z m Z ddlmZmZmZmZmZmZgdZddlmZgd ZeeZd,d dZd-ddZd.ddZddZd.ddZd.ddZeZ d/ddZ!e!Z"  d0ddZ#d1d d!Z$d1d"d#Z%d2d$d%Z&d2d&d'Z'd1d(d)Z(d.d*d+Z)dS)3zMiscellaneous 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 $ N)ndarray)assert_assert_allcloseassert_array_almost_equal_nulp assert_raises build_err_msg)mask_orgetmask masked_arraynomaskmaskedfilled)almostapproxassert_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)TestCase)rrrrrTh㈵>:0yE>c Cstt|t|}t|}t|}|jjdks|jjdkr%t||Stt|d|d| tj }tt|d|dd tj } t t || ||t | } | S)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. OFcopymaskr)r r rdtypecharnpequalravelr astypefloat64Z less_equalumathZabsolute) ab fill_valuertolZatolmd1d2xydr62lib/python3.10/site-packages/numpy/ma/testutils.pyr)s $rc Cstt|t|}t|}t|}|jjdks|jjdkr%t||Stt|d|d| tj }tt|d|dd tj }t t |||d| k} | S)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 Fr!r$@) r r rr$r%r&r'r(r r)r*Zaroundabs) r,r-decimalr.r0r1r2r3r4r5r6r6r7rAs rcCsJtt|t||tt|D]}t||||d|d|qdS)z; Asserts the equality of two non-array sequences. item= N)rlenrange)actualdesirederr_msgkr6r6r7_assert_equal_on_sequencesVs"rEcCsdt|j|j|jjD]$}t||t||}}|tur/|tur/tt||t||q dS)zI Asserts that two records are equal. Pretty crude for now. N)rr$namesoperatorgetitemr )r,r-fafZbfr6r6r7ras rcCs|t|trDt|tsttt|tt|t|||D]!\}}||vr1t|d|t||||d|d|q dSt|tt frYt|tt frYt ||ddSt|t stt|t stt ||g|}||ksrt|dS|t ur||t us|t ur|t urt ||g|ddd}t|t|}t|}|j|j}}|jd kr|jd krt ||ddSt|||S) z, Asserts that two items are equal. z not in key=r>Nr<rCr3r4)headerrFS) isinstancedictAssertionErrorreprtyperr?itemslisttuplerErrr ValueErrorr&Z asanyarrayr$r%tolistr)rArBrCrDimsgZ actual_dtypeZ desired_dtyper6r6r7rps>  "   rcCs.t|trAt|tsttt|tt|t|||D]\}}||vr.tt|t||||d|d|q dSt|tt frtt|tt frttt|t||t t|D]}t||||d|d|q_dSt|t j st|t j rt |||St||g|}||kst|dS)z< Raises an assertion error if two items are equal. rKr>Nr=)rPrQrRrSrT fail_if_equalr?rUrVrWr@r&rrr)rArBrCrDrZr[r6r6r7r\s(   "" r\cCs^t|tjs t|tjrt|||||dSt||g||d}tt|||dks-t|dS)z~ Asserts that two items are almost equal. The test is equivalent to abs(desired-actual) < 0.5 * 10**(-decimal). )r;rCverbose)rCr^rN)rPr&rrrroundr:rR)rArBr;rCr^r[r6r6r7rsrc Cstt|t|}t|d|ddd}t|d|ddd}|tur#|tus+|tur:|tur:t||g|||dd}t|tjj|| || ||||dS)zn Asserts that comparison between two masked arrays is satisfied. The comparison is elementwise. F)r"r#Z keep_maskZsubokrM)rCr^rNrFrCr^rN) r r r r rrXr&Ztestingrr) Z comparisonr3r4rCr^rNr.r0r[r6r6r7rs  rcCttj||||dddS)z@ Checks the elementwise equality of two masked arrays. Arrays are not equalr`N)rrG__eq__r3r4rCr^r6r6r7r  rcCs dd}t|||||dddS)zT Raises an assertion error if two masked arrays are not equal elementwise. cSstt|| S)N)r&allrrMr6r6r7comparesz$fail_if_array_equal..comparerbr`Nr)r3r4rCr^rgr6r6r7rs  rc$fdd}t|||||dddS)| Checks the equality of two masked arrays, up to given number odecimals. The equality is checked elementwise. cst||d dS).compareArrays are not almost equalr`Nrhr3r4r;rCr^rgr6rlr7r   rcri)rjcs t||S)rk)rrMrlr6r7rg s z*assert_array_almost_equal..comparermr`Nrhrnr6rlr7rrorcCra)z7 Checks that x is smaller than y elementwise. zArrays are not less-orderedr`N)rrG__lt__rdr6r6r7rrercCs:|tur t|tu|turt|tut|||ddS)z- Asserts the equality of two masks. rLN)r rr)Zm1Zm2rCr6r6r7rs   r)Trr)r8T)r<)r]r<T)r<Tr<T)r<T)r8r<T)*__doc__rGZnumpyr&rZnumpy._core.umathZ_corer+Z numpy.testingrrrrrcorer r r r r rZ __all__maskedZunittestrZ__some__from_testing__all__rrrErrr\rrrrrrrrrrr6r6r6r7s<        '