fume-manage-python.git

U  
ý°dNUã@sddlmZddlZddlZddlZddlmZmZm    Z    ddl
mZddZ ddZdd    Zd
dZdd ZddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Z d,d-Z!d.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&dS)8é)Úascii_lowercaseN)Ú    DataFrameÚSeriesÚ    TimestampcCsÖtdddddddg}tddddgdddd    gd
}tdddgdddgd
}| d d¡}| |¡}t | dd¡|¡t | dd¡|¡t |jdddd| |j¡¡t |jdddd| |j¡¡dS)Néééééééré©ÚindexéécSs|dS©Nr©©ÚxrrúXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_filters.pyÚ<lambda>óz$test_filter_series.<locals>.<lambda>cSs| ¡dkS©Né
©ÚmeanrrrrrrcSs| ¡dkSrrrrrrrrcSs| ¡dkSrrrrrrrrF©ZdropnacSs| ¡dkSrrrrrrrr)rÚapplyÚgroupbyÚtmÚassert_series_equalÚfilterÚreindexr)ÚsÚexpected_oddÚ expected_evenÚgrouperÚgroupedrrrÚtest_filter_seriess
 
þ
þr)cCsÚtdddddddg}tddddgdddd    gd
}tdddgdddgd
}|d d d¡}| |¡}t | dd¡|¡t | dd¡|¡t |jdddd| |j¡¡t |jdddd| |j¡¡dS)Nrrrr    r
rrrr rrrcSs|dSrrrrrrr&rz.test_filter_single_column_df.<locals>.<lambda>cSs| ¡dkSrrrrrrr(rcSs| ¡dkSrrrrrrr)rcSs| ¡dkSrrrrrrr,rFrcSs| ¡dkSrrrrrrr0r)rrrr Úassert_frame_equalr"r#r)Údfr%r&r'r(rrrÚtest_filter_single_column_df"s
 
þ
þr,cCsrtddddgddddgd}|d dd¡}| |¡}tddgddgdddgd}t | d    d¡|¡dS)
Nré©ÚAÚBr/cSs|dSrrrrrrr7rz-test_filter_multi_column_df.<locals>.<lambda>rrcSs|d ¡|d ¡dkS)Nr/r0r©Úsumrrrrr;r)rrrr r*r"©r+r'r(ÚexpectedrrrÚtest_filter_multi_column_df5s
ÿr5cCsntddddgd ¡d}|d dd¡}| |¡}tddgdd    gddd
gd}t | dd¡|¡dS) Nrr-úa b c dr.r/cSs|dSrrrrrrrArz&test_filter_mixed_df.<locals>.<lambda>ÚbÚcrrcSs|d ¡dkS)Nr/rr1rrrrrDr)rÚsplitrrr r*r"r3rrrÚtest_filter_mixed_df?s
 
r:cCstdddddddg}| dd    ¡}| |¡}t | d
d    ¡|g¡tddddgd ¡d }|d dd    ¡}| |¡}t | dd    ¡|j    g¡dS)Nrrrr    r
rrcSs|dSrrrrrrrIrz,test_filter_out_all_groups.<locals>.<lambda>cSs| ¡dkS)NéèrrrrrrKrr-r6r.r/cSs|dSrrrrrrrMrcSs|d ¡dkS)Nr/r;r1rrrrrOr)
rrrr r!r"rr9r*Úloc)r$r'r(r+rrrÚtest_filter_out_all_groupsGs
 
r=cCstdddddddg}| dd    ¡}| |¡}| d
d    ¡}t ||¡tddddgd ¡d }|d dd    ¡}| |¡}| dd    ¡}t ||¡dS)Nrrrr    r
rrcSs|dSrrrrrrrTrz+test_filter_out_no_groups.<locals>.<lambda>cSs| ¡dkS©NrrrrrrrVrr-r6r.r/cSs|dSrrrrrrrYrcSs|d ¡dkS)Nr/rrrrrrr[r)    rrrr"r r!rr9r*)r$r'r(Úfilteredr+rrrÚtest_filter_out_no_groupsRs
 
