U e{j@sddlZddlZddlZddlmZmZddlZddlZddl m Z ddl m Z ddl mZddlmZddlmZmZddlmZdd lmZd d lmZd d ZddZddZddZGdddeZGdddeZddZ dS)N)BytesIOStringIOfits)_pad) _pad_length) encode_ascii) VerifyError VerifyWarning)AstropyUserWarning)_NOT_OVERWRITING_MSG_MATCH) FitsTestCasecCsHtddg}t|}d|d<d|ks,td|d<|ddksDtdS) zQMake sure that operations on a shallow copy do not alter the original. #4990.)ar )br dcrrr N)rHeadercopyAssertionError)original_headerZ copied_headerr`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/fits/tests/test_header.pytest_shallow_copys   rcCsNtdg}tj|dd}d|d<|ddks2td|d<|ddksJtdS) zVMake sure that creating a Header from another Header makes a copy if copy is True.)r T)rrrrNrrr)r new_headerrrrtest_init_with_header%s  rcCs<dddddd}t|}|D]}||||kstqdS)N )rrrder)dict1h1irrrtest_init_with_dict2s r)csJddtdDt}t|tfddtDsFtdS)NcSsg|]\}}||fqSrr).0jr(rrr ;sz.test_init_with_ordereddict..abcdefghijklmnopqrstuvwxyzc3s&|]\}}||dkVqdS)r Nr)r*r(valr'Zlist1rr @sz-test_init_with_ordereddict..) enumerate collections OrderedDictrrallr)r&rr/rtest_init_with_ordereddict9s  r5c@seZdZdZddZejdddgddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!d;d<Z"d=d>Z#d?d@Z$dAdBZ%dCdDZ&dEdFZ'dGdHZ(dIdJZ)dKdLZ*dMdNZ+dOdPZ,dQdRZ-dSdTZ.dUdVZ/dWdXZ0dYdZZ1d[d\Z2d]d^Z3d_d`Z4dadbZ5dcddZ6dedfZ7dgdhZ8didjZ9dkdlZ:dmdnZ;dodpZdudvZ?dwdxZ@dydzZAd{d|ZBd}d~ZCddZDddZEddZFddZGddZHddZIddZJddZKejdeLMeLNgddZOddZPddZQddZRddZSddZTddZUddZVddZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZfddZgddÄZhddńZiddDŽZjddɄZkdd˄Zldd̈́ZmddτZnddфZoddӄZpddՄZqddׄZrddلZsddۄZtdd݄Zudd߄ZvddZwddZxddZyddZzddZ{ddZ|ddZ}ddZ~ddZddZddZddZddZddZddZddZddZddZddZddZdd Zd d Zd d ZddZddZddZdS(TestHeaderFunctionszTest Header and Card objects.cCshtddg}|ddd|ks&td|ks2t|ddksBt|ddksRt|jddksdtdS)z*Test renaming keyword with rename_keyword.ABCDEFr8r9rr:N)rrZrename_keywordrcommentsselfheaderrrrtest_rename_keywordFs   z'TestHeaderFunctions.test_rename_keywordkeyr8rcCstddg}||kst||dks*t||dkstdd}t|tdks ttjd}|jdks:tdS)z(Test Card constructor with boolean valuerQTzABC = Tz ABC = FFN)rrLrVrrrRrSrNrrrtest_boolean_value_cardss  z+TestHeaderFunctions.test_boolean_value_cardcCs$tdd}t|tdks tdS)z-Test Card constructor with long integer valueZlong_intltNxwb B z,LONG_INT= -467374636747637647347374734737437NrUrNrrrtest_long_integer_value_card|s z0TestHeaderFunctions.test_long_integer_value_cardcCsDtdd}t|tdkr@t|tdkr@t|tdks@tdS)z/Test Card constructor with floating point valueZfloatnumgQ zFLOATNUM= -4.6737463674763E+32zFLOATNUM= -4.6737463674763E+032NrUrNrrrtest_floating_point_value_cards  z2TestHeaderFunctions.test_floating_point_value_cardcCsPtdd}td}td}td}t||krLt||krLt||ksLtdS)z(Test Card constructor with complex valuerQym8R{.FakeHeadercSsdd|DS)NcSsg|] }|dqS)rrr*lrrrr,szLTestHeaderFunctions.test_update..FakeHeader.keys..rrjrrrkeyssz8TestHeaderFunctions.test_update..FakeHeader.keyscstfdd|DS)Nc3s&|]}|dkr|ddVqdS)rr NrrrDrrr0s zRTestHeaderFunctions.test_update..FakeHeader.__getitem__..)next)rArDrrr __getitem__sz?TestHeaderFunctions.test_update..FakeHeader.__getitem__N)__name__ __module__ __qualname__rrrrrr FakeHeadersrFOOBARBAZr8r )r9rergr8r9rg)r this is a commentHELLOrr)NAXIS1NAXIS2rrrr re) listrrupdatesetrrr?values)rArrBZtmphdrrrr test_updates  zTestHeaderFunctions.test_updatec Cs*t|d}|djddi|djddks:t|djjddksRttt |djddiW5QRX| | d | tj| d d d }d |djjd<| t| d }|djjdd kst|djj ddd t|djjddkst| dS)N arange.fitsrrrrrFOO2)rrZEXTRA test.fitsrmodeQUXr{z COMMENT 0)ropendatarBrrr?rarhriwritetotempclose add_commentrVr~striprAhdulrrrtest_update_comments  "z'TestHeaderFunctions.test_update_commentcCsVd}td|}t|td|ks(td}td|d}t|td|ksRtdS)Nz2A commentary card's value has no quotes around it.rvHISTORY z!A commentary card has no comment.COMMENTrgzCOMMENT rU)rAr.rrrrtest_commentary_cardss  z)TestHeaderFunctions.test_commentary_cardscCs,tjd}|jdkst|jdks(tdS)NzLCOMMENT card has no comments. / text after slash is still part of the value.zDcard has no comments. / text after slash is still part of the value.rK)rrLrRrSrrgrNrrr*test_commentary_card_created_by_fromstring(s z>TestHeaderFunctions.test_commentary_card_created_by_fromstringcCs$tjd}t|tdks tdS)NzHISTORY (1, 2)rrLrRrVrrrNrrr3test_commentary_card_will_not_parse_numerical_value4s zGTestHeaderFunctions.test_commentary_card_will_not_parse_numerical_valuecCs$tjd}t|tdks tdS)NzHISTORY = (1, 2)rrNrrrtest_equal_sign_after_column89s z1TestHeaderFunctions.test_equal_sign_after_column8cCsPtdd}t|tdks ttjt|}|jdks>t|jdksLtdS)NrKz / EXPOSURE INFORMATIONz% / EXPOSURE INFORMATION)rrLrVrrrRrMrSrNrrrtest_blank_keyword?s  z&TestHeaderFunctions.test_blank_keywordcCs(tdtjj}t|tdks$tdS)NZundefz UNDEF =)rrLcard UNDEFINEDrVrrrNrrrtest_specify_undefined_valueFsz0TestHeaderFunctions.test_specify_undefined_valuecCs$tjd}t|tdks tdS)NzABC = (8, 9)rrNrrr&test_complex_number_using_string_inputKs z:TestHeaderFunctions.test_complex_number_using_string_inputc CsPtjd}|jdksttjtjjddt |t dksBtW5QRXdS)Nzabc = + 2.1 e + 12g~BVerification reported errorsmatchzABC = +2.1E+12) rrLrRrSrrarbrcr rVrrAcapsysrrrr#test_fixable_non_standard_fits_cardPs z7TestHeaderFunctions.test_fixable_non_standard_fits_cardc Cs>tjd}tjtjjddt|dks0tW5QRXdS)NzGno_quote= this card's value has no quotes / let's also try the commentrrzPNO_QUOTE= 'this card''s value has no quotes' / let's also try the comment ) rrLrRrarbrcr rVrrNrrrtest_fixable_non_fscYs z(TestHeaderFunctions.test_fixable_non_fsccCs$tjd}t|tdks tdS)Nz ABC = z ABC =rrNrrr'test_undefined_value_using_string_inpuths z;TestHeaderFunctions.test_undefined_value_using_string_inputcCsLt}d|d<d|d<t|ddgks0tt|ddgksHtdS)NrrUNDEF)rr)rN)rrrrritemsr@rrrtest_undefined_keys_valuesms z.TestHeaderFunctions.test_undefined_keys_valuesc Cs^tjd}|jdkst|jdks(ttjtjj ddt |t dksPtW5QRXdS)NzXYZ= 100ZXYZrrrzXYZ = 100) rrLrRrMrrSrarbrcr rVrrrrrtest_mislocated_equal_signus z.TestHeaderFunctions.test_mislocated_equal_signc Csxtjd}tjtddt|tdks0tW5QRXtjd}tjtddt|tdksjtW5QRXdS)NzHISTO = (1, 2)zheader keyword is invalidrz HISTORY (1, 2)) rrLrRrarbr rVrrrrrrtest_equal_only_up_to_column_10s   z3TestHeaderFunctions.test_equal_only_up_to_column_10c Csrtjd}tt}|W5QRXd}d}t|dksBt|t |dj ksXt|t |dj ksntdS)NABC= a6zt||d<||d<|d|dksbt|d|dksvtdS)a Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/194 Test that a blank keyword ('') can be assigned a too-long value that is continued across multiple cards with blank keywords, just like COMMENT and HISTORY cards. rrKrfrrrvN)rrrprjoinrstrip)rArSrBrrrtest_blank_keyword_long_valuesz1TestHeaderFunctions.test_blank_keyword_long_valuecCsntddd}t}|j|||dt|d}|djjd}| t |dksjt dS)NrQrrz test_new.fitsrr) rrL PrimaryHDUrBappendrrrr~rrVr)rArhdurrrrtest_long_string_from_files z.TestHeaderFunctions.test_long_string_from_filecCs"tddd}t|dkstdS)NrQZlongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvaluelongstringvalueZnlongcommentlongcommentlongcommentlongcommentlongcommentlongcommentlongcommentlongcommentlongcommentlongcommentaABC = 'longstringvaluelongstringvaluelongstringvaluelongstringvaluelongstr&'CONTINUE 'ingvaluelongstringvaluelongstringvaluelongstringvaluelongstringvalu&'CONTINUE 'elongstringvalue&' CONTINUE '&' / longcommentlongcommentlongcommentlongcommentlongcommentlongcommeCONTINUE '' / ntlongcommentlongcommentlongcommentlongcomment rrNrrr!test_word_in_long_string_too_longs z5TestHeaderFunctions.test_word_in_long_string_too_longc CsRtjtdtdtd}tjtjjddt|dksDt W5QRXdS)Nz;abc = 'longstring''s testing & ' / comments in line 1zLcontinue 'continue with long string but without the ampersand at the end' /zNcontinue 'continue must have string value (with quotes)' / comments with ''. rrzABC = 'longstring''s testing continue with long string but without the &' CONTINUE 'ampersand at the endcontinue must have string value (with quotes)&' CONTINUE '' / comments in line 1 comments with ''. ) rrLrRrrarbrcr rVrrrrr%test_long_string_value_via_fromstrings& z9TestHeaderFunctions.test_long_string_value_via_fromstringcCsNtjtdtdtd}|jdks.t|jdksz:TestHeaderFunctions.test_missing_keyword..rzKeyword 'NAXIS' not found.rN)rrrarhKeyErrorrjrrrtest_missing_keywordsz(TestHeaderFunctions.test_missing_keywordcCs<t}d|d<d|kst|ddks,td|ks8tdS)NrrrlZ ABCDEFGHIrr@rrrtest_hierarch_card_lookups  z-TestHeaderFunctions.test_hierarch_card_lookupcCst}d|d<|d=dS)Nrr)rrr@rrrtest_hierarch_card_deletesz-TestHeaderFunctions.test_hierarch_card_deletec Cst}tjtjjddd|d<W5QRXd|d<d|d<tjtjjdd|ddW5QRX|d =tjtjjdd|ddW5QRX|d=t|dd kst dS) NrrrrlZabcdefghr_re) abcdefghijrr ) rrrarbrcr insertrrupperrr@rrr test_hierarch_card_insert_deletes*z4TestHeaderFunctions.test_hierarch_card_insert_deletec CsPd}t}tt.}|ddit|dks8td|ksDt|ddksTt|ddit|dksrt|dstd|ddit|dkstt|dkst|dstd|d d it|dkstt|dkst|d std |d d it|d kstt|dks0t|d sBtd |ddit|dksbt|t|dj kszt|ddit|dkst|dstd|ddit|dkst|dstd|dd it|dkst|dstd |dd it|dks0t|dsBtd W5QRXdS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/158 Tests several additional use cases for working with HIERARCH cards. a HIERARCH card will be createdzHIERARCH BLAH BLAHTESTArz BLAH BLAHTESTBTESTCr zHIERARCH blah blahTESTDz blah blahTESTErezBLAH BLAH BLAHrfzHIERARCH BLAH BLAH BLAHryzHIERARCH blah blah blahzblah blah blahrN) rrrarbr rrprrVrrAmsgrBrsrrrtest_hierarch_create_and_updatesJ z3TestHeaderFunctions.test_hierarch_create_and_updatec Csd}t}tt}|ddit|dks8td|ksDt|ddksTt|ddit|dksrt|dstd|ddit|dkst|dstd|d d it|dkstt|dkst|d std |d d it|d kstt|dks t|d s2td W5QRXt}tt<}|dditdd|Dt|dkst|t |dj kst|ddit|dkst|dstd|ddit|d kst|dstd|d d it|d ks tt|dks2t|d sDtd |d d it|dksdtt|dksvt|d std W5QRXdS)aD Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/158 Tests several additional use cases for working with HIERARCH cards, specifically where the keyword is fewer than 8 characters, but contains invalid characters such that it can only be created as a HIERARCH card. rzHIERARCH BLA BLArrzBLA BLArrr zHIERARCH bla blarzbla blarrecSsg|] }|jqSr)category)r*xrrrr,bszMTestHeaderFunctions.test_short_hierarch_create_and_update..rfN) rrrarbr rrprprintrVrrrrr%test_short_hierarch_create_and_update7sR  z9TestHeaderFunctions.test_short_hierarch_create_and_updatecs$tfdd}tt|dS)Ncs dd<dS)N)barZbazZquxrrrrrrtest}sz=TestHeaderFunctions.test_header_setitem_invalid..test)rrrarhri)rArrrrtest_header_setitem_invalidzs z/TestHeaderFunctions.test_header_setitem_invalidcCspt}d|d<d|d<|ddks(t|ddks8t|ddksHt|jddksZt|jddksltdS)N)rrrrrrrKrrrr?r@rrrtest_header_setitem_1tuplesz.TestHeaderFunctions.test_header_setitem_1tuplecCst}d|d<d|d<|ddks(t|ddks8t|ddksHt|jddksZt|jddkslt|jddks~tdS) Nrr)NNrrrrrKrr@rrrtest_header_setitem_2tuplesz.TestHeaderFunctions.test_header_setitem_2tuplecCs"t}d|d<|ddks td|d<|ddks8td}tjj|dd}tdtjj}||gtjj|d<|jd d d |d |d dkst|ddkst|ddkst|ddkst|ddkst|d dkstd|d<d|d <|j D]}|j tjjkstqdS)z Setting the value of a card to None should simply give that card an undefined value. Undefined value should map to None. rrNzUNDEF = DEFINED = 42 sepZUNDEF2ZUNDEF3)UNDEF5NzUndefined valueTendZUNDEF6ZDEFINEDrrr$ZUNDEF4) rrrrRrLrrextendrr~rS)rArBhstrrrrrtest_header_set_value_to_nones,    z1TestHeaderFunctions.test_header_set_value_to_nonecCs@tdg}|jddd|ddks*t|jddksr rer8r:r=rrfrr>r<r9rrrprrArBZ newheaderrrrtest_header_slices       z%TestHeaderFunctions.test_header_slicecCstdddg}d|dd<|ddks,t|ddksTest selecting a subsection of a header via wildcard matching.rPrDEFr ZABDreAB*rerr Nr/r0rrrtest_wildcard_slices z'TestHeaderFunctions.test_wildcard_slicecCsPtdddg}t|ddks$tt|ddks8tt|ddksLtd S) zp Regression test for issue where wildcards did not work on keywords containing hyphens. )ZDATEr )zDATE-OBSre)zDATE-FOOrfzDATE*rfzDATE?*rezDATE-*Nr/r@rrrtest_wildcard_with_hyphen sz-TestHeaderFunctions.test_wildcard_with_hyphencCstdddg}d|d<|ddks(t|ddks8td|d<|ddksPt|ddks`td d g|d<|dd ks|t|dd kstd S) z@Test assigning to a header slice selected via wildcard matching.r7r8r:r r;rrer2r3r4Nrr@rrrtest_wildcard_slice_assignments z2TestHeaderFunctions.test_wildcard_slice_assignmentcCs:tdddg}|d=t|dks&t|ddks6tdS)z@Test deleting cards from a header that match a wildcard pattern.r7r8r:r;r rNr/r@rrrtest_wildcard_slice_deletion-sz0TestHeaderFunctions.test_wildcard_slice_deletioncCs2tddddddg}|ddd d d gks.tdS) Nr7rurw)r9rf)rvry)rvrrvr reryrrr@rrrtest_header_history5s z'TestHeaderFunctions.test_header_historycCsBtddg}|d|ks"td|ks.tt|dks>tdS)Nr+r,r8r:r)rrclearrrpr@rrrtest_header_clearBs   z%TestHeaderFunctions.test_header_clearfitsextc Csrtt|g}d|djd<|djtt}|j| dddW5QRXd}|t |j ksnt dS)Nrr r temp.fitsT overwritez'XTENSION' card does not exist.) rZHDUListrrBrArarhr rrrVrSr)rArCZhdulisterrerr_msgrrrtest_header_clear_writeIs z+TestHeaderFunctions.test_header_clear_writecCsptjddg}d|kst|ddks,t|jddks>td|ksJt|ddksZt|jddksltdS)Nr8r9rKrrfromkeysrr?r@rrrtest_header_fromkeysSs  z(TestHeaderFunctions.test_header_fromkeyscCsrtjddgd}d|kst|ddks.t|jddks@td|ksLt|ddks\t|jddksntdS)Nr8r9r:rKrJr@rrrtest_header_fromkeys_with_value\s  z3TestHeaderFunctions.test_header_fromkeys_with_valuecCsBtjdgd}d|kst|ddks,t|jddks>tdS)Nr8r9r:r9r:rJr@rrr+test_header_fromkeys_with_value_and_commentes z?TestHeaderFunctions.test_header_fromkeys_with_value_and_commentcCstjdddgd}d|ks td|ks,td|ks8td|ksDt|ddksTt|ddksdt|ddkstt|ddkst|ddkstdS) Nr8r9r:)r8rr)r8rerre)rrrKrr@rrr$test_header_fromkeys_with_duplicatesks    z8TestHeaderFunctions.test_header_fromkeys_with_duplicatescCs*tddg}t|ddgks&tdSNr+r,)rrrrrr@rrrtest_header_itemswsz%TestHeaderFunctions.test_header_itemscCs6tddg}t||D]\}}||kstqdSrQ)rrziprrrArBrrrrrtest_header_iterkeys{sz(TestHeaderFunctions.test_header_iterkeyscCs:tddg}t|ddgD]\}}||ks tq dS)Nr+r,r9r<)rrrSrrrTrrrtest_header_itervaluessz*TestHeaderFunctions.test_header_itervaluesc CsDt|d*}t|djddddddd gks6tW5QRXdS) NrrSIMPLErrrrZNAXIS3r)rrrrrBrrrrrtest_header_keyss z$TestHeaderFunctions.test_header_keyscCstddddg}|}|dks&tt|dks6tt|ddd gksLt|d }|d ksbtt|d ksrtt|dd gkst|d }|dkstt|d kstt|d gksttt|jddS)Nr+r,r.Gr3r3rfr8r:r=r r<rerr9r) rrpoprrprrarh IndexError)rArBlastmidfirstrrrtest_header_list_like_pops     z-TestHeaderFunctions.test_header_list_like_popcCstddddg}tt|jddd|d}|d ksrfrZr3ry)rr setdefaultrrpr@rrrtest_setdefaults z#TestHeaderFunctions.test_setdefaultcCstddg}|ddd|ddks.t|ddks>td|ksJt|ddksZt|d dksjttddg}|jddd|ddkst|ddkstd|kst|ddkst|d dkstd S) zm Test adding new cards and updating existing cards from a dict using Header.update() r+r,r=rZr8r>r8rr>rN)rrrrr@rrrtest_update_from_dicts  z)TestHeaderFunctions.test_update_from_dictcCsttddg}|dtddg|ddks4t|ddksDtd|ksPt|ddks`t|d dksptd S) zv Test adding new cards and updating existing cards from an iterable of cards and card tuples. r+r,)r8r=r>rZr8r=rrN)rrrrLrr@rrrtest_update_from_iterables z-TestHeaderFunctions.test_update_from_iterablecCst}t}d|jd<|j|j7_t|jdkst|ddksNt|dt|dksht|ddksxt|ddkst|jd d d t|d kst|ddkst|dd kstdS)z Tests that blank cards can be appended, and that future appends will use blank cards when available (unless useblanks=False) r+r,ryrrKryr.r>rYFZ useblanksrr3N)rrrrprr@rrrtest_header_append_use_blankss z1TestHeaderFunctions.test_header_append_use_blankscCstddg}|dt|dks(tt|ddksrrKryr<rYFrrr3)rrrr rprr@rrrtest_header_insert_use_blankss z1TestHeaderFunctions.test_header_insert_use_blankscCs^tdddg}|ddt|ddks4t|ddksDt|jdd ksVt|jdd d d t|d dks~tt|ddkst|ddkst|dd|ddddgkst|jddd d |dddddgkst|ddt|ddkst|jddd d t|ddks@tt|ddksZtd S)!z Test that a keyword name or tuple can be used to insert new keywords. Also tests the ``after`` keyword argument. Regression test for https://github.com/spacetelescope/PyFITS/issues/12 )rr)r Comment 1)r Comment 3r)rreNumber of axesrrrer)rrTr{r rr)rr )r Comment 2rrrr)rre)r Comment 4rr)rTryr)rTrN)rrr rrrr?r@rrr!test_header_insert_before_keywords(    z5TestHeaderFunctions.test_header_insert_before_keywordc Cstddg}|dt|dks(tt|dgks:td|ksFttt|dW5QRX|jdddt|dksttddd g}|jddd d|kstt|dkstt|dgkst|d d kstdS) Nr+r,r:r r8r>TrErj)Z remove_allrr<) rrrIrprrrarhrr@rrr test_removes    zTestHeaderFunctions.test_removecCs$tddg}t|jdks tdS)Nr7)r9rZr3z A C DEF H)rrrr?rr@rrrtest_header_comments sz(TestHeaderFunctions.test_header_commentscCsztdddg}|jdd}t|ddgks2t|jddd}t|dddgksXt|jd }t|ddgksvtdS) NABr:r<EFrZr3AIJKr r3rrr<A*)rrr?rr)rArBsrrrtest_comment_slices_and_filters s z3TestHeaderFunctions.test_comment_slices_and_filterscCstdddg}d|jdd<t|jdddgks6t|j|djdksPt|j|djdksjt|j|d jdkst|jdd|jddd <t|jdddgkstd d g|jd <t|jd dd gkstdS)NrrrLr r<rrrrMNr)rrr?rrr~rGrgr@rrr test_comment_slice_filter_assignsz4TestHeaderFunctions.test_comment_slice_filter_assigncCsHt}ttd}|D] }||d<q|ddd|ddksFt|ddd|ddksft|ddd|ddkst|ddd|ddkst|dddd|dddkst|dddd|dddkst|dddd|dddkst|d d |jd d d t|ddd|ddksTtt|ddd|ddksztt|ddd|ddkstt|ddd|ddkstt|dddd|dddkstt|dddd|dddkstt|dddd|dddksDtdS)Nrrvr rfrzryrrerr7r;Tr%)rrrrangerr r)rArBindicesidxrrrtest_commentary_slicing$s(      $$& &&&&**z+TestHeaderFunctions.test_commentary_slicingcCst}d|d<d|d<d|d<d|d<d|d<|ddddgksFtd|dd <|ddddgkshtd|dd d<|ddddgkstd d g|ddd<|dd d dgkstdS) NrrrPrvBARNEYFREDr9ZGHIrr rrrr@rrrtest_update_commentaryCs z*TestHeaderFunctions.test_update_commentarycCsTt}d|d<d|d<d|d<|d|dks4td|d<|d|dksPtdS)a Regression test for an issue found in *writing* the regression test for https://github.com/astropy/astropy/issues/2363, where comparison of the list of values for a commentary keyword did not always compare correctly with other iterables. z hello worldrvrNrr@rrrtest_commentary_comparisonYsz.TestHeaderFunctions.test_commentary_comparisoncCs<t}d|d<d|d<d}||d<d|d<||d<t|d ksDtt|d dksXtt|jd d |dd ksztt|jdd |d dkst|jd|ddt|dkstt|jdd |dd kstt|jd d |d dks tt}| ddi| ddid}| || ddi| |t|jd ksht|jd j dks~tt|jd d |dd kstt|jdd |d dkst|j |ddt|jdkstt|jdd |dd kstt|jd d |d dks8tdS)NrrrrZZABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCrvrrrrerfrHryr{r^r ) rrrprrrVr~rrrr}rM)rArBZlongvalrrrtest_long_commentary_cardis<"&"(  $($z-TestHeaderFunctions.test_long_commentary_cardc Cs|d}t|dX}|dj|tjtt d|djj|ddW5QRX|djj|ddW5QRXt }|j dd i|jj |j |ddd |j|d d d t|d }t|dtjstW5QRXt }|j dd i|jj |j |ddddd|jks2td|jksBtd|jksRt|j|d d ddt|d ,}t|dkstd|djkstW5QRXdS)Nz header.txtrrrFrETrorr update_firstrignore output_verifyrrr EXTENSIONrWrrFrer )rrrrrB totextfilerarhOSErrorr rsrr' fromtextfiler isinstancerrrprAZ home_is_tempZheader_filenamerrZhdul2rrrtest_totxtfiles>   z"TestHeaderFunctions.test_totxtfilec Cs|d}t|dX}|dj|tjtt d|djj|ddW5QRX|djj|ddW5QRXt }|j dd i|jj |j |ddd |j|d d d t|d }t|dtjstW5QRXt }|j dd i|jj |j |ddddd|jks2td|jksBtd|jksRt|j|d d ddt|d ,}t|dkstd|djkstW5QRXdS)z: Repeat test_totxtfile, but with tofile() z header.fitsrrrFrETrorrrrrrrrWrrer N)rrrrrBtofilerarhrr rsrr'fromfilerrrrrprrrr test_tofiles>   zTestHeaderFunctions.test_tofilecCs*|d}tj|}|ddks&tdS)zBRegression test for https://github.com/astropy/astropy/issues/8711z scale.fitsZDATASETZ2MASSN)rrrrr)rArhdrrrr test_fromfiles  z!TestHeaderFunctions.test_fromfilec Cszt}d|d<d|d<d|d<t|dd"}|d d d |jDW5QRXtj|d}||ksvtd S) zRegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/122 Manually write a text file containing some header cards ending with newlines and ensure that fromtextfile can read them back in. rNr8r,r9)r<r=r:test.hdrrsr!css|]}t|VqdSrrVrr*rrrrr0sz?TestHeaderFunctions.test_header_fromtextfile..N rrrrwriterr~rr)rArBfrrrrtest_header_fromtextfiles&z,TestHeaderFunctions.test_header_fromtextfilec Cs~tddg}t|dd,}|ddd|jD|dW5QRXtj|d}d |ksnt||ksztd S) zRegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/154 Make sure that when a Header is read from a text file that the END card is ignored. r7r;rrsr!css|]}t|VqdSrrrrrrr0szMTestHeaderFunctions.test_header_fromtextfile_with_end_card..z ENDENDNr)rArBrrrrr&test_header_fromtextfile_with_end_cards z:TestHeaderFunctions.test_header_fromtextfile_with_end_cardcsxtddgfdd}tt|ddttjdtjtjdddttjtdttjdd S) z Regression test 2 for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/154 Manually adding an END card to a header should simply result in a ValueError (as was the case in PyFITS 3.0 and earlier). r7r;cs ||<dSrr)rvrrrsetitemsz9TestHeaderFunctions.test_append_end_card..setitemrrKTr%N) rrrarhrirr rpr)rArrrrtest_append_end_cards z(TestHeaderFunctions.test_append_end_cardc s^tjtddjfdd}|dd}tjtdd}tj |}|ksRt W5QRXt |d kslt |d d}tjtd d}tj |}|kst W5QRXt |d kst |d d}tjtd d}tj |}|kst W5QRXt |d kst |dd}tjtdd$}tjj |dd}|ks>t W5QRXt |d ksZt dS)aX Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/217 This tests the case where the END card looks like a normal card like 'END = ' and other similar oddities. As long as a card starts with END and looks like it was intended to be the END card we allow it, but with a warning. rrcsNjdddd}||7}|r0|dtt|7}t|trBt|St|SdS)NrKF)r#Zendcardpaddingr)tostringrrprbytesrr)r&padrZhorigrrinvalid_header.s zBTestHeaderFunctions.test_invalid_end_cards..invalid_headerzEND =Tz+Unexpected bytes trailing END keyword: ' ='rr z END = z/Unexpected bytes trailing END keyword: ' ='z END$%&%^*%*z5Unexpected bytes trailing END keyword: '\$%&%\^\*%\*'rFz(Missing padding to end of the FITS block)rN) rrnparangerBrarbr rrrrp)rArrrsrrrrtest_invalid_end_cards"sL         z*TestHeaderFunctions.test_invalid_end_cardsc Cs t}d|d<d|d<tj|tdd}||dt|dd}|}W5QRX| d d d }| d d d }t|dd}| |W5QRXt j tdd6}t|d}|ddkst|ddkstW5QRXt|dkstdS)z5 Test header with invalid characters rrZhellorr)rBrrrbshellouhéllolatin1sBARuBÀR test2.fitswbz1non-ASCII characters are present in the FITS filerzB?Rzh?llor N)rrrrrrrrreadreplaceencoderrarbr rrrp)rArrroutr\rsrrrtest_invalid_charactersls(z+TestHeaderFunctions.test_invalid_characterscCs^tdddg}|jdddt|dddgks4t|jr>t|jdd d t|dddgksbt|jrlt|jdddt|dddgkst|jrt|jddd t|dddgkst|jrt|jdddt|dddgkst|jrt|jdd dt|dddgkst|jr(t|jdd d t|dddgksNt|jrZtd S) z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/125 Ensures that a header is not modified when setting the position of a keyword that's already in its correct position. r+rNr,r9rebeforer8r:rr{{N)rrrrr _modifiedr@rrrtest_unnecessary_moves,      z)TestHeaderFunctions.test_unnecessary_movec Csd}tjj|dd}|ddks$t|ddks4ttjtjjdd  t|j dt d ksbtW5QRX|j dj s|ttjtjjdd  t|j dt d kstW5QRX|j dj st|j sttjj|dd}tjtjjdd "t|j dt d kstW5QRX|j dj s*ttjtjjdd "t|j dt d ksZtW5QRX|j dj svt|ddkst|ddkst|j std|d<d|d<t|j dt d kstt|j dt d kstd S) zERegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/137z=FOCALLEN= +1.550000000000e+002 APERTURE= +0.000000000000e+000r!r"ZFOCALLENg`c@ZAPERTURErrzFOCALLEN= +1.550000000000E+002APERTURE= +0.000000000000E+000N) rrrRrrarbrcr rVr~rr)rAr(rrrrtest_invalid_float_cardssJ$$ && z,TestHeaderFunctions.test_invalid_float_cardsc Cst}d|jd<||dt|dd}|d|tdW5QRXt|d0}t t }|j|ddd W5QRXW5QRXt |d kst t|d j}d |kst d S)zW Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/140 gEȠ>rrzrb+iZr%rDwarnrrrfz'(invalid value string: '5.0022221e-07')N)rrrBrrrseekrrrarbr rprrVr)rArrrrrsrrrrtest_invalid_float_cards2s  (z-TestHeaderFunctions.test_invalid_float_cards2cCstjd}t|tdks t|jdks.ttjd}t|tdksNt|jdks\ttjd}t|tdks|t|jdkstdS)a8 Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/137, part 2 Ticket https://aeon.stsci.edu/ssb/trac/pyfits/ticket/137 also showed that in float values like 0.001 the leading zero was unnecessarily being stripped off when rewriting the header. Though leading zeros should be removed from integer values to prevent misinterpretation as octal by python (for now Astropy will still maintain the leading zeros if now changes are made to the value, but will drop them if changes are made). rrzAPERTURE= 0.000000000000E+000z APERTURE= 017N)rrLrRrVrrrSrNrrrtest_leading_zeross   z&TestHeaderFunctions.test_leading_zeroscCshtd}td}t}d|d<d|d<|ddks8t|ddksHtt|jd|ks^tt|jd|ksttt}td|d<td|d<|ddkst|ddkstt|jd|kstt|jd|kstt}|tj ||tj ||ddks"t|ddks4tt|jd|ksLtt|jd|ksdtdS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/123 Tests assigning Python and Numpy boolean values to keyword values. zFOO = TzBAR = FTrFrN) rrrrrVr~rbool_rrLrR)rAZfooimgZbarimgrrrrtest_assign_booleans.z'TestHeaderFunctions.test_assign_booleancCsFtdddg}t|dddgks&td|ks2td|ks>t|d d ksNtt|d d ksbtd |d<|dd ksztt|d kst|d=t|ddgkstt|d kst|dd kst|dd |dd kst|jdd ddt|dddgks t|dd kstt|d ks0t|dd d ksFtt|d ksXt|dd d ksntt|d kstt|dddgkst| d ddt|dkstt|ddddgkst|dd kst| dd kst| dd kst| dt|d ks*tt|dddgksBtdS)a  Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/149 Basically ensures that all public Header methods are case-insensitive w.r.t. keywords. Provides a reasonably comprehensive test of several methods at once. )abCr )Defre)GeHrfrPr9ZGEHrQZdEfZgehrfzg*r reZaBcZgEhdefZAbcrrry)rZiJkZIJKZijkrN) rrrrrprFrr[rhrrHrGrIrrrr(test_header_method_keyword_normalization8sB     z.assignrrzFOO = 'BAR 'rzFOO = 'BAZ 'rz$FOO = 'BAR ' / BAZN) rrrrrrarhrir?)rAZerikkurrrrtest_assign_unicodes&   z'TestHeaderFunctions.test_assign_unicodec Cs2t}tjtdd|ddW5QRXdS)a5 First regression test for https://github.com/spacetelescope/PyFITS/issues/37 Although test_assign_unicode ensures that `str` objects containing non-ASCII characters cannot be assigned to headers. It should not be possible to assign bytes to a header at all. zIllegal value: b'Hello'.rrsHelloN)rrrarhrirrrrrtest_assign_non_asciis z)TestHeaderFunctions.test_assign_non_asciic Cs:t}d|d<|ddks ttjd}|||ddksFt|jdjdks^t|jdjdksvttj ddV|ddkst|dd kst|jdjdkst|jdjdkstW5QRX|ddkst|ddkst|jdjdkst|jdjdks6td S) aZ Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/146, and for the solution that is optional stripping of whitespace from the end of a header value. By default extra whitespace is stripped off, but if `fits.conf.strip_header_whitespace` = False it should not be stripped. z Bar rZBarzQUX = 'Bar 'rzFOO = 'Bar 'Zstrip_header_whitespaceFz Bar N) rrrrLrRrr~rqrZconfZset_temprArrrrrtest_header_strip_whitespaces"   "z0TestHeaderFunctions.test_header_strip_whitespacec Csddddgd}t}|D]}||jd<q||dt|d}|djd|ksdtW5QRXtj|jd }|jd|jdkst||d t|d }|djd|kstW5QRXd S) a{ Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/156 When creating a new HDU from an existing Header read from an existing FITS file, if the original header contains duplicate HISTORY values those duplicates should be preserved just as in the original header. This bug occurred due to naivete in Header.extend. zCCD parameters table ...z* reference table oref$n951041ko_ccd.fitsz# INFLIGHT 12/07/2001 25/02/2002z all bias framesrfrvrrrrN)rrrBrrrr)rAhistoryritemrZnew_hdurrr*test_keep_duplicate_history_in_orig_headers$   z>TestHeaderFunctions.test_keep_duplicate_history_in_orig_headerc Csdtjd}tjd}tjd}t}tt*}|j||j||j|W5QRXt |dksvt | | dtt}t | d}t |dkst |dj}d|kst d|kst d |kst |dd kst |dd kst |d d kst tt|jdd tt|jdd tt|jd d W5QRXW5QRXdS)z Test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/109 Allow opening files with headers containing invalid keywords. zCLFIND2D: contour = 0.30zJust some random text.ZPAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAArfrrZCLFIND2DzJust somZAAAAAAAAz: contour = 0.30ze random text.ZHAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAZfooN)rrLrRrrarbr rBrrprrrrrhrir)rAc1c2c3rrsrrBrrrtest_invalid_keyword_cards s.           z.TestHeaderFunctions.test_invalid_keyword_cardsc CsLtjd}tjtjjdd|dW5QRXt|tdksHt dS)z Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/172 Ensures that when fixing a hierarch card it remains a hierarch card. z#HIERARCH ESO DET CHIP PXSPACE = 5e6rrrz#HIERARCH ESO DET CHIP PXSPACE = 5E6N) rrLrRrarbrcr rVrrrrrr$test_fix_hierarch_with_invalid_value8 s z8TestHeaderFunctions.test_fix_hierarch_with_invalid_valuecCsxt}tt|jdtdtt|jdtjtt|jdt dtt|jdtdtt|jdtj dS)a Regression test for https://github.com/spacetelescope/PyFITS/issues/11 For the time being it should not be possible to assign the floating point values inf or nan to a header value, since this is not defined by the FITS standard. rnaninfN) rrrarhrirfloatrrZfloat32r rrrrtest_assign_inf_nanF s z'TestHeaderFunctions.test_assign_inf_nancCstdg}d|d<|ddk s$tt|dts6t|ddksFttd|d<|ddksdtd|d<|ddks|ttd|d<|ddkstd|d<|ddk stt|dtst|ddksttd|d<|ddkstdS)z Regression test for an issue where a value of True in a header cannot be updated to a value of 1, and likewise for False/0. )rTr rTFrN)rrrrintrrrrrrtest_update_boolV s" z$TestHeaderFunctions.test_update_boolcCsHt}d|d<d|d<t|dts*tt|dsr?r@rBrrs CompImageHDUrIrLrMrOrPrRrUrVrXr`rergrirkrlrvr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrr!r$rrrrr6Cs               !  =C  -             $ %(+J(4!9'%#*  K '   r6cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zejd4d5d6d7d8gd9d:Z Z!S);TestRecordValuedKeywordCardsa Tests for handling of record-valued keyword cards as used by the `FITS WCS distortion paper `__. These tests are derived primarily from the release notes for PyFITS 1.4 (in which this feature was first introduced. Note that extra leading spaces in the `value` fields should be parsed on input, but will be stripped in the cards. cstt|_|jdd|jdd|jdd|jdd|jdd|jdd|jdd|jdd dS) NDP1NAXIS: 2z AXIS.1: 1 AXIS.2: 2z NAUX: 2zAUX.1.COEFF.0: 0zAUX.1.POWER.0: 1zAUX.1.COEFF.1: 0.00048828125zAUX.1.POWER.1: 1)super setup_methodrr _test_headerrrj __class__rrr.p s  z)TestRecordValuedKeywordCards.setup_methodc Cs0tjd}|jdkst|jdks(t|jdks6t|jdksDttjd}|jdks^t|jdkslt|jdkszttjd}|jd kst|jd kst|jd ksttd d }|jdkst|jdkst|jdksttd d }|jdkst|jdkst|jdks"ttd d }|jd ks>t|jd ksNt|jd ks^ttdd}|jdkszt|jdkst|jdksttdd}|jdkst|jdkst|jdkstt tj j tdd}W5QRX|jdks t|jdkst|jd ks,td S)zl Test different methods for initializing a card that should be recognized as a RVKC DP1 = 'NAXIS: 2' / A comment DP1.NAXIS@r A commentzDP1 = 'NAXIS: 2.1'g@zDP1 = 'NAXIS: a'r*zNAXIS: aNr+z NAXIS: 2.0rer) rrLrRrMrrSfield_specifierrgrarbrcr rNrrrtest_initialize_rvkc| sL        z1TestRecordValuedKeywordCards.test_initialize_rvkccCsHtjd}|jdkst|jdks(t|jdks6t|jdksDtdS)z Tests that the field_specifier can accessed from a card read from a string before any other attributes are accessed. r2rr3r4r5N)rrLrRr6rrMrSrgrNrrrtest_parse_field_specifier s  z7TestRecordValuedKeywordCards.test_parse_field_specifiercCsptjd}|jdkstd|_|jdks.t|jdksttfdddtjtddjdW5QRXdS)z Accessing a nonexistent RVKC should raise an IndexError for index-based lookup, or a KeyError for keyword lookup (like a normal card). cs j|Sr)r/)rrjrrr rzKTestRecordValuedKeywordCards.test_access_nonexistent_rvkc..rz!Keyword 'DP1\.AXIS\.3' not found.r DP1.AXIS.3N)rarhr\rr/rjrrjrtest_access_nonexistent_rvkc sz9TestRecordValuedKeywordCards.test_access_nonexistent_rvkccCsPd|jd<|jddkstt|jdts0td|jd<|jddksLtdS)z9A RVKC can be updated either via index or keyword access.rfrr3@g? DP1.AXIS.1Nr;rjrrrtest_update_rvkc s   z-TestRecordValuedKeywordCards.test_update_rvkccCs<t}d|d<|ddks td|d<|ddks8tdS)z;Regression test for an issue that appeared after SVN r2412.r z D2IM1.EXTVERrrer4Nrrrrrtest_update_rvkc_2 s z/TestRecordValuedKeywordCards.test_update_rvkc_2cCs|tjd}|jdkst|jdks(ttdd}|jdksBt|jdksPttdd}|jdksjt|jdksxtdS)Nr2r*r+r3rez NAXIS: 2.0r4)rrLrRZ rawkeywordrZrawvaluerNrrrtest_raw_keyword_value s   z3TestRecordValuedKeywordCards.test_raw_keyword_valuecCs<|jjddddd|jddks&t|jddks8td S) z It should be possible to insert a new RVKC after an existing one specified by the full keyword/field-specifier combination.r*z AXIS.3: 1r DP1.AXIS.2r{rfr r?N)r/rrrjrrrtest_rvkc_insert_after( sz3TestRecordValuedKeywordCards.test_rvkc_insert_aftercCs|jd=t|jdkstt|jddks0t|jddksBtt|jddksXt|jd=t|jdksrtt|jddkst|jddkstt|jdd kst|jddkstd S) z Deleting a RVKC should work as with a normal card by using the full keyword/field-spcifier combination. rBrrr3rer rFrzDP1.NAUXNr/rprrrjrrrtest_rvkc_delete1 sz-TestRecordValuedKeywordCards.test_rvkc_deletecCs|jd}t|tjstdd|jDddgks6t|jd}dd|jDdd gks\t|jd }d d|jDd d ddgkst|jd}dd|jDdgkst|jd}dd|jDddgkstdS)z+Test the keyword filter strings with RVKCs. DP1.AXIS.*cSsg|]}t|qSrrrrrrr,K szKTestRecordValuedKeywordCards.test_pattern_matching_keys..DP1 = 'AXIS.1: 1'DP1 = 'AXIS.2: 2'zDP1.N*cSsg|]}t|qSrrrrrrr,Q szDP1 = 'NAXIS: 2'zDP1 = 'NAUX: 2'z DP1.AUX...cSsg|]}t|qSrrrrrrr,W sDP1 = 'AUX.1.COEFF.0: 0'DP1 = 'AUX.1.POWER.0: 1'(DP1 = 'AUX.1.COEFF.1: 0.00048828125'DP1 = 'AUX.1.POWER.1: 1'z DP?.NAXIScSsg|]}t|qSrrrrrrr,_ sz DP1.A*S.*cSsg|]}t|qSrrrrrrr,b sN)r/rrrrr~rAclrrrtest_pattern_matching_keysF s2        z7TestRecordValuedKeywordCards.test_pattern_matching_keyscCsn|jd=t|jdkstt|jddks0t|jddksBtt|jddksXt|jddksjtdS)z'Deletion by filter strings should work. DP1.A*...rerr3r rHNrIrjrrr"test_pattern_matching_key_deletiong s z?TestRecordValuedKeywordCards.test_pattern_matching_key_deletioncCsZ|jd}dd|jDdddddd gks.t|d }d d|jDdddd gksVtd S) zX A card list returned via a filter string should be further filterable. rUcSsg|]}t|qSrrrrrrr,w szQTestRecordValuedKeywordCards.test_successive_pattern_matching..rLrMrNrOrPrQz *.*AUX...cSsg|]}t|qSrrrrrrr, sN)r/r~r)rArSZcl2rrr test_successive_pattern_matchingq s   z=TestRecordValuedKeywordCards.test_successive_pattern_matchingcCs"|jd}t|ddgkstdS)zl The CardList.keys() method should return full keyword/field-spec values for RVKCs. rKrBrFN)r/rrrRrrrtest_rvkc_in_cardlist_keys s z7TestRecordValuedKeywordCards.test_rvkc_in_cardlist_keyscCs&|jd}t|ddgks"tdS)zv The CardList.values() method should return the values of all RVKCs as floating point values. rKrr4N)r/rrrrRrrrtest_rvkc_in_cardlist_values s z9TestRecordValuedKeywordCards.test_rvkc_in_cardlist_valuescCs8|jd}|jdjdkstt|jdjts4tdS)z| Individual card values should be accessible by the .value attribute (which should return a float). rKrrN)r/r~rSrrr rRrrrtest_rvkc_value_attribute s z6TestRecordValuedKeywordCards.test_rvkc_value_attributecCst}d|d<d|d<d|ks$td|ks0td|ks.csdS)Nr_rrrrrr rcsdS)Nr`rrrrrr rN)rrLrRrrrarhr)rArrrrr"test_overly_aggressive_rvkc_lookup s     z?TestRecordValuedKeywordCards.test_overly_aggressive_rvkc_lookupcCsddlm}||d}|jdgd}d|ks6td|ksBt|jdgdgd }d|ks`td |ksltt|d d kst|jd gddgd }d|kstd |kstt|d dkst|jd gdgd }d|kstd|kstd|kst|||d}d|jdgdks*t|||d}d|jd gddks\td|jd gddksvt|dS)z9Tests the basic functionality of the `fitsheader` script.r fitsheaderzerowidth.fitsAIPS FQ extensionszEXTNAME = 'AIPS FQrEXTNAMErgkeywordsz BITPIX =r!rfr z EXTNAME =ryzNAXIS*z NAXIS =z NAXIS1 =z NAXIS2 =rzEXTNAME = 'SCIzSCI,2z comp.fitszXTENSION= 'IMAGEF)rg compressedzXTENSION= 'BINTABLETN) astropy.io.fits.scriptsrcZHeaderFormatterrparserrpsplitr)rArcZhfoutputrrrtest_fitsheader_script s2          z3TestRecordValuedKeywordCards.test_fitsheader_scriptc Cstjdtjdd}tj|d}tj|j|jd}|j | dddt | d<}d |d jkslt d |d jks~t d |d jkst W5QRXd S)zHRegression test for issue https://github.com/astropy/astropy/issues/7312ry)Zdtype)rerer)rrBz tmp2.fitsTrErqr ZPCOUNTZGCOUNTN) rrr rrrr(rrBrrrr)rArZphduZchdurrrr1test_fitsheader_compressed_from_primary_image_ext s zNTestRecordValuedKeywordCards.test_fitsheader_compressed_from_primary_image_extc Csddlm}ddlm}|d}||}||t}|dg}t|t|dj ks^t |jdddgd}t|t|dj t|dj t|d j kst W5QRX|jdgd}t |d |kst t |d dkst |d |d dkdkst |jdgdgd}t|dks,t |d ddksBt |d ddksXt |d ddksnt |jdgd}|dkst |jdgdgd}|dkst | dS)z7Tests the `--table` feature of the `fitsheader` script.rrrbrdrere4rfryrrrSrMrhrir ZDOES_NOT_EXISTN) astropy.iorrlrcrZTableHeaderFormatterrrmrprBrrr4r)rArrcZ test_filename formatterZfitsobjZmytablerrrtest_fitsheader_table_feature s:         z:TestRecordValuedKeywordCards.test_fitsheader_table_featurerrzwb+abzab+c Cs>t|d|d"}tjtdd}||W5QRXdS)Nz mode.fitsrrr)rrrrsrZonesr)rArffrrrrtest_hdu_writeto_mode= sz2TestRecordValuedKeywordCards.test_hdu_writeto_mode)"rrrr%r.r7r8r9r:r<r=r>r@rCrDrErGrJrTrVrWrXrYrZr\rarprqrurar&r'rx __classcell__rrr0rr)d s8 5         !    #( *r)cCsGdddtj}|d}|jddks,t|jddks>t|jdd ksPt|d d }t||kslt|jddks~t|jdd kst|d }t||kst|jdd kst|}t||kst|jddkst|jdd kst|d |jdd kstd S)z?Check that subclasses don't get ignored on slicing and copying.cseZdZfddZZS)ztest_subclass..MyHeadercs2t|trt|dkr|d7}tj|f||S)Nre) no comment)rrrpr-r)rArargskwargsr0rrrI sz&test_subclass..MyHeader.append)rrrrryrrr0rMyHeaderH sr}))rrr_)rr4second)rrArr_rr~rrzr Nzc*))r$r]r$)rrr?rtyperr')r}Z my_headerZslice_Z selectionZcopy_rrr test_subclassE s(   r)!r2rriorrnumpyrrarsrZastropy.io.fits.cardrZastropy.io.fits.headerrZastropy.io.fits.utilrZastropy.io.fits.verifyr r Zastropy.utils.exceptionsr Zastropy.utils.miscr Zconftestrrrr)r5r6r)rrrrrsV         5d