fume-manage-python.git

U  
ý°d·¦ã@s~ddlmZmZddlmZddlZddlZddlZddlm    Z    m
Z
mZmZm Z ddlmZddlmZddZdd    Zd
dZejjde    d d d ddgdddddgdd ddgfe    ddd d gdd dd gddd gfe    dd gid gfe    d d d ddd d dgeddd dgfe    d ddd ddgddddddgdd ddgfe    edddddddd dddg
ddddddddddg
ddddgfe    d ddgdddggdddgdd dgfgddd d!d"d#d$gd%d&d'Zd(d)Zd*d+Zej d,d-d.d/d.d0d.d1d.g¡d2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!d<d=Z"d>d?Z#d@dAZ$dBdCZ%dDdEZ&dFdGZ'dHdIZ(ej dJdKdLg¡dMdNZ)dOdPZ*dQdRZ+dSdTZ,dUdVZ-dWdXZ.ej dYdKdLg¡dZd[Z/d\d]Z0d^d_Z1d`daZ2dbdcZ3dddeZ4dfdgZ5dhdiZ6djdkZ7dldmZ8dndoZ9dpdqZ:drdsZ;ej dte <¡e <¡ ¡e <¡ =¡g¡dudvZ>dwdxZ?ej dydLdKg¡dzd{Z@d|d}ZAej d~e Bed¡e Cdd¡ejDdddde Edddg¡g¡ddZFej ddd.dd gddggfdd.dd hddhgfdd.ddgfdd.dd ddddgfdd.ddid d igddid diggfg¡ddZGddZHej ddd.dd.g¡ddZIddZJej ddd.dd.g¡ej dJdKdLg¡ddZKd d¡ZLd¢d£ZMd¤d¥ZNd¦d§ZOd¨d©ZPej dªdKdLg¡d«d¬ZQej ddLe    d d d gddd gge
dddgeRd®dgdKed d gejEd¯d°gddgd±d²gg¡d³d´ZSdµd¶ZTd·d¸ZUd¹dºZVej dªdKdLg¡d»d¼ZWd½d¾ZXej d¿d ddgdÀdÁdÂgg¡dÃdÄZYej dÅd ddgdÆdÇdÈggd ddgdÆdÈdÇggd ddgdÆdÉdÈggg¡dÊdËZZej dÌdKgifgdÍdKifg¡dÎdÏZ[ej dÐdÑdg¡dÒdÓZ\ej dÔdÕdÖd.fdÕd×d.fdØdÙdÚg¡dÛdÜZ]dS)Ýé)ÚdateÚdatetime)ÚStringION)Ú    DataFrameÚIndexÚ
MultiIndexÚSeriesÚbdate_range)Úget_groupby_method_argscCsÎd}tjt|ddddgddggd}| d¡}| |jj¡ ¡}| |jj¡ dd¡}t     
||¡tjt|ddddgd    }td
ddgtdd }t dddg|d}|jddd dd¡}t     ||¡dS)Na+2011.05.16,00:00,1.40893
2011.05.16,01:00,1.40760
2011.05.16,02:00,1.40750
2011.05.16,03:00,1.40649
2011.05.17,02:00,1.40893
2011.05.17,03:00,1.40760
2011.05.17,04:00,1.40750
2011.05.17,05:00,1.40649
2011.05.18,02:00,1.40893
2011.05.18,03:00,1.40760
2011.05.18,04:00,1.40750
2011.05.18,05:00,1.40649rÚtimeÚvalue)ÚheaderÚnamesZparse_datesÚ    date_timecSs| ¡S©N©Úidxmax©Úx©rúVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_apply.pyÚ<lambda>/óz#test_apply_issues.<locals>.<lambda>)r rz
2011.05.16z
2011.05.17z
2011.05.18©ÚdtypeÚnamez00:00z02:00©ÚindexF©Ú
group_keyscSs|d|d ¡S)Nrrrrrrrr:r)ÚpdZread_csvrÚ    set_indexÚgroupbyrrrÚapplyÚtmÚassert_frame_equalrÚobjectrÚassert_series_equal)ÚsÚdfÚexpectedÚresultZexp_idxrrrÚtest_apply_issuess, ü
ÿÿr,cstdddddgdddddgdd    d
gdtjjddjddgdd dgd}jddjDdd fdd¡}t ||¡dS)NÚaÚbçð?ç@ç@ç@ç@©ÚkeyÚdatar5r6©ÚcolumnséÚfloat64r&©ÚaxisÚkeyscSsg|]}t|qSr©Ústr©Ú.0rrrrÚ
<listcomp>Gsz&test_apply_trivial.<locals>.<listcomp>©r<csjddS©Nr9©Úilocr©r)rrrHrz$test_apply_trivial.<locals>.<lambda>)    rr ÚconcatrFr"Údtypesr#r$r%©r*r+rrGrÚtest_apply_trivial?sþ,
ÿrKcs|tdddddgdddddgdd    d
gdtjgdd dgd}jddjDddd fdd¡}t ||¡dS)Nr-r.r/r0r1r2r3r4r5r6r7r9r:r&r;cSsg|]}t|qSrr>r@rrrrBUsz+test_apply_trivial_fail.<locals>.<listcomp>T)r<rcsSrrrrGrrrVrz)test_apply_trivial_fail.<locals>.<lambda>)rr rHr"rIr#r$r%rJrrGrÚtest_apply_trivial_failNsþ
ÿrLzdf, group_namesr9éér-r.Úc©r-r.éééééé    )r-ÚtwoZ
aaabbbcccc)r-ÚBÚCr7ZGH2936zGH7739 & GH10519ZGH10519ZGH2656ZGH12155ZGH20084ZGH21417)Úidscsgfdd}fdd}fdd}fdd}fd    d
}|||||fD].}dd=|jddd  |¡|ksNtqNdS)Ncs |j¡| ¡Sr)ÚappendrÚcopy©Úgroup©rrrÚf_copysz/test_group_apply_once_per_group.<locals>.f_copycs |j¡|Sr©r[rr]r_rrÚf_nocopysz1test_group_apply_once_per_group.<locals>.f_nocopycs |j¡dS©Nrrar]r_rrÚf_scalarsz1test_group_apply_once_per_group.<locals>.f_scalarcs |j¡dSrrar]r_rrÚf_nonesz/test_group_apply_once_per_group.<locals>.f_nonecs |j¡tdgdgdS)Nr9rP)r[rrr]r_rrÚ f_constant_dfsz6test_group_apply_once_per_group.<locals>.f_constant_dfr-Fr)r"r#ÚAssertionError)r)Zgroup_namesr`rbrdrerfÚfuncrr_rÚtest_group_apply_once_per_group\s'
ric
Cs~d}tddddddddgdddddd    d
dgddddddd    d
dgd }|jddd dd¡| ¡j d¡}||ksztdS)NrMrr9Ú0Ú2Ú4Ú6Ú8Ú10Z12Z14)Úgroup_by_columnZtest_columnrrpFrcSstdS)NÚfunction_called)ÚprintrGrrrrµrz2test_group_apply_once_per_group2.<locals>.<lambda>rq)rr"r#Z
readouterrÚoutÚcountrg)Zcapsysr*r)r+rrrÚ test_group_apply_once_per_group2¦sþûÿrucCs`tdddgtdd}dd}dd}|jd    d
d |¡}|jd    d
d |¡}t ||¡dS)Nrr9rN)ÚAr.cSs|Srrr]rrrÚslowÄsz,test_apply_fast_slow_identical.<locals>.slowcSs| ¡Sr©r\r]rrrÚfastÇsz,test_apply_fast_slow_identical.<locals>.fastrvFr)rÚranger"r#r$r%)r)rwryZfast_dfZslow_dfrrrÚtest_apply_fast_slow_identical½sr{rhcCs|SrrrrrrrÓrrcCs|ddSrrrrrrrÔrcCs|jddS)NF©ÚdeeprxrrrrrÕrcCs|jddS)NTr|rxrrrrrÖrcCsLtddddgddddgddddgd    }|jd
dd |¡}t ||¡dS) Nr9rMrNrRrSrTrUrQ)Úgr-r.r~Fr©rr"r#r$r%)rhr)r+rrrÚ5test_groupby_apply_identity_maybecopy_index_identicalÐs(rcCs¤ttj d¡ddddddgd}|jdddd    j}|j}t ||¡td
dddddgi}|jd |d<|     d¡ 
¡ ¡j}|j    ddd 
¡j}t ||¡dS)NrTÚonerWÚthree©Úfoo1Zfoo2cSs|Srrrrrrrírz-test_apply_with_mixed_dtype.<locals>.<lambda>r9rCÚc1rMrQr0Úc2F©Úas_index) rÚnpÚrandomÚrandnr#rIr$r'rr"ÚmeanÚreset_indexr)r)r+r*Úresult1Úresult2rrrÚtest_apply_with_mixed_dtypeås
þÿrc CsRtddddddgddddddgtdd}|jd    d
d}|jd    dd}| d¡j}| d¡j}td dddg}t ||¡t ||¡| dd¡j}| dd¡j}t     
ddddg¡}ddddg}    t    j
|    d    dgd}
t ||
¡t ||¡ttd}tddgddgddgddgddgg|d}|jd ddd dd¡j}t ||¡dS) Nr.r-rOr9rMrNrT)Zitem_idÚuser_idrrTrFrrRcSs
| d¡S©NrM©Úheadrrrrrrz-test_groupby_as_index_apply.<locals>.<lambda>cSs
| d¡Srrrrrrr r)rr)rrM©r9r9©rMrR©r9r©r9rM)rMr9)rNrRr_ÚabcderSr)rrcSs|Srrrrrrrr)rrzr"rrrr$Úassert_index_equalr#rÚfrom_tuplesÚlist) r)Zg_asZg_not_asZres_asZ
res_not_asÚexpZres_as_applyZres_not_as_applyZexp_not_as_applyÚtpZexp_as_applyÚindÚresrrrÚtest_groupby_as_index_applyùs0ýÿ*r¡cCsx| ddg¡}dd}dd}dd}| |¡}|jjd    ks@t| |¡}|jjd    ksZt| |¡}|jjd
ksttdS)NrvrXcSs| ¡}d|j_|S©NÚstat)Údescriberr©r^r+rrrÚdesc!sz.test_apply_concat_preserve_names.<locals>.desccSs$| ¡}d|j_|dt|}|Sr¢)r¤rrÚlenr¥rrrÚdesc2&sz/test_apply_concat_preserve_names.<locals>.desc2cSs0| ¡}dt|d|j_|dt|}|S)NZstat_Úd)r¤r§rrr¥rrrÚdesc3-sz/test_apply_concat_preserve_names.<locals>.desc3)rvrXr£)rvrXN)r"r#rrrg)Zthree_groupÚgroupedr¦r¨rªr+rZresult3rrrÚ test_apply_concat_preserve_namess
 
 
 
r¬cCstdd}tddd}ttj d¡|d}|jddd    d
}| |¡}t|tsRt    t
|dr`t    t |j |j ¡dS)Nc    Ss:tjddt |¡}W5QRXt||| ¡|dS)NÚignore)Úinvalid)rZdemeanedÚlogged)rZerrstateÚlogrr)Zpiecer¯rrrÚfBs
ÿz%test_apply_series_to_frame.<locals>.fz1/1/2000éd)ÚperiodsrcSs|jSr©ÚmonthrrrrrLrz,test_apply_series_to_frame.<locals>.<lambda>Frr)r    rrrrr"r#Ú
isinstancerrgÚhasattrr$rr)r±ZdrÚtsr«r+rrrÚtest_apply_series_to_frameAs
r¹cCs4| ddg¡d t¡}|jjdddks0tdS)NrvrXrYrM©rvrX)r"r#r§rrrg©r)r+rrrÚ test_apply_series_yield_constantTsr¼cCsl| ddg¡ t¡}t|ts"t|jdks0t| ddg¡ddg t¡}t|tsZt|jdkshtdS)NrvrXrYÚD)r"r#r§r¶rrgrr»rrrÚtest_apply_frame_yield_constantYsr¾cCsH| ddg¡}| t¡}| ¡d}t |j|j¡t |j|j¡dS)NrvrXrY)    r"r#r§rtr$rrÚassert_numpy_array_equalÚvalues©r)r«r+r*rrrÚtest_apply_frame_to_seriesds
 
