HR-e>dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZddlmZddlmZmZmZmZddlmZd d lmZmZd d lmZmZd d lm Z d d l!m"Z"d dl#m$Z$gdZ%gdZ&GddZ'Gdde'Z(Gdde'Z)Gdde'Z*Gdde'Z+Gdde+Z,Gdde'Z-d dZ.dS)!z Facilities for diffing two FITS files. Includes objects for diffing entire FITS files, individual HDUs, FITS headers, or just FITS data. Used to implement the fitsdiff program. N) defaultdict) signature)islice) __version__) diff_valuesfixed_width_indentreport_diff_valueswhere_not_allclose)NOT_OVERWRITING_MSG) BLANK_CARDCard)HDUListfitsopen) _TableLikeHDU)Header) path_like)FITSDiffHDUDiff HeaderDiff ImageDataDiff RawDataDiff TableDataDiff))unitunits)nullz null values)bscalebscales)bzerobzeros)dispzdisplay formats)dim dimensionscdeZdZdZdZdZedZedZ d d Z d Z d Z d Z dS) _BaseDiffaa Base class for all FITS diff objects. When instantiating a FITS diff object, the first two arguments are always the two objects to diff (two FITS files, two FITS headers, etc.). Instantiating a ``_BaseDiff`` also causes the diff itself to be executed. The returned ``_BaseDiff`` instance has a number of attribute that describe the results of the diff operation. The most basic attribute, present on all ``_BaseDiff`` instances, is ``.identical`` which is `True` if the two objects being compared are identical according to the diff method for objects of that type. cf||_||_d|_d|_|dS)a- The ``_BaseDiff`` class does not implement a ``_diff`` method and should not be instantiated directly. Instead instantiate the appropriate subclass of ``_BaseDiff`` for the objects being compared (for example, use `HeaderDiff` to compare two `Header` objects. Nr)ab_fileobj_indent_diff)selfr'r(s 4lib/python3.11/site-packages/astropy/io/fits/diff.py__init__z_BaseDiff.__init__Is3   c|j S)z A ``_BaseDiff`` object acts as `True` in a boolean context if the two objects compared are different. Otherwise it acts as `False`. ) identicalr,s r-__bool__z_BaseDiff.__bool__Ys >!!r/ct|j}i}t|jddD]%}t ||rt ||||<&|||fi|S)aQ Returns a new Diff object of a specific subclass from an existing diff object, passing on the values for any arguments they share in common (such as ignore_keywords). For example:: >>> from astropy.io import fits >>> hdul1, hdul2 = fits.HDUList(), fits.HDUList() >>> headera, headerb = fits.Header(), fits.Header() >>> fd = fits.FITSDiff(hdul1, hdul2, ignore_keywords=['*']) >>> hd = fits.HeaderDiff.fromdiff(fd, headera, headerb) >>> list(hd.ignore_keywords) ['*'] N)rr.list parameterskeyshasattrgetattr)clsotherr'r(sigkwargsargs r-fromdiffz_BaseDiff.fromdiff`s" %%++--..qrr2 2 2Cuc"" 2%eS11s s1a""6"""r/cFtfdjD S)ac `True` if all the ``.diff_*`` attributes on this diff instance are empty, implying that no differences were found. Any subclass of ``_BaseDiff`` must have at least one ``.diff_*`` attribute, which contains a non-empty value if and only if some difference was found between the two objects being compared. c3bK|])}|dt|V*dS)diff_N) startswithr:).0attrr,s r- z&_BaseDiff.identical..sR  $(T__W=U=U D$        r/)any__dict__r2s`r-r1z_BaseDiff.identicalzsD    ,0M       r/NrFc&d}d}t|trttj|}tj|r#|s!t tj||}t|d}n|tj }d}||_ ||_ ||r|n#|r|wwxYw|r|SdS)a Generates a text report on the differences (if any) between two objects, and either returns it as a string or writes it to a file-like object. Parameters ---------- fileobj : file-like, string, or None, optional If `None`, this method returns the report as a string. Otherwise it returns `None` and writes the report to the given file-like object (which must have a ``.write()`` method at a minimum), or to a new file at the path specified. indent : int The number of 4 space tabs to indent the report. overwrite : bool, optional If ``True``, overwrite the output file if it exists. Raises an ``OSError`` if ``False`` and the output file exists. Default is ``False``. Returns ------- report : str or None FNwT) isinstancerospath expanduserexistsOSErrorr formatopenioStringIOr)r*_reportclosegetvalue)r,fileobjindent overwrite return_stringfilepaths r-reportz_BaseDiff.reports4  gy ) ) !g((11Gw~~g&& .y .18AABBB"x-- _kmmG M   LLNNN       &##%% % & &s 4CC8cf|jt||jdzdS)N )r)writerr*)r,texts r-_writelnz_BaseDiff._writelns0 .tT\BBTIJJJJJr/ctNNotImplementedErrorr2s r-r+z_BaseDiff._diff!!r/ctrerfr2s r-rVz_BaseDiff._reportrhr/)NrF)__name__ __module__ __qualname____doc__r.r3 classmethodr@propertyr1r^rcr+rVr/r-r%r%:s   """##[#2   X  2&2&2&2&hKKK""""""""r/r%c BeZdZdZggggdddddf fd ZdZdZxZS)raDiff two FITS files by filename, or two `HDUList` objects. `FITSDiff` objects have the following diff attributes: - ``diff_hdu_count``: If the FITS files being compared have different numbers of HDUs, this contains a 2-tuple of the number of HDUs in each file. - ``diff_hdus``: If any HDUs with the same index are different, this contains a list of 2-tuples of the HDU index and the `HDUDiff` object representing the differences between the two HDUs. Tc \t|ttjfr: t |}n&#t $r} t d|d| d} ~ wwxYwd} nd} t|ttjfr: t |}n&#t $r} t d|d| d} ~ wwxYwd}nd}d|D|_d|D|_d |D|_ d |D|_ ||_ ||_ | |_ | |_| |_t!|_t%|jD]P}|d krHt'j|r4|j||j|Qd |_g|_ t3||| r||r|dSdS#| r||r|wwxYw) a Parameters ---------- a : str or `HDUList` The filename of a FITS file on disk, or an `HDUList` object. b : str or `HDUList` The filename of a FITS file on disk, or an `HDUList` object to compare to the first file. ignore_hdus : sequence, optional HDU names to ignore when comparing two FITS files or HDU lists; the presence of these HDUs and their contents are ignored. Wildcard strings may also be included in the list. ignore_keywords : sequence, optional Header keywords to ignore when comparing two headers; the presence of these keywords and their values are ignored. Wildcard strings may also be included in the list. ignore_comments : sequence, optional A list of header keywords whose comments should be ignored in the comparison. May contain wildcard strings as with ignore_keywords. ignore_fields : sequence, optional The (case-insensitive) names of any table columns to ignore if any table data is to be compared. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rtol : float, optional The relative difference to allow when comparing two float values either in header values, image arrays, or table columns (default: 0.0). Values which satisfy the expression .. math:: \left| a - b \right| > \text{atol} + \text{rtol} \cdot \left| b \right| are considered to be different. The underlying function used for comparison is `numpy.allclose`. .. versionadded:: 2.0 atol : float, optional The allowed absolute difference. See also ``rtol`` parameter. .. versionadded:: 2.0 ignore_blanks : bool, optional Ignore extra whitespace at the end of string values either in headers or data. Extra leading whitespace is not ignored (default: True). ignore_blank_cards : bool, optional Ignore all cards that are blank, i.e. they only contain whitespace (default: True). zerror opening file a ()NTFzerror opening file b (c6h|]}|SrpupperrEks r- z$FITSDiff.__init__..4s ;;;!AGGII;;;r/c6h|]}|Srprwrys r-r{z$FITSDiff.__init__..5 CCCa CCCr/c6h|]}|Srprwrys r-r{z$FITSDiff.__init__..6r}r/c6h|]}|Srprwrys r-r{z$FITSDiff.__init__..7 ???Aaggii???r/*rp)rLstrrMPathLiker ExceptionrQ ignore_hdusignore_keywordsignore_comments ignore_fieldsnumdiffsrtolatol ignore_blanksignore_blank_cardssetignore_hdu_patternsr6glob has_magicremoveadddiff_hdu_count diff_hdussuperr.rW)r,r'r(rrrrrrrrrexcclose_aclose_bname __class__s r-r.zFITSDiff.__init__s~Z a#r{+ , ,  FQKK F F F;q;;;<<#E FGGG a#r{+ , ,  FQKK F F F;q;;;<<#E FGGG<;{;;;CC?CCCCC?CCC?????    *"4$'55 )** 3 3Ds{{t~d33{ ''---(,,T222   GG  Q " " "           s84 AAAB B4B//B4*"G<c0g|]}|jjv|SrprrrEhr,s r- z"FITSDiff._diff..`(RRRA16AQ3Q3Qa3Q3Q3Qr/c0g|]}|jjv|Srprrs r-rz"FITSDiff._diff..arr/cg|] }|j SrprrEhdus r-rz"FITSDiff._diff..c222Csx222r/cg|] }|j Srprrs r-rz"FITSDiff._diff..drr/c&g|] }|jv |Srpr)rEr a_ignoreds r-rz"FITSDiff._diff..g%!O!O!OqvY7N7N!7N7N7Nr/c&g|] }|jv |Srpr)rEr b_ignoreds r-rz"FITSDiff._diff..irr/)lenr'r(rfilename filenamearrjid filenamebrrrfnmatchfilterrangeminrr@r1rverrappend)r,a_namesb_namespatternidxhdu_diffrrs` @@r-r+zFITSDiff._diffRs tv;;#df++ % %#&tv;;DF "CCDDD   W((6$*:#;#;<%?%?@@M MM87<< ;V;VV    W W W    PDI P PTY P P       M   % % % MMD E E E MM9!4Q!799 : : : MM9!4Q!799 : : :>  IJJJ   M   % % % MM1 2 2 2 F.2n D D *C7Faxx ##D))) n---- ##D)));MM"O3"O"O'"O"OV"O"O"OPPPPMM"93"9"9"9::: OODM$,2BO C C C C D Dr/rjrkrlrmr.r+rV __classcell__rs@r-rrs  "  |||||||*P*P*PXDDDDDDDDDDDDDDr/rc@eZdZdZgggdddddffd ZdZdZxZS)ra Diff two HDU objects, including their headers and their data (but only if both HDUs contain the same type of data (image, table, or unknown). `HDUDiff` objects have the following diff attributes: - ``diff_extnames``: If the two HDUs have different EXTNAME values, this contains a 2-tuple of the different extension names. - ``diff_extvers``: If the two HDUS have different EXTVER values, this contains a 2-tuple of the different extension versions. - ``diff_extlevels``: If the two HDUs have different EXTLEVEL values, this contains a 2-tuple of the different extension levels. - ``diff_extension_types``: If the two HDUs have different XTENSION values, this contains a 2-tuple of the different extension types. - ``diff_headers``: Contains a `HeaderDiff` object for the headers of the two HDUs. This will always contain an object--it may be determined whether the headers are different through ``diff_headers.identical``. - ``diff_data``: Contains either a `ImageDataDiff`, `TableDataDiff`, or `RawDataDiff` as appropriate for the data in the HDUs, and only if the two HDUs have non-empty data of the same type (`RawDataDiff` is used for HDUs containing non-empty data of an indeterminate type). rrrsTc Ld|D|_d|D|_d|D|_||_||_||_| |_| |_d|_d|_ d|_ d|_ d|_ d|_ t||dS)a Parameters ---------- a : BaseHDU An HDU object. b : BaseHDU An HDU object to compare to the first HDU object. ignore_keywords : sequence, optional Header keywords to ignore when comparing two headers; the presence of these keywords and their values are ignored. Wildcard strings may also be included in the list. ignore_comments : sequence, optional A list of header keywords whose comments should be ignored in the comparison. May contain wildcard strings as with ignore_keywords. ignore_fields : sequence, optional The (case-insensitive) names of any table columns to ignore if any table data is to be compared. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rtol : float, optional The relative difference to allow when comparing two float values either in header values, image arrays, or table columns (default: 0.0). Values which satisfy the expression .. math:: \left| a - b \right| > \text{atol} + \text{rtol} \cdot \left| b \right| are considered to be different. The underlying function used for comparison is `numpy.allclose`. .. versionadded:: 2.0 atol : float, optional The allowed absolute difference. See also ``rtol`` parameter. .. versionadded:: 2.0 ignore_blanks : bool, optional Ignore extra whitespace at the end of string values either in headers or data. Extra leading whitespace is not ignored (default: True). ignore_blank_cards : bool, optional Ignore all cards that are blank, i.e. they only contain whitespace (default: True). c6h|]}|Srprwrys r-r{z#HDUDiff.__init__..(r}r/c6h|]}|Srprwrys r-r{z#HDUDiff.__init__..)r}r/c6h|]}|Srprwrys r-r{z#HDUDiff.__init__..*rr/rpN)rrrrrrrr diff_extnames diff_extversdiff_extlevelsdiff_extension_types diff_headers diff_datarr.) r,r'r(rrrrrrrrrs r-r.zHDUDiff.__init__sL DC?CCCCC?CCC?????    *"4 $&!  Ar/c|jj|jjkr|jj|jjf|_|jj|jjkr|jj|jjf|_|jj|jjkr|jj|jjf|_|jj d|jj dkrC|jj d|jj df|_ t ||jj |jj |_|jj |jjdS|jjr[|jjrOt" ||jj|jj|_d|j_d|j_dSt'|jt(rit'|jt(rOt* ||jj|jj|_d|j_d|j_dS|j sOt, ||jj|jj|_d|j_d|j_dSdS)NXTENSION)r'rr(rrrlevelrheadergetrrr@copyrdatais_imagerrrLrrrr2s r-r+z HDUDiff._diff<sD 6;$&+ % %"&&+tv{!;D  6: # #!%TVZ 8D  6<46< ' '#'6<">D  6=  Z ( (DFM,=,=j,I,I I I !!*-- !!*--)D % '// $&-$$&& (:(:(<(<   6; $&+"5 D V_ $ $*33D$&+tv{SSDN $DN #DN     . . $:dfm3T3T $*33D$&+tv{SSDN $DN #DN   * $)11$ TV[QQDN $DN #DN    $ $r/c,|jr|d|jr"|dj|j|jr"|dj|j|jr"|dj|j|jr"|dj|j|jjsX|j d|d|j |j|j dz |j f|j js\|j d|d |j |j|j dz dSdSdS) Nz No differences found.z( Extension types differ: a: {} b: {}z( Extension names differ: a: {} b: {}z+ Extension versions differ: a: {} b: {}z) Extension levels differ: a: {} b: {}r`z Headers contain differences:r rz Data contains differences:) r1rcrrRrrrrr)rar^r*rr2s r-rVzHDUDiff._reportis > 4 MM2 3 3 3  $  MMC<C.       MMC<CTEWX       MMF?F&       MMD=D(     * M M   % % % MM9 : : :   $ $T]4 %dn.F % M   % % % MM7 8 8 8 N ! !$- q8H ! I I I I I & % % %r/rrs@r-rrs@  XXXXXXt+$+$+$Z#J#J#J#J#J#J#Jr/rc<eZdZdZggddddffd ZdZdZxZS)ra  Diff two `Header` objects. `HeaderDiff` objects have the following diff attributes: - ``diff_keyword_count``: If the two headers contain a different number of keywords, this contains a 2-tuple of the keyword count for each header. - ``diff_keywords``: If either header contains one or more keywords that don't appear at all in the other header, this contains a 2-tuple consisting of a list of the keywords only appearing in header a, and a list of the keywords only appearing in header b. - ``diff_duplicate_keywords``: If a keyword appears in both headers at least once, but contains a different number of duplicates (for example, a different number of HISTORY cards in each header), an item is added to this dict with the keyword as the key, and a 2-tuple of the different counts of that keyword as the value. For example:: {'HISTORY': (20, 19)} means that header a contains 20 HISTORY cards, while header b contains only 19 HISTORY cards. - ``diff_keyword_values``: If any of the common keyword between the two headers have different values, they appear in this dict. It has a structure similar to ``diff_duplicate_keywords``, with the keyword as the key, and a 2-tuple of the different values as the value. For example:: {'NAXIS': (2, 3)} means that the NAXIS keyword has a value of 2 in header a, and a value of 3 in header b. This excludes any keywords matched by the ``ignore_keywords`` list. - ``diff_keyword_comments``: Like ``diff_keyword_values``, but contains differences between keyword comments. `HeaderDiff` objects also have a ``common_keywords`` attribute that lists all keywords that appear in both headers. rsTc d|D|_d|D|_||_||_||_||_t |_t |_t|jD]d} | } | dkrHtj | r4|j | |j| et|jD]d} | } | dkrHtj | r4|j | |j| eg|_d|_d|_d|_i|_t)t|_t)t|_t/|t0rt3j|}t/|t0rt3j|}t/|t2rt/|t2st7dt9||dS)a> Parameters ---------- a : `~astropy.io.fits.Header` or string or bytes A header. b : `~astropy.io.fits.Header` or string or bytes A header to compare to the first header. ignore_keywords : sequence, optional Header keywords to ignore when comparing two headers; the presence of these keywords and their values are ignored. Wildcard strings may also be included in the list. ignore_comments : sequence, optional A list of header keywords whose comments should be ignored in the comparison. May contain wildcard strings as with ignore_keywords. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rtol : float, optional The relative difference to allow when comparing two float values either in header values, image arrays, or table columns (default: 0.0). Values which satisfy the expression .. math:: \left| a - b \right| > \text{atol} + \text{rtol} \cdot \left| b \right| are considered to be different. The underlying function used for comparison is `numpy.allclose`. .. versionadded:: 2.0 atol : float, optional The allowed absolute difference. See also ``rtol`` parameter. .. versionadded:: 2.0 ignore_blanks : bool, optional Ignore extra whitespace at the end of string values either in headers or data. Extra leading whitespace is not ignored (default: True). ignore_blank_cards : bool, optional Ignore all cards that are blank, i.e. they only contain whitespace (default: True). c6h|]}|Srprwrys r-r{z&HeaderDiff.__init__..r}r/c6h|]}|Srprwrys r-r{z&HeaderDiff.__init__..r}r/rrpz[HeaderDiff can only diff astropy.io.fits.Header objects or strings containing FITS headers.N)rrrrrrrignore_keyword_patternsignore_comment_patternsr6rxrrrrcommon_keywordsdiff_keyword_countdiff_keyword_positions diff_keywordsdiff_duplicate_keywordsrdiff_keyword_valuesdiff_keyword_commentsrLrr fromstring TypeErrorrr.) r,r'r(rrrrrrkeywordrs r-r.zHeaderDiff.__init__s9@ DC?CCCCC?CCC  *"4'*uu$'*uu$D011 : :GmmooG#~~$."9"9~$++G444,00999D011 : :GmmooG#~~$."9"9~$++G444,00999 "#% ')# (*$$/t#4#4 &1%6%6" a   %!!$$A a   %!!$$A1f%% *Q*?*? >  Ar/cjr-djjD}djjD}n2t jj}t jj}fd}||\}}||\}}d|D}d|D} t || _t|t|kr#t|t|f_ | j }| j } j rZj D]R} | tj|| }| tj| | } Sdj vrdSt | | } t | |} | s| r | | f_jD]} | j vr j r(d}j D]} tj | | rd}n|r.=(FFFAQ:1E1Ea1E1E1Er/cBg|]}t|tk|Srprrs r-rz$HeaderDiff._diff..>rr/cNi}i}|D]}|j}jr)t|tr|}||jg|||jg|j||fSre) valuerrLrrstrip setdefaultrrcomment)cardsvaluescommentscardrr,s r-get_header_values_commentsz4HeaderDiff._diff..get_header_values_commentsDsFH K K %+*UC*@*@+!LLNNE!!$,33::5AAA##DL"55<.T 0001QWWYY000r/c6h|]}|Srprwrys r-r{z#HeaderDiff._diff..Ur r/rFTrr)rr'rr(r6r intersectionrrr differencerrrrrrziprrrrrrHrrr)r,cardsacardsbrvaluesa commentsavaluesb commentsb keywordsa keywordsbrleft_only_keywordsright_only_keywordsrskipcountacountbr'r(s` r-r+zHeaderDiff._diff;s  " (FFFFFFFFFFFFF$&,''F$&,''F $ $ $ $ $87??77??10000 00000 %i&<&> (()=>>  ' U7 U U%00 71S1STT %00 71S1STT $& & & F#I$8$8$C$CDD$Y%9%9)%D%DEE  K!4 K"46I!JD +1 81 8G$...+ #;Gw88#)**F)**F9?8H,W5GG,gg.>?? C C1q!$)$)DDDC,W5<>1vFFFF.w7>>tDDDDt1':;; 8.w7c1 81 8r/c|jr[|d|d|jd|d|jd|jr|jdD]M}|tjvr|j|d}n |j|}|d|dd|N|jdD]M}|tjvr|j|d}n |j|}|d|dd |N|jrct|j D]<\}}|d |dd |d j |=|j s|j rP|j D]J}t|jd |j ||jt|jd|j ||jIdSdS)Nz( Headers have different number of cards: a: r b: r z Extra keyword 8z in a: z in b: z$ Inconsistent duplicates of keyword rz) Occurs {} time(s) in a, {} times in (b)r)indr)rrcrr_commentary_keywordsr'r(rritemsrRrrrreport_diff_keyword_attrr)r*)r,rvalcounts r-rVzHeaderDiff._reportsj  " @ MMD E E E MM>$"9!"<>> ? ? ? MM>$"9!"<>> ? ? ?   M-a0 M Md777&/!,CC&/C KKKKCKKLLLL-a0 M Md777&/!,CC&/C KKKKCKKLLLL  ' "()E)K)K)M)M"N"N   SWSSSSTTT F?FN  # t'A /  (M, )M.     r/rrs@r-rrs((\  ||||||Bd8d8d8L)))))))r/rc0eZdZdZdfd ZdZdZxZS)ra Diff two image data arrays (really any array from a PRIMARY HDU or an IMAGE extension HDU, though the data unit is assumed to be "pixels"). `ImageDataDiff` objects have the following diff attributes: - ``diff_dimensions``: If the two arrays contain either a different number of dimensions or different sizes in any dimension, this contains a 2-tuple of the shapes of each array. Currently no further comparison is performed on images that don't have the exact same dimensions. - ``diff_pixels``: If the two images contain any different pixels, this contains a list of 2-tuples of the array index where the difference was found, and another 2-tuple containing the different values. For example, if the pixel at (0, 0) contains different values this would look like:: [(0, 0), (1.1, 2.2)] where 1.1 and 2.2 are the values of that pixel in each array. This array only contains up to ``self.numdiffs`` differences, for storage efficiency. - ``diff_total``: The total number of different pixels found between the arrays. Although ``diff_pixels`` does not necessarily contain all the different pixel values, this can be used to get a count of the total number of differences found. - ``diff_ratio``: Contains the ratio of ``diff_total`` to the total number of pixels in the arrays. rrrsc||_||_||_d|_g|_d|_d|_t||dS)a Parameters ---------- a : BaseHDU An HDU object. b : BaseHDU An HDU object to compare to the first HDU object. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rtol : float, optional The relative difference to allow when comparing two float values either in header values, image arrays, or table columns (default: 0.0). Values which satisfy the expression .. math:: \left| a - b \right| > \text{atol} + \text{rtol} \cdot \left| b \right| are considered to be different. The underlying function used for comparison is `numpy.allclose`. .. versionadded:: 2.0 atol : float, optional The allowed absolute difference. See also ``rtol`` parameter. .. versionadded:: 2.0 rprN) rrrdiff_dimensions diff_pixels diff_ratio diff_totalrr.)r,r'r(rrrrs r-r.zImageDataDiff.__init__s[H!   !  Ar/cjjjjkrjjjjf_dSt jjjtjs.t jjjtjsd}d}nj}j }tjj||}t|d_ j dkrdSj dkrj }nj }fdtt|d|D_t#j t#tjjz _dS)Nr)rrcHg|]}|j|j|ffSrp)r'r()rErr,s r-rz'ImageDataDiff._diff..Hs?   46#;s , -   r/)r'shaper(r(np issubdtypedtypeinexactrrr rr+rrrr)floatflatr*)r,rrdiffsrs` r-r+zImageDataDiff._diff&sI 6<46< ' '$(FL$&,#?D  F M$&, 3 3 }TV\2:66 DDD9D9D"46464HHHeAh-- ?a   F =1  HH}H    c5k1h77    005TV[9I9I3J3JJr/c X|jrddt|jdD}ddt|jdD}|d|d||d||d dS|jsdS|jD]o\}}d t|D}|d |d t |d|d|j|jdz|j|j p|j |j kr|d|d |j |j dS)Nz x c34K|]}t|VdSrerrEds r-rGz(ImageDataDiff._report..P(QQ!s1vvQQQQQQr/rc34K|]}t|VdSrer8r9s r-rGz(ImageDataDiff._report..Qr;r/r z Data dimensions differ:rr& No further data comparison performed.cg|]}|dzS)r rp)rExs r-rz)ImageDataDiff._report.._s444qQU444r/z Data differs at rrY indent_widthrr ...z. {} different pixels found ({:.2%} different).)r(rreversedrcr)r r)r*rrr+rrRr*)r,dimsadimsbindexrs r-rVzImageDataDiff._reportNs   JJQQx8LQ8O/P/PQQQQQEJJQQx8LQ8O/P/PQQQQQE MM4 5 5 5 MM/%// * * * MM/%// * * * MMB C C C F  F!-  ME644HUOO444E MM6e666 7 7 7 q q  !\A-YY       ?T] * * MM& ! ! ! < C C       r/)rrrsrsrrs@r-rrsj>111111f&K&K&KP" " " " " " " r/rc4eZdZdZdfd ZfdZdZxZS)ra `RawDataDiff` is just a special case of `ImageDataDiff` where the images are one-dimensional, and the data is treated as a 1-dimensional array of bytes instead of pixel values. This is used to compare the data of two non-standard extension HDUs that were not recognized as containing image or table data. `ImageDataDiff` objects have the following diff attributes: - ``diff_dimensions``: Same as the ``diff_dimensions`` attribute of `ImageDataDiff` objects. Though the "dimension" of each array is just an integer representing the number of bytes in the data. - ``diff_bytes``: Like the ``diff_pixels`` attribute of `ImageDataDiff` objects, but renamed to reflect the minor semantic difference that these are raw bytes and not pixel values. Also the indices are integers instead of tuples. - ``diff_total`` and ``diff_ratio``: Same as `ImageDataDiff`. rrcld|_g|_t|||dS)aF Parameters ---------- a : BaseHDU An HDU object. b : BaseHDU An HDU object to compare to the first HDU object. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rp)rN)r( diff_bytesrr.)r,r'r(rrs r-r.zRawDataDiff.__init__s9" " A11111r/ct|jr+|jdd|jddf|_d|jD|_|`dS)Nrr c(g|]\}}|d|fSrrp)rEr?ys r-rz%RawDataDiff._diff..s$BBBAAaD!9BBBr/)rr+r(r)rI)r,rs r-r+zRawDataDiff._diffsp     $Q'*$Q'*$D CB1ABBB    r/c Z|jrt|d|d|jdd|d|jdd|ddS|jsdS|jD]V\}}|d|d t|d|d|j|jdz|j|j W|d |d |j |j dS) Nz Data sizes differ:rrz bytesrr r=z Data differs at byte rr@rBz- {} different bytes found ({:.2%} different).) r(rcrIr r)r*rrrRr+r*)r,rFrs r-rVzRawDataDiff._reports[    MM/ 0 0 0 MMA$"6q"9AAA B B B MMA$"6q"9AAA B B B MMB C C C F  F!_  ME6 MM;5;;; < < < q q  !\A-YY       f ; B B       r/)rrrrs@r-rrsso*222222,            r/rc8eZdZdZgdddffd ZdZdZxZS)ra Diff two table data arrays. It doesn't matter whether the data originally came from a binary or ASCII table--the data should be passed in as a recarray. `TableDataDiff` objects have the following diff attributes: - ``diff_column_count``: If the tables being compared have different numbers of columns, this contains a 2-tuple of the column count in each table. Even if the tables have different column counts, an attempt is still made to compare any columns they have in common. - ``diff_columns``: If either table contains columns unique to that table, either in name or format, this contains a 2-tuple of lists. The first element is a list of columns (these are full `Column` objects) that appear only in table a. The second element is a list of tables that appear only in table b. This only lists columns with different column definitions, and has nothing to do with the data in those columns. - ``diff_column_names``: This is like ``diff_columns``, but lists only the names of columns unique to either table, rather than the full `Column` objects. - ``diff_column_attributes``: Lists columns that are in both tables but have different secondary attributes, such as TUNIT or TDISP. The format is a list of 2-tuples: The first a tuple of the column name and the attribute, the second a tuple of the different values. - ``diff_values``: `TableDataDiff` compares the data in each table on a column-by-column basis. If any different data is found, it is added to this list. The format of this list is similar to the ``diff_pixels`` attribute on `ImageDataDiff` objects, though the "index" consists of a (column_name, row) tuple. For example:: [('TARGET', 0), ('NGC1001', 'NGC1002')] shows that the tables contain different values in the 0-th row of the 'TARGET' column. - ``diff_total`` and ``diff_ratio``: Same as `ImageDataDiff`. `TableDataDiff` objects also have a ``common_columns`` attribute that lists the `Column` objects for columns that are identical in both tables, and a ``common_column_names`` attribute which contains a set of the names of those columns. rrrscBt||_||_||_||_g|_t|_d|_d|_d|_ g|_ d|_ g|_ d|_ d|_t||dS)av Parameters ---------- a : BaseHDU An HDU object. b : BaseHDU An HDU object to compare to the first HDU object. ignore_fields : sequence, optional The (case-insensitive) names of any table columns to ignore if any table data is to be compared. numdiffs : int, optional The number of pixel/table values to output when reporting HDU data differences. Though the count of differences is the same either way, this allows controlling the number of different values that are kept in memory or output. If a negative value is given, then numdiffs is treated as unlimited (default: 10). rtol : float, optional The relative difference to allow when comparing two float values either in header values, image arrays, or table columns (default: 0.0). Values which satisfy the expression .. math:: \left| a - b \right| > \text{atol} + \text{rtol} \cdot \left| b \right| are considered to be different. The underlying function used for comparison is `numpy.allclose`. .. versionadded:: 2.0 atol : float, optional The allowed absolute difference. See also ``rtol`` parameter. .. versionadded:: 2.0 rprN)rrrrrcommon_columnscommon_column_names diff_rowsdiff_column_count diff_columnsdiff_column_attributesdiff_column_namesrr*r+rr.)r,r'r(rrrrrs r-r.zTableDataDiff.__init__sP!//     #&55 !#')#"$ Ar/c D jj}jj}t|t|kr#t|t|f_d|D}d|D}djvrdSdjD}|D]$}|}||vr||=||vr||=%t|}t|}t| |tj d_ dj D_d||D}d ||D}|s|r||f_ggf_|rOjjD]B} | j|vr%jd | jC|rOjjD]B} | j|vr%jd | jCtjtjkr/tjtjf_dStjtjcxkrd krnndSj D]i} | j} | |vr!|| } || } t*D]n\} }t-| | d}t-| | d}t/||r7j| j| f||ffoj| jj| jt5jjt4jrBt5jjt4jrt=jj }nVd | j!vs d| j!vr,fdtEtDf}nt5j#k}xj$tt|d z c_$j%d krz'TableDataDiff._diff..K$222q222r/cBi|]}|j|SrprZrs r-r\z'TableDataDiff._diff..Lr]r/rc6h|]}|Srp)r[)rEfs r-r{z&TableDataDiff._diff..Us ???q???r/r)keyc@h|]}|jSrprZrEcols r-r{z&TableDataDiff._diff..fs$#T#T#TCHNN$4$4#T#T#Tr/cBi|]}|j|SrprZrcs r-r\z'TableDataDiff._diff..hs3   &)CHNN  c   r/cBi|]}|j|SrprZrcs r-r\z'TableDataDiff._diff..ks3   &)CHNN  c   r/rr r PQcpg|]2}tj||jj0|3S)r )r/allcloserr)rErarraarrbr,s r-rz'TableDataDiff._diff..sU!{ ItCytyty   r/)*r'columnsr(rrTrr[rrrr operator attrgetterrQrRr rUrWrrrS _COL_ATTRSr:rrVrxr/r0r1floatingr rrrRrwherer+rrfieldsr3r*)r,colsacolsbr fieldname colsa_set colsb_setleft_only_columnsright_only_columnsrd name_lowercolacolbrF_valavalbr5 max_diffs last_seen_idxr total_valuesrkrls` @@r-r+zTableDataDiff._diff@s u::U # #&)%jj#e**%=D "32E22222E222 $$ $ $ F@?D,>??? ' % %I!))IE!!)$E!!)$ ''  '' $  " "9 - -83Fv3N3N   $U#T@S#T#T#T   -6-A-A)-L-L     -6-A-A)-L-L     . 2 .!24F GD &("XD "  ?v~ ? ?8>>##'888*1-44SX>>>  ?v~ ? ?8>>##'999*1-44SX>>> tv;;#df++ % %!$&kk3tv;;7DN F tv;;#df++ * * * * * * * * * F&8 S8 SC))J]**$D$D%  atT400tT400tT**/66(..**D1D$<@6#(#D6#(#D}TZ55 /"- BK;; /+4DIDIVVV ""cSZ&7&7#(T#3#3.. OOs3uQx==11 1OO}!!t'((DM99!MC0@,A,AA aMM  MeAh955 S S-'' #  ''#(C49d3i:P(QRRRR S46{{S)<%=%== 0053F3FFr/c "|jr[|d|d|jd|d|jd|jr|jdD]H}|jd|j}|d|d|dI|jdD]H}|jd|j}|d|d|d It t}|jD]d\}}|\}}|d |d ||d t|d|d|j |j dz|j |j e|jrr|d|d|jd|d|jd|ddS|jsdS|jD]Z\}}|dj|t|d|d|j |j dz|j |j [|jr6|j|jkr&|d|j|jz d|j|jkr|d|d|j|jdS)Nz) Tables have different number of columns:rrrr z Extra column z of format z in az in bz Column  has different rr@z Table rows differ:r=z" Column {} data differs in row {}:rBz additional difference(s) found.z= {} different table data element(s) found ({:.2%} different).)rTrcrWrUr[rRdictrprVr r)r*rrrSrrr+r*) r,rrR col_attrscol_attrvalsrFindxrs r-rVzTableDataDiff._reportsv  ! ? MME F F F MM=$"8";== > > > MM=$"8";== > > >  ! O.q1 O O*1-djjll;B MtMMMMMNNNN.q1 O O*1-djjll;B MtMMMMMNNNN$$ #9  NHd!JD$ MMLTLL)D/LLL M M M QQ !\A-YY       >  MM/ 0 0 0 MM5$."355 6 6 6 MM5$."355 6 6 6 MMB C C C F  F!,  LD& MME>EtL M M M q q  !\A-YY           ? ? MMXt6XXX    ?T] * * MM& ! ! ! K R R       r/rrs@r-rrs--^,.3CCCCCCJPGPGPGd@ @ @ @ @ @ @ r/rc ||vr||}t|D]k\}}||dkrd}n d|dzd}|td|d|d |d |t|d|d||dz jdSdS) zk Write a diff between two header keyword values or comments to the specified file-like object. Nrr[r ]z Keyword rrz: )rYrA) enumeraterarr ) rYrFr5rr rrr$dups r-r#r#s %W~!$ V VHC{axx$#'nnn MM"HHH3HHtHHH    s1vs1vwSSTW U U U U U V Vr/rL)/rmrrrTrnrMos.pathr collectionsrinspectr itertoolsrnumpyr/astropyrastropy.utils.diffrrr r astropy.utils.miscr rr r hdu.hdulistrr hdu.tablerrrutilr__all__rpr%rrrrrrr#rpr/r-rs   ###### 322222""""""""+*******$$$$$$    I"I"I"I"I"I"I"I"X|D|D|D|D|Dy|D|D|D~GJGJGJGJGJiGJGJGJT{{{{{{{{H ] ] ] ] ] I] ] ] @U U U U U -U U U pG G G G G IG G G T VVVVVVr/