ó šßÈ[c@sddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZd dlmZdefd„ƒYZdS(iÿÿÿÿNi(tColumn(tFITSDifft HeaderDifft ImageDataDifft TableDataDifftHDUDifftreport_diff_values(tHDUListt PrimaryHDUtImageHDU(t BinTableHDU(tHeaderi(tcatch_warnings(tAstropyDeprecationWarning(trange(tfitsi(t FitsTestCasetTestDiffcBs‚eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!d „Z"d!„Z#d"„Z$d#„Z%d$„Z&d%„Z'd&„Z(d'„Z)d(„Z*d)„Z+RS(*cCs„tddd gƒ}|jƒ}t||ƒjs9t‚t|jƒ|jƒƒjs]t‚tjtƒtddƒWdQXdS( NtAitBitCi(Ri(Ri(Ri( R tcopyRt identicaltAssertionErrorttostringtpytesttraisest TypeError(tselfthathb((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_identical_headerss  $cCsHtdd d gƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_slightly_different_headers!s  cCs`td d d gƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_common_keywords's    cCs{tddd gƒ}|jƒ}|d=t||ƒ}|j sGt‚|jd ks\t‚|jddgkswt‚dS( NRiRiRi(Ri(Ri(Ri(ii(R RRRRtdiff_keyword_countR#(RRRtdiff((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_keyword_count.s cCstdddgƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_keywords9s     cCsmtdd d gƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_keyword_valuesDs   cCsptd d d gƒ}|jƒ}d|jdlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_keyword_commentsLs  cCs€td d d gƒ}|jƒ}|jd ƒ|jd ƒt||ƒ}|j sZt‚|jiddgd6ks|t‚dS(NRiRiRiii(Ri(Ri(Ri(Ri(Ri(ii(R RtappendRRRR,tNone(RRRR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt,test_different_keyword_values_with_duplicateUs   cCsÕtdddgƒ}|jƒ}|jdƒ|jdƒ|jdƒ|jdƒt||ƒ}|j stt‚|jiks‰t‚|jidd6dd6dd6ks³t‚|jƒ}d |ksÑt‚dS(NRiRiRis comment 1s comment 2is comment 3is comment 4sTInconsistent duplicates of keyword 'A' : Occurs 3 time(s) in a, 1 times in (b)(Ri(Ri(Ri(Ris comment 1(Ris comment 2(Ris comment 3(Ris comment 4(ii(ii(ii( R RR1RRRR,tdiff_duplicate_keywordstreport(RRRR&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt"test_asymmetric_duplicate_keywords^s     $ cCsétd d dgƒ}|jƒ}d|d@Rg½7†@grÄZ| @g zo @trtolgíµ ÷Æ°>gñh㈵øä>(Ri(Rg9b->@(Rg½7†@(g9b->@grÄZ| @(g½7†@g zo @(g9b->@grÄZ| @(R RRRRR,(RRRR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_floating_point_rtolos   #cCsÌtd ddgƒ}|jƒ}d|dR7gñh㈵øä>tatolg¢&ú|”ç>(Ri(Rgð?(Rg(gð?grÄZ| ð?(ggíµ ÷Æ°>(ggíµ ÷Æ°>(gð?grÄZ| ð?(gð?grÄZ| ð?(R RRRRR,(RRRR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_floating_point_atol~s4   #c CsHtddgƒ}|jƒ}d|dt‚Wd QXd S(scVerify uses of tolerance and rtol. This test should be removed in the next astropy version.Rgð?Rgš™™™™™¹?grÄZ| ð?gP:‘`ª™¹?t tolerancegíµ ÷Æ°>iso"tolerance" was deprecated in version 2.0 and will be removed in a future version. Use argument "rtol" instead.NR7gñh㈵øä>(Rgð?(Rgš™™™™™¹?(gš™™™™™¹?gP:‘`ª™¹?(gð?grÄZ| ð?( R RR R RtcategoryRtstrtmessageR,R(RRRt warning_linesR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_deprecation_tolerancešs    c CsÜtjjdtƒÁtd d d gƒ}|jƒ}d|d<|d|dks[t‚t||ƒ}|jsyt‚|j iksŽt‚t||dtƒ}|j s³t‚|j id gd6ksÒt‚WdQXdS( Ntstrip_header_whitespaceRiRiRsA t ignore_blanks(Ri(Ri(RsA (sA R( Rtconftset_temptFalseR RRRRR,(RRRR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_ignore_blanksµs  cCs!td d d gƒ}td ddddgƒ}|jƒ}|jƒ|jƒt||ƒjsht‚t||ƒjs€t‚t||ƒjs˜t‚t||dtƒj s·t‚t||dtƒj sÖt‚t||dtƒjsôt‚|jƒt||dtƒj st‚d S(s`Test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/152 Ignore blank cards. RiRiRittignore_blank_cardsN(Ri(Ri(Ri(Ri(RGRG(Ri(RGRG(Ri(R RR1RRRRE(RRRthc((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_ignore_blank_cardsÆs    cCstdddgƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_ignore_keyword_valuesås    cCs$tdddgƒ}|jƒ}d|jdlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_ignore_keyword_commentsùs    cCsgtjdƒjdƒ}tjdƒjdƒ}t||ƒ}|jsNt‚|jdksct‚dS(Nidi i(i i (i i (tnptarangetreshapeRRRt diff_total(RtiatibR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_trivial_identical_images s cCsctjdƒd}tjdƒd}t||ddƒ}|jsJt‚|jdks_t‚dS( Ni gñh㈵øä>gñh㈵øô>R7g-Cëâ6?i(i i (i i (RQtonesRRRRT(RRURVR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt(test_identical_within_relative_tolerances cCstjd ƒd}tjd ƒd}t||ddƒ}|j sKt‚|jdks`t‚t||ddƒ}|js„t‚|jdks™t‚dS( Ni gñh㈵øä>gñh㈵øô>R7g-Cëâ6?idR9i(i i (i i (RQtzerosRRRRT(RRURVR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt(test_identical_within_absolute_tolerancescCsitjdƒd}tjdƒd}t||ddddƒ}|jsPt‚|jdkset‚dS( Ni gñh㈵øä>gñh㈵øô>R7R9i(i i (i i (RQRZRRRRT(RRURVR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt#test_identical_within_rtol_and_atol%s cCsjtjdƒd}tjd ƒd}t||ddddƒ}|j sQt‚|jdksft‚dS( Ni gñh㈵øä>gñh㈵øô>R7R9gíµ ÷Æ°>id(i i (i i (RQRZRRRRT(RRURVR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt'test_not_identical_within_rtol_and_atol,s cCs’tjdƒjdƒ}tjd|ƒ}|j|jdƒƒtj|jdƒƒ}tj|jdƒƒ}t||ƒ}|j sŽt ‚dS(s¸Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/189 For this test we mostly just care that comparing to compressed images does not crash, and returns the correct results. Two compressed images will be considered identical if the decompressed data is the same. Obviously we test whether or not the same compression was used by looking for (or ignoring) header differences. gY@i tdatas test.fitsN(i i ( RQRRRSRt CompImageHDUtwritetottemptopenRRR(RR^thduthdulathdulbR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_identical_comp_image_hdus3s cCsÌtjdƒjd ƒ}tjdƒd}t||ƒ}|j sJt‚|jd d fkset‚|jdkszt‚|jƒ}d|ks˜t‚d|ksªt‚d|ks¼t‚dsÈt‚dS( Nidi iisData dimensions differs a: 10 x 10sb: 100s%No further data comparison performed.(i i (i i (id( RQRRRSRRRtdiff_dimensionsRTR5(RRURVR&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_dimensionsEs cCsÍtjdƒjd ƒ}tjdƒjd ƒ}d|d lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_pixelsVs  c Csotddddttgƒ}tdddddgdggƒ}td dd d d dddd dgddddggƒ}tddddddddgƒ}tddddddgƒ}tddddddddgƒ}td ddd!d"dddgƒ}td#dd dd8d9gƒ}td&dd'dd:d;gƒ} tddd,dddgd dggƒ} ||||||||| | g } tj| ƒ} tjg| D]} | jƒ^q®ƒ}t| j|jƒ}|jsít ‚t |j ƒd-kst ‚|j t d.d/d0d1d2d3d4d5d6d7g ƒksAt ‚|jdksVt ‚|jdkskt ‚dS(<NRtformattLtarrayRtXiiRt4Itdims(2, 2)iiiiiiR"tJtbscaleg@R(tA3tabctdefR)tunittmggð?tGtbzerogš™™™™™¹¿tHyð?y@tItMg@y@g@y@sPI(2)i taRMtctdtetftgthtitjyð?y@@y@@y@@(RtTrueRER t from_columnsRRR^RRtlentcommon_columnstcommon_column_namestsetRiRT(Rtc1tc2tc3tc4tc5tc6tc7tc8tc9tc10tcolumnsttaRttbR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_identical_tablesbs*$'$$$*$(3cCs…tdddƒ}tdddƒ}tj||gddƒ}tj|gƒ}tj|gƒ}t||ƒ}|jst‚dS(s  Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/178 Ensure that diffing tables containing empty data doesn't crash. R"RlRrR(tnrowsiN(RR RˆRRRRR(RRRŽtthduRdReR&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_diff_empty_tables|sc Cs¥tddddttgƒ}tdddddgdggƒ}td dd d d dddd dgddddggƒ}tdddddgdggƒ}td dd d d ddd ddgddddggƒ}tj|||gƒ}tj|||gƒ}t|j|jddd gƒ}|js>t‚t |j ƒdksYt‚|j t dgƒkswt‚|j dksŒt‚|jdks¡t‚dS(NRRlRmRnRRoiiRRpRqs(2, 2)iiiiiiit ignore_fieldsR~(RR‡RER RˆRR^RRR‰RŠR‹RŒRiRT( RRRŽRRR‘R˜R™R&((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_ignore_table_fieldss$'$'!cCsatddddttgƒ}tddddttgƒ}tddddttgƒ}tj||gƒ}tj||gƒ}t|j|jƒ}|j s©t‚t |j ƒdksÄt‚|j t dgƒksât‚|j dgdgfkst‚|jd kst‚|jd ks-t‚|jƒ}d |ksKt‚d |ks]t‚dS( NRRlRmRnRRiR~isExtra column B of format L in asExtra column C of format L in b(RR‡RER RˆRR^RRR‰RŠR‹RŒtdiff_column_namesRiRTR5(RtcatcbtccR˜R™R&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt test_different_table_field_names£s! cCsvtddddttgƒ}tddddttgƒ}tddddttgƒ}tj|gƒ}tj|||gƒ}t|j|jƒ}|j s©t‚|j dks¾t‚t |j ƒdksÙt‚|j t d gƒks÷t‚|jgddgfkst‚|jd ks-t‚|jd ksBt‚|jƒ}d |ks`t‚d |ksrt‚d S(sh Test tables with some common columns, but different number of columns overall. RRlRmRnRRiiRMis) Tables have different number of columns:s a: 1 b: 3N(ii(RR‡RER RˆRR^RRtdiff_column_countR‰RŠR‹RŒR RiRTR5(RR¡R¢R£R˜R™R&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt!test_different_table_field_counts¸s ! c Csytddddttgƒ}tddddttgƒ}tddddtttgƒ}tddddtttgƒ}tj||gƒ}tj||gƒ}t|j|jƒ}|j sÍt‚|j d ksât‚t |j ƒdksýt‚|j dkst‚|j gks't‚|jƒ}d|ksEt‚d |ksWt‚d |ksit‚d sut‚d S(sh Test tables that are otherwise identical but one has more rows than the other. RRlRmRnRiisTable rows differsa: 2sb: 3s%No further data comparison performed.N((ii(RR‡RER RˆRR^RRR¥R‰RŠt diff_rowst diff_valuesR5( Rtca1tcb1tca2tcb2R˜R™R&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_table_rowsÓs"!! c CsÒtddddttgƒ}tdddddgdggƒ}td dd d d dddd dgddddggƒ}tddddddddgƒ}tddddddgƒ}tddddddddgƒ}td ddd!d"dddgƒ}td#dd dd>d?gƒ}td&dd'dd@dAgƒ} tddd,dddgd dggƒ} tddddttgƒ} tdddddgdggƒ} td dd d d dddd dgdddd-ggƒ} tddddddddgƒ}tdddddd.gƒ}tddddddddgƒ}td ddd!d"ddd/gƒ}td#dd ddBdCgƒ}td&dd'ddDdEgƒ}tddd,ddd gddggƒ}tj||||||||| | g ƒ}tj| | | |||||||g ƒ}t|j|jd1d2ƒ}|j sat‚|j dFksvt‚|j ddGttffks›t‚|j ddHdgdgffksÆt‚|j d ddIksãt‚|j d ddddgddggkj ƒst‚|j d ddddgdd-ggkj ƒsUt‚|j ddJdKfkstt‚|j ddLdMfks“t‚|j ddNdOfks²t‚|j ddPdQfksÑt‚|j ddRdSfksðt‚|j d-dTdUfkst‚|j d3dVdWdXffks4t‚|j d4dYdZd[ffksYt‚|j d5dd\ksvt‚|j d5ddddgkj ƒs£t‚|j d5dddd gkj ƒsÐt‚|j d6dd]ksít‚|j d6ddd dgkj ƒst‚|j d6ddddgkj ƒsGt‚|j d7ks\t‚|j d8ksqt‚|jƒ}d9|kst‚d:|ks¡t‚d;|ks³t‚|jd<ƒdksÎt‚d=S(^sn Test diffing table data on columns of several different data formats and dimensions. RRlRmRnRRoiiRRpRqs(2, 2)iiiiiiR"RrRsg@gR(RtRuRvR)RwRxgð?RyRzgš™™™™™¹¿R{yð?y@R|R}g@y@g@y@sPI(2)itghig@g@tnumdiffsii i i i i gÍÌÌÌÌÌä?s8Column A data differs in row 0: a> True b> Falses;...and at 13 more indices. Column D data differs in row 0:s;13 different table data element(s) found (65.00% different)s more indicesNyð?y@@y@@y@@yð?ð?y@@y@@y@@((Ri(Ri(Ri(R"i(ig@(R(i(RvR®(R)i(ggð?(R)i(gð?g@(Ryi(gg@(Ryi(gð?g@(R{iyð?yð?ð?(R|iy@@y@@(Rri(Rri(RR‡RER RˆRR^RRt diff_columnsR¨tallRTRiR5tcount(RR©R«tca3tca4tca5tca6tca7tca8tca9tca10RªR¬tcb3tcb4tcb5tcb6tcb7tcb8tcb9tcb10R˜R™R&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_different_table_dataðst$'$$$*$'$$$*%+99%%---- cCs#tjdƒjd ƒ}td|ƒ}|j|jdƒƒ|j|jdƒƒt|jdƒ|jdƒƒ}|jsƒt‚|j ƒ}d|ks¡t‚d|ks³t‚d|ksÅt‚tjdƒ}t d|ƒ}t ||ƒ}|jst‚|j ƒ}d|kst‚d S( s5Test identicality of two simple, extensionless files.idi R^s testa.fitss testb.fitss Primary HDUs Extension HDUsNo differences found.N(i i ( RQRRRSRR`RaRRRR5R R(RR~RcR&R5tehdu((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_identical_files_basic8s !  cCsìtjdƒjd ƒ}td|ƒ}td|ƒ}t||gƒ}t|||gƒ}t||ƒ}|j s|t‚|j d ks‘t‚|j gks¦t‚|j ƒ}d|ksÄt‚d|ksÖt‚d|ksèt‚d S( sc Test files that have some identical HDUs but a different extension count. idi R^iis'Files contain different numbers of HDUss a: 2 b: 3s(No differences found between common HDUsN(i i (ii( RQRRRSRR RRRRtdiff_hdu_countt diff_hdusR5(RR~tphduRÄRdReR&R5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_partially_identical_files1Os c CsÉtjdƒjdƒ}td|ƒ}td|ƒ}td|dƒ}t|||gƒ}t|||gƒ}t||ƒ}|j s’t‚|j dks§t‚t |j ƒdksÂt‚|j dddksßt‚|j dd}|j st‚|j dkst‚|j dks*t‚|jdks?t‚|jjsQt‚|jdk sft‚|j} t| tƒs„t‚| j s”t‚| jdks©t‚| jgtdƒD]"} d| f| | dff^q¼ksít‚| jdkst‚| jdkst‚|jƒ} d| ks5t‚d| ksGt‚d | ksYt‚d | kskt‚d | ks}t‚x3tdƒD]%} d j| dƒ| ksŠt‚qŠWd | ksÅt‚dS(sQ Test files that have some identical HDUs but one different HDU. idi R^iigð?s Primary HDUsExtension HDU 2sExtension HDU 1sHeaders contain differencessData contains differencessData differs at [{}, 1]s/100 different pixels found (100.00% different).N(i i ((((((RQRRRSRR RRRRRÆR‰RÇt diff_extnamest diff_extverstdiff_extension_typest diff_headerst diff_dataR2t isinstanceRRgRjRRiRTR5Rl( RR~RÈRÄtehdu2RdReR&thdudifftdatadifftyR5((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_partially_identical_files2gsF > #cCs¦tƒ}tddƒ}t||gƒ}tddƒ}d|jdlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_partially_identical_files3–s0    cCs;tjd dtjƒ}d|d*tj|d)|jƒ}tjjddtjftjtjfgddd gƒjtj ƒ}|jƒ}t ||ƒj s§t ‚t ||ƒj s¿t ‚d|d d lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_diff_nans¸s2   - $!$cCsStjƒ}d}d}t|||ƒ|jƒ}|jdƒdksOt‚dS(sT Regression test for https://github.com/astropy/astropy/issues/4122 gð?s1.0tus7 (float) a> 1.0 (str) b> '1.0' ? + + N(tiotStringIORtgetvaluetlstripR(RR‚R~RMtout((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_diff_typesÝs   cCsntjƒ}tjdƒ}tjdƒ}t|||ƒ|jƒ}d|ksXt‚d|ksjt‚dS(sX Regression test for https://github.com/spacetelescope/PyFITS/issues/21 gA(Ç$wž?g øH‚$wž?sa>sb>N(RñRòRQtfloat32RRóR(RR‚R~RMRõ((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyttest_float_comparisonìs  cCs‡|jdƒ}td d d gƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt!test_file_output_from_path_stringþs   cCsƒ|jdƒ}td d d gƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt!test_file_output_overwrite_safetys  cCs£|jdƒ}td d dgƒ}|jƒ}d|dlib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt"test_file_output_overwrite_successs   c CsÌ|jdƒ}tdddgƒ}|jƒ}d|d(RRûRRRüRýR?((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyt%test_file_output_overwrite_vs_clobbers   (,t__name__t __module__RR R$R'R+R-R0R3R6R8R:R@RFRJRNRPRWRYR[R\R]RfRhRkRšRRŸR¤R¦R­RÃRÅRÉRÔRßRïRöRøRþRRR(((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyRsT                        H   / " %   (RñRtnumpyRQtcolumnRR&RRRRRRRcRRR t hdu.tableR RÜR t tests.helperR tutils.exceptionsR textern.six.movesRRRGRR(((s>lib/python2.7/site-packages/astropy/io/fits/tests/test_diff.pyts   .