IR-eX=PddlmZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZmZdd lmZdd lmZdd lmZegd Ze gd gdgdggdZedZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d Z)d!Z*d"Z+d#Z,d$Z-d%Z.d&Z/ej01d'(d)Z2ej01d'(d*Z3d+Z4ej05d,eegd-Z6dS).)datetimeN)assert_allclose assert_equal)units)ColumnTable)assert_quantity_allclose)Time TimeDelta)BoxLeastSquares LombScargle) TimeSeries)Quantity)get_pkg_data_filename)z2016-03-22T12:30:31z2015-01-21T12:30:32z2016-03-22T12:30:40) )r)rrr)abc)nameszdata/sampled.csvcNt}tgdd|d<dS)N)iQiRiSmjdformattime)rr tss Elib/python3.11/site-packages/astropy/timeseries/tests/test_sampled.pytest_empty_initializationr"s, B+++E:::BvJJJct}tjtd5gd|d<ddddS#1swxYwYdS)NzSTimeSeries object is invalid - expected 'time' as the first column but found 'flux'matchrrrflux)rpytestraises ValueErrorrs r!!test_empty_initialization_invalidr,s B     YY6 s?AActt}|d|jusJt|jjtjdS)Nrr)r INPUT_TIMErrisotrs r!test_initialize_only_timer1)sE  $ $ $B f: z/////r#cttgdgdgddg}t|jjtjt|dgdt|dgddS)N rrrrrrdatar)rr/rrr0rs r!test_initialization_with_datar:0ss :::yyy*A#s T T TBz///C***%%%C)))$$$$$r#ctjtd5tgdgdgddgddddS#1swxYwYdS)Nz1Either 'time' or 'time_start' should be specifiedr%r3r5rrr9r)r)r* TypeErrorrr#r!test_initialize_only_datar?7s M   CC YYY/SzBBBBCCCCCCCCCCCCCCCCCCsAA AcZttt}|jgdksJdS)N)rr9)rrrr)rr/ PLAIN_TABLEcolnamesrs r!test_initialization_with_tablerC>s4 + 6 6 6B ;111 1 1 1 1 1 1r#c ttddddddtddgdgd gd d g }t|jjgd dS)Nrr4rsecrr3r5rr time_start time_deltar9r)z2018-07-01T10:10:10.000z2018-07-01T10:10:13.000z2018-07-01T10:10:16.000)rrr rrr0rs r!#test_initialization_with_time_deltarKCs D!QB33Qu---jj))) $Cj    B      r#c tjtd5tt ddddddgdgdgd d g ddddS#1swxYwYdS) Nz-'time' is scalar, so 'time_delta' is requiredr%rErFrr4r3r5rr)rIr9rr)r*r=rrr>r#r!&test_initialization_missing_time_deltarNTs I      aBB77**iii(*                      s-AAAc tjtd5ttt ddddddgdgdgd d g ddddS#1swxYwYdS) Nz+Cannot specify both 'time' and 'time_start'r%rErFrr4r3r5rr)rrIr9r)r)r*r=rr/rr>r#r!/test_initialization_invalid_time_and_time_startrP_s y(V W W W  aBB77**iii(*                       s3AA #A c tjtd5tt ddddddgdgdgd gd d g ddddS#1swxYwYdS) Nz0'time_delta' should be a Quantity or a TimeDeltar%rErFrr4)rrrr3r5rrrHrMr>r#r!&test_initialization_invalid_time_deltarRis L      aBB77 yy**iii(*                       s0AA Act}t|d<t|}t |jhdksJt |jtksJtgdtgddg}t |jddhksJt |jtksJd}tj t|5t|t dddn #1swxYwYtj t|5ttgdtgddg ddddS#1swxYwYdS) Nr)r9>rrrrr3rr<zA'time' has been given both in the table and as a keyword argumentr%)r9rr8) rAcopyr/rsetrBallrr)r*r=)r9ts1ts2MESSAGEs r!%test_initialization_with_time_in_datarZus     DDL $   C s|   7 7 7 7 7 7 7 sx:% & &&&& :::z23- H H HC s|   - - - - sx:% & &&&&RG y 0 0 0//:..../////////////// y 0 0 0XX ***j)A#vWWWWXXXXXXXXXXXXXXXXXXs$#DD  D ,#EE #E ctjtd5ttt dddddS#1swxYwYdS)NzP'n_samples' has been given both and it is not the same length as the input data.r%i)rr9 n_samples)r)r*r=rr/rAr>r#r!test_initialization_n_samplesr]s     FF  EEEEFFFFFFFFFFFFFFFFFFAA  A ctjtd5ttddgddggddg ddddS#1swxYwYdS) Nz5Length of 'time' \(3\) should match data length \(2\)r%r4rrr6rrr8)r)r*r+rr/r>r#r!#test_initialization_length_mismatchr`s R   NN  2q'Aq6):3*MMMMNNNNNNNNNNNNNNNNNNs A  A A ctjtd5ttt ddddddS#1swxYwYdS)Nz='time_delta' should not be specified since 'time' is an arrayr%rrGr)rrJ)r)r*r=rr/r r>r#r!4test_initialization_invalid_both_time_and_time_deltarbs N   KK  y5/I/I/IJJJJ KKKKKKKKKKKKKKKKKKs&AAAcT tgdd}t|}gd|d<|dtjz}t |jtsJt|jj gd d |dtjztd d }t |jtsJt|jj gdd d |dtjzdtjz}t |jtsJt|jj gdd |dtjzdtjz}t |jtsJt|jj gdd |dtjzdtjzdtjz}t |jtsJt|jj gdd |dtjzd}t |jtsJt|j tj gdd |dtjztd dd}t |jtsJt|j tj gdd d |dtjzdd}t |jtsJt|j tj gdd |dtjzdd }t |jtsJt|j tj gd!d |dtjzddd"}t |jtsJt|j tj gd#d dS)$Nrrr unixrr.rrrrrrr(皙 @period)rrg333333333333?gffffff?gư>)rtol皙?)rl epoch_time)g333333g?rng?皙?)roatolrl wrap_phase)rrrrmrprnrsrl epoch_phase)rsgffffffgٿrngg)rlrxrv)rsg?gffffff@rng333333@皙@T)rlnormalize_phase)r?g׿?g?)rlrqrz)gǿg?r}rrr{?rrlrvrz)rr{g?r|g?r}r~rlrxrz)r~gۿgr}gпgԿ)rlrxrvrz)r~g?g?r}g?g?) r rfoldus isinstancerr rrGrto_valueone)timesr tsfs r! test_foldrsd $$$V 4 4 4E    B###BvJ ''qs' # #C ch * ****CHL">">">TJJJJ ''qstC/G/G/G' H HC ch * ****CHL"A"A"ASWXXXX ''qssQSy' 9 9C ch * ****CHL":":":FFFF ''qsac ' : :C ch * ****CHL"D"D"D4PPPP ''qsac cACi' P PC ch * ****CHL"@"@"@tLLLL ''qsD' 9 9C ch ) )))) !%  BBB  ''QSyT#f%=%=%=t   C ch ) )))) !%  KKK    ''qsq$' G GC ch ) )))) !%  >>>  ''qsd' K KC ch ) )))) !%  NNN  ''QSydq$   C ch ) )))) !%  JJJ r#ctgdd}t|}gd|d<tjtjd5|d dddn #1swxYwYtjtjd5|d tjz dddn #1swxYwYtjtjd 5|d tjzd dddn #1swxYwYtjtjd5|d tjzd tjzddddn #1swxYwYtjtjd5|d tjzd dddn #1swxYwYtjtjd5|d tjzd tjzddddn #1swxYwYtjtd5|d tjzdtjzdddn #1swxYwYtjtd5|d tjzdtjzdddn #1swxYwYtjtd5|d tjzdddddn #1swxYwYtjtd5|d tjzddddddS#1swxYwYdS)Nrdrhrr.rir(z,period should be a Quantity in units of timer%rjrkzLepoch_phase should be a Quantity in units of time when normalize_phase=Falseg?rwzSepoch_phase should be a dimensionless Quantity or a float when normalize_phase=TrueTrzKwrap_phase should be a Quantity in units of time when normalize_phase=Falseruz >>>>  #CE : : :F ;  % % % % ,,..C I  Z-B C C J J O O Q QQQQ ;&,,u-- - - - - HC % % ' '''' y(M N N N%%t$$$%%%%%%%%%%%%%%%     CyyCH y(R S S S$$s###$$$$$$$$$$$$$$$$$$s$G??HHI""I&)I&ctjtd5tjt dddddS#1swxYwYdS)NzS``time_column`` should be provided since the default Table readers are being used\.r%csvrr)r*r+rreadCSV_FILEr>r#r!test_read_time_missingr`s     00 ////000000000000000000sAA  A ctjtd5tjt ddddddS#1swxYwYdS)Nz/Time column 'abc' not found in the input data\.r%abcr time_columnrrr>r#r!test_read_time_wrongrks L   CC eEBBBBCCCCCCCCCCCCCCCCCCr^ctjtdd}|jgdksJt |dksJ|djdksJ|dd ksJdS) NDaterr)rABCDEFGrrisorgp@)rrrrBrrsum) timeseriess r! test_readrrsveLLLJ  "M"M"M M M M M z??b f  $ - - - - c?   E ) ) ) ) ) )r#astropy)sourcecddlm}td}tj|5t j|d}dddn #1swxYwY|djdksJ|djdksJ|d j d ksJt|d ksJt|j d ksJdS) Nr) UnitsWarningz/timeseries/kplr010666592-2009131110544_slc.fitsz kepler.fitsrrr0tdbsap_flux electron / si7) astropy.unitsrrr)warnsrrrrunit to_stringrr)rfilenamers r!test_kepler_astropyrzs>******$%VWWH l # #EE_XmDDD EEEEEEEEEEEEEEE f  $ . . . . f  #u , , , , j ! & 0 0 2 2n D D D D z??e # # # # z! " "b ( ( ( ( ( (sA  AActd}tjtd5t j|d}dddn #1swxYwY|djdksJ|djdksJ|d j d ksJt|d ksJt|j d ksJdS) NzJtimeseries/hlsp_tess-data-alerts_tess_phot_00025155310-s01_tess_v1_lc.fitsz Ignoring 815 rows with NaN timesr%z tess.fitsrrr0rrri=Kr) rr)r UserWarningrrrrrrrr)rrs r!test_tess_astropyrs6$TH k)K L L LCC_XkBBB CCCCCCCCCCCCCCC f  $ . . . . f  #u , , , , j ! & 0 0 2 2n D D D D z??e # # # # z! " "b ( ( ( ( ( (sAAActtgdgdgddg}|d}tjt |d5|tgd d d dddn #1swxYwYtjt |d 5| tgd d tgd d gd dgdddn #1swxYwYtjt |d5| ddgdddn #1swxYwYtjt |d5| ddddn #1swxYwYtjt |d5| ddgdddn #1swxYwYtjt |d5| dddddn #1swxYwYd}|}ddg|_tjt |5| dddddS#1swxYwYdS)Nr3r5rrr8zQTimeSeries object is invalid - expected 'time' as the first column but found '{}'rr%)rrr6)namer)rder)indexesrbananazfTimeSeries object is invalid - expected \['time', 'a'\] as the first columns but found \['time', 'b'\])rr/rTr)r*r+r add_columnr add_columns keep_columns remove_columnremove_columns rename_column_required_columns)r rYts_2cols_requireds r!test_required_columnsrs :::yyy*A#s T T TBGGIII  z)<)< = = =CC  VIIIC888BBBCCCCCCCCCCCCCCC z)<)< = = =     IIIC ( ( (&*E*E*E FQRTUPV                   z)<)< = = =++  Sz***+++++++++++++++ z)<)< = = =((  '''((((((((((((((( z)<)< = = =00    &#///000000000000000 z)A)A B B B22  111222222222222222  &  +13-' z 1 1 1--'',,,------------------s$;B++B/2B/$AEEE9*F//F36F3((HH #H *J  JJ)K99K=K=>M!!M%(M%clscttgdgdgddg}||d}t||sJt |jj|jjt|j |d|j J||dd}t|j |d||dd}t |j ddS)Nr3r5rrr8) uncertaintyg?) rr/from_timeseriesrrtjdrrydyr )rr p1p2p3s r!test_periodogramrs :::yyy*A#s T T TB  R % %B b#  BDGRWZ(((r#w 5===  R#  6 6BRUBsG,,,  R#  6 6BBE3r#)7rr) numpy.testingrrrrr astropy.tablerrastropy.tests.helperr astropy.timer r astropy.timeseries.periodogramsr r astropy.timeseries.sampledrrrastropy.utils.datarr/rArr"r,r1r:r?rCrKrNrPrRrZr]r`rbrrrrrrmark remote_datarrr parametrizerr>r#r!rs  77777777''''''''999999((((((((HHHHHHHH111111""""""444444 TWWW X X eZZZIII6oooNNN !3 4 4;;;   000%%%CCC222 "          XXX0 F F FNNNKKKSSSl?H?H?HD$$$>000CCC*** ** ) )+* ) ** ) )+* )------`+ >??  @?   r#