U
|
ý°dÄ ã @ s¸ d Z ddlmZmZ ddlZddlZddlZddlm Z m
|
Z
|
m Z m Z m Z mZmZmZ ddlmZ ddlmZ G dd dZG dd d ZG d
|
d d ZG d d d Zdd Zdd ZdS )zG
|
test where we are determining what we are grouping, or getting groups
|
é )ÚdateÚ timedeltaN)ÚCategoricalIndexÚ DataFrameÚGrouperÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_range)ÚGroupingc @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestSelectionc C sn t ddggddgd}| d¡}tjtdd |dg W 5 Q R X tjtd d |ddg W 5 Q R X d S )
|
Né é ÚAÚB©Úcolumnsz"Columns not found: 'C'"©ÚmatchÚCz^[^A]+$)r ÚgroupbyÚpytestÚraisesÚKeyError)ÚselfÚdfÚg© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_grouping.pyÚtest_select_bad_cols s
|
z"TestSelection.test_select_bad_colsc C s² t ddddgtdtddtdddgd }d
|
}tjt|d | d¡ W 5 Q R X tjt|d | ddg¡ W 5 Q R X | d¡}| ¡ }|jjd kst |jj
|
d ks®t d S )Nr r r é r é r é )r Údataz!Grouper for 'A' not 1-dimensionalr r é ) r Úranger r Ú
|
ValueErrorr Úcountr ZnlevelsÚAssertionErrorÚsize)r r ÚmsgÚgroupedÚcr r r Ú'test_groupby_duplicated_column_errormsg* s
|
ÿ
|
z5TestSelection.test_groupby_duplicated_column_errormsgc C st | d¡j ¡ }| d¡d ¡ }t ||¡ d|d< | d¡jdd}| d¡dddg tj¡}t ||¡ d S )Nr r g ø?ÚmeanT©Z numeric_onlyÚD) r r ÚsumÚtmÚassert_series_equalr/ ÚaggÚnpÚassert_frame_equal©r r ÚresultÚexpectedr r r Útest_column_select_via_attr; s z)TestSelection.test_column_select_via_attrc
|
C s¼ t ddddddddgddddddddgtj d¡tj d¡tj d¡d}| d¡d d
|
g ¡ }| d¡|jd d
¡ }|jd d
dd d
|
gf d¡ ¡ }t ||¡ t ||¡ d S ) NÚfooÚbarÚoneÚtwoÚthreer# ©r r r r1 ÚEr r r1 r r! )
|
r r6 ÚrandomÚrandnr r/ r Úlocr3 r7 ©r r r9 Úresult2r: r r r Ú test_getitem_list_of_columnsE s
|
|
|
ûÿ
|
" z*TestSelection.test_getitem_list_of_columnsc C sÎ t tdd tj d¡tj d¡tj d¡d}| d¡|jdd
¡ }| d¡ddg ¡ }|jd d
dddgf d¡ ¡ }t
|
||¡ t
|
||¡ t j t d d
|
| d¡d ¡ W 5 Q R X d S ) NÚabcdr r# )r r r! r" r r r% r! ú"Cannot subset columns with a tupler )r r! )r Úlistr6 rC rD r r r/ rE r3 r7 r r r' rF r r r Ú!test_getitem_numeric_column_namesX s
|
|
|
|
üÿ " z/TestSelection.test_getitem_numeric_column_namesc C s0 t jtdd | d¡d ¡ W 5 Q R X d S )NrJ r r )r r1 )r r r' r r/ ©r r r r r Ú+test_getitem_single_tuple_of_columns_raisesn s z9TestSelection.test_getitem_single_tuple_of_columns_raisesc
|
C s¤ t ddddddddgddddddddgtj d¡tj d¡tj d¡d}| d¡d ¡ }|jd d
dd gf d¡ ¡ }|jd d
d
|
f }|}t ||¡ d S ) Nr< r= r> r? r@ r# rA r r r )
|
r r6 rC rD r r/ rE Úilocr3 r4 )r r r9 Zas_frameZ as_seriesr: r r r Útest_getitem_single_columns s
|
|
|
ûÿ
|
z(TestSelection.test_getitem_single_columnc sN t ddd dD i d¡} fdd¡}|j}|j}t ||¡ d S )NÚTuplesc s s, | ]$}t j d dd¡D ]}||fV qqdS ©r% é N©r6 rC Úrandint©Ú.0ÚxÚyr r r Ú <genexpr> s zJTestSelection.test_indices_grouped_by_tuple_with_lambda.<locals>.<genexpr>©r r c s j | df S ©Nr ©rO ©rX ©r r r Ú<lambda> ó zITestSelection.test_indices_grouped_by_tuple_with_lambda.<locals>.<lambda>)r r Úindicesr3 Úassert_dict_equal©r ÚgbZ gb_lambdar: r9 r r_ r Ú)test_indices_grouped_by_tuple_with_lambda s ÿ
|
z7TestSelection.test_indices_grouped_by_tuple_with_lambdaN) Ú__name__Ú
|
__module__Ú __qualname__r r. r; rH rL rN rP rf r r r r r s
|
r c @ s e Zd Zej dejejej ej
|
ej g¡dd Z dd Z dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zej dddddgd d d d gd d! ed"D g¡d#d$ Zd%d& Zd'd( Zd)d* Zd+d, Z d-d. Z d/d0 Z d1d2 Z ej d3d4d5g¡d6d7 Z d8d9 Z!ej d3d4d5g¡d:d; Z"d<d= Z#ej d>d4ddddddddddg
|
gd5ddddddddddg
|
gg¡d?d@ Z$dAdB Z%dCdD Z&ej dEdFe'de(j)dGfdHe'de(j)e*g e(j)ddIdJfdKe'de(j)e*g e(j)ddIdJfg¡dLdM Z+dNdO Z,dPdQ Z-dRdS Z.dTS )UÚ TestGroupingÚindexc C s| t t d¡ dd¡tdd}|t||_|jtddd d d
|
¡ tt |j
|
¡ |_|jtddd d d
|
¡ d S ) Né
|
rS r ZABr ZabcdeF©Z
|
group_keysc S s | S ©Nr r^ r r r r` ª ra z7TestGrouping.test_grouper_index_types.<locals>.<lambda>c S s | S rn r r^ r r r r` ra ) r r6 ÚarangeÚreshaperK Úlenrk r ÚapplyÚreversedÚtolist)r rk r r r r Útest_grouper_index_types s
|
z%TestGrouping.test_grouper_index_typesc C sä t ¡ tdd }t|t ¡ }tj||gddgd}ttj ddd¡|d }|
|
¡ t dd
|
d t dd
|
d g¡ ¡ }tdgd d |_| t dd
|
dt dd
|
dg¡ ¡ }t ||¡ | t dd
|
dt dd
|
dg¡ ¡ }t ||¡ d S )Né )Údaysr< r= ©Únamesr éd éá ©rk ÚW©ÚkeyÚfreqÚint64©Údtype©Úlevelr r )r Útodayr r r Ú from_productr r6 rC rU Ú reset_indexr r r2 r r r3 r7 )r Zd0ÚdatesZ
|
date_indexr r: r9 r r r Ú test_grouper_multilevel_freq¯ s" ÿÿÿ ÿz)TestGrouping.test_grouper_multilevel_freqc C sj t ddddddgddddddgd}| d ¡}| ¡ }| td d
|
¡}| ¡ }t ||¡ | td dd ¡}| ¡ }t ||¡ | d d ¡}dddg|d < |jd d
d dgf }t ||¡ t ddddddgddddddgddddddgd}| d ¡ ¡ }| td d
|
g¡}| ¡ }t ||¡ | d dg¡ ¡ }| td d
|
tdd
|
g¡}| ¡ }t ||¡ | d tdd
|
g¡}| ¡ }t ||¡ | td d
|
dg¡}| ¡ }t ||¡ tt j
|
dddt j t dtdtdddgdddgdd}| tddd ¡ ¡ }td gtjtd gddd d}t ||¡ | tdd ¡ ¡ }|jdd ¡ }t ||¡ d S )!Nr r r r% r! rS r" ©r r r ©r )r Úaxisc S s | ¡ S rn )r2 r^ r r r r` Ù ra z8TestGrouping.test_grouper_creation_bug.<locals>.<lambda>r )r r r r# r r ÚabÚ20130101©Úperiodsr> r? r@ rx r| ÚMr é z
|
2013-01-31)r Úname©r
)r r r2 r r3 r7 rr rE r r6 ro r r rK r& r ÚpdÚ DatetimeIndexr
|
r4 )r r r r: r9 Úsr r r Útest_grouper_creation_bugË s\ &
|
.ÿ þþþ z&TestGrouping.test_grouper_creation_bugc C sZ t ddddddg¡}ddg|_tt d ¡d
|
d
|
d d d
|
d
|
gd |d }| dtddg¡jdd}| ¡ ddg¡jdd}t
|
||¡ | tdddg¡jdd}| ¡ ddg¡jdd}t
|
||¡ | d¡}| dtddg¡jdd}| ¡ ddg¡jdd}t
|
||¡ | tdddg¡jdd}| ¡ ddg¡jdd}t
|
||¡ d S )N)Úar )r r )r r% ©Úbr )r r )r r% ÚouterÚinnerr" r> r? r r| r r Tr0 ) r Ú from_tuplesry r r6 ro r r r/ r r3 r7 )r ÚidxZdf_multir9 r: Z df_singler r r Ú test_grouper_column_and_index s: ÿ
|
þ ÿ ÿ
|
ÿÿ ÿÿz*TestGrouping.test_grouper_column_and_indexc C sz ddg}t jddddg|d}tt d¡ d d
|
¡|d }|j|d ¡ }| ¡ |¡ ¡ }|j
|
tj ¡|_ t ||¡ d S ) NrX rY ©r r ©r r )r% r! )rS r" rx é éÿÿÿÿr% r| r )r r r r6 ro rp r r/ r r Zastyper r3 r7 )r Z idx_namesr r Z by_levelsZ
|
by_columnsr r r Ú test_groupby_levels_and_columns> s z,TestGrouping.test_groupby_levels_and_columnsc
|
C s ddddg}ddg}t ddddgddddgddddgddddgddddggt¡}t||dd}t||d}|jdd|d ¡ }t d
|
dgd
|
dgd
|
dgd
|
dgd
|
dggt¡}t||dd} t|| d}
|
t ||
|
¡ t|j |d }|jdd|d ¡ }t|j | d }
|
t ||
|
¡ d S ) Nr r r r T)Ú
|
categoriesZordered)r$ r r )r r
Úobservedr! r| )r$ rk )
|
r6 ÚarrayÚintr r r r2 r3 r7 ÚT) r r¨ r r§ r$ Z cat_columnsr r9 Z expected_dataZexpected_columnsr: r r r Ú*test_groupby_categorical_index_and_columnsL s* 4 ÿ * ÿ z7TestGrouping.test_groupby_categorical_index_and_columnsc C s t dditjtdtdddgddgd d
|
}| tdd tdd d g¡ ¡ }t dddddddgitjtdtdd ddgddgd d
|
}t ||¡ d S )Nr r r r éP r r> r? rx r| r r r é r é r% )r r )
|
r r r rK r r r r2 r3 r7 r8 r r r Ú#test_grouper_getting_correct_binnerc s$ ÿþÿþþz0TestGrouping.test_grouper_getting_correct_binnerc C s t | d¡jddgks td S )Nr r= r< )Úsortedr Úgrouperr) rM r r r Útest_grouper_itery s z TestGrouping.test_grouper_iterc C s( t jtdd | g ¡ W 5 Q R X d S )NzNo group keys passed!r )r r r' r rM r r r Útest_empty_groups| s z TestGrouping.test_empty_groupsc C s: | d¡}| |j¡jdd}|jdd}t ||¡ d S )Nr Tr0 )r r² r/ r3 r7 )r r r, r9 r: r r r Útest_groupby_grouper s
|
z!TestGrouping.test_groupby_grouperc C sà t ddi}| ddi¡ t¡}| dg¡ t¡}t ||¡ t ddddgtdd }d
|
d
|
d d d }| |¡ ¡ }| |¡ tj¡}tj d
|
d
|
d d gtj
|
d }| |¡ ¡ }| |¡ ¡ }t ||¡ t ||¡ t ||¡ d S )NZT1rS ZT2ç ð?ç @g @g @rI r| r r )r r r- Údr ) r r r5 r2 r3 r4 rK r/ r6 r© r )r r r9 r: ÚmappingrG Zexp_keyZ expected2r r r Útest_groupby_dict_mapping s z&TestGrouping.test_groupby_dict_mappingr r r r% r r r- r¸ c C s g | ]}t d dd| qS )iå é r )r
|
)rW Úir r r Ú
|
<listcomp> s zTestGrouping.<listcomp>r! c C s\ |ddddg|d}t ddddg|dd}| |¡ ¡ }|ddg}d|j_t ||¡ d S ) Nr r r% r! r| r )r r )rk r )r r Úlastrk r r3 Z assert_equal)r Zframe_or_seriesrk ÚobjÚgroupsr9 r: r r r Ú$test_groupby_series_named_with_tuple s
|
z1TestGrouping.test_groupby_series_named_with_tuplec C sR t dddd}ttj d¡|d}d}tjt|d | dd ¡ W 5 Q R X d S )
|
Nz 01-Jan-2013r¤ ZMS©r r r| z4Grouper result violates len\(labels\) == len\(data\)r c S s | dd
S )Nr r" r r r r r r` » ra zDTestGrouping.test_groupby_grouper_f_sanity_checked.<locals>.<lambda>) r r r6 rC rD r r r) r )r r Útsr+ r r r Ú%test_groupby_grouper_f_sanity_checked¬ s
|
z2TestGrouping.test_groupby_grouper_f_sanity_checkedc C s6 d}t jt|d t|j|ddg W 5 Q R X d S )NzEGrouper for '<class 'pandas.core.frame.DataFrame'>' not 1-dimensionalr r )r r r' r rk )r r r+ r r r Ú%test_grouping_error_on_multidim_input½ s z2TestGrouping.test_grouping_error_on_multidim_inputc C sX t dddgdddgddd gg}t d
|
d d g¡|_|jdd dgd ¡ }t ||¡ d S )Nr r r% r! rS r" r» r# é r[ r¢ )r r r ©r r
)r r r r r Úfirstr3 r7 )r r r9 r r r Útest_multiindex_passthru s z%TestGrouping.test_multiindex_passthruc C s¸ |j dd ¡ }|j dd ¡ }t ||¡ |j dd ¡ }|j dd ¡ }t ||¡ |j ddgd ¡ }| ¡ }t ||¡ |j ddgd ¡ }|j ddgd ¡ }t ||¡ d S )Nr¥ r ÚsecondéþÿÿÿrÈ )r r2 r3 r7 Z
|
sort_index)r Úmframer9 r: r r r Ú test_multiindex_negative_levelË s z+TestGrouping.test_multiindex_negative_levelc C s4 t t|j¡|_|jdddd dt ji¡ d S )Nr F©Zas_indexr ÚQ)r6 ro rq r r r5 r/ rM r r r Ú&test_multifunc_select_col_integer_colsÝ s z3TestGrouping.test_multifunc_select_col_integer_colsc C sê ddgddgg}t |¡}tddgg|d}| d¡j}|d dgksJt| d g¡j}|d dgksjttddgd
|
d gg|d}| d¡j}| d g¡j}||ks¨ttddgd
|
dgg|d}| d¡j}| d g¡j}t ||¡ d S ) Nr( Úvaluesú to filterÚ r r r r )rÒ rÓ r r )r r r r rÀ r) r3 rc )r ÚlstZmidxr r, r: r9 r r r Ú#test_multiindex_columns_empty_levelã s
|
z0TestGrouping.test_multiindex_columns_empty_levelc C sø t ddddgddddgddddggt dddd gddddgg¡d
|
}| d g¡j}| d ¡j}t ||¡ t |jt dddd gd d d d gg¡d
|
}| dg¡j}| d ¡j}t ||¡ t |jdddd gd
|
}| dg¡j}| d ¡j}t ||¡ d S )Nr r r% r! rS r" r r r- r r r¸ Úe)r r¸ )r r¸ )r rÖ )r r Ú from_arraysr rÀ r3 rc rÑ )r r r: r9 Zdf2Zdf3r r r Ú test_groupby_multiindex_tupleû s( þ ÿþ z*TestGrouping.test_groupby_multiindex_tupleÚsortTFc C s |}| ¡ }|jd|d ¡ }|jd|d ¡ }|j|d j|d ¡ }|j|d j|d ¡ } d|j_d| j_|jjdkst|jjdkstt ||¡ t || ¡ |jj|jj d ksÀt|jj|jj d ksØt|jd|d ¡ }|jd|d ¡ }t ||¡ t || ¡ |j
|
jdd|d ¡ }|j
|
jdd|d ¡ }t ||j
|
¡ t || j
|
¡ d}
|
t j t |
|
d |jdd
|
W 5 Q R X d S ) Nr ©r
rÙ r rÈ )rÙ rÊ )r
r rÙ z2level > 0 or level < -1 only valid with MultiIndexr r )r r r2 rÑ rk r r) r3 r7 ry r« r r r' ) r rÙ rÌ r ÚframeZ deleveledZresult0Úresult1Z expected0Z expected1r+ r r r Útest_groupby_level s2 z TestGrouping.test_groupby_levelc C s t dgd dgd tdd d¡}|dkr6|j}|jd|d d | |¡ }tjt|d
|
|jd |d W 5 Q R X d S ) Nr r% r r" )ÚexpZvar1rÞ )r r )r
r z&level name foo is not the name of the r r< ) r r& Ú set_indexr« r Z_get_axis_namer r r' )r r r r+ r r r Ú test_groupby_level_index_names= s "ÿz+TestGrouping.test_groupby_level_index_namesc C s t ddgddddggddddddddgddddddddggd}tt d¡|d}|jd|d ¡ }td d
|
gddgd}t ||¡ t ddgddddggddddd dddgddddddddggd}tt d¡|d}|jd|d ¡ }td d gd dgd}t ||¡ d S )Nr r r r% )ÚlevelsÚcodesg @r| rÚ g @g 6@r¥ g 2@g r¶ )r r r6 ro r r2 r3 r4 )r rÙ rk r r9 r: r r r Útest_groupby_level_with_nasI s &þ &þz(TestGrouping.test_groupby_level_with_nasc C sZ |}d}t jt|d | ¡ W 5 Q R X d}t jt|d |jd d d W 5 Q R X d S )Nz*You have to supply one of 'by' and 'level'r )Zbyr
)r r Ú TypeErrorr )r rÌ rÛ r+ r r r Útest_groupby_argsb s z TestGrouping.test_groupby_argsz sort,labelsc C s4 |j d|d}t |tj¡}t |jjd |¡ d S )Nr rÚ )r r6 r© Úintpr3 Úassert_almost_equalr² râ )r rÙ ÚlabelsrÌ r, Ú
|
exp_labelsr r r Útest_level_preserve_ordern s z&TestGrouping.test_level_preserve_orderc C sN | |j d¡¡}tjddddddddddg
|
tjd}t |jj d |¡ d S )Nr r r r% r )
|
r rk Zget_level_valuesr6 r© ræ r3 rç r² râ )r rÌ r, ré r r r Útest_grouping_labels{ s $z!TestGrouping.test_grouping_labelsc C s t dtddddi}tj|jd< tddd}| |g¡}td tt t
|
d
|
i}t |j |¡ | |¡}td di}t |j |¡ d S ) Nr z1/1/2011im r1 r r¥ ZASr~ z
|
2011-01-01il )r r r ZNaTrO r r r
|
r rK r& r3 rc rÀ )r r r² r9 r: r r r Útest_list_grouper_with_nat s
|
z'TestGrouping.test_list_grouper_with_natz func,expectedZ transform©r r r5 ©r r )r r rk rr c C sB t g g d}|jddd}t|d |dd }t ||¡ d S )Nr£ r Frm r c S s | S rn r r^ r r r r` ¬ ra z>TestGrouping.test_evaluate_with_empty_groups.<locals>.<lambda>)r r Úgetattrr3 r4 )r Úfuncr: r r r9 r r r Ú test_evaluate_with_empty_groups s z,TestGrouping.test_evaluate_with_empty_groupsc C sÐ t g ddd}| g ¡}| ¡ }| tg tjd¡}t ||¡ t |j
|
j dksTt t |j
|
jd tjg t tj¡d¡ t |j
|
jd tjg t tj¡d¡ |j
|
jd dks´t | |¡j
|
jdgksÌt d S )Nr Úfloat64rí r r r r )r r r/ Zset_axisr r6 ræ r3 r4 rq r² Ú groupingsr) Zassert_numpy_array_equalZ
|
group_infor© r ry )r r Úgrr9 r: r r r Útest_groupby_empty¯ s
|
|
ÿ
|
ÿz TestGrouping.test_groupby_emptyc C s t dtjdgd tjdggdddgd ddg¡}|jddgd ¡ }t g ttdgd d
|
tg d d
|
gg g gddgd dgd d}t ||¡ d S )NrX rl é r r r r r Úobjectr rò )rá râ ry r )r$ rk r r )
|
r r6 Únanrß r r2 r r r3 r7 r8 r r r Ú%test_groupby_level_index_value_all_naÇ s$ ÿþýø
|
z2TestGrouping.test_groupby_level_index_value_all_nac C sx t dddgdddggdddgd }| ddg¡}||jd
|
k }| d¡ ¡ }t d dgtg tjdd d }t ||¡ d S )Né{ r r¶ r r· ÚidÚcategoryÚvaluer r rò rî )r r rk )
|
r rß rý r r2 r r6 r r3 r7 )r r Úemptyr9 r: r r r Ú#test_groupby_multiindex_level_emptyÙ s ÿýz0TestGrouping.test_groupby_multiindex_level_emptyN)/rg rh ri r ÚmarkZ parametrizer3 ZmakeFloatIndexZmakeStringIndexZ makeIntIndexZ makeDateIndexZmakePeriodIndexru r r r¡ r¦ r¬ r° r³ r´ rµ rº r& rÁ rÄ rÅ rÉ rÍ rÐ rÕ rØ rÝ rà rã rå rê rë rì r r6 rò r rñ rõ rù rÿ r r r r rj s ûþ
|
|
E.
|
|
ýþ
|
|
(
|
þþ
|
þ ÿþ ÿþõþ
|
|
rj c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestGetGroupc C s t t ddddddg¡ddddddgddddd d
|
gd }| d ¡}t|jd }| |¡}| t| ¡ ¡}| t t|¡}t
|
||¡ t
|
||¡ | d dg¡}t|jd }| |¡}| t|d ¡ |d f¡}| t t|d |d f¡}t
|
||¡ t
|
||¡ d}t j t|d | d¡ W 5 Q R X t j t|d | d¡ W 5 Q R X d}t j t|d | d¡ W 5 Q R X d S )Nz 10-Oct-2013z 11-Oct-2013r< r= r r r% r! rS r" )ÚDATEÚlabelZVALr r r z<must supply a tuple to get_group with multiple grouping keysr zHmust supply a same-length tuple to get_group with multiple grouping keys)r< r= Úbaz)r r Ú to_datetimer rK rÀ Ú get_groupr
|
Z to_pydatetimeÚstrr3 r7 r r r' )r r r r rÜ rG Zresult3r+ r r r Útest_get_groupî sH úÿ
|
ôÿ
|
|
|
zTestGetGroup.test_get_groupc C s t ddddg}ddddg}|jt |d |¡|d }| t dd¡¡}t ddgddgd
|
}t ||¡ d }tj t
|
|d | t dd¡¡ W 5 Q R X d S ) Nr% r r» r" r rS rl é )r¨ r| z"Interval\(10, 15, closed='right'\)r ) r r r Zcutr ZIntervalr3 r7 r r r )r r¨ r¸ Zbinsr r9 r: r+ r r r Útest_get_group_empty_bins s z&TestGetGroup.test_get_group_empty_binsc C s t ddddggdgdj}| d¡}t dddgiddgd}| d¡}t ||¡ t ddddg¡}t dd d
|
|D i}| d¡}| d ¡}t d|d f|d fgiddgd}t ||¡ d S ) N)r r£ Úidsr| r r ú
|
2010-01-01z
|
2010-01-02c S s g | ]
|
}|fqS r r )rW rX r r r r½ 5 s z@TestGetGroup.test_get_group_grouped_by_tuple.<locals>.<listcomp>)r )r r« r r r3 r7 r r )r r rô r: r9 Údtr r r Ú test_get_group_grouped_by_tuple, s
|
|
|
|
$z,TestGetGroup.test_get_group_grouped_by_tuplec sr t ddd dD i d¡} fdd¡}| t|j ¡ d ¡}| t|j ¡ d ¡}t ||¡ d S )NrQ c s s, | ]$}t j d dd¡D ]}||fV qqdS rR rT rV r r r rZ > s zKTestGetGroup.test_get_group_grouped_by_tuple_with_lambda.<locals>.<genexpr>r[ c s j | df S r\ r] r^ r_ r r r` B ra zJTestGetGroup.test_get_group_grouped_by_tuple_with_lambda.<locals>.<lambda>r )r r r rK rÀ Úkeysr3 r7 rd r r_ r Ú+test_get_group_grouped_by_tuple_with_lambda; s ÿ
|
z8TestGetGroup.test_get_group_grouped_by_tuple_with_lambdac C sJ t d¡}d}t||td}tdd}| |¡}tt|d d ksFtd S )Nr r r1 ©r ) r r r r÷ r r ÚnextÚiterr) )r rk r$ Zseriesr² r, r r r Útest_groupby_with_emptyI s
|
|
|
z$TestGetGroup.test_groupby_with_emptyc C s¤ t dtdi}t | d¡ d¡|jddg ¡ t tdddgddg d}t | d¡ ¡ |¡ t | d¡ ¡ |¡ |jd d
|
dg }t | d¡
|
d ¡|¡ d S ) Nr Zabssbabr rS r r )r )rk r r% r! r ) r rK r3 r7 r r rO r r( r2 Znth)r r rÞ r r r Ú test_groupby_with_single_columnQ s "z,TestGetGroup.test_groupby_with_single_columnc C s| t ddddgddddgddddggd d
|
d d gd }| d d
|
g¡}| d d
|
d g¡ ¡ }|jd dksft|jd dksxtd S )Nr< r= r r r r r r% rÈ rÊ Úthirdr> r )r< r= r )r< r r )r rß r r* rE r) rM r r r Ú test_gb_key_len_equal_axis_len\ s
|
þz+TestGetGroup.test_gb_key_len_equal_axis_lenN)
|
rg rh ri r r
|
r r r r r r r r r r í s / r c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestIterationc C sÄ | dg¡}|j}||jks t|j ¡ D ]"\}}|j| d |k ¡ s*tq*| ddg¡}|j}||jkspt|j ¡ D ]D\}}|j| d |d k ¡ s t|j| d |d k ¡ sztqzd S )Nr r r r )r rÀ r) ÚitemsrE Úall)r r r, rÀ ÚkÚvr r r Ú test_groupso s zTestIteration.test_groupsc C s, | dd dd g¡}|jjd D ]}q"d S )Nc S s | ¡ S rn )Úweekdayr^ r r r r` ra z9TestIteration.test_grouping_is_iterable.<locals>.<lambda>c S s | j S rn )Úyearr^ r r r r` ra r )r r² ró )r Ztsframer, r r r r Útest_grouping_is_iterable s z'TestIteration.test_grouping_is_iterablec C sà t t d¡}t ddddddg¡}t ddddddg¡}| ||g¡}t|}dd|ddg fdd|dg fdd|d g fdd|d
|
d g fg}t|D ]B\}\\}} }
|
|| \} } } | |ksÂt| | ksÎtt |
|
| ¡ qd S ) Nr" r r Ú1Ú2r r r r! r% rS )
|
r r6 ro r© r rK Ú enumerater) r3 r4 )r r Úk1Úk2r, Úiteratedr: r¼ r> r? r@ Úe1Úe2Úe3r r r Útest_multi_iter s ü z TestIteration.test_multi_iterc C s¼ t ddddddg¡}t ddddddg¡}tt j d¡t j d¡||dddd d
|
d d gd }| ddg¡}t|}|j}dd|j|dg fdd|j|ddg fdd|j|ddg fdd|j|dg fg}t |D ]F\} \\}
|
} } || \} }}| |
|
kst
|
|| kst
|
t | |¡ qÚt ddddddg¡|d< t ddddddg¡|d< | ddg¡}dd |D }t |dkst
|
| dddg¡ ¡ }|jjddd }|D ]
|
\}}q¬d S ) Nr r r! r" r" )Zv1Zv2r$ r% r> r? r@ ZfourZfiveÚsixr| r$ r% r! r% rS r r r c S s i | ]\}}||qS r r )rW r Zgpr r r Ú
|
<dictcomp>¼ s z7TestIteration.test_multi_iter_frame.<locals>.<dictcomp>r r r r£ rÇ )r6 r© r rC rD r rK rk rE r# r) r3 r7 rq r/ r« )r Z three_groupr$ r% r r, r& r r: r¼ r> r? r@ r' r( r) rÀ Z three_levelsr Úgroupr r r Útest_multi_iter_frame s8 þü z#TestIteration.test_multi_iter_framec C s t t| d¡ t t| ddg¡ t t|d |d ¡ t t|d |d |d g¡ t t| d¡d t t| ddg¡d d S )Nr r r )Údictr r rM r r r Ú test_dictifyÅ s "zTestIteration.test_dictifyc C s t ddgddgdt ddg¡d}| tdd d
|
g¡}t|jd ksJt|jd ksXtt d df|jksntt d df|jkst|
|
t d df¡}t ||j dgd d
f ¡ |
|
t d df¡}t ||j dgd d
f ¡ t dddgdddgdt dddg¡d}| tdd d
|
g¡}t|jd ks2t|jd ksBtt d df|jksZtt d df|jksrt|
|
t d df¡}t ||j dd gd d
f ¡ |
|
t d df¡}t ||j dgd d
f ¡ t dddgdddgdt dddg¡d}| tdd d
|
g¡}t|jdks"t|jdks2tt d df|jksJtt d df|jksbtt ddf|jkszt|
|
t d df¡}t ||j dgd d
f ¡ |
|
t d df¡}t ||j dgd d
f ¡ |
|
t ddf¡}t ||j d gd d
f ¡ d S )NÚstartiÒ i. )ÚeventZchangez
|
2014-09-10z
|
2013-10-10r| r r r2 r z
|
2014-09-30z
|
2013-10-31r r i£# z
|
2014-09-15z
|
2014-08-05r% z
|
2014-08-31)r r r r r rq rÀ r) Zngroupsr
|
r r3 r7 rO )r r r, Úresr r r Ú test_groupby_with_small_elemÍ sT þ þ þ z*TestIteration.test_groupby_with_small_elemc C s\ t tdtdg¡}tdddgg|d}| |d ¡}|jjd ¡ }d }||ksXtd S )
|
NZAABZabar r r% r )r r r zGrouping(('A', 'a'))) r r× rK r r r² ró Ú__repr__r) )r Úmir rô r9 r: r r r Útest_grouping_string_repr s z'TestIteration.test_grouping_string_reprN)
|
rg rh ri r r r* r. r0 r4 r7 r r r r r n s )4r c C s t dddgdddgdddgd d¡} | jtddtd d
|
gd d }|jjd jrXt|jjd jsjt| ¡ }t ddgddgd}t
|
||¡ d S )Nr r r% r! rS )r r r- r r r r FrÎ r r» )r r- ) r rß r r r² ró Zin_axisr) r2 r3 r7 )r re r9 r: r r r Ú test_grouping_by_key_is_in_axis s ( r8 c C s
|
t dddgdd} tdd}| |¡}d}tjt|d |j}W 5 Q R X ||jksZtd }tjt|d |j}W 5 Q R X ||jkstd
|
}tjt|d |j }W 5 Q R X ||j ks¾td }tjt|d |j
|
W 5 Q R X d }tjt|d |j W 5 Q R X d S ) Nr r r% )r r r r zUse GroupBy.groups insteadr zUse GroupBy.grouper insteadz-Grouper.obj is deprecated and will be removedzUse Resampler.ax insteadz Grouper.indexer is deprecated) r r r r3 Zassert_produces_warningÚ FutureWarningrÀ r) r² r¿ ZaxZindexer)r Zgrperre r+ r3 r r r Útest_grouper_groups s*
|
|
r: ) Ú__doc__Údatetimer r Únumpyr6 r Zpandasr r r r r r r r
|
r Zpandas._testingZ_testingr3 Zpandas.core.groupby.grouperr r rj r r r8 r: r r r r Ú<module> s& (
|
{ X
|
