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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
U
­ý°dÈOã @s€dZddlZddlmZddlZddlZddlm    Z    ddl
Z ddl
m Z m Z mZmZmZmZmZddlmZddlmZdd„Zd    d
„Zd d „Zd d„Zdd„Zdd„Zdd„Zdd„Zdd„Z dd„Z!dd„Z"dd„Z#dd „Z$d!d"„Z%d#d$„Z&d%d&„Z'ej( )d'd(d)g¡d*d+„ƒZ*d,d-„Z+d.d/„Z,ej( )d0e-e d1d2d3d4œiƒfe.e d1d5d5d5gd6d7d7gd4œiƒfd8d9„e d1d2d3d4œiƒfd:d9„e d1d5d5d5gd6d7d7gd4œiƒfg¡d;d<„ƒZ/ej( )d0e-ed2d3gd5d6gd1d=fe.ed5d5d5gd6d7d7ggd5d6gd1d=fd>d9„ed2d3gd5d6gd1d=fd?d9„ed5d5d5gd6d7d7ggd5d6gd1d=fg¡d@dA„ƒZ0dBdC„Z1dDdE„Z2dFdG„Z3ej( )dHe4e5e6e7e8e9e:e;g¡dIdJ„ƒZ<dS)Kz
test all other .agg behavior
éN)Úpartial)ÚSpecificationError)Ú    DataFrameÚIndexÚ
MultiIndexÚ PeriodIndexÚSeriesÚ
date_rangeÚ period_range)Ú pprint_thingc    Cs–ttj d¡tj d¡dddddgdddddgdœƒ}| d¡}dd    „}tjtd
d | |g¡W5QRXtjtd
d | |¡W5QRXdS) NéÚaÚbÚoneÚtwo)Zdata1Zdata2Úkey1Úkey2rcSs| ¡| ¡S©N)ÚmaxÚmin)Zarr©rú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/aggregate/test_other.pyÚ peak_to_peak)sz5test_agg_partial_failure_raises.<locals>.peak_to_peakzunsupported operand type©Úmatch)    rÚnpÚrandomÚrandnÚgroupbyÚpytestÚraisesÚ    TypeErrorÚagg)ÚdfÚgroupedrrrrÚtest_agg_partial_failure_raisess
 
  üÿ
r%cCsòdddgdddgdddgg}td    d
„|Dƒd d
„|Dƒd d
„|Dƒd œƒ}dd
„|Dƒ}tdd
„|Dƒdd
„|Dƒdd
„|Dƒd œƒ}|d|d ¡|d<| d¡ tj¡}|d|d ¡|d<| d¡ tj¡}t|ƒt|ƒksît‚dS)Néú
2012-01-01çð?éz
2012-01-02g@ég@cSsg|] }|d‘qS©rr©Ú.0ÚxrrrÚ
<listcomp>8sz,test_agg_datetimes_mixed.<locals>.<listcomp>cSsg|] }|d‘qS©r&rr,rrrr/9scSsg|] }|d‘qS©r)rr,rrrr/:s)ÚkeyÚdateÚvaluecSs<g|]4}|d|dr,tj |dd¡ ¡nd|dg‘qS)rr&z%Y-%m-%dNr))ÚdtÚdatetimeÚstrptimer3)r-Úrowrrrr/>s
ü ýcSsg|] }|d‘qSr+rr,rrrr/IscSsg|] }|d‘qSr0rr,rrrr/JscSsg|] }|d‘qSr1rr,rrrr/Ksr4Úweightsr3)rÚsumrÚ    aggregaterÚlenÚAssertionError)ÚdataZdf1Zdf2Zgb1Zgb2rrrÚtest_agg_datetimes_mixed3s*   ýÿú       ýÿr?cCs¬tdddd}ttj dd¡|d}|jdd ¡}t|jt    ƒsDt
‚td    d
dd }t tj  t |ƒ¡|d}t tj  t |ƒ¡|d}t ||d œ¡}| |jj¡}t|ƒdS) Nz2012-1-1ÚMr*)ÚfreqÚperiodsr)©Úindexr)Úlevelz1999-01r )ÚstartrBrA)Ús1Ús2)r
rrrrrr:Ú
isinstancerDrr=rZrandr<Ú    from_dictÚmonthÚlist)Zprngr#ÚrsrDrGrHr$rrrÚtest_agg_period_indexXsrNc    CsÎtddddddddgtddddd    œƒ}d|jd
d d d gdf<|jd
dddg d¡}| d¡}t | ¡|¡t | d¡|¡t | ddi¡|¡t     |j
 ¡|d¡t     |j
 d¡|d¡|jd
