a ߙfb&S@sddlZddlZddlmZddlmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZmZdd lmZdd lmZdd lmZGd d d eZdS)N) FitsTestCase)GLOBAL_TIME_INFO time_to_fitsis_time_column_keyword) EarthLocation)fits)TableQTableColumn)Time TimeDelta)BARYCENTRIC_SCALES)FITS_DEPRECATED_SCALES)AstropyUserWarningc@sLeZdZddZddZejdee fddZ ejdd d d gfejd d d dZ ejdee fddZ ejdee fddZ ejdee fddZejdee fddZejdee fddZejdee fddZejdee fddZejdee fddZejdee fdd Zd!S)" TestFitsTimecCs4tddg|_tddgddgddggg|_dS)Nz1999-01-01T00:00:00.123456789z2010-01-01T00:00:00r)nparraytimetime_3dselfrBlib/python3.9/site-packages/astropy/io/fits/tests/test_fitstime.py setup_classszTestFitsTime.setup_classcCs4tddusJtddus Jtddus0JdS)NZTRPOSFTIMESYSZTRPOS12T)rrrrrtest_is_time_column_keywordsz(TestFitsTime.test_is_time_column_keyword table_typesc CsR|}t|jddd|d<t|jddd|d<tddgdd gd d gd d |d_tjtd dt|\}}Wdn1s0Y|d|djjj dd k sJ|d|djj j dd k sJ|d|djj j dd k sJtjtd d&|j |ddddWdn1s20Yt|dd}|dddusdJtjtdd&|j|dddd}Wdn1s0Y|dj|djk sJ|dj|djksJtddd|d_tddd|d_tt0}t|\}}dt|jvs2JWdn1sH0Yd|d_tjtd d}t|\}}Wdn1s0Yt|dksJtddd|d_t|\}}tD]j}|dt|d|tjtd d}t|\}}Wdn1s0Yt|dksJq|}tddgdd gd d gggddgdd gd d gggddgdd gd d gggd d } t|jd!| d"|d<t|\}}|d|djjj dd k sJ|d|djj j dd k sJ|d|djj j dd k sJ|j |dddd|j|dddd}|dj|djk sfJt|jtdgggdgggd gggd d d#|d<t|\}}|d|djjj dd ksJ|d|djj j dd ksJ|d|djj j dd ksJ|j |dddd|j|dddd}|dj|djksNJdS)$zk Test all the unusual conditions for locations of ``Time`` columns in a ``Table``. isotutcformatscaleattbg?g@g@g@ZMmunitNTime Column "b" has no specified location, but global Time Position is presentmatchNOBSGEO-XmOBSGEO-YOBSGEO-Z time.fitsrTr$ overwriterZTRPOS2zTime column reference position "TRPOSn" is not specified. The default value for it is "TOPOCENTER", and the observatory position has been specified.r$astropy_nativerrrz/Multiple Time Columns with different geocentricz+Earth Location "TOPOCENTER" for Time ColumnZjd)r$location)r7)r rrr7pytestwarnsrrxto_valueallyzwritetemprZ getheadergetreadZraises ValueErrorstrvaluelenrZreplace_columngetattrr) rr ttablehdrtmerrwr%r7rrrtest_time_to_fits_loc"s *$$$6&  4 , , &&&  """z"TestFitsTime.test_time_to_fits_locmaskFTserialize_method)jd1_jd2formatted_valuecCsttjj|j|d}ttjjddgddggt|dddd}|j|jur^|d uusdnJt||gd d gd }|j| d dd|dtj | d ddd}|d urt |d j |j ksJt |d j |j ksJ|dkr>t |d trt |d |ksJt |d tr8t |d |ksJnTt |d trdt |d |jkshJt |d trt |d |jksJdS)zO Test the data returned by ``time_to_fits`` for masked values. )rOrrrr)rrcxcsecr$Fr&r()namesr2rT)r$r4rPr5rQN)r rZmaZ MaskedArrayrZ broadcast_toZmaskedr r?r@rBr<rO isinstancer rE)rrPrOr&r(rHrKrrr"test_time_to_fits_serialize_methods$$" (**z/TestFitsTime.test_time_to_fits_serialize_methodc Cs|}t|jddtdddddd|d <td d gd d d|d<|d jjj|d jjj|d jjjd}tj t ddt |\}}Wdn1s0Yt D]<\}}|||dksJ|j||d ksJ||q| D]$\}}|||ks J||q|jd}|D]<} || d|| jksHJ|| ddks$Jq$|d ddksxJ|dddksJt|dksJdS)zL Test the header and metadata returned by ``time_to_fits``. r!r"隧@9>r/r)r$r%r7r&rrrSr'r#r()r.r0r1r+r,NrZ__coordinate_columns__ coord_type coord_unitd time_ref_pos TOPOCENTER)r rrr7r:rEr=r>r8r9rrritemsZcommentsremovemetar%upperrF) rr rHZ ideal_col_hdrrIrJkeyrEZ coord_infocolnamerrrtest_time_to_fits_headers6     *    z%TestFitsTime.test_time_to_fits_headercCs|}t|jddd|d<d|jd<d|jd<|j|d d d d |j|d d d d }t|jdtsnJ|jdj|jdksJ|jdjd ksJ|jdj dksJt|jdtsJ|jdj|jdksJ|jdjdksJ|jdj dksJd|jd<|j|d d d d |j|d d d d }t|jdtsTJ|jdj|jdkspJ|jdj dksJt|jdtsJ|jdj|jdksJ|jdj t |jdksJd|dj j d <|j|d d d d |j|d d d}t|jdtr(J|jd|jdksBJt|jdtrXJ|jd|jdksrJ|d|djk sJdS)zn Test that the relevant global time metadata is read into ``Table.meta`` as ``Time``. r!r"r#r&z1999-01-01T00:00:00ZDATEi^zMJD-OBSr2rTr3r5mjdZETrrRrTN)r rrcr?r@rBrVrEr$r%rinforPr<)rr rHrKrrrtest_fits_to_time_metasF    z#TestFitsTime.test_fits_to_time_metac CsX|}t|jddtdddddd|d <t|\}}|d |d jjjd dksTJ|d |d jjjd dkstJ|d |d jjjd dksJ|j | dddd|j | dddd}|d j|d jk sJ|d jjj |d jjjd dksJ|d jjj |d jjjd dks,J|d jjj |d jjjd dksTJdS)z Test that ``location`` specified by using any valid unit (length/angle) in ``Time`` columns gets stored in FITS as ITRS Cartesian coordinates (X, Y, Z), each in m. Test that it round-trips through FITS. r!r"rrrZkmr)r[r&r.r/r0r1r2rTr3r5N)r rrrr7r:r;r=r>r?r@rBr<rE)rr rHrIrJrKrrrtest_time_loc_unit s       ((zTestFitsTime.test_time_loc_unitcCs|}t|jddd|d<ddg|d<dd g|d <|d|d|j|d d d d|j|d d d d}t|dtsJt|jdksJt|jdksJdD]4}t||j jdksJt||j jdksJqdS)zg Ensure that fits_to_time works correctly if the time column is also an index. r!r"r#r&rrr(rrcr2rTr3r5r)r&r(N) r rZ add_indexr?r@rBrVrFindicesri)rr rHrKnamerrrtest_fits_to_time_index's     z$TestFitsTime.test_fits_to_time_indexcCs|d}tjtdd|j|dd}Wdn1s<0Yt|dtsXJ|djdksjJ|djd ks|J||}t|j d d |j d  d }t |d}|||dk sJd D]}t||trJqdS)a Test that FITS table with time columns (standard compliant) can be read by io.fits as a table with Time columns. This tests the following: 1. The special-case where a column has the name 'TIME' and a time unit 2. Time from Epoch (Reference time) is appropriately converted. 3. Coordinate columns (corresponding to coordinate keywords in the header) other than time, that is, spatial coordinates, are not mistaken to be time. zchandra_time.fitsz5Time column "time" reference position will be ignoredr,Tr6Nrr'rhZMJDREFrr#)ZchipxZchipyZdetxZdetyr:r=) datar8r9rrBrVr r%r$rclowerr r<)rr filenamerKZ non_nativeZref_timeZ delta_timerfrrrtest_io_time_read_fitsGs ,    z#TestFitsTime.test_io_time_read_fitscCstjdddd|jd}tj|g}|j|ddd|j|ddd }t|dt s^J|dj d kspJ|dj d ksJ|d|jk sJd S) zP Test that ISO-8601 Datetime String Columns are read correctly. datetimeA29ZTCGZ GEOCENTERrnr$r\r_rr2Tr4rpZtcgrN) rr r BinTableHDU from_columnswritetor@rBrVr r%r$r<)rr rlbhdurKrrrtest_io_time_read_fits_datetimeks z,TestFitsTime.test_io_time_read_fits_datetimecCsdtjdddd|jd}gd}tjj|gt|d}|j|dd d |j|dd d }t |dt spJ|dj j j d ksJ|dj jj d ksJ|dj jj dksJgd}tjj|gt|d}|j|dd d |j|dd d }t |dt sJ|dj jj dks&J|dj jj dks>Jtj|dj jj dddds`JdS)z{ Test that geocentric/geodetic observatory position is read properly, as and when it is specified. rurvTTr`rw))r.rX)r0rY)r1rZheaderr2TrxrprXrYrZ)zOBSGEO-Lr)zOBSGEO-Br)zOBSGEO-Hrrg& .>)ZrtolZatolN)rr rryrzHeaderr{r@rBrVr r7r:rEr=r>ZlonZlatrZiscloseZheight)rr rlcardsr|rKrrrtest_io_time_read_fits_locations* z,TestFitsTime.test_io_time_read_fits_locationc Cstddg}tjdddddd|d}gd }tjj|gt|d }|j|d d d t j t dd$}|j |d d d}Wdn1s0Yt |dksJt|dtsJ|djdksJ|djdksJ|dj|ksJtddg}tjdddddd|d}tj|g}|j|d d d |j |d d d}t|dtshJ|djdks|J|djdksJ|dj|ksJdS)z Test handling of 'GPS' and 'LOCAL' time scales which are recognized by the FITS standard but are not native to astropy. i %i%gps_timeDsZGPSr`)rnr$r*r\r]r_rrrr2Trxz&FITS recognized time scale value "GPS"r,rpNrZgpsZtair local_timer^ZLOCALZ RELOCATABLErhlocal)rrrr ryrzrr{r@r8r9rrBrFrVr r$r%rEr<) rr rrlrr|rMrKrrrrtest_io_time_read_fits_scales4 2 z)TestFitsTime.test_io_time_read_fits_scalecCstjdddd|jd}tj|g}|j|dddtjt d d $}|j |ddd Wd n1sp0Yt |d ksJtjddd|jd}tj|g}|j|ddd|j |ddd d S)zC Test warnings for time column reference position. rurvr~r`rwr2Trxz.observatory position is not properly specifiedr,rpNr)rnr$r\r) rr rryrzr{r@r8r9rrBrF)rr rlr|rMrrr(test_io_time_read_fits_location_warningss 2 z5TestFitsTime.test_io_time_read_fits_location_warningsN)__name__ __module__ __qualname__rrr8ZmarkZ parametrizer r rNrWrgrjrkrortr}rrrrrrrrs2  e   ' ?   #  ' )r)r8ZnumpyrrZastropy.io.fits.fitstimerrrZastropy.coordinatesrZ astropy.iorZ astropy.tabler r r Z astropy.timer r Zastropy.time.corerZastropy.time.formatsrZastropy.utils.exceptionsrrrrrrs