HR-eTddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZddlmZmZddlmZddlmZdd lmZd d lmZGd d eZdS)N) EarthLocation)fits)GLOBAL_TIME_INFOis_time_column_keyword time_to_fits)ColumnQTableTable)Time TimeDelta)BARYCENTRIC_SCALES)FITS_DEPRECATED_SCALES)AstropyUserWarning) FitsTestCaseceZdZdZdZejdee fdZ ejddddgfejdd d Z ejdee fd Z ejdee fd Z ejdee fd Zejdee fdZejdee fdZejdee fdZejdee fdZejdee fdZejdee fdZdS) TestFitsTimectjddg|_tjddgddgddggg|_dS)Nz1999-01-01T00:00:00.123456789z2010-01-01T00:00:00r)nparraytimetime_3dselfs Clib/python3.11/site-packages/astropy/io/fits/tests/test_fitstime.py setup_classzTestFitsTime.setup_classsEH=?TUVV x1a&1a&1a&!9 :;; cxtddusJtddusJtddusJdS)NTRPOSFTIMESYSTRPOS12T)rrs rtest_is_time_column_keywordz(TestFitsTime.test_is_time_column_keywordsW%g..%7777&i00E9999&i00D888888r table_typesc |}t|jdd|d<t|jdd|d<tddgdd gd d gd |d_t jt d 5t|\}}dddn #1swxYwY|d|djj d k sJ|d|djj d k sJ|d|djj d k sJt jt d 5| |ddddddn #1swxYwYtj|dd}|ddJt jt d5||ddd}dddn #1swxYwY|dj|djk sJ|dj|djksJtddd|d_tddd|d_t jt(5}t|\}}dt+|jvsJ dddn #1swxYwYd|d_t jt d 5}t|\}}dddn #1swxYwYt/|dksJtddd|d_t|\}}t0D]}|dt5|d|t jt d 5}t|\}}dddn #1swxYwYt/|dksJ|}tddgdd gd d gggddgdd gd d gggddgdd gd d gggd } t|jd!| "|d<t|\}}|d|djj d k sJ|d|djj d k sJ|d|djj d k sJ| |ddd||ddd}|dj|djk sJt|jtdgggdgggd gggd #|d<t|\}}|d|djj d ksJ|d|djj d ksJ|d|djj d ksJ| |ddd||ddd}|dj|djksJdS)$zk Test all the unusual conditions for locations of ``Time`` columns in a ``Table``. isotutcformatscaleattbg?g@g@g@MmunitNTime Column "b" has no specified location, but global Time Position is presentmatchNOBSGEO-XmOBSGEO-YOBSGEO-Z time.fitsrTr+ overwriterTRPOS2zTime 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 Columnjd)r+location)rA)r rrrApytestwarnsrrxto_valueallyzwritetempr getheadergetreadraises ValueErrorstrvaluelenr replace_columngetattrr) rr&ttablehdrtmerrwr,rAs rtest_time_to_fits_locz"TestFitsTime.test_time_to_fits_loc%s KMMdie<<<#did;;;#(c S#Jc QUVVV# \ G    ) ) &aJE3  ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) j!QsV_%6%?%?S%?%I%IINNPPPPPj!QsV_%6%?%?S%?%I%IINNPPPPPj!QsV_%6%?%?S%?%I%IINNPPPPP \ G    K K GGDIIk**6TG J J J  K K K K K K K K K K K K K K KnTYY{33Q77wwx&&... \ N      !! +&&vd"B                3 AcFO388:::::#w1S6?2222(1a00#'1a00# ]: & & W#%aJE3DCIVVVVV W W W W W W W W W W W W W W W # \ @    ) %aJE3 ) ) ) ) ) ) ) ) ) ) ) ) ) ) )1vv{{{{(1a00#!!__ s(  E  S'!C&%"8"8 9 9 9"C -)!__ s  - - - - - - - - - - - - - - - q66Q;;;;; KMM Cj3*sCj 1 2Cj3*sCj 1 2Cj3*sCj 1 2     dl4(CCC#!!__ sj!QsV_%6%?%?S%?%I%IINNPPPPPj!QsV_%6%?%?S%?%I%IINNPPPPPj!QsV_%6%?%?S%?%I%IINNPPPPP  +&&vFFF    IIk " "6$  3 AcFO388::::: I"cUG9wiC5'NNN   # "!__ s:!C&/"3"<"<#"<"F"FFFFF:!C&/"3"<"<#"<"F"FFFFF:!C&/"3"<"<#"<"F"FFFFF  +&&vFFF    IIk " "6$  #w1S6?222222sl>BB!$B!,GG G -,I%%I),I)+MM M :NN N5QQ Q maskFTserialize_method)jd1_jd2formatted_valuec ttj|j|}ttjddgddggtj|dd}|j|jcxur|d uusnJt||gd d g }|| d dd|tj | d dd}|d urTtj |d j |j ksJtj |d j |j ksJ|dkrxt|d trtj |d |ksJt|d trtj |d |ksJdSt|d tr#tj |d |jksJt|d tr#tj |d |jksJdS)zO Test the data returned by ``time_to_fits`` for masked values. )r\rrrr)rrcxcsecr+Fr-r/)namesr:rT)r+r<r]r>r^N)r rma MaskedArrayr broadcast_tomaskedr rIrJrMrFr\ isinstancerrQ)rr]r\r-r/rUrXs r"test_time_to_fits_serialize_methodz/TestFitsTime.test_time_to_fits_serialize_methodsE ""494"88 9 9  E  1v1v.R_T65R5R  S S   x18::::E(9:::::: Aq6#s , , ,  IIk " "-    [;//t T T T u  6"S',!&011 1 1 16"S',!&011 1 1 1 y ( (bgt,, E31 1E1E E E Ebgt,, E31 1E1E E E E E Ebgv.. M26"S'QW:L3M3M M M Mbgv.. M26"S'QW:L3M3M M M M M Mr c |}t|jddtdddd|d <td d gd d |d<|d jjj|d jjj|d jjjd}tj td5t|\}}dddn #1swxYwYtj D]G\}}|||dksJ|j||d ksJ||H| D](\}}|||ksJ||)|jd}|D]G} || d|| jksJ|| ddksJH|d ddksJ|ddJt'|dksJdS)zL Test the header and metadata returned by ``time_to_fits``. r(r)隧@9>r7r1r+r,rAr-rrrar.r*r/)r6r8r9r3r4Nr__coordinate_columns__ coord_type coord_unitd time_ref_pos TOPOCENTER)r rrrArDrQrGrHrBrCrrritemscommentsremovemetar,upperrR) rr&rU ideal_col_hdrrVrWkeyrQ coord_infocolnames rtest_time_to_fits_headerz%TestFitsTime.test_time_to_fits_headers KMM I"8WgCHHH    # q!fXT:::##)/#)/#)/  \ G    ) ) &aJE3  ) ) ) ) ) ) ) ) ) ) ) ) ) ) )+022  JCs8uQx''''<$a0000 JJsOOOO'--//  JCs8u$$$$ JJsOOOOZ 89 ! < >>>#~.6663xx1}}}}}}s.C  CCc|}t|jdd|d<d|jd<d|jd<||d d d ||d d d }t |jdtsJ|jdj|jdksJ|jdjd ksJ|jdj dksJt |jdtsJ|jdj|jdksJ|jdjdksJ|jdj dksJd|jd<||d d d ||d d d }t |jdtsJ|jdj|jdksJ|jdj dksJt |jdtsJ|jdj|jdksJ|jdj t|jdksJd|dj j d <||d d d ||d d }t |jdtrJ|jd|jdksJt |jdtrJ|jd|jdksJ|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:00DATEi^zMJD-OBSr:rTr;r>mjdETr#r_rbN)r rrxrIrJrMrhrQr+r,rinfor]rF)rr&rUrXs rtest_fits_to_time_metaz#TestFitsTime.test_fits_to_time_metas KMMdie<<<#.v!y  +&&vFFF    IIk " "6$   "'&/400000wv$v6666wv%////wv$----"'),d33333wy!'16)+<<<<<wy!(E1111wy!'50000!y  +&&vFFF    IIk " "6$   "'&/400000wv$v6666wv$----"'),d33333wy!'16)+<<<<<wy!'+A!&BS+TTTTT0A# $V,  +&&vFFF   dii 44V  D Dbgfot44444wv!&.0000bgi0$77777wy!QVI%6666631S6<',,.......r c T|}t|jddtdddd|d <t|\}}|d |d jjd ksJ|d |d jjd ksJ|d |d jjd ksJ| | ddd| | ddd}|d j|d jk sJ|d jjj |d jjd ksJ|d jjj |d jjd ksJ|d jjj |d jjd ksJdS)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)rrrkmr1rnr-r6r7r8r9r:rTr;r>N)r rrrrArDrErGrHrIrJrMrFrQ)rr&rUrVrWrXs rtest_time_loc_unitzTestFitsTime.test_time_loc_unit4s KMM I"1a666    #"!__ s:!C&/"3"<"<#"<"F"FFFFF:!C&/"3"<"<#"<"F"FFFFF:!C&/"3"<"<#"<"F"FFFFF  +&&vFFF    IIk " "6$   3 AcFO388:::::#w!'1S6?+<+E+E3+E+O+OOOOO#w!'1S6?+<+E+E3+E+O+OOOOO#w!'1S6?+<+E+E3+E+O+OOOOOOOr c|}t|jdd|d<ddg|d<dd g|d <|d|d||d d d ||d d d }t |dtsJt|jdksJt|jdksJdD]L}t||j jdksJt||j jdksJMdS)zg Ensure that fits_to_time works correctly if the time column is also an index. r(r)r*r-rrr/rrcr:rTr;r>r)r-r/N) r r add_indexrIrJrMrhrRindicesr)rr&rUrXnames rtest_fits_to_time_indexz$TestFitsTime.test_fits_to_time_indexVsl KMMdie<<<#Q#Q# C C  +&&vFFF    IIk " "6$  "S'4((((( 19~~""""2:!#### 3 3Dqw|+,,1111r$x},--22222 3 3r c|d}tjtd5||d}dddn #1swxYwYt |dt sJ|djdksJ|djd ksJ||}t |j d d |j d  }t|d}||z|dk sJd D]}t ||t rJ dS)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 ignoredr4Tr?Nrr.rMJDREFr#r*)chipxchipydetxdetyrDrG) datarBrCrrMrhr r,r+rxlowerr rF)rr&filenamerX non_nativeref_time delta_timer}s rtest_io_time_read_fitsz#TestFitsTime.test_io_time_read_fitsws99011 \ N    A A!!(4!@@B  A A A A A A A A A A A A A A A"V*d+++++&z4''''&z E)))) %%h//  OH %/),2244    z&122 :%F388:::::D 5 5G!"W+t44 4 4 4 4 5 5sAAActjdddd|j}tj|g}||dd||dd }t|dtsJ|dj d ksJ|dj d ksJ|d|jk sJd S) zP Test that ISO-8601 Datetime String Columns are read correctly. datetimeA29TCG GEOCENTERrr+rprsrr:Tr<rtcgrN) rrr BinTableHDU from_columnswritetorJrMrhr r,r+rF)rr&rbhdurXs rtest_io_time_read_fits_datetimez,TestFitsTime.test_io_time_read_fits_datetimes K$)    ,,aS11 TYY{++t <<<   dii 44T  J J"Z.$/////*~#u,,,,*~$....:$)+002222222r cftjdddd|j}gd}tj|gtj|}||dd ||dd }t|dtsJ|dj j j d ksJ|dj jj d ksJ|dj jj dksJgd}tj|gtj|}||dd ||dd }t|dtsJ|dj jj dksJ|dj jj dksJt%j|dj jj dddsJdS)z{ Test that geocentric/geodetic observatory position is read properly, as and when it is specified. rrTTrtr))r6rk)r8rl)r9rmheaderr:Trrrkrlrm)zOBSGEO-Lr)zOBSGEO-Br)zOBSGEO-Hrrg& .>)rtolatolN)rrrrrHeaderrrJrMrhr rArDrQrGrHlonlatriscloseheight)rr&rcardsrrXs rtest_io_time_read_fits_locationz,TestFitsTime.test_io_time_read_fits_locations" K%)    WVV,,aSU9K9K,LL TYY{++t <<<   dii 44T  J J"Z.$/////*~&(.(::::*~&(.'9999*~&(.'9999DCC,,aSU9K9K,LL TYY{++t <<<   dii 44T  J J"Z.$/////*~&*0A5555*~&*0A5555z"Z.18>PTUUUUUUUUr c tjddg}tjdddddd|}gd }tj|gtj| }||d d tj td5}| |d d }dddn #1swxYwYt|dksJt|dtsJ|djdksJ|djdksJ|dj|ksJtjddg}tjdddddd|}tj|g}||d d | |d d }t|dtsJ|djdksJ|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_timeDsGPSrt)rr+r2rprqrsrrrr:Trz&FITS recognized time scale value "GPS"r4rNrgpstair local_timerrLOCAL RELOCATABLErlocal)rrrrrrrrrJrBrCrrMrRrhr r+r,rQrF) rr&rrrrrZrXrs rtest_io_time_read_fits_scalez)TestFitsTime.test_io_time_read_fits_scales8Y 233 K%   DCC,,aSU9K9K,LL TYY{++t <<< \ &N    O !!$))K"8"8!NNB O O O O O O O O O O O O O O O1vv{{{{"Z.$/////*~$----*~#u,,,,:$05577777Xq!f%% K&   ,,aS11 TYY{++t <<<   dii 44T  J J"\*D11111,&%////,%0000< &*499;;;;;;;s/+C&&C*-C*ctjdddd|j}tj|g}||ddtjtd 5}| |dd d d d n #1swxYwYt|d ksJtjddd|j}tj|g}||dd| |dd d S)zC Test warnings for time column reference position. rrrrtrr:Trz.observatory position is not properly specifiedr4rNr)rr+rpr) rrrrrrrJrBrCrrMrR)rr&rrrZs r(test_io_time_read_fits_location_warningsz5TestFitsTime.test_io_time_read_fits_location_warningss K%)    ,,aS11 TYY{++t <<< \ B    J   TYY{33D  I I I  J J J J J J J J J J J J J J J 1vv{{{{ KZ$di X X X,,aS11 TYY{++t <<<;//EEEEEs+B;;B?B?N)__name__ __module__ __qualname__rr%rBmark parametrizer r r[rir~rrrrrrrrr rrrs<<<999 []UFO<<z3z3=<z3x [VedE]%;<< [/1OPPNNQP=<N< []UFO<<--=<-^ []UFO<<?/?/=<?/B []UFO<<PP=<PB []UFO<<33=<3@ []UFO<<&5&5=<&5P []UFO<<33=<30 []UFO<<(V(V=<(VT []UFO<<6<6<=<6rs: ------ 0/////////((((((((000000777777777777""""""fFfFfFfFfF<fFfFfFfFfFr