r@cCs¸tdddgdddgd}| d¡}|jdddd    }ttjgd
tjgd
d}t ||¡tdddgdddgd}| d¡}|jdddd    }tggdd d}t ||¡dS)Nrrr)Úar7rAcSs|d ¡dkS©Nr7r    r1rrrrrcrz2test_filter_out_all_groups_in_df.<locals>.<lambda>FrrcSs|d ¡dkSrBr1rrrrrirTÚint64©Údtype)rrr"ÚnpÚnanr r*)r+Úresr4rrrÚ test_filter_out_all_groups_in_df_s
 
rIc    Cs\dd}tddddg}| dd¡}| |¡}d    }tjt|d
| |¡W5QRXdS)NcSs| ¡dkrt| ¡dkSr>)r2Ú
ValueErrorrrrrÚraise_if_sum_is_zeroosz:test_filter_condition_raises.<locals>.raise_if_sum_is_zeroéÿÿÿÿrrrcSs|dSrrrrrrrurz.test_filter_condition_raises.<locals>.<lambda>ú'the filter must return a boolean result©Úmatch)rrrÚpytestÚraisesÚ    TypeErrorr")rKr$r'r(ÚmsgrrrÚtest_filter_condition_raisesns
rTcCsrtj tdddgg¡}tt d¡ dd¡|dd}|jddd         d
d¡}|j
ddddf}t ||¡dS) NrrrédrLrrC)ÚcolumnsrE)ÚlevelZaxiscSs|jddkS)N)rrr)Úilocrrrrrrz2test_filter_with_axis_in_groupby.<locals>.<lambda>r-) ÚpdZ
MultiIndexZfrom_productÚrangerrFÚarangeZreshaperr"rXr r*)rÚdataÚresultr4rrrÚ test_filter_with_axis_in_groupby|s
r^c    CsHtt d¡tdt d¡d}|d}| d¡}| |¡}dd}d}tjt|d| |¡W5QRXd    }tjt|d| |¡W5QRXd
d}d}tjt|d| |¡W5QRXd    }tjt|d| |¡W5QRXdd}d}tjt|d| |¡W5QRXd    }tjt|d| |¡W5QRXdS) NéÚaabbbbcc©r/r0ÚCr0cSs|S©Nrrrrrrrz(test_filter_bad_shapes.<locals>.<lambda>z@filter function returned a DataFrame, but expected a scalar boolrNrMcSs|dkS©NrrrrrrrrcSst ||¡Src)rFÚouterrrrrrrz0can't multiply sequence by non-int of type 'str')    rrFr[ÚlistrrPrQrRr")r+r$Úg_dfÚg_sÚfrSrrrÚtest_filter_bad_shapess2 
 
rjcCsztt d¡tdt d¡d}|d}| |d¡}| |¡}dd}t | |¡|jg¡t     | |¡|g¡dS)Nr_r`rar0cSstjSrc)rFrGrrrrrªrz*test_filter_nan_is_false.<locals>.<lambda>)
rrFr[rfrr r*r"r<r!)r+r$rgrhrirrrÚtest_filter_nan_is_false¤s 
rkcCstt d¡tdt d¡d}|d}| |d¡}| |¡}dd}| |¡}t ||jg¡| |¡}t     ||g¡dS)Nr_r`rar0cSstjSrc)rYZNArrrrr¶rz+test_filter_pdna_is_false.<locals>.<lambda>)
rrFr[rfrr"r r*r<r!)r+ÚserrgrhÚfuncrHrrrÚtest_filter_pdna_is_false¯s 
 
 
rnc    stj d¡ttj ddd¡}| dd¡}| |¡}dd}|| |¡ d¡}|     |¡}t
 | ¡| ¡¡dttj d¡}| dd¡}| |¡}d    d}|| |¡ d¡}|     |¡}t
 | ¡| ¡¡t  tt¡}d| tj dd
