o hs< @sddlmZddlZddlZddlmZddlZddlmZm 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%ddZ&ej"#dddgej"#dee!gddZ'ej"#dee!gddZ(e$ddZ)dd Z*e$ej"#dgd!d"d#Z+e$ej"#dej,d$ej"j-d%d&d'd(d)gd*d+Z.e$ej"#dgd!ej"#d,d-d.gd/d0Z/e$ej"#dgd!d1d2Z0e$ej"#dgd!d3d4Z1e$ej"#dgd!d5d6Z2ej"#d7e gd8d9d:e gd9d;egd9d;gej"#dd?d@Z7e$ej"#dgd!dAdBZ8e$dCdDZ9e$dEdFZ:ej"#dGdHdIgdJdKZ;dS)L)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_endcsfdd}|S)Ncs|i|S)z8return the _index_factory created using the args, kwargs)argskwargs_index_factoryrQ/var/www/html/env/lib/python3.10/site-packages/pandas/tests/resample/test_base.py _create_index-sz#create_index.._create_indexr)rrrrr create_index+s rfreq2D1hcCsF|}||}||jd|jd|d}||}t||dS)Nrr)resampleasfreqindexreindextmassert_almost_equal)series_and_framerrobjresult new_indexexpectedrrr test_asfreq4s  r/cCs|}|d}||jd|jddd}||}t|||dd}d|jd<|djdd }||jd|jddd}|j|dd }t ||dS) Nr!rr"r#floatvaluer@) fill_value) r$r%r&r'r(assert_series_equalastypeto_frameilocassert_frame_equal)seriesrserr,r-r.framerrrtest_asfreq_fill_valueAs   r<cCsz|}d}t|jtr t}d}tj||d|d}|d}Wdn1s0wYt ||dS)N+Resampling with a PeriodIndex is deprecatedmatch1min) isinstancer&r FutureWarningr(assert_produces_warningr$r% interpolater8)r;dfwarnmsgr,r.rrrtest_resample_interpolateWs rHcCsHt}d}tjt|d|dWddS1swYdS)NzaOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'RangeIndex'r>YE)rpytestraises TypeErrorr$)xprGrrr%test_raises_on_non_datetimelike_indexes  "rN)MEDhc Csd|}|dkr-t|jtr-d}tjt|d||WddS1s&wYdS|dkr9t|jtr9d}d}t|jtrCt}d}t j ||d ||}Wdn1s\wYt ||}|dkrt g|jdd gdd }t|j||_t j||d d n| }t|j||_t j||d d t |j|j|jj|jjksJdS) NrOaResampling on a TimedeltaIndex requires fixed-duration `freq`, e.g. '24h' or '3D', not r>Mr=ohlcropenhighlowcloser&columnsF check_dtype)rAr&r rJrK ValueErrorr$r rBr(rCgetattrrcopyrr8r4assert_index_equalr) rempty_series_dtiresample_methodr:rGrFrsr,r.rrrtest_resample_empty_seriesps>     rerOzDon't know why this fails)reason)marksrPrQcCs|}ttgt||d|_d}tjt|d ||}Wdn1s)wYt ||}|dkrQt g|jddgdd}tj ||dd n|dd}tj ||dd t |j|j|jj|jjkssJdS) Nr#r=r>rTrrUrZFr\)r`r rlenr&r(rCrBr$r_rr8r4rar)rr9rcr:rGrdr,r.rrrtest_resample_nat_index_seriess    rirccountsizec Cs|}|dkr-t|jtr-d}tjt|d||WddS1s&wYdS|dkr9t|jtr9d}d}t|jtrCt}d}t j ||d ||}Wdn1s\wYt ||}t |j|}t gd||jd} t || dS)NrOrRr>rSr=int64)dtyper&name)rAr&r rJrKr^r$r rBr(rCr_rr rnr4) rrbrcr:rGrFrdr,r&r.rrr test_resample_count_empty_seriess.      roc Cs|}|dkr/t|jtr/d}tjt|d|j|ddWddS1s(wYdS|dkr;t|jtr;d}d}t|jtrEt}d}t j ||d|j|dd}Wdn1s`wYt ||}|dkrt |jgd g}tg|jdd |tjd } t|j|| _n|d kr|} ntgtjd } t|j|| _t |j| j|jj| jjksJt || dS)NrOrRr>F group_keysrSr=rTrUr)r&r[rmrkrm)rAr&r rJrKr^r$r rBr(rCr_r from_productr[rr`npfloat64rr rlrarr)) empty_frame_dtirrcrErGrFrdr,mir.rrrtest_resample_empty_dataframesB    rxcCsg|d<|dkr/t|jtr/d}tjt|d||WddS1s(wYdS|dkr;t|jtr;d}d}t|jtrEt}d}t j ||d ||}Wdn1s^wY| }t |j|}t d|tdgtdd }t ||dS) NarOrRr>rSr=rlrr)rmr&r[)rAr&r rJrKr^r$r rBr(rCrjrrrobjectr8rrvrGrFrdr,r&r.rrr#test_resample_count_empty_dataframe s.     r|cCsg|d<|dkr/t|jtr/d}tjt|d||WddS1s(wYdS|dkr;t|jtr;d}d}d}t|jtrGt}t j ||d ||}Wdn1s^wY| }t |j|}t gd|d}t ||dS) NryrOrRr>rSResampling with a PeriodIndexrl)rmr&)rAr&r rJrKr^r$r rBr(rCrkrr r4r{rrr"test_resample_size_empty_dataframe-s.     r~r&rSryrrn)rnrmzdatetime64[ns]z3ignore:PeriodDtype\[B\] is deprecated:FutureWarningcCsd}t|trtgd|jd}t}d}tg||}tj||d|jddd}Wdn1s2wYz t||WdSt yJYdSw)NBrr=r>dFrp) rAr rnrBr r(rCr$r_r )r&rmrcrFrGrbrdrrrtest_resample_empty_dtypesOs   rcCs&|}|dkr-t|jtr-d}tjt|d||WddS1s&wYdS|dkr9t|jtr9d}d}d}t|jtrEt}t j ||d|j|dd}Wdn1s^wY| dd }t j ||d|| d }Wdn1swYt j ||dd dS) NrOrRr>rSr}FrpcSsdS)Nrrxrrrsz,test_apply_to_empty_series..sumr\) rAr&r rJrKr^r$r rBr(rCapplyr4)rbrr:rGrFrdr,r.rrrtest_apply_to_empty_seriesos0   rc Csd}t|dd}d}d}t|jtrt}tj||d ||}Wdn1s+wYtj||d ||}Wdn1sGwYt ||D]\\}}\} } || ks_Jt || qQdS)NrQstart)r conventionr}r>) r rAr&r rBr(rCgroupbyr$zipr4) r9rtgrGrFgrouped resampledrkrvgkgvrrrtest_resampler_is_iterables      rcs|}dd}d}d}t|jtrt}tj||d ||}||fdd |j }Wdn1scs |S)N)quantilerqrrrs z(test_resample_quantile..) rAr&r rBr(rCr$raggrenamernr4)r9r:rrGrFr,r.rrrtest_resample_quantiles "rhowfirstlastc Cst|r t|djj}ntj}tgd|d|dg|d|dgdtdddd |d }|d }t ||}||d }| |j d t dg}t |||d } d | j_t|| dS)Nrr)rrrg@r2)rybcz 2020-01-01rP)periodsr)r&rmrO)skipnarz 2020-01-31)rr rmna_valuertnanrrr$r_rshapepd to_datetimer&rr(r8) any_real_nullable_dtyperrrrErdmethodr,gbr.rrrtest_first_last_skipnas$     r)s  (               '  .