rÂcCs^|jddgdd}| t¡}| ¡jdtjidjdd}t     |j
|j
¡t |j|j¡dS)NrvrXFrrYr7r½) r"r#r§rtÚrenamerÚnanÚdropr$rrr¿rÀrÁrrrÚ)test_apply_frame_not_as_index_column_namels
 
rÆcsdd}fdd}ttj ddd¡tj ddd¡tj d¡d d    ¡ |¡} d    ¡d
 |¡}tj||dd|j    d
kst
dS) NcSs | d¡d ¡ ¡jddS)NrXrYrM)r"ÚsumÚsort_valuesrFr]rrrÚtranswsz-test_apply_frame_concat_series.<locals>.transcs,|  |j¡d¡}| ¡ ¡jddS)NrXrM)r"ÚreindexrrÇrÈrF)r^r«rGrrÚtrans2zsz.test_apply_frame_concat_series.<locals>.trans2rrSéè©rvrXrYrvrYF)Zcheck_names)rrrÚrandintrr"r#r$r'rrg)rÉrËr+rrrGrÚtest_apply_frame_concat_seriesvs
ýÿrÏcCs>|jdddd}| dd¡}| dd¡}t ||¡dS)NcSs|jSrr´rrrrrrz&test_apply_transform.<locals>.<lambda>FrcSs|dSrrrrrrrrcSs|dSrrrrrrrr)r"r#Ú    transformr$r')r¸r«r+r*rrrÚtest_apply_transformsrÑcCsP| ddddg¡}dd}| |¡}|D]\}}t |j|||¡q,dS)NcSs|jSr©Úyearrrrrrrz,test_apply_multikey_corner.<locals>.<lambda>cSs|jSrr´rrrrrrcSs| d¡ddS)Nrvéûÿÿÿ©rÈr]rrrr±sz%test_apply_multikey_corner.<locals>.f)r"r#r$r%Úloc)Útsframer«r±r+r5r^rrrÚtest_apply_multikey_corners
 
