ehs< ddlmZddlZddlZddlmZddlZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZed ed d d ed d dfZeded d d ed d dfZ edddfZ!ejDjGdee e!gZ$ejJdZ&ejDjGdddgejDjGdee!gdZ'ejDjGdee!gdZ(e$dZ)dZ*e$ejDjGdgddZ+e$ejDjGdejXdejDj[d !d"d#gd$Z.e$ejDjGdgdejDjGd%d&d'gd(Z/e$ejDjGdgdd)Z0e$ejDjGdgdd*Z1e$ejDjGdgdd+Z2ejDjGd,e gd-d./e gd.0egd.0gejDjGd1e3e4e5d2gejDjmd3d4Z7e$ejDjGdgdd5Z8e$d6Z9e$d7Z:ejDjGd8d9d:gd;Z;y)<)datetimeN)is_extension_array_dtype) DataFrame DatetimeIndexIndex MultiIndexNaT PeriodIndexSeriesTimedeltaIndex) DataError)Grouper) date_range) period_range)timedelta_range)_asfreq_compatdtii pitdiz1 dayz10 dayz3_index_factory,_series_name,_index_start,_index_endcfd}|S)Nc|i|S)z8return the _index_factory created using the args, kwargs)argskwargs_index_factorys ]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/resample/test_base.py _create_indexz#create_index.._create_index-st.v..r)rrs` r create_indexr!+s/ r freq2D1hc|}|j|j}||jd|jd|}|j|}t j ||y)Nrr")resampleasfreqindexreindextmassert_almost_equal)series_and_framer"r!objresult new_indexexpecteds r test_asfreqr34s\ C \\$  & & (FSYYq\399R=tDI{{9%H68,r c|}|jdj}||jd|jdd}|j|}t j |||j djd}d|jd<|jdjd }||jd|jdd}|j|d }t j||y) Nr$rr&r'floatvaluer@) fill_value) r(r)r*r+r,assert_series_equalastypeto_frameilocassert_frame_equal)seriesr!serr0r1r2frames rtest_asfreq_fill_valuerAAs C \\$  & & (FSYYq\399R=tDI{{9%H68, JJw  ( ( 1EEJJqM ^^D ! ( (C ( 8FU[[^U[[_4HI}}Y3}7H&(+r cl|}d}t|jtrt}d}t j ||5|j djj}|j dj}dddt jy#1swY xYw)N+Resampling with a PeriodIndex is deprecatedmatch1min) isinstancer*r FutureWarningr,assert_produces_warningr(r) interpolater=)r@dfwarnmsgr0r2s rtest_resample_interpolaterNWs B D"((K( 7C # #D 45V$++-99;;;v&2245&(+55s A B**B3ct}d}tjt|5|j ddddy#1swYyxYw)NzaOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'RangeIndex'rDYE)rpytestraises TypeErrorr()xprMs r%test_raises_on_non_datetimelike_indexrUesD B . y , Ds AA )MEDhc|}|dk(rRt|jtr8d}tjt |5|j |dddy|dk(rt|jtrd}d}t|jtrt}d}tj||5|j |}dddt|}|dk(r_tg|jddjgd }t|j||_tj||d nC|j}t|j||_tj ||d tj"|j|j|jj$|jj$k(sJy#1swYyxYw#1swY(xYw) NrVaResampling on a TimedeltaIndex requires fixed-duration `freq`, e.g. '24h' or '3D', not rDMrCohlcropenhighlowcloser*columnsF check_dtype)rGr*r rQrR ValueErrorr(r rHr,rIgetattrrcopyrr=r9assert_index_equalr") r"empty_series_dtiresample_methodr?rMrLrsr0r2s rtest_resample_empty_seriesrmps C t| 399n= 1 ]]:S 1  LL   *SYY < D#))[) 7C # #D 4 \\$  )WR ) +F&  ciim((*4T ( 48 fhEB88:' 48 vxUC&,,7 <<   3 3 33 37   sG"6G."G+.G8rVzDon't know why this fails)reason)marksrWrXc|j}ttgt|z||_d}t j t|5|j|}dddt|}|dk(rDtg|jddjgd}t j||d n+|ddj}t j||d t j|j|j|jj|jjk(sJy#1swYxYw) Nr'rCrDr\rr]rbFrd)rhr r lenr*r,rIrHr(rgrr=r9rir")r"r>rkr?rMrlr0r2s rtest_resample_nat_index_seriesrrs ++-CSECH,48CI 7C # #M = \\$  )WR ) +F&  ciim((*4T  fhEBr7<<> vxUC&,,7 <<   3 3 33 3  s E  Erkcountsizec~|}|dk(rRt|jtr8d}tjt |5|j |dddy|dk(rt|jtrd}d}t|jtrt}d}tj||5|j |}dddt|}t|j|}tgd||j} tj|| y#1swYyxYw#1swYlxYw)NrVrZrDr[rCint64)dtyper*name)rGr*r rQrRrfr(r rHr,rIrgrr rxr9) r"rjrkr?rMrLrlr0r*r2s r test_resample_count_empty_seriesrys  C t| 399n= 1 ]]:S 1  LL   *SYY < D#))[) 7C # #D 4 \\$  *WR ) +F 399d +Ebu388DH68,)   sD'6D3'D03D<cl|}|dk(rTt|jtr:d}tjt |5|j |ddddy|dk(rt|jtrd}d}t|jtrt}d}tj||5|j |d}dddt|}|dk(rwtj|jgd g}tg|jdd j!|t"j$ } t'|j|| _n1|d k7r|j!} nt)gt"j* } t'|j|| _tj,|j| j|jj.| jj.k(sJtj0|| y#1swYyxYw#1swY_xYw)NrVrZrDF group_keysr[rCr\r]r)r*rcrwrtrw)rGr*r rQrRrfr(r rHr,rIrgr from_productrcrrhnpfloat64rr rvrir"r-) empty_frame_dtir"rkrKrMrLrlr0mir2s rtest_resample_empty_dataframers B t| 288^< 1 ]]:S 1 0 KKK / 0 *RXX{; D"((K( 7C # #D 41 [[%[ 01 )WR ) +F&  $ $bjj2R%S T bhhrl'')2RZZ ($7 F "779"BHH-#BHHd3HN&,,7 <<   3 3 33 368,A 011sH8H)H&)H3cg|d<|dk(rRt|jtr8d}tjt |5|j |dddy|dk(rt|jtrd}d}t|jtrt}d}tj||5|j |}dddj}t|j|}td|tdgt }tj ||y#1swYyxYw#1swYpxYw) NarVrZrDr[rCrvr})rwr*rc)rGr*r rQrRrfr(r rHr,rIrsrrrobjectr=r"rrMrLrlr0r*r2s r#test_resample_count_empty_dataframer s OC t| ?#8#8.I 1 ]]:S 1 +  $ $T * + *_%:%:KH D/''5 7C # #D 4,  % %d +, XXZF ?00$ 7EweUC5PV=WXH&(+' +,,sD.9D:.D7:Eclg|d<|dk(rRt|jtr8d}tjt |5|j |dddy|dk(rt|jtrd}d}d}t|jtrt}tj||5|j |}dddj}t|j|}tgd|}tj||y#1swYyxYw#1swY`xYw) NrrVrZrDr[Resampling with a PeriodIndexrv)rwr*)rGr*r rQrRrfr(r rHr,rIrtrr r9rs r"test_resample_size_empty_dataframer-s  OC t| ?#8#8.I 1 ]]:S 1 +  $ $T * + *_%:%:KH )C D/''5 # #D 4,  % %d +, WWYF ?00$ 7Ebu5H68,' +,,sD9D*D'*D3r*r[rr"rx)rxrwzdatetime64[ns]z3ignore:PeriodDtype\[B\] is deprecated:FutureWarningcBd}t|trtgd|j}t}d}t g||}t j ||5|jdd}ddd t|y#1swYxYw#t$rYywxYw)NBrrCrDdFr{) rGr rxrHr r,rIr(rgr )r*rwrkrLrMrjrls rtest_resample_empty_dtypesrOs D%%BSuzz: 7Cb%/ # #D 4>  & &su & => $O$&>>    sB4BB BBc|}|dk(rRt|jtr8d}tjt |5|j |dddy|dk(rt|jtrd}d}d}t|jtrt}tj||5|j |d}dddjd}tj||5|j |jd }dddtj|d y#1swYyxYw#1swYxYw#1swY:xYw) NrVrZrDr[rFr{cy)Nrr)xs rz,test_apply_to_empty_series..sr sumrd) rGr*r rQrRrfr(r rHr,rIapplyr9)rjr"r?rMrLrlr0r2s rtest_apply_to_empty_seriesros3 C t| #3#9#9>J 1 ]]:S 1 ,  % %d + , *%5%;%;[I )C D"((+6 # #D 42 \\$5\ 12XXk "F # #D 43<<%++E2368?' ,2233s#D=6E ;!E=E EEcd}t|d}d}d}t|jtrt}t j ||5|j|}dddt j ||5|j|}dddtD](\\}}\} } || k(sJt j|| *y#1swYrxYw#1swYMxYw)NrXstart)r" conventionrrD) rrGr*r rHr,rIgroupbyr(zipr9) r>r"tgrMrLgrouped resampledrkrvgkgvs rtest_resampler_is_iterablers D dw /B )C D&,, , # #D 4%..$% # #D 4*OOD) *!)W5'R(2rRxx r2&' %%**s C<CCC#c|}dd}d}d}t|jtrt}t j ||5|j |j}|j |jfdj|j}dddt jy#1swY xYw)Ng?rXrrDc&|jS)N)quantile)rqs rrz(test_resample_quantile..sAJJqMr ) rGr*r rHr,rIr(raggrenamerxr9)r>r?r"rMrLr0r2rs @rtest_resample_quantilers C A D )C D&,, , # #D 4Td#,,Q/<<%))*ABII#((ST68,TTs AB??Chowfirstlastct|r!t|jj}ntj }t gd|d|dg|d|dgdtddd | }|jd }t||}|| }|j|jd tjdgz}t||| } d | j_t!j"|| y)Nr})rrrg@r7)rbcz 2020-01-01rW)periodsr")r*rwrV)skipnarz 2020-01-31)rr rwna_valuernanrrr(rgrshapepd to_datetimer*r"r,r=) any_real_nullable_dtyperrrrKrlmethodr0gbr2s rtest_first_last_skipnars 78 78>>GG66 C3/C3/ qs;% B T B R F 6 "F BHHQK2>>,#?"@@ ABwr3v.HHNN&(+r )rs >   1/43:/%$1!5xa7L M dHT1a$8(4B:OP "E7H=  9/  $.9J;X-/-9J;X,,& , ,!12$43$4N  T!2!2:U!2!VW  44.!12*Wf,=>-?3->!12)-3)-\!12,3,@!12-3-@ BSs+bs#r$5#v7G"HIRS TJ ,!12@3@>''& - - & 12,3,r