dddg d¡}| d¡}t |  ¡|¡t | d¡|¡t | ddi¡|¡t     |j
  ¡|d¡t     |j
 d¡|d¡t d d d d gt tdƒdddd}t     |j
 t¡|¡t     |j
 ¡|¡t d
d d d gt tdƒdddd}t     |j
 ¡|¡dS)NÚAÚBÚCÚDz1/1/2011éÚH)rBrA)ÚclassÚtimerr&r)r rVr*éérUÚfirstéÚlastÚABCD©Úname©rDr^)rr    ÚlocZ    set_indexrÚtmÚassert_frame_equalrYr"Úassert_series_equalrVr[rrrLr<ÚsizeÚcount)r#Úexpr$rrrÚ*test_agg_dict_parameter_cast_result_dtypesgs2 þÿ
 
""rgcCs^dd„tdƒDƒ}tdƒ}t||dœƒ}| d¡d t¡}| d¡d ¡}t ||¡dS)NcSsg|]}t d|dd¡‘qS)ißr&)r5r6r,rrrr/sz0test_agg_cast_results_dtypes.<locals>.<listcomp>é Z aaabbbbbbccd)ÚXÚYrirj)    ÚrangerLrrr"r<rerarc)ÚuÚvr#ÚresultÚexpectedrrrÚtest_agg_cast_results_dtypess rpcCsÎtdddddgdddddgdddddgdœƒ}tdddddgiddddgd    }d
|j_| d
¡dg ¡}t ||¡tddddgddddgd œddddgd    }d
|j_| d
¡dd g ¡}t ||¡dS) Nr&r)r*rWr )r rÚcr g@rCr)r rqrq)rrDr^rÚmeanrarb)r#rornrrrÚtest_aggregate_float64_no_int64™s.  *rsc    
Cstddddddddgddddddddgtj d¡dt d¡dœƒ}| dd    g¡}|d
 ¡}|d
 ¡}|d  ¡}|d  ¡}|d  d d g¡}t    j