rØrTFc
Cstdddddddddg    tdd}|jd|d dd    ¡}| d
ddddd g¡}|r|tjddddddg|jgddgd|_t     ||¡dS)Nr9rMrNrV)r5rr5rcSs|jddSrrErrrrr£rz'test_apply_chunk_view.<locals>.<lambda>rrRrTrUr_)
rrzr"r#ZtakerÚfrom_arraysrr$r%)rr)r+r*rrrÚtest_apply_chunk_views$ÿrÚcCs`tddddddddddg
ddddddddddg
tdddd}| ddg¡}| d    d
¡dS)Nr9rMrrVéÿÿÿÿ)rÚname2rrrÜcSs|jdddS)NrT©ZinplacerÕrrrrr¸rz4test_apply_no_name_column_conflict.<locals>.<lambda>)rrzr"r#)r)r«rrrÚ"test_apply_no_name_column_conflicts
ýÿ    rÞcCs|tddddddgt dddgd¡t dd¡d}d    d
}|jddd  |¡}| ¡}t dddgd¡|d<t ||¡dS)Nr/r0r-r.rOrMç@©r©rOÚvcSs,|d}|| ¡| ¡| ¡|d<|S©NráÚv2©ÚminÚmax©r^rárrrr±Äs z#test_apply_typecast_fail.<locals>.fr©Frççà?r9rã)    rrÚtileÚaranger"r#r\r$r%)r)r±r+r*rrrÚtest_apply_typecast_fail»s
ýÿrìc    Cs¦t ddddddgddddddgg¡}tddddddgt ddd    gd¡t dd
¡d|d}d d}|jddd |¡}| ¡}t dddgd¡|d<t     
||¡dS)Nrr9rMrNr/r0r-r.rOrßràrcSs,|d}|| ¡| ¡| ¡|d<|Srârärçrrrr±Üs z%test_apply_multiindex_fail.<locals>.fr©Frrèrérã)rrÙrrrêrër"r#r\r$r%)rr)r±r+r*rrrÚtest_apply_multiindex_failÑs&
ýú    rícCs4|jdddd dd¡}|d}t ||¡dS)NcSs|jSrrÒrrrrrêrz#test_apply_corner.<locals>.<lambda>FrcSs|dSrrrrrrrêrrM)r"r#r$r%)r×r+r*rrrÚtest_apply_cornerésrîcCshtddddgddddgddd    d
gd}dd }dd}| d¡ |¡}| d¡ |¡}t ||¡dS)Nr²éÈi,r-r.rOr9rMrNrR)Úid_fieldÚcategoryrcSs(|jddkr| ¡S||jdkSdS©Nrr9rO)Úshaper\rñrrrrÚfilt1ûsz&test_apply_without_copy.<locals>.filt1cSs$|jddkr|S||jdkSdSrò)rórñrrrrÚfilt2sz&test_apply_without_copy.<locals>.filt2rðr)r6rôrõr*r+rrrÚtest_apply_without_copyïs
 
 
ýÿröÚtest_seriescCs®tddgddgddggddgdddgd}|rr| d¡d}|jd    d
d dd ¡}| ¡}| ¡}t ||¡n8|jdd
d dd ¡}| d¡}| d¡}t ||¡dS)NrÚpÚoÚXÚYr9rM©r8rrF)ÚlevelrcSs|Srrrrrrrrz<test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>rcSs|Srrrrrrrr)    rr!r"r#Ú
sort_indexr$r'rÈr%)r÷r)Úserr+r*rrrÚ*test_apply_with_duplicated_non_sorted_axissÿ
 
