a ߙfbw@sJddlZddlZddlmZddlZddlZddlmZddl m Z ddl m Z ddl mZmZddlmZmZmZmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddZ+ddZ,ddZ-ddZ.ddZ/ej0j1ej02deefdd Z3ej02deefd!d"Z4d#d$Z5d%d&Z6d'd(Z7d)d*Z8d+d,Z9d-d.Z:d/d0Z;d1d2Zd7d8Z?d9d:Z@d;d<ZAd=d>ZBd?d@ZCdAdBZDdCdDZEdEdFZFdGdHZGdIdJZHdKdLZIdMdNZJdOdPZKej02dQeegdRdSZLdTdUZMdS)VN)StringIO)represent_mixins_as_columns)ParentDtypeInfo) ArrayWrapper) EarthLocationSkyCoord)TableQTablejoinhstackvstackColumn NdarrayMixin) serialize)time) coordinates)units) BaseColumn) table_helpers)AstropyUserWarning)MergeConflictWarning)representation_equal)skycoord_equal) MIXIN_COLScCsL|d}d|j_|jjdks Jd|j_|jjdks8Jt|tjtjtj tj tj frt ttj|j_Wdq1s0Yntj|j_|jjtjusJd|j_|jjdksJddi|j_|jjddiksJt td|j_Wdn1s 0Yt t|jjWdn1s>0YdS)z6 Required attributes for a column can be set. maNr)infoname description isinstanceuQuantityrrrTime TimeDelta BaseRepresentationOrDifferentialpytestraisesAttributeErrorrunitformatmetaZbad_attr) mixin_colsrr-=lib/python3.9/site-packages/astropy/table/tests/test_mixin.pytest_attributess( *   ( r/cCs`t|jtjrt|tus$t|tr8t||jusLJnt|t|usLJ|jjdus\JdSN) r rr!Z QuantityInfotyper r ColumnClassr)tableZ table_colZin_colr-r-r.check_mixin_type@s r4cCsn||}t||d|dt|}|j|dd}t||d|d||}t||d|ddS)z 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. r)irbrnamesZcol3N)rr4listvalues) table_typesr,tZcolsr-r-r.test_make_tableNs   r=cCsPddlm}tt}|D]0}|drd|dvrt}|j||ddqdS)z 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. r)_get_connectors_tableZWritez.fast_ZFormatr*N)Zastropy.io.ascii.connectr>r rrwrite)r>r<Zfmtoutr-r-r.test_io_ascii_write^s   rBcCsrt}tjgddd|d<t|d}|j|dddtj|dd }t|dtjs\J|djdksnJd S) a Test that table with Quantity mixin column can be round-tripped by io.votable. Note that FITS and HDF5 mixin support are tested (much more thoroughly) in their respective subpackage tests (io/fits/tests/test_connect.py and io/misc/tests/test_hdf5.py). )rZnmr)r table-tmpvotableTr*Z overwriter?N) r r!r"strr r@readr r))tmpdirr<filenameqtr-r-r.test_votable_quantity_writelsrNr;c Cs~|ddgddgg}tjD]P}tjddgddggd|tdd d d d d |d|<tjddg|d|d|<qddg|d<t|d}tjtdd |j |dddWdn1s0Ytjtdd |j |ddd}Wdn1s0YtjD]}dD]}||}t ||tjs2J||j ||j ksLJ||j d ks`J||j||jkszJ||||ksJqqd!D]4}t ||tsJ||||ksJqtjD]*}dD]}||}d"||jjd<qq|j |ddd|j |dd#}tjD]L}dD]@}||}t ||tjrVJ||||jks4Jq4q,dS)$a} Test that table with Time mixin columns can be written by io.fits. Validation of the output is done. Test that io.fits writes a table containing Time mixin columns that can be partially round-tripped (metadata scale, location). Note that we postpone checking the "local" scale, since that cannot be done with format 'cxcsec', as it requires an epoch. rrCstringcolumnrDZcxcsec隧@9>rrEr*scalelocationr1999-01-01T00:00:00.1234567892010-01-01T00:00:00rVr6@@crFzQTime Column "btai" has no specified location, but global Time Position is presentmatchfitsTrHNz8Time column reference position "TRPOSn" is not specifiedr*Zastropy_nativerr6jdcol0Zcol1r]formatted_valuer?)rZSTANDARD_TIME_SCALESr#rrIr r&warnsrr@rJr rVr*rWallr rserialize_methodvalue)rKr;r<rVrLtmabrr-r-r. test_io_time_write_fits_standard}s\   .. "  rmc Cs<|ddgddgg}tjddgddggd d td d d ddd|d<tjddgd d|d<ddg|d<t|d}tjtdd |j|dddWdn1s0Ytjtd d |j |ddd!}Wdn1s0Yd"D]}|d#}t ||tjsJ||j ||j ks.J||j d$ksBJ||j ||j ks\J||||ksJqd%D]4}t ||tsJ||||kszJqzd"D]}|d#}d&||jjd<q|j|ddd|j |dd'}d"D]@}|d#}t ||tjrJ||||jksJqdS)(a Test that table with a Time mixin with scale local can also be written by io.fits. Like ``test_io_time_write_fits_standard`` above, but avoiding ``cxcsec`` format, which requires an epoch and thus cannot be used for a local time scale. rrCrOrPiQiRiSiTZmjdlocalrRrSrTrrErUZa_localrXrYrZZb_localr[r\r]rFz/Time Column "b_local" has no specified locationr^r`TrHNz9Time column reference position "TRPOSn" is not specified.rarb_localrcrdrfr?)rr#rrIr r&rgrr@rJr rVr*rWrhr rrirj)rKr;r<rLrkrlrr-r-r.test_io_time_write_fits_localsR   ..rpcCs|t|}|jttjf}|s(tdt}t t }|j |ddWdn1s\0Ydt |j vsxJdS)zf Test that table with mixin columns (excluding Quantity) cannot be written by io.votable. zno unsupported column typesrGr?Nz'cannot write table with mixin column(s))r columnsZnot_isinstancerr!r"r&skiprr' ValueErrorr@rIrj)r,r<Zunsupported_colsrAerrr-r-r.test_votable_mixin_write_fails  ,ruc Cs2|}|gd|d<|gd|d<tD]\}}|||<q4||}gd|d<tD]$\}}|||j_|d||j_qddD]}t||d|d}|d }|d }tD]f\}}|d } |d } t|| ||t|| |||| jj|ksJ|| jj|dksJqqd D]V}t t  } t||d|d}Wdn1s^0Ydt | j vs(Jq(t t "} t||ddgd}Wdn1s0Ydt | j vsJtjtdd t||dgd}Wdn1s0Yt|d|dks.JdS)zq Join tables with mixin cols. Use column "i" as proxy for what the result should be for each mixin. )rr6r6r]rrrrCrQr5)r6r]rd2)innerleft)keys join_typei_1i_2_1_2)outerrightNzjoin requires masking columnZskycoord)r{z(one or more key columns are not sortablezFIn merged column 'quantity' the 'description' attribute does not matchr^quantityZa_1)rr ritemsrrr assert_table_name_col_equalr&r'NotImplementedErrorrIrj TypeErrorrgrnprh) r;t1rcolt2r|t12idx1idx2name1name2excr-r-r. test_join"sB     0 20rc Cs|}|gd|d<tD],\}}|||<|||j_ddi||j_q"dD]:}dD].}||}|r|dd}|d krtt  }t ||g|d }Wdn1s0Yd t |j vs^Jq^t ||g|d }|d } |d } tD]\}}|d} |d} t || || t || || dD]R} t||j| t|| j| ksbJt||j| t|| j| ks8Jq8qq^qTdS)zs Hstack tables with mixin cols. Use column "i" as proxy for what the result should be for each mixin. rvr5rr)ryr)TFNr)r|zhstack requires masking columnr}r~rr)rr+)rr rrrrr+r&r'rr rIrjrgetattr)r;rrrr|Zchoprrrrrrrattrr-r-r. test_hstackQs6      .&rcCst|tjrBt||j|jks&Jt||j|jksJnt|tjrhtt|||sJnnt|t j rt |t urt|||ksJn>t|t jrt||j|jksJnt|||ksJdS)zQ Assert all(t[name] == col), with special handling for known mixin cols. N)r rrrrhradecr%rr!r"r1r rrdata)r<rrr-r-r.rus     rcCsd}|d}d|j_d|j_d|j_ddi|j_t|g}ddgtdd gtddfD]j}||}||}t |d|||D]@}t |dj|t |j|ksJt |j|t |j|ksJqqXd S) zU Test that slicing / indexing table gives right values and col attrs inherit rr)dtyper*rr+r{0}rwrrrQrrCN) rrr*rr+r rarrayslicerr)r,attrsrr<itemrm2rr-r-r.test_get_itemss  " rc Csdd}dd}d}dD]}||}||j_d|j_d|j_d d i|j_tjtj||fD]~}||}|D]l}|d kst|jjd d s|jjj ds|tjtjfvrt|j|} n|jj d} t|j|| kshJqhqXqdS)z Test copy, pickle, and init from class roundtrip preserve info. This tests not only the mixin classes but a regular column as well. cSstt|Sr0)pickleloadsdumpsr]r-r-r.pickle_roundtripsz>test_info_preserved_pickle_copy_init..pickle_roundtripcSs ||Sr0) __class__rr-r-r.init_from_classsz=test_info_preserved_pickle_copy_init..init_from_classr)r5rrrwrrrZisnativeTZvoid=N) rrr*rr+copydeepcopyrr startswithZ newbyteorder) r,rrrZcolnamerfuncrrZoriginalr-r-r.$test_info_preserved_pickle_copy_inits, rcCsd}|d}|jjdusJt|gdgd}|jjdus Table => QTable rN) r rrrrrrrhrjr)r r2r)rMr8rr<Zqt2r-r-r.test_conversion_qtable_tableAs rcCsRt}ddg|d<d|d<t|dddgks4Jt|dddgksNJdS)z? Test for mixin-related regression described in #3321. xyrr6N)rrrhr<r-r-r.test_setitem_as_column_nameZs  rcCs,tddgtjg}|gdks(JdS)zI Test that table representation of quantities does not have unit rrC)rez m z----z 1.0z 2.0N)r r!rpformatrr-r-r.test_quantity_representationesrcCsHtjdgdgdgtjd}t|g}|gdks8Jtjdgdgdgdd}t|g}|gdksnJtdgtjdgtjdgtj}t|g}|gd ksJt dgtjdgtj}t|g}|gd ksJt dgtj tj d gtj tj dgtj tj}t|g}|gd ksDJd S)z> Test that Representations are represented correctly. rrrE) col0 ------------ (0., 1., 0.)r)rz m rr ) col0 z deg, deg, pc --------------z(10., 20., 1.))z col0 z deg z ----------z (10., 20.)rC)z col0 zmas / yr, mas / yr, km / sz--------------------------z (10., 2., 10.)N)rZCartesianRepresentationr!ZonerrZSphericalRepresentationdegZpcZUnitSphericalRepresentationZSphericalCosLatDifferentialZmasZyrZkmrr]r<r-r-r."test_representation_representationqs"  &  0 rcCstjdgdgdgdd}t|g}|gdks6Jtjdgdgdgddd}t|g}|gdksnJd |d _|gd ksJd |d _|gd ksJdS)z Test that skycoord representation works, both in the way that the values are output and in changing the frame representation. rrZ cartesian)representation_type)rzNone,None,Nonerz 0.0,1.0,0.0r)r)r)z col0 z m,m,m z -----------z 0.0,1.0,0.0Z unitsphericalre) col0 zdeg,deg --------z90.0,0.0Z cylindrical)rz m,deg,m rz 1.0,90.0,0.0N)rrrrrrr-r-r.test_skycoord_representations    rcCstjgddd}tjgdddgd}tjjgddd gd }td d d t}t|gdgd }||d<||d<||d<t |dtsJ|ddd|ddksJ|dd d|d dksJ|ddd|ddksJ|d dd|d dksJt |dts(J|ddd|ddksJJ|ddd|ddkslJ|ddd|ddksJ|ddd|ddksJt |dtsJ|ddd|ddksJ|ddd |dd ksJ|ddd|ddks*J|ddd |dd ksLJt |dts`J|ddd|ddksJ|ddd|ddksJ|ddd|ddksJ|ddd|ddksJ| gdksJdS)z Test directly adding a plain structured array into a table instead of the view as an NdarrayMixin. Once added as an NdarrayMixin then all the previous tests apply. ))rr)rCr6)rQr])rDrwz.MyArrayWrapperN)__name__ __module__ __qualname__rrr-r-r-r.MyArrayWrapperfsr)rrrCrkzSfailed to represent column 'tm' \(MyArrayWrapper\) as one or more Column subclassesr^N)rrrr&r'rr)rr<rAr_r-r-r.test_bad_info_classcsr)Nrriorr&ZnumpyrZastropy.table.serializerZastropy.utils.data_inforZastropy.table.table_helpersrZastropy.coordinatesrrZ astropy.tablerr r r r r rrZastropyrrrr!Zastropy.table.columnrrZastropy.utils.exceptionsrZastropy.utils.metadatarZ-astropy.coordinates.tests.test_representationrZ!astropy.io.misc.asdf.tags.helpersrZconftestrr/r4r=rBrNZmarkZ remote_dataZ parametrizermrprurrrrrrrrrrrrrrrrrrrrrrrrr-r-r-r.sp    $           ! K D/$!5     )"<