||gdd}d d g|_ t j ||dd| tjtjg¡}t    j
||||gdd}t d
d gd d gg¡|_ t j ||dd|d d
g tjtjg¡}t    j
||||gdd}t d d
gd d gg¡|_ t j ||dd| d d dœ¡}t    j
||gdd}t j ||dd| d d gd d gdœ¡}t    j
||||gdd}t d
d gd d gg¡|_ d}tjt|d"|d d
g tjtjdœ¡W5QRXdS)NÚfooÚbarrrrSr(©rOrPrQrRrOrPrQrRr:rrr&)ZaxisT)Z
check_like©rQrRz&Column\(s\) \['r', 'r2'\] do not existr)ÚrÚr2)rrrrÚarangerrrr:r"ÚpdÚconcatÚcolumnsrarbrZ from_productrr ÚKeyError)    r#r$Zc_meanZc_sumZd_meanZd_sumrnroÚmsgrrrÚtest_aggregate_api_consistencyªsBüÿ        
r€c    CsØtdddddgtdƒtdƒdœƒ}d}tjt|d,| d¡ dd    d
gid d d gidœ¡W5QRXd}tjt|d"| d¡ddg dd
i¡W5QRXd}tjt|d| d¡j dd i¡W5QRXdS)Nr&r)r ©rOrPrQúnested renamer is not supportedrrOrtr:rrurer)rPrQz!Column\(s\) \['ma'\] do not existrPrQÚma)    rrkrr rrr"r~rP)r#rrrrÚ"test_agg_dict_renaming_deprecationÙs"
ÿ&r„c
Cs¸tddddddddgddddddddgtj d¡dt d¡dœƒ}| dd    g¡}d
}tjt|d |d      d ddgi¡W5QRXtjt|d |d      dddœ¡W5QRXdS)NrtrurrrSr(rvrOrPr‚rrRrQr:Ústdrw)
rrrrrzrrr rr"©r#ÚgrrrrÚtest_agg_compatìsüÿ     rˆc
Cs2tddddddddgddddddddgtj d¡dt d¡dœƒ}| dd    g¡}d
}tjt|d &|     d d dgidd dgidœ¡W5QRXtjt|d &| 
dd dgidd dgidœ¡W5QRXtjt|d |d 
tj tj dœ¡W5QRXtjt|d |d 
tj tj dœ¡W5QRXdS)NrtrurrrSr(rvrOrPr‚rrQrrr:rR)Zr1ryÚrar…Úrbrw)Úresult1Úresult2)rRrŒ) rrrrrzrrr rr;r"r:rrr†rrrÚtest_agg_nested_dictss"üÿ    **"rc    CsJttjjdddƒ}dd„}tjtdd| d¡ |¡W5QRXdS)    Né
)érŽ)rdcSs tdƒt| ¡ƒtdƒ‚dS)Nz(----------------------------------------Útest)r Z    to_stringr!)r#rrrÚraiseException!s z=test_agg_item_by_item_raise_typeerror.<locals>.raiseExceptionrrr)    rrrÚrandintrr r!rr")r#r‘rrrÚ%test_agg_item_by_item_raise_typeerrorsr“cCsBt ¡}| dd„dd„g¡}| tj¡}| ¡}t ||¡dS)NcSs|jSr)Úyear©r.rrrÚ<lambda>,óz*test_series_agg_multikey.<locals>.<lambda>cSs|jSr)rKr•rrrr–,r—)raZmakeTimeSeriesrr"rr:rc)Útsr$rnrorrrÚtest_series_agg_multikey*s
 r™cCs´tdddddddddddg dddddddddddg dddddddddddg tj d¡tj d¡tj d¡dœƒ}d    d
„}| d d g¡ |¡}| d d g¡ d d„¡}t ||¡dS)NrtrurrZdullZshinyé )rOrPrQrRÚEÚFcSst|jjƒdkst‚dS)Nrrt)r<ÚvaluesÚbaser=r•rrrÚbadcsz.test_series_agg_multi_pure_python.<locals>.badrOrPcSsdS)Nrtrr•rrrr–hr—z3test_series_agg_multi_pure_python.<locals>.<lambda>)rrrrrr"rarb)r>rŸrnrorrrÚ!test_series_agg_multi_pure_python3s\õõõ
 
 
Öÿ/r c Cs–dd„}tddddgddd    d
gt d dd¡t d dd¡t d dd ¡t d dd ¡gd œƒ}| d¡}| |g¡}|jjd|_| |¡}t ||¡dS)NcSstj| ¡ddS)Nr&)Úq)rZ
percentileZdropna)r rrrÚP1osz test_agg_consistency.<locals>.P1r&r)r*rWrŽéééiÝrš)Zcol1Zcol2r3r3r)    rr5r3rr"r}Úlevelsrarb)r¢r#r‡rornrrrÚtest_agg_consistencyls 
 
    üýÿ
 
