ó šßÈ[c@s#yddlZWnek r)eZnXeZyddlZeZWnek r_eZnXddlZddlZddl Z ddl m Z ddl mZmZddlmZmZmZmZmZmZmZddlmZddlmZddlmZd d lm Z d d lm!Z!d d l"m#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/d„Z0d„Z1d„Z2d„Z3d„Z4d„Z5d „Z6d!„Z7d"„Z8d#„Z9d$„Z:d%„Z;d&„Z<dS('iÿÿÿÿNi(tsix(tcPicklet cStringIO(tTabletQTabletjointhstacktvstacktColumnt NdarrayMixin(ttime(t coordinates(tunitsi(t BaseColumn(t table_helpersi(t MIXIN_COLScCsa|d}d|j_|jjdks.t‚d|j_|jjdksRt‚t|tjtjt j fƒr›t j t ƒtj|j_WdQXn*tj|j_|jjtjksÅt‚d|j_|jjdksét‚idd6|j_|jjidd6kst‚t j t ƒd|j_WdQXt j t ƒ|jjWdQXdS(s6 Required attributes for a column can be set. tmtaNi(tinfotnametAssertionErrort descriptiont isinstancetutQuantityR tSkyCoordR tTimetpytesttraisestAttributeErrorRtunittformattmetatbad_attr(t mixin_colsR((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_attributes!s$   ! cCsŽt|jtjƒr't|ƒtk s6t|tƒrTt|ƒ|jksrt‚nt|ƒt|ƒksrt‚|jj dksŠt‚dS(N( RRRt QuantityInfottypeRRt ColumnClassRRtNone(ttablet table_coltin_col((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pytcheck_mixin_typeAs 'cCs‘|j|ƒ}t||d|dƒt|jƒƒ}|j|ddƒ}t||d|dƒ|j|ƒ}t||d|dƒdS( sŸ Make a table with the columns in mixin_cols, which is an ordered dict of three cols: 'a' and 'b' are table_types.Column type, and 'm' is a mixin. RtnamestiRtbtcol3N(R-RR.R(RR+tlisttvalues(t table_typesR"tttcols((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_make_tableOscCsŽddlm}ttƒ}xk|ƒD]`}|ddkrIt rIq&n|dr&d|dkr&tƒ}|j|d|dƒq&q&WdS( s´ Test that table with mixin column can be written by io.ascii for every pure Python writer. No validation of the output is done, this just confirms no exceptions. i(t_get_connectors_tabletFormats ascii.ecsvtWrites.fast_RN(tio.ascii.connectR6RRtHAS_YAMLtStringIOtwrite(R6R3tfmttout((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_io_ascii_write_s  c Cstƒ}tjdddgddƒ|d<|jdƒj}t|dƒjƒxqdD]i}|j|d |d tƒtj |d |ƒ}t |dtjƒs§t ‚|dj dksWt ‚qWWt rtjtƒ }|j|d d d tƒWdQXdt|jƒkst ‚ndS(s Test that table with Quantity mixin column can be written by io.fits, io.votable but not by io.misc.hdf5. Validation of the output is done. Test that io.fits writes a table containing Quantity mixin columns that can be round-tripped (metadata unit). iiiRtAngstromRs table-tmptwtfitstvotableRt overwritethdf5Ns'cannot write table with mixin column(s)(RBRC(RRRRtstrpathtopentcloseR<tTruetreadRRRtHAS_H5PYRRt ValueErrortstrtvalue(ttmpdirR3tfilenameR=tqtterr((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_io_quantity_writeos " c Csºt|ƒ}|jjttjfƒ}|s=tjdƒnxvdD]n}|dkrct rcqDnt ƒ}tj t ƒ}|j |d|ƒWdQXdt |jƒksDt‚qDWdS( s€ Test that table with mixin column (excluding Quantity) cannot be written by io.votable, io.fits, and io.misc.hdf5. sno unsupported column typesRBRCRERNs'cannot write table with mixin column(s)(RBRCRE(Rtcolumnstnot_isinstanceR RRRtskipRKR;RRLR<RMRNR(R"R3tunsupported_colsR=R>RR((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_io_write_failŠs   c Cs˜|jƒ}|jddddgƒ|d<|jddddgƒ|dtjƒD]0\}}|||j_|d ||j_q£WxÒdD]Ê}t||d dd|ƒ}|d}|d}x’tjƒD]„\}}|d} |d} t|| ||ƒt|| ||ƒ|| jj|ks„t‚|| jj|d ks t‚q WqÞWx\dD]T}t j t ƒ"} t||d dd|ƒ}WdQXdt | j ƒks³t‚q³Wt j tƒ"} t||d ddgƒ}WdQXdt | j ƒksYt‚t||d dgƒ}tj|d|dkƒs”t‚dS(sq Join tables with mixin cols. Use column "i" as proxy for what the result should be for each mixin. RR.tciiiiR-tdt2tinnertlefttkeyst join_typeti_1ti_2t_1t_2toutertrightNsjoin requires masking columntskycoordsnot allowed as a key columntquantityta_1(R\R](RdRe(RRRtitemsRRRtassert_table_name_col_equalRRRtNotImplementedErrorRMRNRLtnptall( R2tt1Rtcoltt2R_tt12tidx1tidx2tname1tname2texc((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyt test_joinžs<      ( !!c Cs|jƒ}|jddddgƒ|d|D]6}||jj|ks2t‚t ||||ƒqWdS(sF Test that a table round trips from QTable => Table => QTable RgN(RRtcolnamesRRRRRlRmRNRRR&Rj(RQR,RR3tqt2((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_conversion_qtable_table°s       &)  cCsqtƒ}ddg|ds&   cCst|ƒ}|jƒ}|jddƒ|jddddgksIt‚t|dtjƒr‹tj |dj |dj kƒst‚nŽt|dt j ƒrötj |dj |dj kƒsÊt‚tj |dj|djkƒst‚n#tj |d|dkƒst‚dS(s Rename a mixin column. RtmmR-RR.N(RRŽt rename_columnR¬RRRRRlRmR€R RR}R~(R"R3ttc((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyttest_rename_mixin_columns\s  !,),(=th5pyt ImportErrorRxRKRItyamlR:RŽRtnumpyRltexternRtextern.six.movesRRˆRR;R(RRRRRRR tR R R RtcolumnR RtconftestRR#R+R5R?RSRXRwR|RjR‡R’R˜R™R›RœR£R¨R«R®R±R´R¹RÍRÎRÒ(((s=lib/python2.7/site-packages/astropy/table/tests/test_mixin.pyts\         4      , $    5    " <