U e{j6~@spddlZddlZddlmZddlZddlZddlmZm Z ddlm Z ddl m Z mZddlmZddlmZddlmZmZmZmZmZmZmZmZmZdd lmZdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*ddZ+ddZ,ddZ-ddZ.ddZ/ej0j1ej02deefddZ3ej02deefddZ4d d!Z5d"d#Z6d$d%Z7d&d'Z8d(d)Z9d*d+Z:d,d-Z;ej02d.d/d0gd1d2Zd7d8Z?d9d:Z@d;d<ZAd=d>ZBd?d@ZCdAdBZDdCdDZEdEdFZFej02dGd/d0gdHdIZGdJdKZHdLdMZIdNdOZJdPdQZKdRdSZLej02d.d/d0gej02dTeegdUdVZMdWdXZNdS)YN)StringIO) coordinatestime)units) EarthLocationSkyCoord)skycoord_equal)representation_equal) Column NdarrayMixinQTableTablehstackjoin serialize table_helpersvstack) BaseColumn)represent_mixins_as_columns) ArrayWrapper)ParentDtypeInfo)AstropyUserWarning)MergeConflictWarning) MIXIN_COLSc Cs |d}d|j_|jjdks td|j_|jjdks8tt|tjtjt j t j tj frxt ttj|j_W5QRXntj|j_|jjtjkstd|j_|jjdkstddi|j_|jjddikstt td|j_W5QRXt t|jjW5QRXdS)z6 Required attributes for a column can be set. marN)infonameAssertionError description isinstanceuQuantityrrrTime TimeDelta BaseRepresentationOrDifferentialpytestraisesAttributeErrorrunitformatmetaZbad_attr) mixin_colsrr.]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/tests/test_mixin.pytest_attributes%s4     r0cCs`t|jtjrt|tk s$t|tr8t||jksLtnt|t|ksLt|jj dks\tdSN) r!rr"Z QuantityInfotyper r ColumnClassrr)tableZ table_colZin_colr.r.r/check_mixin_typeNs  r5cCsn||}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)r r5listvalues) table_typesr-tcolsr.r.r/test_make_table]s   r?cCsPddlm}tt}|D]0}|drd|dkrt}|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=fmtoutr.r.r/test_io_ascii_writems   rEcCsnt}tjdddgdd|d<|d}|j|dd d tj|dd }t|dtjsXt|djdksjtd 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). rnmr*r table-tmpvotableTr+ overwriterAN)r r"r#rBreadr!rr*)tmp_pathr=filenameqtr.r.r/test_votable_quantity_write|srRr<c CsJ|ddgddgg}tjD]P}tjddgddggd|tdd d d d d |d|<tjddg|d|d|<qddg|d<|d}tjtdd|j|dddW5QRXtjtdd|j|ddd}W5QRXtjD]}dD]}||}t ||tjst ||j ||j kst ||j dks2t ||j ||j ksLt ||||kst qqd D]4}t ||tst ||||kslt qltjD]*}dD]}||}d!||jjd<qq|j|ddd|j|dd"}tjD]L}dD]@}||}t ||tjr"t ||||jkst qqd#S)$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. rrFstringcolumnrGZcxcsec隧@9>rrIr+scalelocationr1999-01-01T00:00:00.1234567892010-01-01T00:00:00rZr7@@crJzQTime Column "btai" has no specified location, but global Time Position is presentmatchfitsTrLz8Time column reference position "TRPOSn" is not specifiedr+Zastropy_nativerr7jdcol0col1raformatted_valuerAN)rZSTANDARD_TIME_SCALESr$rr'warnsrrBrNr!rrZr+r[allr rserialize_methodvalue)rOr<r=rZrPtmabrr.r.r/ test_io_time_write_fits_standards\      rrc Cs |ddgddgg}tjddgddggd d td d d ddd|d<tjddgd d|d<ddg|d<|d}tjtdd|j|dddW5QRXtjtdd|j|ddd }W5QRXd!D]~}|d"}t||tjst ||j ||j kst ||j d#kst ||j ||j ks*t ||||k st qd$D]4}t||ts`t ||||k sHt qHd!D]}|d"}d%||jjd<q|j|ddd|j|dd&}d!D]@}|d"}t||tjrt ||||jk st qd'S)(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. rrFrSrTiQiRiSiTZmjdlocalrVrWrXrrIrYZa_localr\r]r^Zb_localr_r`rarJz/Time Column "b_local" has no specified locationrbrdTrLz9Time column reference position "TRPOSn" is not specified.rerf_localrgrhrkrAN)rr$rr'rlrrBrNr!rrZr+r[rmr rrnro)rOr<r=rPrprqrr.r.r/test_io_time_write_fits_localsT   ruc Csht|}|jttjf}|s(tdt}t t }|j |ddW5QRXdt |j ksdtdS)zf Test that table with mixin columns (excluding Quantity) cannot be written by io.votable. zno unsupported column typesrKrAz'cannot write table with mixin column(s)N)r columnsZnot_isinstancerr"r#r'skiprr( ValueErrorrBstrror)r-r=Zunsupported_colsrDerrr.r.r/test_votable_mixin_write_fail-s  r{c Cs|}|ddddg|d<|ddddg|d<tD]\}}|||<q<||}dddd g|d<tD]$\}}|||j_|d ||j_qpd D]}t||d|d }|d }|d}tD]f\}}|d} |d} t|| ||t|| |||| jj|kst|| jj|d kstqqdD]@}t t } t||d|d }W5QRXdt | j ks4tq4t t} t||ddgd}W5QRXdt | j kstt jtddt||dgd}W5QRXt|d|dkstdS)zq Join tables with mixin cols. Use column "i" as proxy for what the result should be for each mixin. rr7rarrrFrUr6d2)innerleft)keys join_typei_1i_2_1_2)outerrightzjoin requires masking columnZskycoord)rz(one or more key columns are not sortablezFIn merged column 'quantity' the 'description' attribute does not matchrbquantityZa_1N)r r ritemsrr rassert_table_name_col_equalrr'r(NotImplementedErrorryro TypeErrorrlrnprm) r<t1rcolt2rt12idx1idx2name1name2excr.r.r/ test_join@sD     rc Cs|}|ddddg|d<tD],\}}|||<|||j_ddi||j_q&dD]&}dD]}||}|r|d d }|d krtt }t ||g|d }W5QRXd t |j ksbt qbt ||g|d }|d} |d} tD]\}}|d} |d} t|| || t|| || dD]R} t||j| t|| j| ksRt t||j| t|| j| ks(t q(qqbqXd S)zs Hstack tables with mixin cols. Use column "i" as proxy for what the result should be for each mixin. rrrFrUr6r)r~r)TFNr)rzhstack requires masking columnrrrr)r r,)r r rrrr r,r'r(rrryrorrgetattr)r<rrrrZchoprrrrrrrattrr.r.r/ test_hstackpsB       rcCst|tjrBt||j|jks&tt||j|jkstnt|tjrhtt |||stnnt|t j rt |t krt|||kstn>t|tjrt||j|jkstnt|||kstdS)zQ Assert all(t[name] == col), with special handling for known mixin cols. N)r!rrrrmrardecr&r r"r#r2r rrdata)r=rrr.r.r/rs     rcCsd}|d}d|j_d|j_d|j_ddi|j_t|g}ddgtdd gtddfD]j}||}||}t |d|||D]@}t |dj|t |j|kst t |j|t |j|kst qqXd S) zU Test that slicing / indexing table gives right values and col attrs inherit rr*dtyper+r r,r{0}r|rrrUrrFN) rrr+r r,r rarrayslicerrr)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|tjtjfkrt|j|} n|jj d} t|j|| ksht qhqXqdS)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|Sr1)pickleloadsdumpsrar.r.r/pickle_roundtripsz>test_info_preserved_pickle_copy_init..pickle_roundtripcSs ||Sr1) __class__rr.r.r/init_from_classsz=test_info_preserved_pickle_copy_init..init_from_classr)r6rrr|rrrZisnativeTvoid=N) rrr+r r,copydeepcopyrr startswithZ newbyteorderr) r-rrrZcolnamerfuncrroriginalr.r.r/$test_info_preserved_pickle_copy_inits. rcCsvt|tr$|jj}|jj}n|j}|j}ddt||D}|rft|rrtn t |srtdS)zCheck whether data attributes in col1 and col2 share memory. If copy=True, this should not be the case for any, while if copy=False, all should share memory. cSs.g|]&\}}t|tjr|jrt||qSr.)r!rZndarrayshapeZmay_share_memory).0Zv1Zv2r.r.r/ s z&check_share_memory..N) r!rrrZ_represent_as_dictzipr;anyrrm)rjZcol2rZmap1Zmap2Zsharedr.r.r/check_share_memorys    rrTFcCsd}|d}|jjdkstt|gdg|d}|jjdks>tt||d|d||d<|jjdkshtt||dddd|j_d |j_d |j_dd i|j_t|g|d}|jdgkstt||d|dt|gd g|d}|jjdkst|jd gkstt||d |d||d <t||d ddd|j_|j |g|dt||d|ddD]N}t ||||D]6}|dkrpt |d j|t ||j|ksptqpq\|d}t |t |kr Table => QTable rN)r rrrr r rrrmror*r!r3r)rQr9rr=Zqt2r.r.r/test_conversion_qtable_tables rcCsRt}ddg|d<d|d<t|dddgks4tt|dddgksNtdS)z? Test for mixin-related regression described in #3321. xyrr7N)r rrmrr=r.r.r/test_setitem_as_column_names  rcCs2tddgtjg}|dddddgks.tdS) zI Test that table representation of quantities does not have unit rrFriz m z----z 1.0z 2.0N)r r"rpformatrrr.r.r/test_quantity_representationsrcCsZtjdgdgdgtjd}t|g}|dddgks:ttjdgdgdgdd}t|g}|ddddgkstttd gtjd gtjdgtj }t|g}|d d d dgkstt d gtjd gtj}t|g}|ddddgkstt d gtj tj dgtj tj d gtjtj}t|g}|ddddgksVtdS)z> Test that Representations are represented correctly. rrrI col0 ------------z (0., 1., 0.)rz m  col0 z deg, deg, pc --------------z(10., 20., 1.)z col0 z deg z ----------z (10., 20.)rFz col0 zmas / yr, mas / yr, km / sz--------------------------z (10., 2., 10.)N)rZCartesianRepresentationr"oner rrZSphericalRepresentationdegZpcZUnitSphericalRepresentationZSphericalCosLatDifferentialZmasyrkmrrar=r.r.r/"test_representation_representationsV    &     rcCstjdgdgdgdd}t|g}|ddddgks:ttjdgdgdgd dd }t|g}|d d d dgksvtd|d_|ddddgkstd|d_|ddddgkstdS)z Test that skycoord representation works, both in the way that the values are output and in changing the frame representation. rrZ cartesian)representation_typerzNone,None,Nonerz 0.0,1.0,0.0r)r*rz col0 z m,m,m z -----------z 0.0,1.0,0.0Z unitsphericalri col0 zdeg,deg --------z90.0,0.0Z cylindricalrz m,deg,m rz 1.0,90.0,0.0N)rrr rrrrr.r.r/test_skycoord_representations<       ras_ndarray_mixincCsTtjddddgdd}tjddd d gd d gd}tjjd dddgddgd}tjddddd}|r|t}|t}|t}|t}t}nt}t |gdgd}||d<||d<||d<t |d|st |ddd|ddkst |ddd|ddkst |ddd|ddks6t |ddd|ddksXt t |d|slt |ddd|ddkst |ddd|ddkst |ddd|ddkst |ddd|ddkst t |d|st |ddd|ddks*t |ddd|ddksLt |ddd|ddksnt |ddd|ddkst t |d|st |ddd|ddkst |ddd|ddkst |ddd|ddks t |ddd|ddks,t |j d d!d"d#d$d%d&d'd(gksPt d)S)*a' Test directly adding various forms of structured ndarray columns to a table. Adding as NdarrayMixin is expected to be somewhat unusual after #12644 (which provides full support for structured array Column's). This test shows that the end behavior is the same in both cases. )rr)rFr7)rUra)rGr|z.MyArrayWrapperN)__name__ __module__ __qualname__rrr.r.r.r/MyArrayWrappersrrrrFrpzSfailed to represent column 'tm' \(MyArrayWrapper\) as one or more Column subclassesrbN)rr rr'r(rr)rr=rDrcr.r.r/test_bad_info_classsr)Orriornumpyrr'Zastropyrrrr"Zastropy.coordinatesrrZ astropy.coordinates.tests.helperrZ-astropy.coordinates.tests.test_representationr Z astropy.tabler r r r rrrrrZastropy.table.columnrZastropy.table.serializerZastropy.table.table_helpersrZastropy.utils.data_inforZastropy.utils.exceptionsrZastropy.utils.metadatarZconftestrr0r5r?rErRmarkZ remote_dataZ parametrizerrrur{rrrrrrrrrrrrrrrrrrrrrrrrrr.r.r.r/sr    ,       ) S J0($ B    4* L$