¡¡}tttj dd¡dttj ¡t|d}| d ¡}||j fdd¡ d¡}|     fdd¡}t
 ||¡|j dd¡}| |¡}||j fdd¡ d¡}|     fdd¡}t
 ||¡| d¡}||j fdd¡ d¡}|     fdd¡}t
 ||¡dS)NrrUr;cSst |d¡S©NrL©rFÚroundrrrrrÁrz0test_filter_against_workaround.<locals>.<lambda>cSs| ¡dkSrrrrrrrÃrÚboolcSst |d¡SrorprrrrrËrcSs| ¡dkSrrrrrrrÍrér)ÚintsÚfloatsÚlettersrtcs| ¡dkS©Nrrr©ÚNrrràrcs|d ¡dkS)NrurrrrxrrrárcSst |d¡Srorprrrrrårcst|dkSr©Úlenrrxrrrçrcst|jdkSr)r{rvrrxrrrèrrvcs| ¡dkSrwrrrxrrrírcs|d ¡dkS)Nrtrrrrxrrrîr)rFÚrandomÚseedrÚrandintrrÚ    transformZastyper"r r!Zsort_valuesÚarrayrfrÚtakerrur*rvrt)    r$r'r(riZold_wayZnew_wayrvZrandom_lettersr+rrxrÚtest_filter_against_workaround½sJ
 
 
 