r§cCs†tddgddgdœƒ tj¡}Gdd„dƒ}ttjdd    „d
d    „ttƒ|ƒg}| d ¡ t¡}|D] }| d ¡ |¡}t     ||¡q`dS) Nr&r)r*rW)rtruc@seZdZdd„ZdS)z$test_agg_callables.<locals>.fn_classcSst|ƒSr©r:)Úselfr.rrrÚ__call__sz-test_agg_callables.<locals>.fn_class.__call__N)Ú__name__Ú
__module__Ú __qualname__rªrrrrÚfn_classŒsr®cSst|ƒSrr¨r•rrrr–“r—z$test_agg_callables.<locals>.<lambda>cSs| ¡Srr¨r•rrrr–”r—rt)
rZastyperÚint64r:rrr"rarb)r#r®Zequiv_callablesroZecallrnrrrÚtest_agg_callablesˆsú    r°c    CsÎtdt dddg¡gdt dddg¡gdt dddg¡ggd    d
gd }| d    ¡}t dd d g¡gt dddg¡gg}tddgd    d}d
g}t|||d}|jdd}t ||¡|jddd}t ||¡dS)Nr&rŽréé(é2é<r)ÚcategoryZ    arraydata)r}éFéZr]©rDr}F©Z numeric_onlyr:)    rrÚarrayrrr:rarbr")r#ÚgbZ expected_dataZexpected_indexZexpected_columnroZaltrnrrrÚtest_agg_over_numpy_arraysŸs ýú
$  r¼Ú    as_periodTFcCs,tdddd}|r"| d¡ d¡}tddddg|dœƒ}| d    ¡}|d
 d d „¡}t|ddd …d
d}d    |j_t     
||¡|d
 dd „¡}tddgd
d}d    |j_t     
||¡|d
 dd „¡}tt j ddt j ddgd
d}d    |j_|rtt j  d¡t j  d¡gd
d}d    |j_t     
||¡dS)Nr'rWÚUTC)rBÚtzrRrr&©r rr rcSs
|jdS©Nr©Úilocr•rrrr–Ær—z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>r)r]cSs |jdjSrÁ)rÃr”r•rrrr–Ìr—iÜcSs|jd|jdS)NéÿÿÿÿrrÂr•rrrr–Ñr—)Údays)r    Z tz_localizeZ    to_periodrrr"rrDr^rarcr{Z    TimedeltaÚoffsetsZDay)r½Zdtir#r»rnrorrrÚ$test_agg_tzaware_non_datetime_resultºs(
    rÇcs¨tjddd‰td‡fdd„tdƒDƒdœƒ}| d    ¡d
 tj¡jd }| d    ¡d
 d d „¡jd }| d    ¡d
 ¡jd }|ˆksŒt    ‚|ˆks˜t    ‚|ˆks¤t    ‚dd„tddƒDƒ}td    d
gd|dœƒ}| d¡}|djd ‰ˆ| 
d ¡djd kst    ‚ˆ|  d¡djd ks"t    ‚ˆ|  ¡djd ks>t    ‚ˆ|  dd „¡jdks\t    ‚|djd‰ˆ| ¡djd ks†t    ‚ˆ|  dd „¡jdks¤t    ‚dS)Nz2016-01-01 12:00:00ú
US/Pacific©r¿r&csg|]}ˆtj|d‘qS))Úminutes)r5Ú    timedelta)r-Únn©r˜rrr/Ýsz0test_agg_timezone_round_trip.<locals>.<listcomp>rŽrÀr rrcSs
t |¡Sr)rrr•rrrr–àr—z.test_agg_timezone_round_trip.<locals>.<lambda>cSs$g|]}tjd|d›ddd‘qS)z    2016-01-0Údz     12:00:00rÈrÉ)r{Ú    Timestamp)r-Úirrrr/çsr r)©rOrPrOrPcSs
|jdSrÁrÂr•rrrr–ór—)rr&cSs
|jdS)NrÄrÂr•rrrr–ùr—)r{rÏrrkrr"rrrÃr=ZnthÚheadrYÚapplyr[)r#r‹rŒZresult3Údatesr$rrÍrÚtest_agg_timezone_round_tripÚs*    ÿ
rÕcCs¢tddgddgddggtd}|d}td    d
d gtjd}tdd d dgi|td}d|j_| d¡jdd}t     
||¡| d¡jdd}|g}t     
||¡dS)Nr&r)r*rWr rX)Údtypelÿÿÿÿllllll)rDrÖrFr¹T) rÚobjectrrZuint64rDr^rr:rarb)r#rDrornrŒZ    expected2rrrÚtest_sum_uint64_overflowüs"ÿ ý rØzstructure, expectedrQ)r&r&r&)r*rWrW))r&r&)r*rWr&r*rWcCst|ƒSr©Útupler•rrrr–r—r–cCst|ƒSr©rLr•rrrr–r—c    Csdtddddddgddddddgddddddgdœƒ}| ddg¡ |¡}ddg|j_t ||¡dS)Nr&r*rWrrOrP)rrr;rDÚnamesrarb©Z    structureror#rnrrrÚtest_agg_structs_dataframes .ÿ rÞr_cCst|ƒSrrÙr•rrrr–3r—cCst|ƒSrrÛr•rrrr–4r—c    Cs`tddddddgddddddgddddddgdœƒ}| d¡d |¡}d|j_t ||¡dS)Nr&r*rWrrOrQ)rrr;rDr^rarcrÝrrrÚtest_agg_structs_series.s .ÿrßcCsŒdddg}ttjdddg|ddddgdœƒ}|jd    |d
j tj¡}tddd gtj    dddg|d    d d d}|r|||d k}t
  ||¡dS)Nr rrq)Ú
