zmc
2023-10-12 ed135d79df12a2466b52dae1a82326941211dcc9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
U
­ý°d:"ã@sˆddlmZddlZddlZddlmZmZmZmZddl    m
Z ddl m Z ddlmZddlmZddlmZddlmZdd    lmZed
ed d d ƒed d d ƒfZeded d d ƒed d d ƒfZedddfZej deeeg¡Zejdd„ƒZej dddg¡ej deeg¡dd„ƒƒZ ej deeg¡dd„ƒZ!edd„ƒZ"dd„Z#eej dd d!d"g¡d#d$„ƒƒZ$eej dd d!d"g¡d%d&„ƒƒZ%eej dd d!d"g¡ej d'd(d)g¡d*d+„ƒƒƒZ&eej dd d!d"g¡d,d-„ƒƒZ'eej dd d!d"g¡d.d/„ƒƒZ(eej dd d!d"g¡d0d1„ƒƒZ)ej d2e  *d¡¡ej d3e+e,e-d4g¡d5d6„ƒƒZ.eej dd d!d"g¡d7d8„ƒƒZ/ed9d:„ƒZ0ed;d<„ƒZ1dS)=é)ÚdatetimeN)Ú    DataFrameÚNaTÚ PeriodIndexÚSeries)Ú    DataError)ÚGrouper)Ú
date_range)Ú period_range)Útimedelta_range)Ú_asfreq_compatZdtiiÕéé
ÚpiZtdiz1 dayz10 dayz3_index_factory,_series_name,_index_start,_index_endcs‡fdd„}|S)Ncs
ˆ||ŽS)z8return the _index_factory created using the args, kwargs©)ÚargsÚkwargs©Ú_index_factoryrúVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/resample/test_base.pyÚ _create_index&sz#create_index.<locals>._create_indexr)rrrrrÚ create_index$s rÚfreqZ2DÚ1HcCsF|}| |¡ ¡}||jd|jd|d}| |¡}t ||¡dS)Nréÿÿÿÿ©r)ÚresampleÚasfreqÚindexÚreindexÚtmÚassert_almost_equal)Zseries_and_framerrÚobjÚresultÚ    new_indexÚexpectedrrrÚ test_asfreq-s
 
r&cCs¦|}| d¡ ¡}||jd|jddd}| |¡}t ||¡| d¡ d¡}d|jd<| d¡jdd    }||jd|jddd}|j|dd    }t     ||¡dS)
NrrrrÚfloatÚvaluer g@)Z
fill_value)
rrrrr Úassert_series_equalZastypeZto_frameZilocÚassert_frame_equal)ÚseriesrÚserr#r$r%ÚframerrrÚtest_asfreq_fill_value:s
 
r.cCs,|}t | d¡ ¡ ¡| d¡ ¡¡dS)NZ1T)r r*rrZ interpolate)r-ÚdfrrrÚtest_resample_interpolatePs
 ÿr0c    Cs6tƒ}d}tjt|d| d¡ ¡W5QRXdS)NzaOnly valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'RangeIndex')ÚmatchÚA)rÚpytestZraisesÚ    TypeErrorrZmean)ZxpÚmsgrrrÚ%test_raises_on_non_datetimelike_indexYs
ÿr6ÚMÚDÚHcCs–|dkr2t|jtƒr2|j tjjd|›dd¡|}t|     |¡|ƒƒ}| 
¡}t |j|ƒ|_t   |j|j¡|jj|jjks‚t‚t j||dddS)NÚohlcz    GH13083: z fails for PeriodIndex©ÚreasonF©Z check_dtype)Ú
isinstancerrÚnodeÚ
add_markerr3ÚmarkÚxfailÚgetattrrÚcopyr r Úassert_index_equalrÚAssertionErrorr))rÚempty_series_dtiÚresample_methodÚrequestr,r#r%rrrÚtest_resample_empty_seriesds
ÿÿrJcCsÚ|dkr|j tjjdd¡| ¡}ttgt|ƒ|d|_    | 
|¡}t ||ƒƒ}|dkr’t g|j    dd… ¡ddd    d
gd }t j||d d n |dd… ¡}t j||d d t  |j    |j    ¡|j    j|j    jksÖt‚dS)Nr7zDon't know why this failsr;rr:rÚopenÚhighÚlowÚclose)rÚcolumnsFr=)r?r@r3rArBrDrrÚlenrrrCrr r*r)rErrF)rIrr+rHr,Úrsr#r%rrrÚtest_resample_nat_index_series{s"
 
ÿrRrHÚcountÚsizecCs@t| |¡|ƒƒ}t|j|ƒ}tgd||jd}t ||¡dS)NÚint64)ÚdtyperÚname)rCrr rrrWr r))rrGrHr#rr%rrrÚ test_resample_count_empty_series”s rXcCs||}t|j|dd|ƒƒ}|dkr,| ¡}ntgtjd}t|j|ƒ|_t     |j|j¡|jj
|jj
kslt ‚t  ||¡dS)NF©Z
group_keysrT)rV) rCrrDrÚnprUr rr rErrFr!)Úempty_frame_dtirrHr/r#r%rrrÚtest_resample_empty_dataframe¢s
r\cCsDg|d<| |¡ ¡}t|j|ƒ}tdgid|d}t ||¡dS©NÚarU)rVr)rrSr rrr r*©rr[r#rr%rrrÚ#test_resample_count_empty_dataframe¸s
 r`cCs@g|d<| |¡ ¡}t|j|ƒ}tgd|d}t ||¡dSr])rrTr rrr r)r_rrrÚ"test_resample_size_empty_dataframeÈs
 rarrVzdatetime64[ns]cCs@tg||ƒ}zt|jddd|ƒƒWntk
r:YnXdS)NÚdFrY)rrCrr)rrVrHrGrrrÚtest_resample_empty_dtypesØs
 rccCsB|}|j|dd dd„¡}| |¡ tj¡}tj||dddS)NFrYcSsdS)Nr r©ÚxrrrÚ<lambda>ìóz,test_apply_to_empty_series.<locals>.<lambda>r=)rÚapplyrZÚsumr r))rGrr,r#r%rrrÚtest_apply_to_empty_seriesçsrjc    Cs\d}t|dd}| |¡}| |¡}t||ƒD](\\}}\}}||ksJt‚t ||¡q.dS)Nr9Ústart)rZ
convention)rÚgroupbyrÚziprFr r))    r+rÚtgZgroupedZ    resampledZrkÚrvZgkÚgvrrrÚtest_resampler_is_iterableòs 
 
 rqcsL|}d‰d}| |¡ ˆ¡}| |¡ ‡fdd„¡ |j¡}t ||¡dS)Ngè?r9cs
| ˆ¡S)N)Úquantilerd©Úqrrrfrgz(test_resample_quantile.<locals>.<lambda>)rrrZaggÚrenamerWr r))r+r,rr#r%rrsrÚtest_resample_quantileþs  rv)2rÚnumpyrZr3ZpandasrrrrZpandas._testingZ_testingr Zpandas.core.groupby.groupbyrZpandas.core.groupby.grouperrZpandas.core.indexes.datetimesr    Zpandas.core.indexes.periodr
Zpandas.core.indexes.timedeltasr Zpandas.core.resampler Z
DATE_RANGEZ PERIOD_RANGEZTIMEDELTA_RANGErAZ parametrizeZall_tsZfixturerr&r.r0r6rJrRrXr\r`raZall_timeseries_index_generatorr'ÚintÚobjectrcrjrqrvrrrrÚ<module>s|         þ
ÿ     ÿ