ýÿ    
 
 
rcCstt d¡tdt d¡d}| d¡}| dd¡}tt dd¡td    t dd¡dtjddtjd
d}t ||¡| dd¡}|j    g}t ||¡|d}| |¡}| d d¡}t
ddgtjddtjd
dd}t ||¡| dd¡}|g}t ||¡dS)Nr_r`rar0cSst|dkSrrzrrrrrörz'test_filter_using_len.<locals>.<lambda>rr ZbbbbrDrcSst|dkS©NrrzrrrrrýrcSst|dkSrrzrrrrrrrr7)rÚnamecSst|dkSrrzrrrrrr)rrFr[rfrr"rCr r*r<rr!)r+r(Úactualr4r$rrrÚtest_filter_using_lenòs& 
þ
 
"rc
CsÒtddddddddgddddddd    dgd
}|d}| d¡}| d d¡}|jddddg}t ||¡| |d¡}| dd¡}|jddddg}t ||¡t t    |ddd¡|_
|d}| d¡}| dd¡}|jddddg}t ||¡| |d¡}| dd¡}|jddddg}t ||¡ddddddddg}|j
||_
|d}| d¡}| dd¡}|jddddg}t ||¡| |d¡}| dd¡}|jddddg}t ||¡dS)Nrrréé-é>ré"é©ÚpidÚtagrrcSst|dkSrdrzrrrrrrz0test_filter_maintains_ordering.<locals>.<lambda>rrcSst|dkSrdrzrrrrrrrLcSst|dkSrdrzrrrrr!rcSst|dkSrdrzrrrrr&rr rr    cSst|dkSrdrzrrrrr/rcSst|dkSrdrzrrrrr4r)rrr"rXr r*r!rFr[r{r)r+r$r(rr4ZSHUFFLEDrrrÚtest_filter_maintains_ordering sB(ÿ
 
 
rcCsþttjddddddddgtdd}| dd    g¡}|d
 dd¡}t |d
|¡|d
 t    ¡}t
d dd ddgd
d}t ||¡| dd¡}t ||¡| d¡}td
d dd ddgi}t ||¡| t    ¡}td
d dd ddgi}t ||¡dS)Nr    rCrDZfooÚbarZ20130101rar0rbr/cSsdS©NTrrrrrrErz0test_filter_multiple_timestamp.<locals>.<lambda>rr©rcSsdSrrrrrrrLrr2r_)rrFr[rrr"r r!rr{rr*)r+r(r]r4rrrÚtest_filter_multiple_timestamp9s(ýÿ
 
rc
Csddddddddg}tddddddddgdddddd    d
dgd|d}| d ¡}|d}| |d ¡}ddddg}| dd¡}|j|}t ||¡|jdddd}| ¡}tj|jddddg<t ||¡| dd¡}|     |¡}t 
||¡|jdddd}ttjddtjdtjtjdg|dd}t 
||¡| t ¡}tddddddddg|dd}t 
||¡|j t ¡}t 
||¡dS)NrrrrrrrrrrrrrrrrcSst|dkSrdrzrrrrrerzEtest_filter_and_transform_with_non_unique_int_index.<locals>.<lambda>cSst|dkSrdrzrrrrrirFrr    r cSst|dkSrdrzrrrrrorcSst|dkSrdrzrrrrrsrr©rrr"rXr r*ÚcopyrFrGrr!rrr{r©rr+Ú
grouped_dfrlÚgrouped_serÚexpected_indexesrr4rrrÚ3test_filter_and_transform_with_non_unique_int_indexXs6(þ
 
 
&
rc
Csddddddddg}tddddddddgdddddd    d
dgd|d}| d ¡}|d}| |d ¡}ddddg}| dd¡}|j|}t ||¡|jdddd}| ¡}tj|jddddg<t ||¡| dd¡}|     |¡}t 
||¡|jdddd}ttjddtjdtjtjdg|dd}t 
||¡| t ¡}tddddddddg|dd}t 
||¡|j t ¡}t 
||¡dS)NrrrrrrrrrrrrrrrrcSst|dkSrdrzrrrrrrzNtest_filter_and_transform_with_multiple_non_unique_int_index.<locals>.<lambda>cSst|dkSrdrzrrrrrrFrr    r cSst|dkSrdrzrrrrrrcSst|dkSrdrzrrrrrrrrrrrrÚ<test_filter_and_transform_with_multiple_non_unique_int_indexs6(þ
 
 
&
rc
Cstjddddddddgtd}tddddddddgdddd    dd
ddgd|d }| d¡}|d}| |d¡}ddddg}| dd¡}|j|}t ||¡|jdddd}|     ¡}tj
|jddddg<t ||¡| dd¡}| |¡}t ||¡|jdddd}t tj
ddtj
dtj
tj
dg|dd}t ||¡| t¡}t ddddddddg|dd}t ||¡|j t¡}t ||¡dS)NrrrrDrrrrrrrrrrrrrcSst|dkSrdrzrrrrr¹rzGtest_filter_and_transform_with_non_unique_float_index.<locals>.<lambda>cSst|dkSrdrzrrrrr½rFrr    r cSst|dkSrdrzrrrrrÃrcSst|dkSrdrzrrrrrÇrr)rFrÚfloatrrr"rXr r*rrGrr!rrr{rrrrrÚ5test_filter_and_transform_with_non_unique_float_index¬s6(þ
 
 
&
rc
Cs td}td}td}||||||||g}tddddddddgddd    d
dddd    gd |d}| d¡}|d}| |d¡}ddddg}| dd¡}    |j|}
t |    |
¡|jdddd}    | ¡}
tj    |
jddddg<t |    |
¡| dd¡}    | 
|¡}
t |    |
¡|jdddd}    ttj    ddtj    dtj    tj    dg|dd}
t |    |
¡|  t¡}    tddddddddg|dd}
t |    |
¡|j  t¡}    t |    |
¡dS)Nz2013-09-30 00:05:00z2013-10-30 00:05:00z2013-11-30 00:05:00rrrrrrrrrrrrrrrcSst|dkSrdrzrrrrrærzKtest_filter_and_transform_with_non_unique_timestamp_index.<locals>.<lambda>cSst|dkSrdrzrrrrrêrFrrr    r cSst|dkSrdrzrrrrrðrcSst|dkSrdrzrrrrrôrr)rrrr"rXr r*rrFrGrr!rrr{r)Út0Út1Út2rr+rrlrrrr4rrrÚ9test_filter_and_transform_with_non_unique_timestamp_indexÖs<(þ
 
 
&
r¡c
Cs|td}tddddddddgdddddd    d
dgd|d}| d ¡}|d}| |d ¡}ddddg}| dd¡}|j|}t ||¡|jdddd}| ¡}tj    |jddddg<t ||¡| dd¡}| 
|¡}t ||¡|jdddd}ttj    ddtj    dtj    tj    dg|dd}t ||¡|  t¡}tddddddddg|dd}t ||¡|j  t¡}t ||¡dS)NZbbbcbbabrrrrrrrrrrrrrrrcSst|dkSrdrzrrrrrrzHtest_filter_and_transform_with_non_unique_string_index.<locals>.<lambda>cSst|dkSrdrzrrrrrrFrrr    r cSst|dkSrdrzrrrrrrcSst|dkSrdrzrrrrrrr)rfrrr"rXr r*rrFrGrr!rrr{rrrrrÚ6test_filter_and_transform_with_non_unique_string_indexs6(þ
 
 
&
r¢cCsTtddgddgddggddgd}| d¡}| d    d
¡}t ||jddg¡dS)Nrrrr    r r/r0©rVcSs|d ¡dkS)Nr/rr1rrrrr1rz8test_filter_has_access_to_grouped_cols.<locals>.<lambda>r)rrr"r r*rX)r+ÚgZfiltrrrÚ&test_filter_has_access_to_grouped_cols-s"
r¥c
Cs|tdddgdddgdddgdddgdddgdddgddd    ggdddgd
}tjtdd| d¡ d d¡W5QRXdS)NÚbestrArÚworstr7Úyr8ÚdÚzr£úfilter function returned a.*rNcSs|ddkS)NrAr¦r©r¤rrrrCrz1test_filter_enforces_scalarness.<locals>.<lambda>©rrPrQrRrr"©r+rrrÚtest_filter_enforces_scalarness5sù    ör¯c
Cs|tdddgdddgdddgdddgdddgdddgdddggdddgd}tjtd    d
| d¡ dd¡W5QRXdS) Nr¦rArr§r7r8r©r£r«rNcSs
|j ¡Src)r8rr¬rrrrTrz-test_filter_non_bool_raises.<locals>.<lambda>rr®rrrÚtest_filter_non_bool_raisesFsù    ör°cCs¨ttj d¡t dddgd¡d}|jdd}|jdd    d
d}ttjgdt dddgd¡d}t     ||¡|jdd    d d}tt
jgtdtj d}t     ||¡dS)Né    rrrrr)rWcSs| ¡dkSrdrrrrrr[rz6test_filter_dropna_with_empty_groups.<locals>.<lambda>FrcSs| ¡dkSrdrrrrrr_rTrD)rrE)rrFr|ZrandÚrepeatrr"rGr r!rYZIndexÚintÚfloat64)r\ZgrouppedZresult_falseZexpected_falseZresult_trueZ expected_truerrrÚ$test_filter_dropna_with_empty_groupsWs""rµcCsvttdtdd}| d¡}| dd¡}ttdtdd}t ||¡| ¡| dd¡}t ||¡dS)Nr ZABCABC)r\Úkeyr¶cSsdSrrrrrrrhrzEtest_filter_consistent_result_before_after_agg_func.<locals>.<lambda>cSsdSrrrrrrrmr)rrZrfrr"r r*r2)r+r'r]r4rrrÚ3test_filter_consistent_result_before_after_agg_funcds
r·)'ÚstringrÚnumpyrFrPZpandasrYrrrZpandas._testingZ_testingr r)r,r5r:r=r@rIrTr^rjrkrnrrrrrrrr¡r¢r¥r¯r°rµr·rrrrÚ<module>s>
     5,***-*