rcCsnddddg}ddddg}tddgd|d|d}t||d    d
}dd}|jd ddj |¡}t ||¡dS)Nr9rMrNrRZGroup1ZGroup2©r^rrr©rrcSs"| t |j ¡|j ¡d¡¡SrD)rÊrrërrårærrrrÚreindex_helper-sz1test_apply_reindex_values.<locals>.reindex_helperr^Fr)rrr"rr#r$r')rÀÚindicesr)r*rr+rrrÚtest_apply_reindex_values#srcCsnd}tjjdd|d}t|tj |¡ddddg|d    d
}|jddd }dd}| |¡}d|ksjtdS)NrÌrr²)ÚsizeÚfooÚbarZbazZquxrR)r5Úvalue1Zvalue2r5FrcSs|dd|d<|S)Nr    rMÚvalue3r©r~rrrr±Dsz"test_apply_corner_cases.<locals>.fr
)rrrÎrrr"r#rg)ÚNÚlabelsr)r«r±r+rrrÚtest_apply_corner_cases5s
ýÿ
rcCsNtddgdgdddgd}| dg¡ dd    ¡}t |j¡|_| dg¡ d
d    ¡}t |d|d¡tdd dgdddgt d¡gdd}dd}| d¡ |¡d}|j    }|j
|_t ||¡dd}tddddgddddgddddgdd d!d!gd"}| ¡}t |j ¡|_ | d#¡ |¡j}| d#¡ |¡j}t ||¡dS)$Nr9rMz
2017-03-02rÚinf)ÚNumberÚDateÚStrrcSs
|jdSrcrErrrrrVrz;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>cSs
|jdSrcrErrrrrXrré
ééÚ3rlz12:31:22rN)rvrXÚTcSs|jddgS)NrrXrErrrrÚget_B`sz8test_apply_numeric_coercion_when_datetime.<locals>.get_BrvrXcSstdddgtd}dt|jkr>t||jdkjjd|d<dt|jkrt||jdkjjd|d<t||jdkjjd|d<|S)NÚp1Úp2ZuseTime)rrÚstep1rÚstep2)rr&rÚStater?ÚMachinerÀÚoTime)ZtoolrsrrrÚpredictionsisz>test_apply_numeric_coercion_when_datetime.<locals>.predictionsrrÚz2016-09-19 05:24:33z2016-09-19 23:59:04Z23Z36LZ36R)ÚKeyrrrr")rr"r#r Úto_datetimerr$r'Ú    TimestamprXrvrr\rr)r)r*r+rr Údf1Údf2rrrÚ)test_apply_numeric_coercion_when_datetimeLs8ÿ"ÿ    
 
 
 
üÿr'cCstdddgt d¡gdd}|j|j|d<| d¡ d    d
¡}tdddgt dd¡gdt d¡gdd  d¡}t     ||¡dS)NrvrXrYz2017-02-01 00:00:00rN)ÚclientidrÚtime_delta_zeror(cSst|j ¡|j ¡dS)N)Úclientid_ager)rr)rår)Zddfrrrrsÿz?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>rr½)r(r*r)
rrZ
datetime64rr"r#Ztimedelta64r!r$r%©r)r+r*rrrÚ-test_apply_aggregating_timedelta_and_datetimes$þÿ
ÿýÿúr,cCsddgddgddgddgddgg}t|d    d
gt dd¡d }| d    ¡ ¡}tdddgdddgd}|jd    ddt ||¡dS)NrvrrXrrrYé(é2ÚNameÚValuez
2020-09-01z
2020-09-05rüéZ)r/r0TrÝ)rr Ú
date_ranger"rÇr!r$r%)r6r)r+r*rrrÚ test_apply_groupby_datetimeindexs"
ÿr3cCs®tdddtdDd}dd}dd    }|jd
gd |¡}tdd idgd}d
|j_|jd
gd |¡}ttdddd ddgd}d
|j_t ||¡t ||¡dS)Nr9cSsg|]}t ¡qSr)rÚnow)rAÚnnrrrrB³sz'test_time_field_bug.<locals>.<listcomp>rrPcSstddiS)NrOrM©r©ÚbatchrrrÚfunc_with_no_dateµsz.test_time_field_bug.<locals>.func_with_no_datecSsttdddddS)Néßr9rM©r.rO)rrr7rrrÚfunc_with_date¸sz+test_time_field_bug.<locals>.func_with_dater-©ZbyrOrMrr:r;)    rrzr"r#rrrr$r%)r)r9r<Zdfg_no_conversionZdfg_no_conversion_expectedZdfg_conversionZdfg_conversion_expectedrrrÚtest_time_field_bugsr>cs°tddddddddddddgddddddddddddgdddd    d
ddd    ddd dgdddddddddd
d    dgd}| ddg¡}|jddgdd}dd| fdd¡dS)Nr-r.rOr©Úegñ?rMrNrRrSrTrQr9gffffff@rVréérU)Úgroup1Úgroup2ÚweightrrBrCT)rýÚsortcSst ||¡ d¡}|S)NrN)rÚarrayÚrepeat)rrDrsrrrÚnoddyÔsz7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddycs|j|jSr)rrDr©rHrrrÞrz:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)rr!r"r#)r)Z
df_groupedrrIrÚ(test_gb_apply_list_of_unequal_len_arraysÇsüÿ
rJcCsLtddddgddddgd}dd    }| d
¡ |¡}t}t ||¡dS)Nrr9rQrUrRrS)ÚgroupsZrandom_varscSsdSrrrrrrÚ    test_funcæsz.test_groupby_apply_all_none.<locals>.test_funcrKr)Ztest_dfrLr+r*rrrÚtest_groupby_apply_all_noneás
rMc    Csðtddddgddddgd}tddddgddddgd}dd}| d¡ |¡}| d¡ |¡}tjddgddggddgd    }tjddgddggddgd    }tddgddgd|d
}tddgddgd|d
}t ||¡t ||¡dS)Nr9rMrrN)rKÚvarscSs |jddkrdS|jddgS)NrrMrÛ)rórFrrrrrLósz0test_groupby_apply_none_first.<locals>.test_funcrKr_r)rr"r#rrÙr$r%)    Ztest_df1Ztest_df2rLrrZindex1Úindex2Z    expected1Z    expected2rrrÚtest_groupby_apply_none_firstîsrPcCsltddgddgd}| d¡}| dd¡}tdgd    tjddgdggddgd
 d¡d}t ||¡dS)Nrr9ZfilledÚempty)rr^r^cSs||jdkdS)Nr9r)rr]rrrrrz7test_groupby_apply_return_empty_chunk.<locals>.<lambda>rr_)rr)    rr"r#rrÚfrom_productrÅr$r')r)rKr+r*rrrÚ%test_groupby_apply_return_empty_chunks
ÿþýrScCstd ¡dddgdddgd}|jd    d
d}| dd ¡}tdddgdddgd}t ||¡| dd ¡}t ||¡dS)Nza a br9rMrNrRrTrSrÍrvFrcSs|| ¡Sr©rÇrrrrrrz-test_apply_with_mixed_types.<locals>.<lambda>gUUUUUUÕ?gUUUUUUå?çÙ?g333333ã?r/©rXrYcSs|| ¡SrrTrrrrrr)rÚsplitr"rÐr$r%r#)r)r~r+r*rrrÚtest_apply_with_mixed_typess rXcCshtdddgitddgd}| d¡ dd¡}ttdgtdggtddgddd}t ||¡dS)Nr-r9rMrcSs|jSrrrrrrr!rz*test_func_returns_object.<locals>.<lambda>©r)rrr"r#rr$r'r+rrrÚtest_func_returns_objects(rZÚgroup_column_dtlikecCsTtdg|gd}| d¡ dd¡}tdgtdgdddd    gd
}t ||¡dS)NrrPr-cSstdgdgdS)NÚspamé*rr6rrrrr2rz+test_apply_datetime_issue.<locals>.<lambda>r\r&rr]r7©rr"r#rr$r%)r[r)r+r*rrrÚtest_apply_datetime_issue's
ÿr_c    Cstt d¡t d¡t d¡t d¡t d¡dddddddddddddd}dd}| d    ¡ |¡d
}tdgtjdgd    dd
d }t ||¡dS)Nz2015-02-24 00:00:00)rr9rMrNrRzsome UA stringzanother UA stringZ17661101)ÚdayZ    userAgentÚuserIdcSstdd| ¡DS)NcSs i|]\}}|| ¡jdqS)r)Zvalue_countsr)rArOr(rrrÚ
<dictcomp>WszYtest_apply_series_return_dataframe_groups.<locals>.most_common_values.<locals>.<dictcomp>)rÚitemsrGrrrÚmost_common_valuesVszEtest_apply_series_return_dataframe_groups.<locals>.most_common_valuesr`raz
2015-02-24rYr)    rr r$r"r#rÚ DatetimeIndexr$r')Ztdfrdr+r*rrrÚ)test_apply_series_return_dataframe_groups:s:ûûûñÿÿrfrñcCsàddgd}|rLtj|dddgd}tjdddgdddgdd}dd    d
g}ntddgdd}dd g}t||d|d}tt d¡|ttdttdd     ddg¡}| 
d¡ dd¡}t  ||¡|jjddgksÜtdS)Nr9rMrSrN)Ú
categoriesrX)rgrg4@g9@rèrYré)rYr½rr)rvrXrYr½rvcSs| ¡SrrTrrrrrsrz-test_apply_multi_level_name.<locals>.<lambda>)r ZCategoricalÚCategoricalIndexrrrrërrzr!r"r#r$r%rrrg)rñr.Zexpected_indexÚexpected_valuesr*r)r+rrrÚtest_apply_multi_level_name`s&ÿ"ÿþrkc    Cs¾tjt ddd¡ddddft ddd¡ddd    d
ft ddd¡dd ddft ddd¡ddddfgdddddgd}| d¡ dd¡j}tt     d¡t
t
tjt
gdddddgd}t  ||¡dS)Niàr9ÚredZdarkrnr:ÚgreenZstormyrMÚ9iÞÚblueZbrightrNroiÝZcalmrRZpotatoZobservationÚcolorZmoodZ    intensityZscorer7cSs
|jdSrcrErrrrrrz;test_groupby_apply_datetime_result_dtypes.<locals>.<lambda>zdatetime64[ns]r)rZfrom_recordsr r$r"r#rIrrrr&Úint64r$r')r6r+r*rrrÚ)test_groupby_apply_datetime_result_dtypesxsüù    þrrrÚabcZ2020r½)r³Úfreq)r-r)r-r9)r.rcCsFtdddgdddgd|d}|jddd d    d
¡}t ||¡dS)Nr9rMrrrr^FrcSs|Srrrrrrrrz8test_apply_index_has_complex_internals.<locals>.<lambda>r)rr)r+rrrÚ&test_apply_index_has_complex_internalssruzfunction, expected_valuescCs
|j ¡Sr)rÚto_listrrrrrrcCst|j ¡Sr)ÚsetrrvrrrrrrcCst|j ¡Sr)Útuplerrvrrrrr r)rr9©rMrNcCstt|j ¡Sr)ÚdictÚ    enumeraterrvrrrrr¢rcCsddt|j ¡DS)NcSsg|]\}}||iqSrr)rAÚnÚirrrrB¦sz<lambda>.<locals>.<listcomp>)r{rrvrrrrr¦rcCsNtddddgdgd}| d¡ |¡}t|tddgddd}t ||¡dS)NrvrXrKr7rYr©rr"r#rrr$r')Úfunctionrjr)r+r*rrrÚ1test_apply_function_returns_non_pandas_non_scalarsrcCsrdd}tddddgdddtjgd    }| d
¡ |¡}tddgd gtjggtdddgd
dd}t ||¡dS)NcSs|dj ¡S)NrX)rÀÚflattenr]rrrÚfctµsz4test_apply_function_returns_numpy_array.<locals>.fctr-r.Únoner9rMrNrºrvr/r0r1rYr)    rrrÄr"r#rrr$r')rr)r+r*rrrÚ'test_apply_function_returns_numpy_array³s ÿrrcCs|jSrr©ZgrrrrrÁrcCs|jddSrDrrrrrrÁrcCstddddddddddg
dgd}| d¡ |¡}ttdddd    gtdddd
gtddggtdddgdd d}t ||¡dS)Nr9rMrNÚidr7rrRrUrVrSrTrQrYrr~)rr)r+r*rrrÚ test_apply_function_index_returnÁs"(þrcCshtddddddgddddddgd    }|jd
dd d d¡}tdddgdddgd    }t ||¡dS)NrrWrr9rMrRrSrTrrFrcSs| ¡Sr)rrrrrrÕrzAtest_apply_function_with_indexing_return_column.<locals>.<lambda>r1r2rr+rrrÚ/test_apply_function_with_indexing_return_columnÍsþÿþÿrÚudfcCs| ¡SrrxrrrrrárcCs| ¡ dd¡S)NcSs|dSrDr)Úyrrrrárz<lambda>.<locals>.<lambda>)r\rÃrrrrrárcCstddgddgd}|jd|d |¡}|jj|j|d |¡}|rh|jjdksVt|jjdkstn |jjdksxt|jjdkstdS)Nr-r.r9rMrºrvr)rr"r#rXrvrZnlevelsrg)rrr)Z    df_resultZ series_resultrrrÚtest_apply_result_typeßs    rcCsZtdddgdddgd}|jddd dd    ¡}|jddd d
d    ¡}t ||¡dS)NrMr9rNrºrvFrcSs|Srrrrrrrørz4test_result_order_group_keys_false.<locals>.<lambda>cSs| ¡Srrxrrrrrùrrr+rrrÚ"test_result_order_group_keys_falseôsrcCs¶dgddgddgd}t |¡}tj|dd}tttddtd|d    }tttddtd|d    }|jd
dd d d¡}|jd
dd dd¡}t ||¡dS)Nz
2001-01-01rMz
2001-01-02z
2001-01-03ÚUTC)ÚtzrNrT)rrÚtrFrcSs|ddg ¡S©NrrrxrGrrrrrz1test_apply_with_timezones_aware.<locals>.<lambda>cSs|ddg ¡SrrxrGrrrrr)    r rerrrzr"r#r$r%)ÚdatesZindex_no_tzZindex_tzr%r&rrrrrÚtest_apply_with_timezones_awareýs
  rc    CsÀtddddddgddddddgd    d
