gL i-ddlmZddlZddlZddlmcmZddl Z ddl m Z m Z ddl mZddlmZdZdZdZdZd Zd Zd Zd Zd Zej6j9dgddZej6j9dddgdZej6j9dgddZdZ ejBddZ"y)) timedeltaN) DataFrameSeries)timedelta_rangec&tddgttdg}|jdj}tdtj tj dgt ddd}tj||y) N)minutesdataindex1min0 dayperiodsfreq) rrresampleasfreqnpnanrtmassert_frame_equal)dfresultexpecteds j/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/resample/test_timedelta.pytest_asfreq_bugrsq Ay{Ia4H&I JB [[ ' ' )F #gqv>H&(+c.tjdtjdg}tdgdi|j dj }tddt jdgitdd d  }tj||y) N0s2svalue)r 1sg@g@rr rr ) pd to_timedeltaNaTrrmeanrrrrr)r rrs rtest_resample_with_natr,s{ OOT26640 1E +U 3 < ?)//=9rc tdditjtt dd}|dj d|d<|j d jd }tjtjd d gd d jd}tddgddgd|}|jddgd}|dj d|d<tj||y)N Group_objr5rAr;r'categoryGroup10sc<|jjdS)Nr) value_countsr xs rzDtest_resample_categorical_data_with_timedeltaindex..fsq~~/?/E/Ea/Hrrr0zm8[s]dtyper/ns)rgrjr)axis)rr(r)rErFrLraggTimedeltaIndexrarrayas_unitreindexrr)rrexp_tdirs r2test_resample_categorical_data_with_timedeltaindexr{bs K%R__T%)_SV-W XB[/((4BwK [[  # #$I JF!R @uMUU GCjC:6Hg 6Q?H -44Z@HW&(+rctddd}td|i|}tddd}t||d}tj|j d <|j djd}tj|||dj dj}tj||y) NrDz6 day4Dr/timer'2D)r namer) rrrr(r*ilocrfirstrr@)timesrtimes2express rtest_resample_timedelta_valuesrss GW4 8E FE?% 0B WgD 9F vF 3C&&CHHQK ++d  ! ! #F +C3$ V*  d # ) ) +C3$rzstart, end, freq, resample_freq))8h 21h59min50srk3h)r22h1h5h)527D5006D3D10D)1Drrr)rrrk2h)0hrrkr)r85DDrct|||}ttjt ||}|j |j }t|||}tj|j||jj|jk(sJtj|jdrJy)NrWr')rrSrX) rrrr=r]rr:rr^r risnanr)rSrXr resample_freqidxrArexpected_indexs r!test_resample_timedelta_edge_casers 3T :CryyS"#.A ZZ & * * ,F$-u#NN&,,7 <<   3 3 33 3xx B(( ((r duplicatesTFcttjjdj dt ddd}|r gd |_|jd dddfjd jd }td gdzgdzdgdzgzt d dd }|j |_tj||y)Nr$)'r)sizer!r 3906250nsrRr')r5Br5Cr&3sct|S)N)r]rns rrpzEtest_resample_with_timedelta_yields_no_empty_groups..s SVrir i ) rrrZr[normalrcolumnslocrapplyrr)rrrrs r3test_resample_with_timedelta_yields_no_empty_groupsrs  a ''Z'8D%kJ B) VVDE1H  & &t , 2 23C DF  bSEAI;&D"4@HzzH&(+rr<)rAmsusrsc tjd|d}tdtjtj ddj |itjddd  }|jd jd }tdtjd tjdgitjdddd  j |}tj||y)Nzm8[]r#rrAr;20200101UTC)rtzr'rgGz?z0 days 00:00:00.990000z0 days 00:00:02.990000r$)rrr) rrrrr(r)r=rL date_rangerquantile Timedeltarr)r<rrrrrs r test_resample_quantile_timedeltars HHs4&] #E  "//"))A,S9@@GHmmJe< B[[  ' ' -F  56 56 mmJe$G fUm &(+rc tjtdDcgc]}tjd|dzz c}}t td|}|j dddj }t gd tjtd Dcgc]}tjd|d zz c}d }tj||ycc}wcc}w) Nr0xr6)secondsr'r:right)closedlabel)rr  rJ<r/) r(IndexrFrrrr?rvrr@)irserrrs rtest_resample_closed_rightrs ((b J1BLLq2v6J KC r# &C \\%w\ ? C C EF9>q BAR\\#B, / B H 68,K Cs #C+##C0pyarrowctjtddd}tt j dtj |}|jdj}tj||y)NrDr%)rzduration[ns][pyarrow]rqr'r) r(rrrrr=float64rr+rr@)rrrs rtest_arrow_duration_resamplers` ((?7A6>U VCbii4C@H   t $ ) ) +F68,r)#datetimernumpyrpytestpandas.util._test_decoratorsutil_test_decoratorstdpandasr(rrpandas._testing_testingrpandas.core.indexes.timedeltasrrr,r3rBrGrOrer{rmark parametrizerrrr skip_if_norrrrs )):,, ,2"-- :,"%$%   )  )e}5,6,(!89,:,( -y--r