bMhs< 6ddlmZddlZddlZddlmZddlZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZed ed d d ed d dfZeded d d ed d dfZ edddfZ!ej"#dee e!gZ$ej%dZ&ej"#dddgej"#dee!gdZ'ej"#dee!gdZ(e$dZ)dZ*e$ej"#dgddZ+e$ej"#dej,dej"-d !d"d#gd$Z.e$ej"#dgdej"#d%d&d'gd(Z/e$ej"#dgdd)Z0e$ej"#dgdd*Z1e$ej"#dgdd+Z2ej"#d,e gd-d./e gd.0egd.0gej"#d1e3e4e5d2gej"6d3d4Z7e$ej"#dgdd5Z8e$d6Z9e$d7Z:ej"#d8d9d:gd;Z;dS)<)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/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/resample/test_base.py _create_indexz#create_index.._create_index-s~t.v...r)rrs` r create_indexr!+s$///// r freq2D1hc|}||}||jd|jd|}||}t j||dS)Nrr")resampleasfreqindexreindextmassert_almost_equal)series_and_framer"r!objresult new_indexexpecteds r test_asfreqr34sr C \\$   & & ( (F SYq\39R=tDDDI{{9%%H68,,,,,r cJ|}|d}||jd|jdd}||}t j|||dd}d|jd<|dd }||jd|jdd}||d }t j ||dS) Nr$rr&r'floatvaluer@) fill_value) r(r)r*r+r,assert_series_equalastypeto_frameilocassert_frame_equal)seriesr!serr0r1r2frames rtest_asfreq_fill_valuerAAs  C \\$   & & ( (F SYq\39R=tDDDI{{9%%H68,,, JJw   ( ( 1 1EEJqM ^^D ! ! ( (C ( 8 8F U[^U[_4HHHI}}Y3}77H&(+++++r c|}d}t|jtrt}d}t j||5|d}|d}dddn #1swxYwYt j ||dS)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 4 4 455V$$++--99;;;;v&&2244555555555555555&(+++++sA!B++B/2B/ct}d}tjt|5|dddddS#1swxYwYdS)NzaOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'RangeIndex'rDYE)rpytestraises TypeErrorr()xprMs r%test_raises_on_non_datetimelike_indexrUes B . y , , , DsAAA)MEDhc|}|dkrft|jtrLd}tjt |5||dddn #1swxYwYdS|dkrt|jtrd}d}t|jtrt}d}tj ||5||}dddn #1swxYwYt||}|dkretg|jdd gd }t|j||_tj||d nE| }t|j||_tj||d tj|j|j|jj|jjksJdS) 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_seriesrmpsf C t|| 39n==| 1 ]:S 1 1 1   LL                   *SY << D#)[)) 7C #D 4 4 4  \\$                 )WR ) ) + +F&   cim((**4T4T4T   ( 488 fhEBBBBB88::' 488 vxUCCCC&,777 <  3 3 3 3 3 3 3$A""A&)A& C..C25C2rVzDon't know why this fails)reason)marksrWrXc|}ttgt|z||_d}t jt|5||}dddn #1swxYwYt||}|dkrKtg|jddgd}t j ||d n3|dd}t j ||d t j |j|j|jj|jjksJdS) Nr'rCrDr\rr]rbFrd)rhr r lenr*r,rIrHr(rgrr=r9rir")r"r>rkr?rMrlr0r2s rtest_resample_nat_index_seriesrss ++--CSECHH,4888CI 7C #M = = =  \\$                 )WR ) ) + +F&   cim((**4T4T4T    fhEBBBBBrr7<<>> vxUCCCC&,777 <  3 3 3 3 3 3 3sBBBrkcountsizec|}|dkrft|jtrLd}tjt |5||dddn #1swxYwYdS|dkrt|jtrd}d}t|jtrt}d}tj ||5||}dddn #1swxYwYt||}t|j|}tgd||j} tj|| dS)NrVrZrDr[rCint64)dtyper*name)rGr*r rQrRrfr(r rHr,rIrgrr ryr9) r"rjrkr?rMrLrlr0r*r2s r test_resample_count_empty_seriesrzs C t|| 39n==| 1 ]:S 1 1 1   LL                   *SY << D#)[)) 7C #D 4 4 4  \\$                 *WR ) ) + +F 39d + +Ebu38DDDH68,,,,,rncb|}|dkrht|jtrNd}tjt |5||ddddn #1swxYwYdS|dkrt|jtrd}d}t|jtrt}d}tj ||5||d}dddn #1swxYwYt||}|dkrttj |jgd g}tg|jdd |t"j } t'|j|| _n6|d kr|} nt)gt"j } t'|j|| _tj|j| j|jj| jjksJtj|| dS)NrVrZrDF group_keysr[rCr\r]r)r*rcrxrurx)rGr*r rQrRrfr(r rHr,rIrgr from_productrcrrhnpfloat64rr rwrir"r-) empty_frame_dtir"rkrKrMrLrlr0mir2s rtest_resample_empty_dataframers B t|| 28^<<| 1 ]:S 1 1 1 0 0 KKK / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 *RX{;; D"(K(( 7C #D 4 4 411 [[%[ 0 0111111111111111 )WR ) ) + +F&   $bj2R2R2R%S T T bhrrl''))2RZ   ($77 F " "7799"BH---#BHd33HN&,777 <  3 3 3 3 368,,,,,s$A$$A(+A(C22C69C6cg|d<|dkrft|jtrLd}tjt |5||dddn #1swxYwYdS|dkrt|jtrd}d}t|jtrt}d}tj ||5||}dddn #1swxYwY| }t|j|}td|tdgt }tj||dS) NarVrZrDr[rCrwr~)rxr*rc)rGr*r rQrRrfr(r rHr,rIrtrrrobjectr=r"rrMrLrlr0r*r2s r#test_resample_count_empty_dataframer s OC t|| ?#8.II| 1 ]:S 1 1 1 + +  $ $T * * * + + + + + + + + + + + + + + + *_%:KHH D/'55 7C #D 4 4 4,,  % %d + +,,,,,,,,,,,,,,, XXZZF ?0$ 7 7EweUC5PV=W=W=WXXXH&(+++++$A%%A),A)C11C58C5cg|d<|dkrft|jtrLd}tjt |5||dddn #1swxYwYdS|dkrt|jtrd}d}d}t|jtrt}tj ||5||}dddn #1swxYwY| }t|j|}tgd|}tj||dS) NrrVrZrDr[Resampling with a PeriodIndexrw)rxr*)rGr*r rQrRrfr(r rHr,rIrurr r9rs r"test_resample_size_empty_dataframer-s OC t|| ?#8.II| 1 ]:S 1 1 1 + +  $ $T * * * + + + + + + + + + + + + + + + *_%:KHH )C D/'55 #D 4 4 4,,  % %d + +,,,,,,,,,,,,,,, WWYYF ?0$ 7 7Ebu555H68,,,,,rr*r[rr"ry)ryrxzdatetime64[ns]z3ignore:PeriodDtype\[B\] is deprecated:FutureWarningctd}t|trtgd|j}t}d}t g||}t j||5|dd}dddn #1swxYwY t||dS#t$rYdSwxYw)NBrrCrDdFr|) rGr ryrHr r,rIr(rgr )r*rxrkrLrMrjrls rtest_resample_empty_dtypesrOs D%%%BSuz::: 7Cb%// #D 4 4 4>>  & &su & = =>>>>>>>>>>>>>>> $O$$&&&&&      s$BB BB)) B76B7c|}|dkrft|jtrLd}tjt |5||dddn #1swxYwYdS|dkrt|jtrd}d}d}t|jtrt}tj ||5||d}dddn #1swxYwY| d}tj ||5|| d }dddn #1swxYwYtj ||d dS) NrVrZrDr[rFr|cdS)Nrr)xs rz,test_apply_to_empty_series..sr sumrd) rGr*r rQrRrfr(r rHr,rIapplyr9)rjr"r?rMrLrlr0r2s rtest_apply_to_empty_seriesros7 C t|| #3#9>JJ| 1 ]:S 1 1 1 , ,  % %d + + + , , , , , , , , , , , , , , , *%5%;[II )C D"(+66 #D 4 4 422 \\$5\ 1 1222222222222222XXkk " "F #D 4 4 433<<%%++E2233333333333333368??????s6A""A&)A& C00C47C4')EE #E cd}t|d}d}d}t|jtrt}t j||5||}dddn #1swxYwYt j||5||}dddn #1swxYwYt||D](\\}}\} } || ksJt j || )dS)NrXstart)r" conventionrrD) rrGr*r rHr,rIgroupbyr(zipr9) r>r"tgrMrLgrouped resampledrkrvgkgvs rtest_resampler_is_iterablers D dw / / /B )C D&, ,, #D 4 4 4%%..$$%%%%%%%%%%%%%%% #D 4 4 4**OOD)) ***************!)W55''R(2rRxxxx r2&&&&''s$A11A58A5B44B8;B8c|}dd}d}d}t|jtrt}t j||5||}||fd |j }dddn #1swxYwYt j ||dS)Ng?rXrrDc.|S)N)quantile)rqs rrz(test_resample_quantile..sAJJqMMr ) rGr*r rHr,rIr(raggrenameryr9)r>r?r"rMrLr0r2rs @rtest_resample_quantilers& C A D )C D&, ,, #D 4 4 4TTd##,,Q//<<%%))*A*A*A*ABBII#(SSTTTTTTTTTTTTTTT68,,,,,sA,B;;B?B?howfirstlastct|rt|jj}n tj}t gd|d|dg|d|dgdtddd | }|d }t||}|| }| |j d tj dgz}t||| } d | j_t!j|| dS)Nr~)rrrg@r7)rbcz 2020-01-01rW)periodsr")r*rxrV)skipnarz 2020-01-31)rr rxna_valuernanrrr(rgrshapepd to_datetimer*r"r,r=) any_real_nullable_dtyperrrrKrlmethodr0gbr2s rtest_first_last_skipnars% 788 7888>G6 C3/C3/  qs;;;%   B T  B R  F V6 " " "F BHQK2>,#?#?"@@ A ABwr3v...HHN&(+++++r )rs >>>>>>                    111111//////444444333333:::::://////%$1!5!5xxa7L7L M dHHT1a$8$8((4B:O:OP "E7H=  9/   $..9J;X--/.-9J;X,,,& , , ,!1!1!122$4$432$4N  T!2!2:U!2!V!VWWW  444.!1!1!122*Wf,=>>--?>32->!1!1!122)-)-32)-\!1!1!122,,32,@!1!1!122--32-@  BSs+++ bs###r$$$5#v7G"HIIRSS  TSJI ,!1!1!122@@32@>'''& - - - & 122,,32,,,r