ddd dgd}tddgddgddgdtddgddd}|jdd}| t¡}t ||¡|jdd}t||}t|||}| t¡}t ||¡dS)NécéXr9rMrNrRrSrTrrrr-r.é<©r-r.rOii)éér-rYrr=)    rrr"r#rÇr$r%r
Úgetattr)Zreduction_funcr)r*Úgrpr+ÚargsÚ_rrrÚ;test_apply_is_unchanged_when_other_methods_are_called_firsts$ýÿþ
 
 
rc    Csätddddgtdddtdddtdddtdddgddddgd    td
ddd gddd}| ddg¡}| dd¡}|jdddg}| ¡}t |dddg¡|_    |j
dd}t ||¡|j    j dD]}t|tksÊtqÊdS)Nr-r.iär9rrMrNrRrÍr²éeéfégÚidxrYrrvrXcSs
| d¡SrDrrrrrr:rzRtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp.<locals>.<lambda>rr7)rrrr"r#rFrrZ
from_framerrÅr$r%ÚlevelsÚtyperg)r)rr+r*ÚvalrrrÚ@test_apply_with_date_in_multiindex_does_not_convert_to_timestamp(s(
 
 
 
 
ü
ø
õr¥cCszttj ddg¡t ddgddgg¡d}|jjddd     d
d¡}|jddd     dd¡}t     ||j¡t     ||¡dS) NrTrRrvrXr9rMr7r©r<rýcSs|jdddS)Nrr¦©Z    droplevelrrrrrRrzDtest_apply_by_cols_equals_apply_by_rows_transposed.<locals>.<lambda>cSs|jdddS)Nr9rr¦r§rrrrrTr)
rrrrrRrr"r#r$r%)r)Zby_rowsZby_colsrrrÚ2test_apply_by_cols_equals_apply_by_rows_transposedFsþÿr¨ÚdropnacCsxtdddddgdtjtjddgdtd    d
}|jd|dd  dd¡}|rT| ¡n|jdddddg}t     ||¡dS)Nr9rMrNrRrSr-r.)Úcolr^Zxxyxzrr^F)r©rcSs|Srrrrrrrerz5test_apply_dropna_with_indexed_same.<locals>.<lambda>r)
rrrÄrr"r#r©rFr$r%)r©r)r+r*rrrÚ#test_apply_dropna_with_indexed_sameZsþû r«zas_index, expected)rr)rMrMr_rcCsTtddddgddddgddddgd}|jddg|d dd¡}t ||¡dS)    Nr9rMrr-r.rcSsdSrDrrrrrr~rz5test_apply_as_index_constant_lambda.<locals>.<lambda>)rr"r#r$Zassert_equal)rr*r)r+rrrÚ#test_apply_as_index_constant_lambdajs(r¬cCstdddddgdddd    d
gdddddgdtdd}| d ¡ dd¡}ttddtjdddddgd dgddd}t ||¡dS)Nr9rMrNrRrSrTrUrQrVrrÍrrYcSs
|j ¡Sr)rvrþrrrrrrz(test_sort_index_groups.<locals>.<lambda>rrrryrr_rvr)    rrzr"r#rrrr$r'r+rrrÚtest_sort_index_groupss(þÿûrcCsXttjddddtdtdd}|j|j|jjjgdd     d    d
¡}t
 ||¡dS)Nz
