ó šßÈ[c@ s¥ddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZdd lmZmZmZd d lmZmZd d lmZd dlm Z yddl!Z!Wne"k rAe#Z$nXe%Z$de fd„ƒYZ&de fd„ƒYZ'd„Z(d„Z)d„Z*d„Z+d„Z,dS(iÿÿÿÿ(tdivisiontwith_statementN(t assert_equali(trange(tfits(t AstropyPendingDeprecationWarning(t NUMPY_LT_1_12(traisestcatch_warningstignore_warningsi(tSUBTRACTIVE_DITHER_1tDITHER_SEED_CHECKSUMi(tcomparerecords(t FitsTestCasetTestImageFunctionscB s˜eZd„Zd„Zd„Zd„Zd„Zd„Zej j e j ƒdko[e ddƒd „ƒZd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zej jdeefƒej jdefƒd„ƒƒ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+d&„Z,RS('cC s tjƒ}|jdks!t‚d|jks6t‚d|_|jdksTt‚|jddksmt‚tjddƒ}|jdks”t‚|jddks­t‚tjƒ}d|d½siýÿÿÿsòEXPFLAG = 'NORMAL ' / Exposure interruption indicator FILENAME= 'vtest3.fits' / File name XXX = 1.234E+56 tfilenametfnamethistorytsimpleiii]i\i[i^tdtypes test_new.fitstmodetappendicstest_append.fitsstest_update.fitstupdatetrootnamet U2EQ0201Ttabci;iR%RR/gð?Rtint32cs s|]}t|ƒVqdS(N(R)(RKR4((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pys 1ssCBITPIX = 32 / array data type NAXIS = 1 / number of array dimensions NAXIS1 = 2 PCOUNT = 0 / number of parameters (RHi(ii(ii(ii(ii(ii(ii(ii(RR$R%RRtjointcardstrename_keywordR'Rt ValueErrortnpt array_equaltarrayR0R@RRtflushtostrenameR?treadalltonestfloat32R(RR+tntatuR%Rthdu2((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_io_manipulation¬sr$& '! !!    # ( #cC s,tj|jdƒddƒ}|jƒdS(Ns test0.fitstmemmapi(RR$R%R&(Rtf1((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_memory_mapping7scC sàtjƒ}tj|ƒ}tƒ}|jƒWdQXd}t|ƒdksUt‚|t|djƒkstt‚tƒ }|j |j dƒdƒWdQXd}t|ƒdks½t‚|t|djƒksÜt‚dS(Ns+HDUList's 0th element is not a primary HDU.iistest_new2.fitstfixsOHDUList's 0th element is not a primary HDU. Fixed by inserting one as 0th HDU.( RRRRtverifyR8RR)tmessageR?R@(RR4Rtwttext((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_verification_on_output<s   cC s<tj|jdƒƒ}tj|djdPdƒs;t‚tj|djdddd…ftjddd d d dd d dddg ƒƒs—t‚tj|djdddd…ftjd dd d dddgƒƒsçt‚tj|djdddd…ftjddd d d dd d gƒƒs:t‚tj|djdddd…ftjd d dd d gƒƒs„t‚tj|djddd…dd…ftjddd d d dd d dddg dddddddddddg dd d!d"d#d$d%d&d'd(d)g gƒƒs1t‚tj|djddd…dd…fdd…dd…ftjd*d+d,gd-d.d/gddd ggƒƒs¬t‚|dj}tj|djddd…dd…f|ddd…dd…fƒs t‚tj|djddd…df|ddd…dfƒsOt‚tj|djdd0…dd…dd…fdd…dd…dd…ftjd*d+d,gd-d.d/gddd ggd1d2d3gd4d5d6gd7d8d9ggd:d;d<gd=d>d?gd@dAdBgggƒƒs*t‚tj|djdd…dd…dd…fdd…dd…dd…ftjddCgdDdEggdFdGgdHdIggdJdKgdLdMgggƒƒsÏt‚tj|djdd…ddd…f|dd…ddd…fƒs t‚tj|djdd…dd…dd…f|dd…dd…dd…fƒs}t‚tj|djdd0…ddd…f|dd0…ddd…fƒsÎt‚tj|djdd0…ddN…dd…f|dd0…ddN…dd…fƒs+t‚tj|djdd…ddd…f|dd…ddd…fƒs|t‚tj|djdd…dCddgdf|dd…dCddgdfƒsÓt‚tjtt ttt t ttt tg ƒ}tj|djdd…|dd…f|dd…|dd…fƒsQt‚tj|djdd0…ddd…dOf|dd0…ddd…dOfƒs¨t‚tj|djdOddd…f|dOddd…fƒsít‚tj|djdOdCddgdf|dOdCddgdfƒs8t‚dS(QNs arange.fitsiiiiiei`iaibicidifigihiiijiiiøÿÿÿikiliminioipiqirisitiuiviwixiyizi{i|i}i~ii€iJiKiLiUiViWii¸i¹iºiÃiÄiÅiÎiÏiÐi&i'i(i1i2i3i<i=i>ii i inioiyiziÜiÝiçièi.(iii( RR$R%R\R]tsectionRR^R:R3(Rtfstdatt bool_index((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt test_sectionNsp#9#-#0#'#*$6?  QEN     !N    Q]Q]Q,+)(QE&cC sítjdgƒ}tj|ƒ}|j|jdƒƒtj|jdƒƒ}|dj}|dj}tj |d|dƒs‰t ‚tj |d|dƒs©t ‚tj |d|dƒsÉt ‚tj |d|dƒsét ‚dS( Nis test_new.fitsi.(.i(.i(i.(i.( R\R^RRR?R@R$RsR%R]R(RRfRRBtsecRu((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_section_data_singles     cC sKtjdƒjdƒ}tj|ƒ}|j|jdƒƒtj|jdƒƒ}|d}|dj}|j dd…dd…f|dd…dd…fkj ƒs³t ‚|j ddd…f|ddd…fkj ƒsît ‚|j ddd…f|ddd…fkj ƒs)t ‚|j dd…df|dd…dfkj ƒsdt ‚|j dd…df|dd…dfkj ƒsŸt ‚|j d|dkj ƒsÂt ‚|j d |d kj ƒsåt ‚|j d |d kj ƒst ‚|j d |dkj ƒs+t ‚|j dd…dd…f|dd…dd…fkj ƒsrt ‚|j dd…dd…f|dd…dd…fkj ƒs¹t ‚|j dd…dd…f|dd…dd…fkj ƒst ‚|j dd…dd…f|dd…dd…fkj ƒsGt ‚dS(Niis test_new.fitsii(ii(ii(ii(ii(ii(ii(ii(ii(ii( R\tarangetreshapeRRR?R@R$R%RstallR(RRfRRBtdRu((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_section_data_squares&  G;;;;####GGGcC sR"tjdƒjdƒ}tj|ƒ}|j|jdƒƒtj|jdƒƒ}|d}|dj}|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÅt ‚|j dd…dd…f|dd…dd…fkj ƒs t ‚|j |kj ƒs)t ‚|j ddd…dd…f|ddd…dd…fkj ƒsvt ‚|j ddd…dd…f|ddd…dd…fkj ƒsÃt ‚|j dddd…f|dddd…fkj ƒst ‚|j dddd…f|dddd…fkj ƒsEt ‚|j dddd…f|dddd…fkj ƒs†t ‚|j dddd…f|dddd…fkj ƒsÇt ‚|j dddd…f|dddd…fkj ƒst ‚|j dddd…f|dddd…fkj ƒsIt ‚|j d|d kj ƒslt ‚|j d |d kj ƒst ‚|j d |d kj ƒs²t ‚|j d|dkj ƒsÕt ‚|j d|dkj ƒsøt ‚|j d|dkj ƒst ‚|j d|dkj ƒs>t ‚|j d|dkj ƒsat ‚|j d|dkj ƒs„t ‚|j d|dkj ƒs§t ‚|j d|dkj ƒsÊt ‚|j d|dkj ƒsít ‚|j d |d!kj ƒst ‚|j d"|d#kj ƒs3t ‚|j d$|d%kj ƒsVt ‚|j d&|d'kj ƒsyt ‚|j d(|d)kj ƒsœt ‚|j d*|d+kj ƒs¿t ‚|j dd…ddf|dd…ddfkj ƒst ‚|j dd…ddf|dd…ddfkj ƒsAt ‚|j dd…ddf|dd…ddfkj ƒs‚t ‚|j dd…ddf|dd…ddfkj ƒsÃt ‚|j dd…ddf|dd…ddfkj ƒst ‚|j dd…ddf|dd…ddfkj ƒsEt ‚|j dd…ddf|dd…ddfkj ƒs†t ‚|j dd…ddf|dd…ddfkj ƒsÇt ‚|j dd…ddf|dd…ddfkj ƒst ‚|j ddd…df|ddd…dfkj ƒsIt ‚|j ddd…df|ddd…dfkj ƒsŠt ‚|j ddd…df|ddd…dfkj ƒsËt ‚|j ddd…df|ddd…dfkj ƒs t ‚|j ddd…df|ddd…dfkj ƒsM t ‚|j ddd…df|ddd…dfkj ƒsŽ t ‚|j dd…dd…df|dd…dd…dfkj ƒsÛ t ‚|j dd…dd…df|dd…dd…dfkj ƒs( t ‚|j dd…dd…df|dd…dd…dfkj ƒsu t ‚|j dd…ddd…f|dd…ddd…fkj ƒs t ‚|j dd…ddd…f|dd…ddd…fkj ƒs t ‚|j dd…ddd…f|dd…ddd…fkj ƒs\ t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsµ t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsg t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÀ t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsr t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsË t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs$t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs}t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÖt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs/t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsˆt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsát ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs:t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs“t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsìt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsEt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsžt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs÷t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsPt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs©t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒst ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs[t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs´t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsft ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs¿t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒst ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsqt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÊt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs#t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs|t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÕt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs.t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs‡t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsàt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs9t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs’t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsët ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsDt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒst ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsöt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsOt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs¨t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒst ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsZt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs³t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒset ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs¾t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒst ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒspt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÉt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs"t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs{t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsÔt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs-t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs†t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsßt ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs8 t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs‘ t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsê t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsC!t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsœ!t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsõ!t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒsN"t ‚dS(,Niiis test_new.fitsii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii(iii( R\RzR{RRR?R@R$R%RsR|R(RRfRRBR}Ru((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_section_data_cube³sô  YGMMAAAAAA##################AAAAAAAAAAAAAAAMMMMMMYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYcC s[tjdƒjdƒ}tj|ƒ}|j|jdƒƒtj|jdƒƒ}|d}|dj}|j dd…dd…dd…dd…f|dd…dd…dd…dd…fkj ƒs×t ‚|j dd…dd…dd…f|dd…dd…dd…fkj ƒs0t ‚|j dd…dd…f|dd…dd…fkj ƒswt ‚|j |kj ƒs”t ‚|j ddd…dd…dd…f|ddd…dd…dd…fkj ƒsót ‚|j ddd…ddd…f|ddd…ddd…fkj ƒsFt ‚|j dd…dd…ddd…f|dd…dd…ddd…fkj ƒs¥t ‚|j dd…dddd…f|dd…dddd…fkj ƒsøt ‚|j dd…dd…dd…df|dd…dd…dd…dfkj ƒsWt ‚dS(Niis test_new.fitsii(iiii( R\RzR{RRR?R@R$R%RsR|R(RRfRRBR}Ru((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_section_data_four4s  kYG_S_ScC stj|jdƒƒ}|d}|dj}|jdd…dd…f|dd…dd…fkjƒsvt‚|jddd…f|ddd…fkjƒs±t‚|jddd…f|ddd…fkjƒsìt‚|jdd…df|dd…dfkjƒs't‚|jdd…df|dd…dfkjƒsbt‚|jd|dkjƒs…t‚|jd|d kjƒs¨t‚|jd |d kjƒsËt‚|jd |d kjƒsît‚|jdd…dd…f|dd…dd…fkjƒs5t‚|jdd…dd…f|dd…dd…fkjƒs|t‚|jdd…dd…f|dd…dd…fkjƒsÃt‚|jdd…dd…f|dd…dd…fkjƒs t‚tj|jdƒƒ}|d}|jdd…dd…f|dd…dd…fkjƒsst‚|jddd…f|ddd…fkjƒs®t‚|jddd…f|ddd…fkjƒsét‚|jdd…df|dd…dfkjƒs$t‚|jdd…df|dd…dfkjƒs_t‚|jd|dkjƒs‚t‚|jd|dkjƒs¥t‚|jd|dkjƒsÈt‚|jd|dkjƒsët‚|jdd…dd…f|dd…dd…fkjƒs2t‚|jdd…dd…f|dd…dd…fkjƒsyt‚|jdd…dd…f|dd…dd…fkjƒsÀt‚|jdd…dd…f|dd…dd…fkjƒst‚|j st‚dS(s÷ Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/143 This is like test_section_data_square but uses a file containing scaled image data, to test that sections can work correctly with scaled data. s scale.fitsiNii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(RR$R%RsR|Rt _data_loaded(RRBR}Ru((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_section_data_scaledFs@  G;;;;####GGGG G;;;;####GGGGcC s„tj|jdƒdtƒ}|djjtjdƒksCt‚tj|jdƒƒ}|djjtjdƒks€t‚dS(Ns scale.fitstdo_not_scale_image_datais>i2Rd(RR$R%R:RPR\R(RRB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_do_not_scale_image_dataqs%cC s£tj|jdƒdtjgddƒƒtjddgƒjdƒ}tj|jdƒd|ƒtj|jdƒdt ƒ}|dj j dksŸt ‚d S( s§Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/56 (BZERO and BSCALE added in the wrong location when appending scaled data) s test_new.fitsR%RPtuint8idtuint16tuintiN( RR?R@R\R^tzerostastypeRRR$R:R%RPR(RR}tf((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_append_uint_dataws ! cC sJtjƒ}tjtjjddƒƒ}|jddddddƒdS( sT Regression test for https://github.com/astropy/astropy/issues/6399 idttypeR…tbscaleitbzeroiN(RRRR\trandomtrandtscale(Rthdu1Rh((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt%test_scale_with_explicit_bzero_bscale„s c C sãxÜdD]Ô}d|d}|dkr9tj|ƒ}ndj|ƒ}tjdd|ƒ}|j|ƒ|tjdd|ƒ8}tjd |ƒ}tj|j |kƒs³t ‚|j j j dj|ƒks×t ‚d |j ksìt ‚|j d d|dks t ‚d j|ƒ}|j|j|ƒƒtj|j|ƒd tƒ‰}|d }tj|j |kƒs{t ‚|j j j dj|ƒksŸt ‚d |j ks´t ‚|j d d|dksÕt ‚WdQXqWdS(sÓ Regression test for https://github.com/astropy/astropy/issues/2305 This ensures that an HDU containing unsigned integer data always has the apppriate BZERO value in its header. ii i@iisuint{}idRPR%tBZEROs uint{}.fitsR‡iN(ii i@(R\tuint64tformattemptytfillRzRRR|R%RRPRRR?R@R$R:( Rtint_sizetmax_uintRPtarrtuint_hduRLRBt new_uint_hdu((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_uint_header_consistencyŽs*   $!! $t from_filet do_not_scalec C s§tjdddƒ}|rƒtj|ƒ}d}|j|j|ƒƒtj|j|ƒd|ƒ}|d}|j}WdQXntj|d|ƒ}d|jks­t ‚d |jksÂt ‚|jdd ksÛt ‚|jd d ksôt ‚|jd |_|jd dkst ‚d|jks2t ‚d |jksGt ‚d}|j|j|ƒƒtj|j|ƒƒ&} | djdkj ƒst ‚WdQXdS(s< Regression test for https://github.com/astropy/astropy/issues/4974 BZERO/BSCALE should be removed if data is converted to a floating point type. Currently excluding the case where do_not_scale_image_data=True because it is not clear what the expectation should be. idRPR†sunsigned_int.fitsRƒiNtBSCALER”ii€gð?tBITPIXstest_uint_to_float.fits( R\RˆRRR?R@R$R%RRR|( RRŸR R›ttmp_uintRLRŠRœt_RB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt5test_uint_header_keywords_removed_after_bitpix_change³s.    cC s™tjd dtjƒ}d|df4R¢iàÿÿÿR”R¡i*i N(i*i (i*i (R·R`tstatR@tst_mtimettimetsleepRR$R&RR%RPR\RR|tshape(RtmtimeRBRµ((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_open_scaled_in_update_mode˜s0  "'   '% %c C s˜|jdƒtj|jdƒdddtƒ^}|djd}|djd}|djd}|djjƒ}d|djdii RPtRICE_1itGZIP_1g{®Gáz„¿tGZIP_2idit HCOMPRESS_1c C sR|j|ƒ}tjƒ}tj|ƒ}tj|ddd|d|ƒ}|j|ƒ|j|jdƒdtƒ|j ƒtj |jdƒƒ¶}|dj |kj ƒs¸t ‚|djd|jdksÜt ‚|djd |jd kst ‚|djd |jd ks$t ‚|djd |jd ksHt ‚WdQXdS( NRR/tcompressionTypet quantizeLevels test_new.fitsR²iRÊRËRÌR¢(t newbyteorderRRRRåRRR?R@R:R&R$R%R|RR( RR%RçRèRéR!R"tchdutfd((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_comp_image_s     $$$s not HAS_SCIPYc C sàddl}tjjdƒ|jjƒtjjddƒd}tj|ƒj |j dƒƒtj |ddd d d dd d ƒj |j dƒƒtj |ddd d d dd d ƒj |j dƒƒtj |j dƒƒ}tj |j dƒƒ}tj |j dƒƒ}tj ||ƒ s0t‚tjtj||ƒdddƒs[t‚tjtj||ƒdddƒs†t‚tjtj||ƒdddƒs±t‚tjtj||ƒdddƒsÜt‚dS(s€ Regression test for https://github.com/astropy/astropy/issues/5969 Test that quantize_level is used. iÿÿÿÿNi*ii sim1.fitsRçRìtquantize_methodiRèt dither_seedisim2.fitsiœÿÿÿsim3.fitsgà¿tatolgü©ñÒMbP?gà?iÎÿÿÿgš™™™™™¹?i2(t scipy.miscR\RtseedtmisctascenttrandnRRR?R@RåtgetdataR]Rtisclosetmintmax(RtscipyR%tim1tim2tim3((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_comp_image_quantize_levelxs$ &+++cC sMtjttjtjd dtjƒdddddd d dddgƒd S( s Tests compression with the HCOMPRESS_1 algorithm with data that is not 2D and has a non-2D tile size. ii RPRR/RðRïRñittileSizeN(ii i (R'RR[RRåR\RˆRd(R((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt+test_comp_image_hcompression_1_invalid_data–s c C s»tjddtjƒjdƒ}tjd|dddd d d d d d dgƒ}|j|jdƒƒtj|jdƒƒ3}tj |dj |ƒj ƒdks±t ‚WdQXdS(s Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/171 Tests that data containing more than two dimensions can be compressed with HCOMPRESS_1 so long as the user-supplied tile size can be flattened to two dimensions. i,RPii R%RR/RðRïRñiRiis test.fitsgð?g.@N(ii i g±?( R\RzRdR{RRåR?R@R$tabsR%RR(RtcubeRRB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt%test_comp_image_hcompress_image_stack¢s !cC s?tjdƒjddƒ}|djdƒjƒdd}tjd|dtd tƒ}|j |j d ƒƒtj |j d ƒƒ±}t |dtjƒs§t ‚d |djksÀt ‚|djd d ksÝt ‚d |djksöt ‚|djd |kst ‚tj|dj|kƒs5t ‚WdQXdS(sF Regression test for https://github.com/spacetelescope/PyFITS/issues/32 Ensure that when floating point data is compressed with the SUBTRACTIVE_DITHER_1 quantization method that the correct ZDITHER0 seed is added to the header, and that the data can be correctly decompressed. gY@i iR…i'iR%RöR÷s test.fitstZQUANTIZR tZDITHER0N(R\RzR{tviewtsumRRåR R R?R@R$R7Rt_headerR|R%(RR^tcsumRRB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_subtractive_dither_seed·s ! c C s¾tƒrtjdƒtj|jdƒdtƒ@}t|dtjƒsTt ‚t|dtj ƒ sqt ‚WdQXWdQXtj|jdƒƒ#}t|dtj ƒs´t ‚WdQXdS(Nterrors comp.fitstdisable_image_compressioni( Rtwarningst simplefilterRR$R%R:R7t BinTableHDURRå(RRB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_disable_image_compressionÐs   )cC s‚|jdƒtj|jdƒƒj}tjdƒtj|jdƒddƒj ƒ|tj|jdƒƒjks~t ‚dS(s¾ Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/167 Similar to test_open_scaled_in_update_mode(), but specifically for compressed images. s comp.fitsiRQRSN( R·R`R»R@R¼R½R¾RR$R&R(RRÀ((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt#test_open_comp_image_in_update_modeÞs  "c C stj|jdƒdtƒC}tjd|djd|djƒ}|j|jdƒƒWdQXtj |jdƒƒj }t j dƒtj|jdƒdd ƒj ƒ|tj |jdƒƒj ksÔt‚t j dƒtj|jdƒd ƒ}|dj|j ƒ|tj |jdƒƒj ks8t‚tj|jdƒdd ƒ}|djjtjd ƒks{t‚|djd d ks˜t‚d |djks±t‚d|djksÊt‚d|dj_|j ƒtj|jdƒƒ}|djdkst‚|djjtjd ƒks:t‚|djd d ksWt‚d |djkspt‚d|djks‰t‚dS(sË Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/88 2 Identical to test_open_scaled_in_update_mode() but with a compressed version of the scaled image. s scale.fitsRƒR%iRNiRQRSRdR¢iàÿÿÿR”R¡i*i (i*i (i*i (RR$R%R:RåRR?R@R`R»R¼R½R¾R&RRPR\R¿(RRBRóRÀ((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt*test_open_scaled_in_update_mode_compressedòs6   "'   '% %c C sÍtj|jdƒƒ!}|dj|jdƒƒWdQXtj|jdƒƒt}tj|jdƒƒS}tj|dj|djkƒsšt‚t|dj |dj ƒs½t‚WdQXWdQXdS(Ns comp.fitsis test.fits( RR$R%R?R@R\R|RR tcompressed_data(RRBthdul1R­((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt(test_write_comp_hdu_direct_from_existing's ) c C sŒtj|jdƒdtƒC}tjd|djd|djƒ}|j|jdƒƒWdQXtj|jdƒƒ}|dj}tƒ!|j|jd ƒd tƒWdQX|j ƒtj|jd ƒƒ}|dj|kj ƒsõt ‚|j ƒtj|jdƒƒ}tƒ!|j|jd ƒd tƒWdQX|j ƒtj|jd ƒƒ}|dj|kj ƒs„t ‚|j ƒtj|jdƒdtƒ}|j|jd ƒd td d ƒ|j ƒtj|jd ƒƒ}|dj}|j ƒtj|jd ƒd dƒ}|j ƒtj|jd ƒƒ}|dj|kj ƒsft ‚tj|jd ƒƒ}|j ƒdS(s· Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/88 1 Identical to test_rewriting_large_scaled_image() but with a compressed image. sfixed-1890.fitsRƒR%iRsfixed-1890-z.fitsNis test_new.fitsR²R³R´RQRS( RR$R%R:RåRR?R@R R&R|R(RRBRóRµ((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt,test_rewriting_large_scaled_image_compressed1sD   "   "       c C sFtj|jdƒdtƒC}tjd|djd|djƒ}|j|jdƒƒWdQXtj|jdƒddd tƒ^}|d jd }|d jd }|d jd }|d jjƒ}d|d jd}d|djd<|d j|djkj ƒs6t ‚WdQXWdQXdS(s Regression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/88 3 Identical to test_scale_back() but uses a compressed image. s scale.fitsRƒR%iRNRQRSR¬iR¢R”R¡(RR$R%R:RåRR?R@R¸RRÂRÃRÄR|( RRBRóRÅRÆRÇRµRÈR­((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_scale_back_compressedcs0   )'c C s!tjjdd ƒ}tjd|ddƒ}|j|jdƒƒtj|jdƒƒ3}tj||dj ƒj ƒdksŠt ‚Wd QX~tjd|ddd dƒ}t ƒ!|j|jdƒd t ƒWd QXtj|jdƒƒ&}||dj kjƒst ‚Wd QXd S( sERegression test for https://aeon.stsci.edu/ssb/trac/pyfits/ticket/198tsizeièR%RðRís test.fitsigNRñR²(ièiè(R\RtnormalRRåR?R@R$R R%RRR R:R|(Rtnoisetchdu1Rªtchdu2((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_lossless_gzip_compressionŒs2  "cC s9tjdddtjƒddd…tjftjddƒ}tjjdƒtjjdddfƒ}||d|jd …d|jd…f}|jdƒt|ƒdksft‚d|ksxt‚WdQX|jd dt ƒd|ks£t‚|ddks¹t‚d|ksËt‚|ddksát‚|jd ƒd |kst‚|d d kst‚d |ks(t‚d |ks:t‚|d d ksPt‚WdQXdS(Ns comp.fitsiR@RR5R4tendiÿÿÿÿtCHECKSUMtabcd1234tZHECKSUM(RR5R4(RHRI( RR$R%RRRRRR8RR:(RRBtimghdrttblhdrRp((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_compression_header_append s$     cC s|tj|jdƒƒ^}|dj}x t|ƒdkrJ|jƒq+W|jd ƒ|jd ƒ|jd ƒWd QXd S(sY Regresion test for issue https://github.com/astropy/astropy/issues/5827 s comp.fitsiiètQ1_OSAVGs[adu] quadrant 1 overscan meantQ1_OSSTDs [adu] quadrant 1 overscan stddevtQ1_OSMEDs [adu] quadrant 1 overscan medianN(RNis[adu] quadrant 1 overscan mean(ROis [adu] quadrant 1 overscan stddev(RPis [adu] quadrant 1 overscan median(RR$R%RR8RR(RRBR((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_compression_header_append2!s   c C s°tj|jdƒƒ’}|dj}|dj}tƒ\}|jddƒt|ƒdksit‚d|ks{t‚|j dƒdks–t‚WdQX|jdd ƒd|ks¾t‚|j dƒ|j dƒdksæt‚d|ksøt‚|j dƒ|j dƒdks t‚|j dƒ}|jdd ƒd|ksQt‚|j dƒ|kslt‚d|ks~t‚|j dƒ|j dƒdks¦t‚WdQXdS( Ns comp.fitsiièR@tTELESCOPtOBSERVERs Phil PlaitR(RSs Phil Plait(R( RR$R%RRRtinsertR8RtcountR8(RRBRKRLRpR;((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_compression_header_insert/s&   !((cC stj|jdƒƒc}|dj}|dj}tƒ{}|jdddddƒt|ƒdksrt‚d|ks„t‚|j dƒdksŸt‚|ddksµt‚WdQX|jdd d d ƒ|j dƒ|j d ƒdksùt‚|dd kst‚|j d ƒ|j d ƒdks7t‚|d d ksMt‚|j d ƒdksht‚|j dƒdksƒt‚|ddks™t‚|jdddd ƒ|j dƒ|j d ƒdks×t‚|ddksít‚|j d ƒ|j d ƒdkst‚|d dks+t‚|j d ƒdksFt‚|j dƒdksat‚|ddkswt‚WdQXdS(Ns comp.fitsitZBITPIXiMtasdftaftertXTENSIONtGCOUNTictbeforetPCOUNTtZGCOUNTtZPCOUNTiii( RR$R%RRRtsetR8RRUR8(RRBRKRLRp((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt(test_compression_header_set_before_afterJs2   ((((cC s¨tjtjdgdtjƒƒ}d|jd<|jddgksMt‚|j|jdƒƒtj |jdƒƒ'}|djddgksžt‚WdQXdS(sT Regression test for https://github.com/astropy/astropy/issues/2363 iRPs hello worldR6s test.fitsiN( RRåR\R^RWRRR?R@R$(RRRB((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyt)test_compression_header_append_commentaryjs $ cC sÄtjd ddƒ}tjdddƒ|dtexpectedR¡R”cC s«d}tjtjddd|ƒƒ}||j|<|j|jdƒƒ~tj|jdƒƒ@}|dj||ks…t‚|dj j |ks¡t‚WdQXdS(sÚ If BSCALE,BZERO is set to floating point values, the image should be floating-point. https://github.com/astropy/astropy/pull/6492 Parameters ---------- keyword : `str` Keyword to set to a floating-point value to trigger floating-point pixels. dtype : `numpy.dtype` Type of original array. expected : `numpy.dtype` Expected type of uncompressed array. gê46¼ó?ii RPs test.fitsiN( RRåR\RzRR?R@R$RR%RP(RR>RPRoR*R((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyttest_compressed_scaled_float×s! (R%RçRèRé(RêRë(ii i (ii i (ii i (idid(R>RPRo(,RÝRÞRæR'RßRãRtboR\RˆRdRõtskipifRR RRR RRRRRRRR%R.R<RFRMRQRVRaRbReRhRiRjRnR…R0RWR©Rp(((s?lib/python2.7/site-packages/astropy/io/fits/tests/test_image.pyRäGsD  §     5 2 )   ?         $$cC s |jdƒj}|jdƒj}tjjdƒd}tjƒ}tjtj|ddtƒdt ƒ}d|j dsF          ÿÿÿÿ)ÿÿ²