categoriesr&r)r*rÑrO)Úobservedr)ràr^rPr_) rr{Z CategoricalrrPr"rZnansumrZCategoricalIndexrarc)ráràr#rnrorrrÚtest_agg_category_nansumBs
ÿý râcCs†tdd„tdƒDƒdd„tdƒDƒdœƒ}|jdddd}| dd    d
„i¡}td d„tdƒDƒd d„tdƒDƒdœƒ}t ||¡dS) NcSsg|] }t|ƒ‘qSr©Ústrr,rrrr/Tsz+test_agg_list_like_func.<locals>.<listcomp>r*rÑrOF)Zas_indexÚsortrPcSst|ƒSrrÛr•rrrr–Vr—z)test_agg_list_like_func.<locals>.<lambda>cSsg|] }t|ƒ‘qSrrãr,rrrr/XscSsg|]}t|ƒg‘qSrrãr,rrrr/Xs)rrkrr"rarb)r#r$rnrorrrÚtest_agg_list_like_funcRs*$ÿræcCsxtddgtjdddtjdddgdœƒ}| d¡ dd    d
„i¡}ttjdddgtdgdd dgd }t ||¡dS) Nr&z
2018-01-01r¾rÉz
2018-01-02)Útagr3rçr3cSs
| d¡S)Nr&)rÒ)Úerrrr–hr—z/test_agg_lambda_with_timezone.<locals>.<lambda>r]r¸)rr{rÏrr"rrarb)r#rnrorrrÚtest_agg_lambda_with_timezone]s  þþÿ     ýréÚerr_clsc    sšddlm}m}m}|ƒdd…}tdddddgdddddg||ƒdœƒ}t||d|dgƒƒ}‡fdd„}|d     |d
¡ |¡}tj    ||d d dS) Nr)Ú DecimalArrayÚ    make_dataÚ
to_decimalr r&)Úid1Zid2Údecimalsr*cst|ƒdkrˆ‚|jdSrÁ)r<rÃr•©rêrrÚ
weird_funcs z1test_groupby_agg_err_catching.<locals>.weird_funcrïrîF)Z check_names)
Z$pandas.tests.extension.decimal.arrayrërìrírrrr"rarc)    rêrërìrír>r#rorñrnrrðrÚtest_groupby_agg_err_catchingqs"ÿ rò)=Ú__doc__r6r5Ú    functoolsrÚnumpyrrZ pandas.errorsrZpandasr{rrrrrr    r
Zpandas._testingZ_testingraZpandas.io.formats.printingr r%r?rNrgrprsr€r„rˆrr“r™r r§r°r¼ÚmarkZ parametrizerÇrÕrØrÚrLrÞrßrâræréÚNotImplementedErrorÚ RuntimeErrorr~Ú
IndexErrorÚOSErrorÚ
ValueErrorÚArithmeticErrorÚAttributeErrorròrrrrÚ<module>s†  $      %& /     9
" þþùþ
 
$(üþ    
 øþ