U
|
ý°d¨Ø ã " @ sn d dl Z d dlmZ d dlZd dlZd dlmZ d dlm Z d dl
|
Z d dl
|
m Z m Z mZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ ej ej ej ej ej d d
|
gd d d dd d
|
gddd Z!dd Z"ej# $de%e&e'g¡ej# $ddddgg¡dd Z(G dd dZ)G d d d Z*d d Z+d d! Z,d"d# Z-ej# $d$d%d&d'd(d)d*d+g¡ej# $d,d-d.d/d/d0d1d2d0gifd3d.d/d1d0d1d4d0gifd5d.d/d/d0d1d2d0gifd6d.d/d1d0d1d4d0gifd7d/d1d0d1d1d0gd(d8fg¡d9d: Z.ej# $d;ed<ed=fd>g¡d?d@ Z/ej# $dAdBd d1gd/d2gd/d1gdCfdDd/d2gd d1gd d2gdCfg¡ej# $dEdFdGg¡dHdI Z0dJdK Z1ej# $dEdFdGdg¡dLdM Z2dNdO Z3dPdQ Z4dRdS Z5dTdU Z6ej# $dVdWej7fdXej8fdYej9fdZej:fd[ej'fd\ej;fd5ej&fd6ej%fd-d]d^ fd3d_d^ fd7ej<fej=d`e6ej>dag ¡dbdc Z?ddde Z@dfdg ZAdhdi ZBdjdk ZCej# $dld d/d1d2gd d d/d/gfd gd gfg¡ej# $d$dóejD¡ej# $dmdndog¡dpdq ZEej# $drdsdtg¡dudv ZFdwdx ZGej# $dmdydzg¡ej# $d$d{d d
|
d|d}g¡d~d ZHdd ZIdd ZJej# $dmdydzg¡ej# $d$d|d d
|
g¡ej# $dd/d/d/gd/ddgfd/d1d2gd/dd1gfd/d2d2gd/ddgfg¡d
d ZKej# $dmdydzg¡dd ZLejMej# $dmdydzg¡ej# $dd{e Nd+¡j%fdg¡dd ZOej# $dd/d1dd2d1d d4ddd/d/g dFdGdGdFgfd/dejPd2d1ejPejP ddd/ejP g dFdGdFdGgfd/d1ejQd2d1ejQejQdejP d/ejQg dGdGdGdGgfg¡dd ZRej# $ddddd2d4ddd1d d/d/g dFdGdGdFgfejPd/ejP ejPd1dejP ddejP ejP g dFdFdGdFgfd/d1ejQd2d1ejQejQdejP d/ejQg dGdGdGdGgfg¡dd ZSdd ZTdd ZUdd ZVej# $dd d d¡gg¡d¢d£ ZWd¤d¥ ZXd¦d§ ZYd¨d© ZZdªd« Z[ej# \d¬¡ej# $ddFdGg¡ej# $dd®gd®d¯gg¡d°d± Z]d²d³ Z^d´dµ Z_ej# $d¶d/d/d/d1d1d1d2d2d2g d/e j`d1d/e j`d1d/e j`d1g d0d/d/d1d1d2d2gd/d1d/d1d/d1gd0g¡ej# $d·dWdXdZg¡d¸d¹ Zadºd» Zbd¼d½ Zcd¾d¿ Zdej# $dÀdÁdÂdÃdÄdÅdÆdÇdÈdÉdÊdËdÌdÍdÎdÏdÐdÑdÒdÓdÔdÕdÖd×dØdÙdÚdÛdÜdÝdÞdßg ¡ej# $dEdFdGejeg¡ej# $dd®gd®d¯gg¡dàdá Zfej# $d$egeheiejg¡dâdã Zkej# $d$eheiejg¡ej# $dädådædçgdèddédådædçgddêgdédådædçgdêgddég¡dëdì Zldídî Zmdïdð Zndñdò ZodS )ôé N)ÚStringIO)Úlib)ÚUnsupportedFunctionCall)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_range)Únanops)Úget_groupby_method_args)Ú_test_decoratorsÚInt64ÚFloat64znp.int32znp.int64z
|
np.float32z
|
np.float64)ÚparamsÚidsc C s | j }|}|dkrtj}n|dkr(tj}t |¡jdkrDt |¡jn
|
t |¡j}t |¡jdkrlt |¡j n
|
t |¡j }|||fS )zZ
|
Fixture of dtypes with min and max values used for testing
|
cummin and cummax
|
r r Úi)
|
ÚparamÚnpÚint64Úfloat64ÚdtypeÚkindÚiinfoÚminZfinfoÚmax)Úrequestr Znp_typeÚmin_valÚmax_val© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_function.pyÚdtypes_for_minmax s ÿ
|
ýÿ
|
ýr! c C s` t ddtjdg} | ddddg¡}| tj¡}| tj¡}| ¡ }t ||¡ t ||¡ d S )Nç ð?ç @ç @r é é )
|
r r ÚnanÚgroupbyÚaggÚbuiltinsÚsumÚapplyÚtmÚassert_series_equal)ÚsÚgroupedÚresultZresult2Úexpectedr r r Útest_intercept_builtin_sum8 s r3 ÚfÚkeysÚjimÚjoec s6 t tj ddd¡ddgd}tj d¡|d< | | ¡}|j | |¡}t|j | d }d
|
|j
|
d | d }|j
|
|d fkst | fdd| ¡}t ||¡ t d ¡ | fdd¡}W 5 Q R X t ||¡ |tkr| ¡ ¡ }|j| ddd t j ||dd t t| ddt| dd¡ d S )Nr% é2 )éè r& r6 r7 ©Úcolumnsr9 Zjolie)Zsubsetzinvalid frame shape: z (expected (z, 3))é c s t t | ddS )Nr ©Úaxis)Úgetattrr ©Úx)Úfnamer r Ú<lambda>S ó z%test_builtins_apply.<locals>.<lambda>c s | S ©Nr r@ )Únpfuncr r rC X rD TF)ÚinplaceÚdrop)Z check_dtyper r= )r r ÚrandomÚrandintÚrandnr( Ú__name__r, ÚlenZdrop_duplicatesÚshapeÚAssertionErrorr- Úassert_frame_equalÚassert_produces_warningr+ r) Ú reset_indexÚ set_indexr. r? ) r5 r4 ÚdfÚgbr1 ZngroupsZ
|
assert_msgr2 Z expected2r )rB rF r Útest_builtins_applyC s&
|
|
|
|
rV c @ s¾ e Zd Zejdd Zej dddg¡dd Zej ddd g¡d
|
d Z ej dd d g¡dd Z
|
ej dddg¡dd Z ej dddg¡dd Z ej dddg¡dd Z d d Zd S ) ÚTestNumericOnlyc C s~ t dddgdddgdddgtdttd d¡d d
|
d gtd dd td dddtjddddd ddddddddd g d }|S ) Nr% r& r< ç @ç @ç @ÚabcÚcategoryé é é Z20130101©Úperiodsú
|
US/Eastern)ra Útzz1 sr/ )ra Úfreq) ÚgroupÚintÚfloatÚstringÚcategory_stringÚ category_intÚdatetimeÚ
|
datetimetzÚ timedeltare rf rg rh ri rj rk rl rm r: )r Úlistr Úastyper
|
ÚpdZtimedelta_range©ÚselfrT r r r rT f s.
|
÷ ÷ôzTestNumericOnly.dfÚmethodÚmeanÚmedianc C s¾ t dddg}| d¡}tddgddgt d ¡t d
|
¡gd d gtd tdgtd ddtdddgdt ddgdddddgd}t||dd}t | |¡|¡ |j
|
}| ||||¡ d S )Nrf rg rj re g @r_ ç @rZ z1.5sZ3sç ø?r< z2013-01-01 12:00:00z2013-01-03 00:00:00rb ©rc )rj rg rm rf rk rl r% r& ©Úname©Úindexr; T©Ú numeric_only) r r( r rp Ú Timedeltar r? r- rP Z reindex_liker; Ú_check)rr rT rs Úexpected_columns_numericrU r2 r1 Úexpected_columnsr r r Ú test_averages s0
|
þ
|
|
þ÷ýðz TestNumericOnly.test_averagesr r c C s. t dddddddg}|}| ||||¡ d S )Nrf rg rh rj rk rl rm ©r r ©rr rT rs r r r r r Ú test_extrema¨ s ùÿ z TestNumericOnly.test_extremaÚfirstÚlastc C s0 t ddddddddg}|}| ||||¡ d S ) Nrf rg rh ri rj rk rl rm r r
r r r Útest_first_last¼ s øÿ z TestNumericOnly.test_first_lastr+ Úcumsumc C sL t dddg}t dddddg}|dkr8t ddddg}| ||||¡ d S )Nrf rg rj rh rm r r )rr rT rs r r r r r Útest_sum_cumsumÎ s ÿz TestNumericOnly.test_sum_cumsumÚprodÚcumprodc C s& t dddg}|}| ||||¡ d S )Nrf rg rj r r
r r r Útest_prod_cumprodÚ s z!TestNumericOnly.test_prod_cumprodÚcumminÚcummaxc C s, t ddddddg}|}| ||||¡ d S )Nrf rg rj rk rl rm r r
r r r Útest_cummin_cummaxá s
|
ÿz"TestNumericOnly.test_cummin_cummaxc
|
C s4 | d¡}| d¡rtnt}|dkrdd ddd| dg¡}tj||d t|| W 5 Q R X nZ|d
|
kr¤d d d d g¡}tj||d t|| W 5 Q R X nt|| }t |j
|
|¡ |dkrd d ddddd| dg¡}tj||d t||dd W 5 Q R X n t||dd}t |j
|
|¡ d S )Nre Úcum)r r r r r r ú|zCategorical is not orderedú*function is not implemented for this dtypezCannot perform z with non-ordered Categorical©Úmatch)r+ rt ru r z-category type does not support sum operationsz[Cc]ould not convertz0can't multiply sequence by non-int of type 'str'©r r z category type does not supportzcan't multiply sequenceFr} ) r( Ú
|
startswithÚNotImplementedErrorÚ TypeErrorÚjoinÚpytestÚraisesr? r- Úassert_index_equalr; ) rr rT rs r r rU Ú exceptionÚmsgr1 r r r r í sH
|
|
ýÿýÿ
|
|
úÿ
|
zTestNumericOnly._checkN)rL Ú
|
__module__Ú __qualname__r ÚfixturerT ÚmarkÚ parametrizer r r r r r r r r r r rW c s
|
|
#
|
|
|
|
|
rW c @ s> e Zd Zejdd Zejdd Zejdd Zdd Zd S )
|
ÚTestGroupByNonCythonPathsc C s2 t dddgdtjdgdtjdggddd gd
|
}|S ) Nr% r& ÚfooÚbarr< ZbazÚAÚBÚCr: ©r r r' rq r r r rT # s
|
þz TestGroupByNonCythonPaths.dfc C s | d¡}|S )Nr© ©r( )rr rT rU r r r rU + s
|
z TestGroupByNonCythonPaths.gbc C s |j ddd}|S )Nr© F©Úas_indexr )rr rT Úgnir r r r° 0 s z TestGroupByNonCythonPaths.gnic
|
C sº t ddgdd}tdgdddd d
|
d d d ggdgd ttdgd}tddtjdddddgdtjtjtjtjtjtjtjgg||d}| ¡ }t ||¡ |
|
¡ }| ¡ }t ||¡ d S )Nr% r< r© ry rª Úcountrt Ústdr ú25%ú50%ú75%r r r^ ©ÚlevelsÚcodesr" r# ç r{ ) r r rn Úranger r r' Údescriber- rP rR )rr rT rU r° Zexpected_indexZ expected_colr2 r1 r r r Ú test_describe5 s" þ þú z'TestGroupByNonCythonPaths.test_describeN) rL r¡ r¢ r r£ rT rU r° r¼ r r r r r¦ s
|
|
|
r¦ c C sä t ddtjgdtjdgdddggdddgd } t dtjgtjdgddggddgd }| d¡ ¡ }t ||¡ | jdd
|
d ¡ }t ||¡ | d¡jdd }| jdd }t ||¡ | d¡jdd }| jdd }t ||¡ d S ) Nr% r& r_ r< é r© rª r« r: Fr® r= )r r r' r( r r- rP r )rT r2 r1 r r r Útest_cython_api2L s .& r¾ c C sº t j d¡} t j| d d d
< t| }t jjdddd t¡}t j|d d d
< | |¡ ¡ }| |¡
|
t j ¡}t ||¡ tt j dd¡}| |¡
|
t j ¡}| |¡ ¡ }t ||¡ d S )Nr9 r& r r8 ©Úsizeé é )r rI rK r' r rJ ro rg r( ru r) r Z nanmedianr- rP )ÚarrrT Úlabelsr1 ÚexpÚrsZxpr r r Útest_cython_medianb s rÇ c C sj t tj ddd¡}tddd}t |d |¡}|j|| d ¡ }|j|| d dd ¡}t
|
||¡ d S ) Nr é, éô é7 r )Úobservedc S s | ¡ S rE )ru r@ r r r rC { rD z(test_median_empty_bins.<locals>.<lambda>) r r rI rJ rº rp Zcutr( ru r) r- rP )rË rT ZgrpsZbinsr1 r2 r r r Útest_median_empty_binst s rÌ r Zint8Úint16Úint32r Úfloat32r Zuint64z method,datar rT r% ©ÚaÚbr& r< r r½ r r r± )rT Úout_typec C s¸ t ddddddddddddg}|j | ¡|d< d|krHg |d< d|krZ|d }n| }|d }t |}|j |¡|d< |jd
|
d d | d
|
¡}t|||d }t ||¡ d S ) Nr% rÐ r& r< r½ rÒ ÚargsrÓ rT rÑ T)rG )r rÒ ro rS r( r? r- rP ) r rs ÚdatarT rÓ rÅ Zdf_outZgrpdÚtr r r Ú%test_groupby_non_arithmetic_agg_types s "ÿ
|
|
r× r z2011-01-15 12:50:28.502376z2011-01-20 12:50:28.593448)l m;-L¼l m;-L¼c C s¼ t d| d dd| d dg}d| d id| d id| d id| d i| d dgdddid}| ¡ D ]L\}}d|krg |d< | d ¡}t|||d }|jd j|d ksjtqjd S )
|
Nr% r rÐ r2 )r2 rÔ r& )r r r r Únthr± rÔ rÑ )r Úitemsr( r? ÚilocrÒ rO )r rT Zgrp_exprs rÕ r0 Úresr r r Ú2test_groupby_non_arithmetic_agg_int_like_precision§ s
|
|
|
|
ú
|
rÜ z func, valuesÚidxmin)Úc_intÚc_floatÚc_dateÚidxmaxr~ TFc C s8 t ddddgddddgddd d
|
gd d d dgd}t |d ¡|d< |d j d¡|d< |d |d jd |d< |d j d¡|d< |d d¡|d< |d d¡|d < t| d ¡| |d }t |t
|
ddgd d d }|rì|j dgd!}n$|d |d< |d |d< |d |d< |d |d< |d |d < t ||¡ d S )"Nr© rª r% r& r< r½ g®Gáz@g=
|
×£p=@gR¸
ëQ @gÍÌÌÌÌÌð?Z2019Z2018Z2016Z2017)rz rÞ rß rà rà z
|
US/PacificZ c_date_tzr Z c_timedeltaÚWZc_periodrÞ r Z c_Integerrß r Z
|
c_Floatingrz r} ry ©r| r: )r rp Ú to_datetimeÚdtZ tz_localizerÚ Z to_periodro r? r( r rH r- rP )ÚfuncÚvaluesr~ rT r1 r2 r r r Ú$test_idxmin_idxmax_returns_int_typesÈ s.
|
|
|
|
|
üÿ rè c
|
C sØ t tj dd¡ddddgd} dd d
|
dd d
|
dd d
|
dg
|
| d< | d¡}|jdd }| jd d
dd
f jdd }|j d¡}t
|
|| | d¡¡ t d dd | d< | d¡}d}t jt|d |jdd W 5 Q R X d S )Né
|
r½ r© rª r« ÚDr: r% r& r< r= z
|
2016-01-01r` ÚEz7reduction operation 'argmax' not allowed for this dtyper )r r rI rK r( rá rÚ r| Zget_level_valuesr- r. Z droplevelr
|
r r r )rT rU rÛ ZaltZindexerZgb2r r r r Útest_idxmin_idxmax_axis1ð s
|
|
rì c
|
C sÎ |dkrt d¡ |dkr4d}| j t jj|d¡ ttj dd¡dd d
|
d gd }d |d< ddddddddddg
|
}|
|
|¡}t ||}t ||}ddi} |d k r¨|| d< d}
|
d} |d k rê||
|
krêd}t j t|d ||| W 5 Q R X nà|| kr d}t j t|d ||| W 5 Q R X n¬|d ks.|sp|dkrpd} t j tdd | ¡ d d ||| W 5 Q R X nZ||| } |r|jdd jn|j}t ||| j}|d kr¾|s¾| t¡}t | |¡ d S ) N©rá rÝ z5idxmax and idx_min tested in test_idxmin_idxmax_axis1)ÚcorrwithÚskewzDGH#47723 groupby.corrwith and skew do not correctly implement axis=1©Úreasonré r½ r© rª r« rê r: rA rë r% r& r< r> r~ )r r ÚdiffÚfillnaÚ
|
pct_changeÚrankÚshift) r r rò rô rõ rö r r rÝ rá ró ú1got an unexpected keyword argument 'numeric_only'r z)got an unexpected keyword argument 'axis')ró rö )ú"not supported between instances ofz2can't multiply sequence by non-int of type 'float'úunsupported operand typeú(r ú)rö )r ÚskipÚnodeÚ
|
add_markerr¤ Úxfailr r rI rK r( r? r r r r rH ÚTro rg r- Ú assert_equal)r Ú groupby_funcr~ r rT ÚgroupsrU rs rÔ ÚkwargsZno_argsZhas_axisZmsgsr1 Z df_expectedr2 r r r Útest_axis1_numeric_only sD
|
|
|
|
|
|
|
r c C s¾ t dgd dd} | d¡d ¡ }| jdddd d d
|
¡}d|_t ||¡ t dgd dd} | d t¡| d< | d¡d ¡ }| jdddd d d
|
¡}d|_t ||¡ d S ) NrÒ ré r& ©ÚkeyÚvaluer r F©Z
|
group_keysc S s | ¡ S rE ©r r@ r r r rC Q rD z&test_groupby_cumprod.<locals>.<lambda>éd c S s | ¡ S rE r
|
r@ r r r rC X rD ) r r( r r, rz r- r. ro rg )rT Úactualr2 r r r Útest_groupby_cumprodL s r c C sx t dgd dd} | d¡d ¡ }tdddd gdd
|
}t ||¡ | jdd d d d d ¡}d|_t ||¡ d S )NrÒ r½ i r r r l d( l Iú5 l Fµx:^ ry Fr c S s | ¡ S rE r
|
r@ r r r rC h rD z/test_groupby_cumprod_overflow.<locals>.<lambda>)r r( r r r- r. r, rz )rT r r2 Z numpy_resultr r r Ú test_groupby_cumprod_overflow] s
|
þ ÿr c C s` t ddtjdgdddgd} | d¡jddd}t dtjtjgddd gd
|
}t ||¡ d S ) Nr% r& r$ ©rÑ rÒ ÚcrÑ TF)r~ ÚskipnarZ ©rÒ r )r r r' r( r r- rP ©rT r1 r2 r r r Ú1test_groupby_cumprod_nan_influences_other_columnsn s
|
ýÿ r c O s ddl m} || ddi|S )Nr )ÚsemZddofr% )Z scipy.statsr )rÔ r r r r r Ú scipy_sem| s r z op,targoprt ru r² Úvarr+ r c C s
|
| j d S )Nr ©rÚ r@ r r r rC rD rC c C s
|
| j d S )Néÿÿÿÿr r@ r r r rC rD r )Zmarksc C sZ t tj d¡}tjjdddd t¡}t| |¡| }| |¡ |¡}t
|
||¡ d S )Nr9 r r8 r¿ ) r r rI rK rJ ro rg r? r( r) r- rP )ÚopZtargoprT rÄ r1 r2 r r r Útest_ops_general s
|
r c C s~ d} t jtdd tjt| dgd}W 5 Q R X | d¡}|dg ¡ }|d ¡ ¡ }t ||¡ |d
|
¡ ¡ rzt d S )Nz,Date,app,File
|
-04-23,2013-04-23 00:00:00,,log080001.log
|
-05-06,2013-05-06 00:00:00,,log.log
|
-05-07,2013-05-07 00:00:00,OE,xlsxzCould not infer formatr r )Z parse_datesÚDateÚFile) r- rQ Ú UserWarningrp Zread_csvr r( r Úto_framerP ZisnaÚanyrO )ÚrawrT rU ÚrÚer r r Útest_max_nan_bug s
|
r$ c C sì t ddddddddd d
|
g
|
} t td }| |¡}| d¡}t dddd
|
dd gt td dddddd gg¡d}t ||¡ t ddddddddddg
|
} | |¡}t ddddddgt td dddd ddgg¡d}t |jddd|¡ d S ©Nr% r< r r] r& r_ r r½ é ré Z
|
aaaaabbbbbZaaabbbr^ rã r )Zkeep)r rn r( Únlargestr Ú from_arraysr- r. ©rÑ rÒ rU r" r# r r r Ú test_nlargest« s
|
|
þ
|
þr* c C sF t j d¡} tddd}|ddgg}tj|ddgd }t| d
|
¡|d }| d¡ d ¡}tj
|
|d |d df|d |d df|d |d df|d |d df|d |d df|d |d df|d |d df|d |d df|d |d df|d |d dfg
|
dddgd }ddddddd d d d g
|
}t||d }t j ||d d!d" d S )#NiÍ[Z20180101ré r` ÚoneÚtwor Úsecond)Únamesé rã r% r r& r< r½ r r& r] r^ r_ g-rö´@g-µ`¦wý?g@Ñf¡|ë?gép¿ö?g=±J-î?g©],l;óà?g]£f$ê?gM8r ë¿gÅ«4YØ?g¯
|
F;±é¿Fgü©ñÒMbP?)Ú check_exactZrtol) r rI Z RandomStater
|
r Z from_productr rK r( r' Ú from_tuplesr- r. ) ZnprZdtsÚ iterablesÚidxr/ r1 Zexp_idxZ
|
exp_valuesr2 r r r Útest_nlargest_mi_grouper¿ sB ö óö r4 c C sì t ddddddddd d
|
g
|
} t td }| |¡}| d¡}t dddddd gt td dddd dd gg¡d}t ||¡ t ddddddddddg
|
} | |¡}t ddddddgt td ddddd dgg¡d}t |jddd|¡ d S r% )r rn r( Ú nsmallestr r( r- r. r) r r r Útest_nsmallestì s
|
|
þ
|
þr6 z data, groupsrs r' r5 c C s |d k rt j| |d} |dkr*tt| } t| dd}t| |¡|dd}t|trft j|t jdn|}t| t
|
||j g¡dd}t ||¡ d S )N©r r' rÑ ry r& )Ún©r| rz )r Úarrayrn Úreversedr r? r( Ú
|
isinstanceÚint_r r( r| r- r. )rÕ r r rs Úserr1 Zexpidxr2 r r r Ú test_nlargest_and_smallest_noop s
|
r? ræ r r c C s t dddgdddgd}| d¡}d}tjt|d t|| ddd W 5 Q R X tjt|d t|| dd W 5 Q R X d S ) Nr% r& r< ©r© rª r© z+numpy operations are not valid with groupbyr )r§ )r r( r r r r? )ræ rT Úgr r r r Útest_numpy_compat s
|
rB c
|
C sx | d }| d }t ddddddddgddddddddgd}ddddddddg}| |¡}t d|i |¡}| d¡ ¡ }t ||¡ |jdd d
|
j d d ¡ ¡ }t ||¡ ||j dd gdf< |d |j ddgdf< ||j ddd dgdf< |d |j ddgdf< | d¡ ¡ }tj||dd |jdd d
|
j dd ¡ ¡ }tj||dd | ddi¡}t
|
j |j dddd gdf< t dt
|
j dt
|
j dt
|
j dt
|
j dgi}| d¡ ¡ }t ||¡ |jdd d
|
j dd ¡ ¡ }t ||¡ t dgt dg¡d}tt d¡dgdd}| d¡d ¡ }t ||¡ t dddgdddgd}| d¡j ¡ }tdddgdd}t ||¡ d S )Nr r% r& r< r½ r@ rª r© Fr c S s | ¡ S rE ©r r@ r r r rC 0 rD z test_cummin.<locals>.<lambda>r& r r] T)r0 c S s | ¡ S rE rC r@ r r r rC ; rD rg c S s | ¡ S rE rC r@ r r r rC G rD Ú2001rÐ rÒ r9 rÑ ry )r ro r( r r- rP rª r, r Úlocr r' rp rä r r. rÒ )r! r r Úbase_dfZ expected_minsrT r2 r1 r r r Ú test_cummin# sF .
|
ÿ$ ÿ rG r r ZUInt64rg Úbooleanc C s t ddddddddgtjgd d}|d |¡|d< | d¡}t dtjgd i|d}t|| }t ||¡ t|d | ¡ }t ||¡ d S )Nr% r& r^ r@ rª r© r7 ) r r r' ro r( r? r- rP r )rs r rF r0 r2 r1 r r r Ú test_cummin_max_all_nan_columnY s &
|
rI c
|
C sD | d }| d }t ddddddddgddddddddgd}ddddddddg}| |¡}t d|i |¡}| d¡ ¡ }t ||¡ |jdd d
|
j d d ¡ ¡ }t ||¡ ||j dd gdf< ||j ddd dgdf< | d¡ ¡ }t ||¡ |jdd d
|
j dd ¡ ¡ }t ||¡ | ddi¡}t
|
j |j dddd gdf< t dt
|
j dt
|
j dt
|
j dt
|
j dgi}| d¡ ¡ }t ||¡ |jdd d
|
j dd ¡ ¡ }t ||¡ t dgt dg¡d}tt d¡dgdd}| d¡d ¡ }t ||¡ t dddgdddgd}| d¡j ¡ }tdddgdd}t ||¡ d S )Nr r& r% r< r½ r@ rª r© Fr c S s | ¡ S rE ©r r@ r r r rC u rD z test_cummax.<locals>.<lambda>r& r] c S s | ¡ S rE rJ r@ r r r rC ~ rD rg c S s | ¡ S rE rJ r@ r r r rC rD rD rÐ rÒ r9 rÑ ry )r ro r( r r- rP rª r, r rE r r' rp rä r r. rÒ )r! r r rF Z expected_maxsrT r2 r1 r r r Ú test_cummaxh sB .
|
ÿ $ ÿ rK c C sZ t dd tdD } td| | d¡d}| d¡}| ¡ }|dd g }t ||¡ d S )
|
Nc S s g | ]}t jj| qS r )rp ÚNaTÚ_value)Ú.0r8 r r r Ú
|
<listcomp> s z:test_cummax_i8_at_implementation_bound.<locals>.<listcomp>r r% zM8[ns]©r© rª r« r© rª r« )r rº r Úviewr( r r- rP )r> rT rU rÛ rÅ r r r Ú&test_cummax_i8_at_implementation_bound s
|
rR zgroups,expected_datac C sj t dtdd dg|di}| ¡ }| |¡d }t|| dd}t||dd}t ||¡ t ||¡ d S )NrÑ r% r& r7 F©r )r rz )r r Úcopyr( r? r- rP r. ) rs r r Z expected_datarT ÚorigrU r1 r2 r r r Útest_cummin_max_skipna¨ s rV c C st t tjddgdddgd}| dddg¡ddg }t|| dd}t tjtjtjgdddgd}t ||¡ d S )Nr# rÐ r% rÑ rÒ FrS )r r r' r( r? r- rP )rs rT rU r1 r2 r r r Ú$test_cummin_max_skipna_multiple_colsÁ s
|
rW z dtype,val)r l c C sT |t jg}tddg|d|d}| d¡}| | ¡}td|i|d}t ||¡ d S )Nr% )ÚgrprÒ r7 rX rÒ )rp ÚNAr r( Z transformr- rP )rs r ÚvalrÕ rT r0 r1 r2 r r r Ú#test_nullable_int_not_cast_as_floatÍ s
|
|
|
r[ zin_vals, out_valsr iúÿÿÿgÍÌÌÌÌÌ @é c C s ddddddddd d
|
d g d d d d d d dddddg | d}t |}| d¡jj}ttddd}t||dd}t ||¡ | dg¡j dd ¡}t ||¡ d S )NÚ1Ú2Ú3Ú4Ú5Ú6Ú7Ú8Ú9Ú10Ú11rÑ rÒ r ÚdrP rª Úabcdry r« ©r| rÕ rz c S s | j S rE )Úis_monotonic_increasingr@ r r r rC ý rD z.test_is_monotonic_increasing.<locals>.<lambda>)
|
r r( r« rk r rn r r- r. r, ©Zin_valsZout_valsZ source_dictrT r1 r| r2 r r r Ú test_is_monotonic_increasingÝ s ý rm ré r_ r] éýÿÿÿc C s| ddddddddd d
|
d g d d d d d d dddddg | d}t |}| d¡jj}ttddd}t||dd}t ||¡ d S )Nr] r^ r_ r` ra rb rc rd re rf rg rÑ rÒ r rh rP rª ri ry r« rj ) r r( r« Zis_monotonic_decreasingr rn r r- r. rl r r r Ú test_is_monotonic_decreasing s ýro c C s | j dd}| ¡ d S )Nr )Úlevel)r( r» )Zmframer0 r r r Útest_apply_describe_bug& s rq c C sr t ¡ } | dd dd g¡}| ¡ }t j|d | ¡ dd t j|d | ¡ dd t j|d | ¡ dd d S ) Nc S s | j S rE ©Úyearr@ r r r rC - rD z/test_series_describe_multikey.<locals>.<lambda>c S s | j S rE ©Úmonthr@ r r r rC - rD rt F)Z check_namesr² r )r- ÚmakeTimeSeriesr( r» r. rt r² r )Útsr0 r1 r r r Ú test_series_describe_multikey+ s rx c C s@ t ¡ } | dd ¡}| dd ¡}| ¡ ¡ }t ||¡ d S )Nc S s | j S rE rt r@ r r r rC 6 rD z-test_series_describe_single.<locals>.<lambda>c S s | ¡ S rE )r» r@ r r r rC 7 rD )r- rv r( r, r» Ústackr. )rw r0 r1 r2 r r r Útest_series_describe_single4 s
|
rz Úkey1Úkey2c C sÚ t dddddgdddddgdddddgd}|j|| d d
|
}| ¡ }t dddgd d d gd d d gtjtjdgd d dgd d d gd d d gd d dgd d dgd }t|dkr¼| dd|d ¡ | rÊ| |¡}t ||¡ d S )Nr+ r, Zthreer% r& r½ r& )r{ r| Úfoo2r® r} r" r$ rX r# rY rZ ) r{ r± rt r² r r³ r´ rµ r r| r{ )
|
r r( r» r r' rM ÚinsertrS r- rP )r¯ r5 rT rU r1 r2 r r r Ú test_series_describe_as_index< s2 ýÿ ÷ÿ
|
r c C s@ | j d d
dgf | d ¡}| dd ¡}|jjdks<td S )Nr« r© c S s | ¡ S rE )rt r@ r r r rC ^ rD z(test_series_index_name.<locals>.<lambda>)rE r( r) r| rz rO )rT r0 r1 r r r Útest_series_index_name\ s r c C s | dd dd g¡}| ¡ }g }| D ]Z}|| ¡ }t|g|jgdgt|j tt|jgd}t|j||jd}| |¡ q&t
|
j |dd}t ||¡ | j ddddd dd}| ¡ }| ¡ j}tddg|jgddddgtt|jgd|_t ||¡ d S )
|
Nc S s | j S rE rr r@ r r r rC c rD z.test_frame_describe_multikey.<locals>.<lambda>c S s | j S rE rt r@ r r r rC c rD r r¶ )r; r| r% r= )r© rª r« rê )r( r» r r; rM rº r rç r| Úappendrp Úconcatr- rP r ) Ztsframer0 r1 Z desc_groupsÚcolre Z group_colr2 ZgroupedTr r r Ú test_frame_describe_multikeyb s*
|
þ
|
þr c C s´ t dddddgd dddd d
|
gd d d d ddgd d} dddgd | d< | jddid}d}tjt|d | d¡ ¡ W 5 Q R X tjt|d | d¡ ¡ W 5 Q R X d S )Nr% r& r< r½ r ré r/ é é( r8 r éÈ i, i rÉ )rA ÚyÚz)r r r% )r r% r )r% r r Úkr r: z*Names should be list-like for a MultiIndexr )r Úrenamer r Ú
|
ValueErrorr( r» )Zdf1Zdf2r r r r Ú test_frame_describe_tupleindex} s ýÿr c C sÜ t dd ddt dd ddt dd ddi} t dd ddt dd ddt dd dd i}t| |d
|
}| d ¡j ¡ }||jdk j ¡ j ¡ ||jdk j ¡ j ¡ g}t|tddgd d d dddddddgd}t
|
||¡ d S )Nz2011-01-06 10:59:05rx ia z2011-01-06 12:43:33ic z2011-01-06 12:54:09i /hYl r T i áõ)ÚPRICEÚVOLUMEr ry r± rt r² r r³ r´ rµ r r{ ) r r r( r r» r rç Útolistr r- rP )ZpricesZvolumesrT r1 rÕ r2 r r r Ú$test_frame_describe_unstacked_format s2
|
|
|
ý
|
|
|
ýþýr zZignore:indexing past lexsort depth may impact performance:pandas.errors.PerformanceWarningr¯ Úa1Úa2c C s$ t ddddddgddddddgddddddgd d
|
d d d dgdddddgdd}|dgkrf|jdd}t ddddddd d ddddddd d g¡ d dg¡j}d d g|j_t|dkrâtddgddggd dgd dggddgd |_ nt
|
ddgdd |_ | s| ¡ }|j || d! ¡ }t ||¡ d S )"Néc éX r% r& r< r½ r r& ré r/ r
r r8 é< )r r rÒ r r r rÒ F)r; rT r: )rÒ r± r$ r$ )rÒ rt rY r# )rÒ r² r" r" )rÒ r rX r" )rÒ r³ rv rw )rÒ r´ rY r# )rÒ rµ g @g @)rÒ r rZ r$ r )r· r¸ r. ry r® )r rH Z from_recordsrS r r; r. rM r r| r rR r( r» r- rP )r¯ r5 rT r2 r1 r r r Ú0test_describe_with_duplicate_output_column_names© sZ ü
|
ø
|
|
ðÿìÿ ÿ
|
r c sÄ t ddddgg} ddddg| _| | d ¡}|jg d}dddd d
|
d g fd d dD }tj|dd}tddg gddg ddg ddg dttd gd|_dg|j _
|
t ||¡ d S )Nr r% r& r< )Ú percentilesr± rt r² r r´ r c s, g | ]$}t d |tj|||ggdg dqS )r" r% r{ r¬ )rN rZ r: r r rO ï s ÿz3test_describe_duplicate_columns.<locals>.<listcomp>)r¹ r# r$ r= r& r¶ ) r r; r( r» rp r r rn rº r| r. r- rP )rT rU r1 Úframesr2 r r: r Ú test_describe_duplicate_columnsç s
|
þ
|
,þ
|
r c C sD t dddddgdddddgd} | d¡d ¡ d d
|
ks@td S ) Nr© ij i io i` l @Ü~ÿÿ )ÚuserÚ connectionsr r l ,fÌL )r r( rt rO )rT r r r Ú test_groupby_mean_no_overflowü s þÿr rç Úfunctionc C s¼ |dkr dnd}t j|gd td}tdddgdd d
|
}td |i|d d ¡}t| d d d¡}t|| }t ||¡ |
|
|¡}t ||¡ |
|
|g¡}t d |fg¡|_ t ||¡ d S )Nr g à?rw r< r7 r% r& rÑ r ©rz r rÒ rã r )r r: rg r r ro r( r? r- rP r) r r1 r; )rç r Úoutputrà r3 r2 r r1 r r r Ú,test_apply_to_nullable_integer_returns_float s
|
r¡ c C s¸ t dddgdddgdddgddd} | d¡}tdddgddd}|d jdd
|
}ttjgd d|d d }t ||¡ |jdd
|
}t tjgd tjgd d d|d}t ||¡ d S )Nr r% r& r r r7 rÑ r rÒ ©Z min_countr< )r r| rz r )r r| )
|
r r( r r+ r rp rY r- r. rP )rT r0 r3 r1 r2 r r r Ú0test_groupby_sum_below_mincount_nullable_integer& s &
|
$r£ c C sf t t td¡ddgd d} | d¡d ¡ }tt ddg¡dtddgdd d
|
}t ||¡ d S ) Nré r© rª r )ÚtimeÚcatr¥ r¤ r½ ry )rz r| )
|
r rp Z to_timedeltarº r( rt r r r- r. r r r r Útest_mean_on_timedelta5 s ÿr¦ c C sÐ t ddddgt d¡t d¡t d¡tjgd} tjdd}| d ¡}| ¡ }t d
|
||gitddgd d d }t ||¡ |d
|
¡ }t ||d
|
¡ |d
|
jdd }t
|
|tjgdd
|
|j d}t ||¡ d S )Nr% r& Z1dZ2dZ3drÐ r< )ÚdaysrÑ rÒ ry rã r¢ zm8[ns])r rz r| ) r rp r rL r( r+ r r- rP r. r r| )rT Ztd3rU rÛ r2 r r r Ú#test_groupby_sum_timedelta_with_nat? s
|
þÿ
|
r¨ zkernel, has_arg)ÚallF)r F)ÚbfillF)ÚcorrT)rî T)ÚcovT)r T)r T)r T)r T)rò F)ÚffillF)ró F)r T)rá T)rÝ T)r T)r T)rt T)ru T)r T)rØ F)ÚnuniqueF)rô F)r T)ÚquantileT)r T)rï T)r² T)r+ T)r Tc C s t ddgddgddgdtg d}t| |}|tjkr<i nd|i}| |¡}t|| }|r|dkr|||} d| jks|tn| d ks| d
|
kr´|tjkr´|||} d| jks²tnä|sÂ| d kr |dk sÎt| d ¡rÜt
|
nt }
|
d dddddddg¡} t j|
|
| d ||| W 5 Q R X nz|sV|tjk rVt jt dd ||| W 5 Q R X nB| dksdt|tjksttt jt dd ||| W 5 Q R X d S )Nr% r& r r& )r r Úa3rÒ r~ TrÒ r )r r© rª r ró rØ r® rí r r únot allowed for this dtypez must be a string or a numberú+cannot be performed against 'object' dtypesz!must be a string or a real numberrù rø r r r÷ )rò rô )r Úobjectr r Ú
|
no_defaultr( r? r; rO r r r r r r ) ÚkernelÚhas_argr~ r5 rT rÔ r rU rs r1 r r r r r Útest_numeric_onlyW sR *$
|
|
|
|
þûú
|
ùÿ ÿr· c C s |dkr"d}|j tjj|d¡ dddg}tdddg| d}| |¡}t||}tdddg}| |¡} t| |}
|
t||} d} d} || krÚ| t krÚ|d kr¤d
|
}n|d kr²d }nd }tj
|
t |d || W 5 Q R X n:| t kr|| }|
|
| }|| kr| t ¡}t ||¡ d}||krPd}tj
|
t |d || ddi W 5 Q R X nn| t krd ddddg¡}tj
|
t |d || ddi W 5 Q R X n(|| ddi}|| ddi}t ||¡ d S )Nrî z,corrwith is not implemented on SeriesGroupByrð r r% r7 ) r« r¬ r r r r rá rÝ r¯ )r r rØ rª r rö r+ rò rô r rt ru r r r rí r± r¯ r² z!is not supported for object dtyper )r r r rt ru r r r¯ r rï r² r+ r r r r r r÷ r~ Tr z@SeriesGroupBy.sem called with numeric_only=True and dtype objectz=Series.skew does not allow numeric_only=True with non-numericz7cum(sum|prod|min|max) is not supported for object dtypezCCannot use numeric_only=True with SeriesGroupBy\..* and non-numericF)rý rþ r r¤ rÿ r r( r? r r³ r r ro r- r. r )r r r r Zgrouperr> rU rs Z expected_serÚ expected_gbZexpected_methodrÔ Zfails_on_numeric_objectZ
|
obj_resultr1 r2 Zhas_numeric_onlyr r r Ú"test_deprecate_numeric_only_series¹ sZ
|
|
|
|
|
|
|
|
|
|
|
üÿr¹ r g¹?gÉ?g333333Ó?r© )r ÚincludeÚexcluderf c C sê t dddggdddg| d}|d t¡|d< |d t¡|d< |jd d
d¡jf |}| d¡jf |jd d
|
jd d
}t ||¡ |jd d
d¡j
|
jf |}| d¡j
|
jf |jd d
|
jd d
}t g |_ t ||¡ d S ) Nr% r& r< r© rª r« )r; r r T)rH ) r ro rf rg rÚ r( r» rR r- rP rª r r| )r r rT r1 r2 r r r Útest_groupby_empty_dataset$ s $ &
|
r¼ c C sj t dddgdddgd} | d¡j| dd}td d
|
d d d dgdd}ttjgd |d}t ||¡ d S )Nr% r& r< r] r½ rÐ rÑ r= )r% r )r% r% )r% r& )r& r& )r& r )r& r% )rÑ N)rÕ rz r& rã ) r r( rî r r r r' r- r. )rT r1 r| r2 r r r Útest_corrwith_with_1_axis= s þr½ c C sd t g g g d dddg¡}|jdddgdd}t|| }t| |}|| j}|j}t ||¡ d S )Nr rÑ rÒ r Fr )r rS r( r? r r| r- r )r rT rU rs rÔ r1 r2 r r r Ú,test_multiindex_group_all_columns_when_emptyI s
|
|
|
r¾ c C sÒ |dkr"d}| j tjj|d¡ tdddgdddgd d
|
d ggtd d }t||}|jd|d}t ||| }|j
|
dddgdd}t||} |jd|d}
|
t |
|
|| } |dkrÂ| j ddid } t || ¡ d S )Nrî z:GH#50845 - corrwith fails when there are duplicate columnsrð r% r< r& r½ r] r& r r^ Zabbr: rÑ r® rÒ r r= )rÀ ZngroupZcumcount)rý rþ r r¤ rÿ r rn r r( r? Zset_axisr r- r ) r r r¯ r rT rÔ rU r1 Z expected_dfZ expected_argsr¸ r2 r r r Útest_duplicate_columnsU s (
|
|
r¿ )N)pr* Úior Únumpyr r Z pandas._libsr Z pandas.errorsr Zpandasrp r r r r r r
|
Zpandas._testingZ_testingr- Z pandas.corer Zpandas.tests.groupbyr Z pandas.utilr Útdr£ rÎ r rÏ r r! r3 r¤ r¥ r r r+ rV rW r¦ r¾ rÇ rÌ r× rÜ rè rì r r r r r rt ru r² r r rÀ r Zskip_if_no_scipyr r$ r* r4 r6 ZALL_INT_NUMPY_DTYPESr? rB rG rI rK rR rV rW Z skip_if_32bitr r[ Úinfr' rm ro rq rx rz r r r r r Úfilterwarningsr r r rY r¡ r£ r¦ r¨ r´ r· Úboolrf rg r³ r¹ r¼ r½ r¾ r¿ r r r r Ú<module> s þ
|
=- ÿ ûþ
|
þûþ
|
|
þþ
|
F
|
|
ôþ
|
-"þ
|
6 4 ýþ
|
ÿ $$
|
þ$
|
þöþ
|
$,
|
þ$
|
þöþ
|
|
ÿ7 þ ûþ
|
|
áþ$<
|
jýþ
|
