U
|
¬ý°d8 ã @ s¤ d dl Zd dlZd dlmZmZ dd Zdd Zdd Zd d
|
Z d d Z
|
ej d dddddgdddddgfddddgddddgfdddgddd gfg¡d d Z dS ) é N)Ú_make_unique_kwarg_listÚmaybe_mangle_lambdasc C sF t ddkstt dd jdks&tt dd gd jdksBtd S )NÚmeanc S s | S ©N© ©Úxr r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_aggregation.pyÚ<lambda> ó z7test_maybe_mangle_lambdas_passthrough.<locals>.<lambda>ú<lambda>c S s | S r r r r r r r
|
r r )r ÚAssertionErrorÚ__name__r r r r Ú%test_maybe_mangle_lambdas_passthrough
|
s r c C sx dd dd g} t | }|d jdks*t|d jdks<t| d d |d d ksXt| d d |d d ksttd S )Nc S s dS ©Né r r r r r r
|
r z4test_maybe_mangle_lambdas_listlike.<locals>.<lambda>c S s dS )Né r r r r r r
|
r r ú
|
<lambda_0>r ú
|
<lambda_1>©r r r )ZaggfuncsÚresultr r r Ú"test_maybe_mangle_lambdas_listlike s r c C sL ddd dd gi} t | }|d d jdks2t|d d jdksHtd S ) NÚAc S s dS ©Nr r r r r r r
|
r z+test_maybe_mangle_lambdas.<locals>.<lambda>c S s dS r r r r r r r
|
r r r r r r ©Úfuncr r r r Útest_maybe_mangle_lambdas s r c C s¢ dddddd gi} t | }|d d jdks4t|d d jdksJt| d d ddd ksdt| d d ddd
|
d kst| d d dd
|
d d dkstd S )Nr r c S s
|
d||fS r r )r ÚaÚbr r r r
|
" r z0test_maybe_mangle_lambdas_args.<locals>.<lambda>c S s dS 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 r Ú test_maybe_mangle_lambdas_args! s r c C s. t jt jt jdd} t| }|| ks*td S )N)ZfooÚbar)ÚCÚD)Únpr r r r r r r Ú test_maybe_mangle_lambdas_named, s r% zorder, expected_reorder)Úheightr )r& Úmax)Úweightr' )r( r )r& ú
|
<lambda>_0)r& ú
|
<lambda>_1)Zcol2Úmin)Úcol1r )r, r) )r, r* )r, ú
|
<lambda>_2)Úcolr )r. r) )r. r* )r. r- c C s t | }||kstd S r )r r )ÚorderZexpected_reorderr r r r Útest_make_unique2 s )r0 )Únumpyr$ ZpytestZpandas.core.applyr r r r r r r% ÚmarkZ parametrizer0 r r r r Ú<module> sN ûûøüüùþáþ'
|
