zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
U
®ý°d…ã@s^ddlZddlZddlmZddlZddlmZmZddl    m
Z ddl m Z Gdd„dƒZdS)éN)Ú
is_integer)ÚIndexÚSeries)Ú    Timestampc @sleZdZdd„Zdd„Zdd„Zdd„Zd    d
„Zej     
d e d ƒe d ƒe dƒge d dde d dde dddge   d¡e   d¡e   d¡ge d ƒe d ƒe dƒe jge d dde d dde ddde jge   d¡e   d¡e   d¡e jgg¡dd„ƒZdd„Zdd„Zej     
dddddddgdfd d!d"d#gd$fg¡d%d&„ƒZd'd(„Zej     
d)eed*g¡d+d,„ƒZd-d.„Zd/d0„Zd!S)1ÚTestSeriesQuantilec
Cs"| d¡}|t | ¡d¡ks"t‚| d¡}|t | ¡d¡ksDt‚t|td d¡}|t | ¡d¡ksnt‚|j ¡}| d¡}|t    dƒks’t‚| 
¡}| d¡}|t   d    ¡ks¶t‚tt  d
¡gƒ ¡}|t  d ¡ksÜt‚d }d ddd gddgfD](}tjt|d| |¡W5QRXqôdS)N皙™™™™¹?é
çÍÌÌÌÌÌì?éZ©Údtype皙™™™™É?ú2000-01-10 19:12:00çÐ?z24:00:00ÚNaTrz2percentiles should all be in the interval \[0, 1\]éÿÿÿÿéçà?)Úmatch)ÚquantileÚnpÚ
percentileÚdropnaÚAssertionErrorrÚobjectÚindexÚ    to_seriesrZdiffÚpdZ to_timedeltaZ timedelta64ÚsumÚ    TimedeltaÚpytestZraisesÚ
ValueError)ÚselfÚdatetime_seriesÚqÚdtsZtdsÚresultÚmsgÚinvalid©r)ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_quantile.pyÚ test_quantiles$
 
 
 
 
z TestSeriesQuantile.test_quantilecCsÂddg}| |¡}tt | ¡d¡t | ¡d¡g||jd}t ||¡|j     ¡}d|_| d¡}tt
dƒt
dƒgd    d    gdd}t ||¡| g¡}tg|jt gt d
d d }t ||¡dS) Nrr    rr
©rÚnameZxxx)r r rr r Úfloat64)r-rr ) rrrrrr-ÚtmÚassert_series_equalrrrrÚfloat)r"r#Úqsr&Úexpectedr%r)r)r*Útest_quantile_multi.s6
þú 
 
ý 
 
ÿz&TestSeriesQuantile.test_quantile_multicCsX|jddd}|t | ¡d¡ks&t‚| d¡}|t | ¡d¡ksHt‚||ksTt‚dS)NrZlinear©Ú interpolationr)rrrrr)r"r#r$Zq1r)r)r*Útest_quantile_interpolationKs
 
z.TestSeriesQuantile.test_quantile_interpolationcCsŒtdddgƒjddd}|t t dddg¡d¡ks8t‚t|ƒsDt‚tdddgƒjddd}|t t dddg¡d¡ks|t‚t|ƒsˆt‚dS)    NééérÚlowerr5é2Zhigher)rrrrÚarrayrr)r"r$r)r)r*Ú!test_quantile_interpolation_dtypeWs    z4TestSeriesQuantile.test_quantile_interpolation_dtypecCsÈtddddtjgƒ}| d¡}d}||ks.t‚tgtd}|ttjtjgƒg}|D]p}| d¡}t |¡snt‚| dg¡}t |ttjgdgd¡| d    d
g¡}t |ttjtjgd    d
gd¡qRdS) Nr8rr9r:rg@r ©rr g333333Ó?)    rrÚnanrrrÚisnanr/r0)r"Úsr&r3Ús1ZcasesÚresr)r)r*Útest_quantile_nancs
 
 z$TestSeriesQuantile.test_quantile_nanÚcasez
2011-01-01z
2011-01-02z
2011-01-03z
US/Eastern)Útzz1 daysz2 daysz3 dayscCsXt|dd}| d¡}||dks&t‚| dg¡}t|dgdgdd}t ||¡dS)NZXXX)r-rr8r,)rrrr/r0)r"rFrBrDÚexpr)r)r*Útest_quantile_boxxs $ 
 z$TestSeriesQuantile.test_quantile_boxcCs<t tgdd d¡¡st‚t tgdd d¡¡s8t‚dS)NzM8[ns]r rzm8[ns])rZisnarrr)r"r)r)r*Ú!test_datetime_timedelta_quantiles¤sz4TestSeriesQuantile.test_datetime_timedelta_quantilescCsZttjtjgƒ d¡}|tjks$t‚ttjtjgƒ dg¡}t |ttjgdgd¡dS)Nrr?)rrrrrr/r0)r"rDr)r)r*Útest_quantile_nat©sz$TestSeriesQuantile.test_quantile_natz values, dtyperr8rr9z Sparse[int]gNgð?g@ú Sparse[float]cCsDt||d}| dg¡}tt |¡ƒ dg¡ d¡}t ||¡dS)Nr rrL)rrrZasarrayÚastyper/r0)r"Úvaluesr Úserr&r3r)r)r*Útest_quantile_sparse°s  z'TestSeriesQuantile.test_quantile_sparsecCsîtgdd}| d¡}t |¡s$t‚| dg¡}ttjgdgd}t ||¡tgdd}| d¡}t |¡srt‚| dg¡}ttjgdgd}t ||¡tgdd}| d¡}|tj    ksÀt‚| dg¡}ttj    gdgd}t ||¡dS)Nr.r rr?Úint64zdatetime64[ns])
rrrrArr@r/r0rr)r"rBrDrHr)r)r*Útest_quantile_emptyºs$ 
 
 
 z&TestSeriesQuantile.test_quantile_emptyr ÚInt64cCsdtdddg|d t ddd¡¡}tt ddd¡t ddd¡d}|d    krT| d
¡}t ||¡dS) Nr8rr9r rrrr?rSZFloat64)rrrZarangerMr/r0)r"r r&r3r)r)r*Útest_quantile_dtypesÙs
" 
z'TestSeriesQuantile.test_quantile_dtypesc    Csbttjtjg|d}t d¡| ddg¡}W5QRXttjtjg|ddgd}t ||¡dS)Nr rr©r r)rrÚNAr/Zassert_produces_warningrr0©r"Zany_int_ea_dtyperOr&r3r)r)r*Útest_quantile_all_naás
 z'TestSeriesQuantile.test_quantile_all_nacCsJttjtjdg|d}| ddg¡}tddg|ddgd}t ||¡dS)Nr8r rrrU)rrrVrr/r0rWr)r)r*Útest_quantile_dtype_sizeész+TestSeriesQuantile.test_quantile_dtype_size)Ú__name__Ú
__module__Ú __qualname__r+r4r7r>rEr ÚmarkZ parametrizerrrrrIrJrKrPrRÚintr1rTrXrYr)r)r)r*rs\  ý
 
 
ýü
 
 
üüçþ#
    "þ
 
r)Únumpyrr Zpandas.core.dtypes.commonrZpandasrrrZpandas._testingZ_testingr/Zpandas.core.indexes.datetimesrrr)r)r)r*Ú<module>s