2010-01-01Z12HrS)rtr³r)rÚvalsÚletFrcSs|jddSrcrErrrrrrz;test_positional_slice_groups_datetimelike.<locals>.<lambda>)rr r2rzrr"r¯rÚdtr#r$r%rJrrrÚ)test_positional_slice_groups_datetimelikesýÿÿþr±cCsxtdddgdddgdddgd    }| d
¡}|ddg d d¡}tddgddgdtddgd
dd}t ||¡dS)Nr-r.r9rMrNrRrTrSrÍrvrXrYcSs| t¡ ¡| ¡Sr)ZastypeÚfloatrærårrrrr¦rz7test_groupby_apply_shape_cache_safety.<locals>.<lambda>r/rèr0rVrYrr^)r)Úgbr+r*rrrÚ%test_groupby_apply_shape_cache_safety¢s"
ÿr´cCsftddddgddddgddtjtjgd}|jd|d}| dd    ¡}| d
d    ¡}t ||¡dS)Nr9rMrrS)rrÚzr)r©cSs| dd¡S)Nr9rµ)Únlargest)Zgrp_dfrrrrµrztest_apply_na.<locals>.<lambda>cSs|jddd d¡S)NrµF)Z    ascendingr9)rÈrrrrrr¶r)rrrÄr"r#r$r%)r©r)Zdfgrpr+r*rrrÚ test_apply_na®s&ÿr·cCs tddddgddddgtjddddgddd d    d
g¡ dd¡}tddtjdddgddtjdddggd    d
d gtjddgd    d
gdd}t ||¡dS)Nr9rMr!rNrRr()Úunitrr-r.cSs
|jdS©NrÛrErGrrrrÅrz8test_apply_empty_string_nan_coerce_bug.<locals>.<lambda>rO)r9r!)rMr!r_rü)    rr r#r"r#rrr$r%)r+r*rrrÚ&test_apply_empty_string_nan_coerce_bugºs"
 
