IR-eddlZddlZddlZddlZddlZddlmZgdZej edZ d dZ ej dddfd Z dd ZdS)N)indent)fixed_width_indent diff_valuesreport_diff_valueswhere_not_allclose)widthct|trXt|trCtj|rtj|rdStj|||| S||kS)a Diff two scalar values. If both values are floats, they are compared to within the given absolute and relative tolerance. Parameters ---------- a, b : int, float, str Scalar values to compare. rtol, atol : float Relative and absolute tolerances as accepted by :func:`numpy.allclose`. Returns ------- is_different : bool `True` if they are different, else `False`. Frtolatol) isinstancefloatnpisnanallcloseabrrs 2lib/python3.11/site-packages/astropy/utils/diff.pyrrsp(!U 1e 4 4 8A;; 28A;; 5;q!$T:::::Av c t|tjrt|tjr|j|jkr_|t d|t t|jt|j||dzdStj|j tj rJtj|j tj r&tj t||||}n*tj tj ||k}|jd}|ddD]{}|} |t d | d |t |t||t|||dz|| ||dkr-|t d |dz d d|dS|dkSt!|} t!|} | | kr`d} d} d}t|t"jr t'|}t'|}nKt|}t|}n+t)t+| jt+| jdz}|dzdz} d| jzdz|dz} d| jzdz|dz}t|t}t|t}|r|r|s)t|t"jrt'|nt|}|r|r|s)t|t"jrt'|nt|}d}t1j||D]}|ddkrd}| |ddz}n!|ddkrd}||ddz}n| |z}|t d|d||S)a} Write a diff report between two values to the specified file-like object. Parameters ---------- a, b Values to compare. Anything that can be turned into strings and compared using :py:mod:`difflib` should work. fileobj : object File-like object to write to. The default is ``sys.stdout``, which writes to terminal. indent_width : int Character column(s) to indent. rtol, atol : float Relative and absolute tolerances as accepted by :func:`numpy.allclose`. Returns ------- identical : bool `True` if no diff, else `False`. z Different array shapes: r)fileobj indent_widthFr rNz at z: )rrrrz ...and at dz more indices.  za>zb>(z) T-+z {}  )rrndarrayshapewriterrstr issubdtypedtypefloating transposerwheretolisttupletypenumbersNumberreprmaxlen__name__rjustdifflibndiff splitlinesformatrstrip)rrrrrr diff_indices num_diffsidxlidxtypeatypeblnpadsign_asign_bpaddingis_a_stris_b_str identicallines rrr2s|6!RZ  (Z2:%>%>( 7ag   MM"#@,OO    AG AG )A-      5 ="+ . . :2="+3V3V :<(:1adQU(V(V(VWWLL<a(8(899L &q) #  C::<+>??!C1#&-44W==D&-44W==Da%%a%% "* ! &07>%B%B DGGG Q  "* ! &07>%B%B DGGG Q I allnnallnn==    7c>>IDH$DD !W^^IDH$DD4:0yE>ctjtj|s$tj|j}tjtj|s$tj|j}|dkr|dkrtj||kStjtj||z ||tj|zzkS)a A version of :func:`numpy.allclose` that returns the indices where the two arrays differ, instead of just a boolean value. Parameters ---------- a, b : array-like Input arrays to compare. rtol, atol : float Relative and absolute tolerances as accepted by :func:`numpy.allclose`. Returns ------- idx : tuple of array Indices where the two arrays differ. r )rallisfinitema fix_invaliddatar,absrs rrrs, 6"+a.. ! !& E  a % 6"+a.. ! !& E  a % s{{ts{{xQ 8BF1q5MMTD26!99,<%<= > >>r)r r )rJrK)r7 functoolsr0sysnumpyrmiscr__all__partialrrstdoutrrrrr[s    'Y&vQ7778&)ZacPSzzzzz??????r