ó šßÈ[c@sˆddlZddlZddlZddlZddlmZddlmZm Z m Z ddlm Z ddlm Z dd lmZmZdd lmZmZdd lmZdd lmZejjejjeƒd ƒZyddlZWnek reZ nXe!Z d„Z"de#fd„ƒYZ$de#fd„ƒYZ%d„Z&d„Z'd„Z(d„Z)d„Z*d„Z+dS(iÿÿÿÿN(tassert_allclosei(tHDUListt PrimaryHDUt BinTableHDUi(tfitsi(tunits(trangetzip(tTabletQTable(tcatch_warnings(tUnitScaleErrortdatacCs<x5|jjD]'}tj||||kƒs tSq WtS(N(tdtypetnamestnptalltFalsetTrue(tatbtname((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt equal_datastTestSingleTablecBs˜eZd„Zd„Zejjdƒd„ƒZd„Zd„Z d„Z ejj de e fƒd„ƒZd „Zd „Zd „Zd „ZRS( c Cs…tjttddddgddddgd d d d gƒƒd tdƒtftdƒtdƒftdƒtfgƒ|_dS(NiiiiRRtctdgffffff@g@gÍÌÌÌÌÌ@gÍÌÌÌÌÌ!@R tU1(RtarraytlistRtstrtinttfloatR (tself((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt setup_class%scCs_t|jdƒƒ}t|jƒ}|j|dtƒtj|ƒ}t||ƒs[t‚dS(Nstest_simple.ftst overwrite( RtjoinRR twriteRtreadRtAssertionError(R ttmpdirtfilenamett1tt2((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt test_simple+s snot HAS_PATHLIBcCshtjt|jdƒƒƒ}t|jƒ}|j|dtƒtj|ƒ}t ||ƒsdt ‚dS(Nstest_simple.fitR"( tpathlibtPathRR#RR R$RR%RR&(R R'R(R)R*((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_simple_pathlib2s cCsMt|jdƒƒ}t|jƒ}d|jdR&tmessaget startswith(R R'R(R)tl((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_simple_meta_conflictingLs  cCset|jdƒƒ}t|jƒ}|j|dtddƒtj|ƒ}t||ƒsat‚dS(sE Test that file type is recognized without extension R+R"tformatRN( RR#RR R$RR%RR&(R R'R(R)R*((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_simple_noextensionVs t table_typecCsÅt|jdƒƒ}||jƒ}tj|d_tjtj|d_|j|dt ƒ|j |ƒ}t ||ƒs‚t ‚|djtjksžt ‚|djtjtjksÁt ‚dS(Nstest_with_units.fitsRRR"( RR#R tutmtunittkmtsR$RR%RR&(R RIR'R(R)R*((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_with_units`scCsèt|jdƒƒ}t|jdtƒ}ddddg|jdR&RCRDRRm(R R'R(RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt test_readµs cCspt|jdƒƒ}|jj|ƒtjtƒ}tj|ddƒWdQX|j j ddkslt ‚dS(Nstest_read_with_hdu_0.fitsR\isNo table found in hdu=0( RR#RoRZRftraisest ValueErrorRR%tvaluetargsR&(R R'R(texc((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_read_with_hdu_0¿s R\iR8cCs€t|jdƒƒ}|jj|ƒtƒ}tj|d|ƒ}WdQXt|ƒdksdt‚t ||j ƒs|t‚dS(Nstest_read_with_hdu_1.fitsR\i( RR#RoRZR RR%R>R&RRm(R R'R\R(RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_read_with_hdu_1Æs  iR9cCs€t|jdƒƒ}|jj|ƒtƒ}tj|d|ƒ}WdQXt|ƒdksdt‚t ||j ƒs|t‚dS(Nstest_read_with_hdu_2.fitsR\i( RR#RoRZR RR%R>R&RRn(R R'R\R(RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_read_with_hdu_2Ïs  cCsztƒ}tj|jƒ}WdQXt|ƒdks<t‚t|djƒjdƒs^t‚t ||j ƒsvt‚dS(Niis`hdu= was not specified but multiple tables are present, reading in first available table (hdu=1)( R RR%RoR>R&RRCRDRRm(R RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_read_from_hdulistØs  cCsNtjtƒ}tj|jddƒWdQX|jjddksJt‚dS(NR\isNo table found in hdu=0( RfRzR{RR%RoR|R}R&(R R'R~((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt!test_read_from_hdulist_with_hdu_0àscCs^tƒ}tj|jd|ƒ}WdQXt|ƒdksBt‚t||jƒsZt‚dS(NR\i(R RR%RoR>R&RRm(R R'R\RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt!test_read_from_hdulist_with_hdu_1ås cCs^tƒ}tj|jd|ƒ}WdQXt|ƒdksBt‚t||jƒsZt‚dS(NR\i(R RR%RoR>R&RRn(R R'R\RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyt!test_read_from_hdulist_with_hdu_2ìs cCs\tƒ}tj|jdƒ}WdQXt|ƒdks@t‚t||jƒsXt‚dS(Nii(R RR%RoR>R&RRm(R RER^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_read_from_single_hduós (RdReR!RsRxRyRRfRgRiR€RR‚RƒR„R…R†(((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyRjŸs    $ $  $$cCsœtjtjjtdƒƒ}tj|djtj t t gƒkƒsOt ‚tj|djtj t t gƒkƒs€t ‚tj|djtj t t gƒkƒs±t ‚tj|djtj t t gƒkƒsât ‚tj|dj tj ddgƒkƒst ‚tj|dj tj dd gƒkƒsDt ‚t |dj tj d d gƒƒtj|dj tj t tgƒkƒs˜t ‚d S( sŒ Regression test for #1795 - this bug originally caused columns where TNULL was not defined to have their first element masked. stb.fitstc1tc2tc3tc4iitabcsxy g"Z3£™™ @g±£™¹ÌÌ@N(RR%tostpathR#tDATARRRQRRR&R RR(R^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_masking_regression_1795ús111111#c Cs³dddg}dddg}ddd g}t|||gd ddidd6ƒ}d|d _tjtƒ }|jddddtƒWdQX|jjddks¯t ‚dS(Niiig@g@gffffff @txtytzRRRRR<s first tableRs1.2st.fitsRGRR"is¨The column 'a' could not be stored in FITS format because it has a scale '(1.2)' that is not recognized by the FITS standard. Either scale the data or change the units.(RRR( RRLRfRzR R$RR|R}R&(RRRR^R~((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_scale_error s( cCsÚtjddtƒ}|ddd…tjkt|gƒ}|jt|jdƒƒdtƒt j t|jdƒƒƒV}|dj dj tj d ƒksªt ‚tj|dj d|kƒsÐt ‚WdQXdS( s› Regression test for https://github.com/astropy/astropy/issues/1953 Ensures that Table columns of bools are properly written to a FITS table. iR Nis test.fitsR"itcol0tbool(RtonesR•tFalse_RR$RR#RRR[R R R&R(R'tarrR^thdul((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_bool_columns"!)c Cs ttjdddgƒgƒ}|jt|jdƒƒdtƒtjt|jdƒƒƒS}tj |dj ddd d gkƒs“t ‚|dj d d ks°t ‚Wd QXttjdgƒgƒ}t jtƒ'|jt|jdƒƒdtƒWd QXd S(s Test that a column of unicode strings is still written as one byte-per-character in the FITS table (so long as the column can be ASCII encoded). Regression test for one of the issues fixed in https://github.com/astropy/astropy/pull/4228 uaubucds test.fitsR"iR”RRtcdtTFORM1t2ANu☃(RRRR$RR#RRR[RR R&theaderRfRztUnicodeEncodeError(R'R^R™R*((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_unicode_column's !"!/#c Csùt|jdƒƒ}tddgddggdddgƒ}d |d_d |d_tƒT}|j|d tƒt|ƒdkst‚t|d j ƒj d ƒs²t‚WdQXtƒ}tj |ddƒWdQXt|ƒd ksõt‚dS(Nstest_unit.fitsiiiiRRRsm/ss not-a-unitR"is''not-a-unit' did not parse as fits unitR\( RR#RRLR R$RR>R&RCRDR%(R'R(R)RE((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_unit_warnings_read_write>s*   ( c Cs\tjjtdƒ}tj|ƒ}|jddddddddddg ksXt‚dS(sL Regression test for https://github.com/astropy/astropy/issues/6079 s stddata.fitsR7ts *** End of mandatory fields ***s *** Column names ***s *** Column formats ***N(RŒRR#RŽRR%R<R&(R'R(R^((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyttest_convert_comment_conventionNs (,RŒRuRftnumpyRt numpy.testingRR¢RRRRRRJtextern.six.movesRRttableRR t tests.helperR tunits.format.fitsR RR#tdirnamet__file__RŽR,t ImportErrorRt HAS_PATHLIBRRtobjectRRjRR“RšR R¡R£(((sAlib/python2.7/site-packages/astropy/io/fits/tests/test_connect.pyts4    !   |[