ýÿùøÿ&ýrºÚindex_valuesr/r0r1cCsjtdddgdddgdt|d}td    d
ddgitdddgd dd}| d ¡ dd¡}t ||¡dS)NZaaZa2Úa3r9rMrNrPrÚb_meanr0r1r/r-rYcSst|d ¡gdgdS)Nr.r½r)rrrGrrrràrz0test_apply_index_key_error_bug.<locals>.<lambda>)rrr"r#r$r%)r»r+r*rrrÚtest_apply_index_key_error_bugÏs þûÿü
ÿr¾zarg,idxg¹?g333333Ó?gÉ?rUcCs8td|i|d}|jddd dd¡}t ||¡dS)NrªrFrcSs|Srrrrrrrrz5test_apply_nonmonotonic_float_index.<locals>.<lambda>r)Úargr¡r*r+rrrÚ#test_apply_nonmonotonic_float_indexås+rÀzargs, kwargsÚnumeric_onlycCs6| d¡}|jd||}|jdd}t ||¡dS)NrvrÇT)rÁ)rÇ)r"r#rÇr$r%)r)rÚkwargsr³r+r*rrrÚtest_apply_str_with_argss
rÃrZ    some_namecCsLtddg|d}|jddgdd dd¡}tddg|d}t ||¡dS)    Nr9rMrYr-FrcSs|Srrrrrrr"rz1test_result_name_when_one_group.<locals>.<lambda>)rr"r#r$r')rrÿr+r*rrrÚtest_result_name_when_one_groupsrÄz
method, opr#cCs
|jdSr¹)rÀ©r³rrrr+rcCs|djdS)Nr.rrErÅrrrr,r)ÚaggZskew)rÆÚprod)rÆrÇc    Cs^tggd}|jddd}t|d}t|||}tgddtgdddd}t ||¡dS)    NrPr-Trr.r:r)rrr)rr"rrrr$r')ÚmethodÚopZempty_dfr³r^r+r*rrrÚ test_empty_df(s
ÿrÊ)^rrÚiorÚnumpyrZpytestZpandasr rrrrr    Zpandas._testingZ_testingr$Zpandas.tests.groupbyr
r,rKrLÚmarkZparametrizerzrrirur{rrr¡r¬r¹r¼r¾rÂrÆrÏrÑrØrÚrÞrìrírîrörrrr'r,r3r>rJrMrPrSrXrZÚtodayrr_rfrkrrriZinterval_rangeZperiod_rangerrurrrrrrrrr¥r¨r«r&r¬rr±r´r·rºr¾rÀrÃrÄrÊrrrrÚ<module>s)*$(.ýÿø
(ðùì
,üþ    
%#
 
 
5      þ
&
 
üþ    
þþøþ
 
þ    
ÿþÿþùþ
 
 
ýýúýýúýýúçþ)
 
 
    
 
ûþ