U d{j, @stddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZddlmZe dd d d d gZd dZddZddZddZddZddZddZddZddZejdedd ed ddfed!dedd ed"d!fedgedd dfedgeddd dfgd#d$Zejd%d"ejd&ejd'ejd(ejgd)d*Z dS)+N) assert_equal)units)Time)aggregate_downsamplereduceat) TimeSeries)AstropyUserWarning2016-03-22T12:30:31z2016-03-22T12:30:322016-03-22T12:30:332016-03-22T12:30:342016-03-22T12:30:35c Cs tjtddddddddd g}ttddddddddd gtj}t||ttdtddddtj}t|td d d d gttddddddddd gtj}t|tddd dddd dgtttdtddddtjttdtddddtjdS)Nrg?@g@g@g?g @g@) npaddrZarangesumrZmeanarrayZnanmean)Z add_outputZ sum_outputZ mean_outputZnanmean_outputrg/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/timeseries/tests/test_downsample.py test_reduceats$$ $$ ""rc Cs&tjtddtdW5QRXdS)Nz"time_series should be a TimeSeriesmatch)pytestraises TypeErrorrrrrrtest_timeseries_invalid*sr"c Cs,tjtddttddW5QRXdS)Nz3'time_bin_size' should be a Quantity or a TimeDeltarr) time_bin_size)rr r!rrrrrrtest_time_bin_invalid/s r$c CsBttdddddggdgd}tjtdd t|W5QRXdS) Nrrrrratimedatanamesz_With single 'time_bin_start' either 'n_bins', 'time_bin_size' or time_bin_end' must be providedr)r INPUT_TIMErr r!r)tsrrrtest_binning_arg_invalid8s r,cCslttdddddggdgd}t|dgdgtjd }t|jjd gt|dd gd d }t|jjddgdS)Nrrrrrr%r&r )time_bin_startr#2016-03-22T12:30:31.000r r r- time_bin_end2016-03-22T12:30:33.0002016-03-22T12:30:34.000) rr*rusrr-isotr0)r+Z down_startZdown_endrrrtest_time_bin_conversionBs  r6cCsBttdddddggdgd}t|dd gd }t|jjd d gdS) Nrrrrrr%r&r r )r-r12016-03-22T12:30:35.000)rr*rrr0r5)r+Z down_auto_endrrrtest_time_bin_end_autoVsr8cCsBttdddddggdgd}t|dd gd }t|jjd d gdS) Nrrrrrr%r&r r )r0r.r1)rr*rrr-r5)r+Zdown_start_arrayrrrtest_time_bin_start_arraydsr9cCs~ttdddddggdgd}t|dd}t|jjd d gtt}d |d||dfD]"}t|t|d }t||ksVtqVdS) Nrrrrrr%r&)n_binsr.r1r)r-r:)rr*rrr-r5lenAssertionError)r+Z down_nbinsZn_timesr:rrr test_nbinsts r=c Csttdddddggdgd}ttdddddgtjgdgd}tjdkrXdtj}d}nd tj}|jd dtj}t |||d }t |j dddddg|t |j jtd d dddgt |djjtdddddgt |d||d }t |j dddg|t |j jtd ddgt |djjtdddgt |d||d }t |j ddg|t |j jtd dgt |djjtddgt |d||d }t |j ddg|t |j jtd dgt |djjtddgt |d||d }t |j ddg|t |j jtd dg|djjdksFtt |djtddgt |dddg||d } t | j dddg|t | j jtd ddgt | djjtdddgt |td dgtd dgd} t | j ddgtjt | j jtd dgt | djjtddgtjtdd>t |td dgtddgd} t | djtddgW5QRXdS)Nrrrrrr%r&lg !?r)r#r-r.z2016-03-22T12:30:32.000r1r2r7ctrg@r/zaOverlapping bins should be avoided since they can lead to double-counting of data during binning.rr z2016-03-22T12:30:36.000)rr*r3countsysmaxsizer4r'nsriscloser#rr-r5rr(rrunitnamer<secondrZwarnsr) r+Zts_unitsZ time_bin_incrr-Zdown_1Zdown_2Zdown_3Zdown_4Z down_unitsZdown_uneven_binsZdown_time_arrayZdown_overlap_binsrrrtest_downsamples    "           rGz"time, time_bin_start, time_bin_endrrrcCst|tt|gdgd}t|||d}t|t|ks>tt|ddksRt|j|dksl|dk r|dj stn.|j|dkr|dd|ddkstdS)ztRegression test for #12527: allow downsampling even if all bins fall before or beyond the time span of the data.r%r&r/r#rNr) rrZonesr;rr<allr'minmask)r'r-r0r+Zdownrrrtest_downsample_edge_casess rKdiff_from_base 2dcCsXdtj}ddlm}tdddd}||}||}|||}|||}||ksTtdS) ae A test on time precision limit supported by downsample(). It is related to an implementation details: that time comparison (and sorting indirectly) is done with relative time for computational efficiency. The relative time converted has a slight loss of precision, which worsens as the gap between a time and the base time increases, e.g., when downsampling a timeseries that combines current observation with archival data years back. This test is to document the acceptable precision limit. see also: https://github.com/astropy/astropy/pull/13069#issuecomment-1093069184 ir)_to_relative_longdoublez1980-01-01T12:30:31.000r5Ztdb)formatscaleN)r3rBastropy.timeseries.downsamplerPrr<)rLZprecision_limitrPZt_baset2Zt3Zr_t2Zr_t3rrrtest_time_precision_limit!s    rU)!r@numpyrrZ numpy.testingrZastropyrr3Z astropy.timerrSrrZastropy.timeseries.sampledrZastropy.utils.exceptionsrr*rr"r$r,r6r8r9r=rGmarkZ parametrizerKyearrUrrrrsN          "