B @` @sddlZddlZddlmZmZddZddZddZd d Z d d Z ej d dddddgdddddgfddddgddddgfdddgdddgfgddZ dS)N)_make_unique_kwarg_listmaybe_mangle_lambdascCsFtddksttddjdks&ttddgdjdksBtdS)NmeancSs|S)N)xrrA/tmp/pip-unpacked-wheel-q9tj5l6a/pandas/tests/test_aggregation.py z7test_maybe_mangle_lambdas_passthrough..zcSs|S)Nr)rrrrr r r)rAssertionError__name__rrrr%test_maybe_mangle_lambdas_passthroughsr cCsxddddg}t|}|djdks*t|djdks.cSsdS)Nr)rrrrrr rz r z )rr r )Zaggfuncsresultrrr"test_maybe_mangle_lambdas_listlikes rcCsLdddddgi}t|}|ddjdks2t|ddjdksHtdS) NAcSsdS)Nrr)rrrrrr z+test_maybe_mangle_lambdas..cSsdS)Nr r)rrrrrr rz r z )rr r )funcrrrrtest_maybe_mangle_lambdassrcCsddddddgi}t|}|ddjdks4t|ddjdksJt|ddddd ksdt|ddddd d kst|dddd d d dkstdS)Nrr cSs d||fS)Nrr)rabrrrrr z0test_maybe_mangle_lambdas_args..cSsdS)Nr r)rrrrrr rz z )rr r r)rr r)r)rrr)r )rr r )rrrrrtest_maybe_mangle_lambdas_argssrcCs.tjtjtjdd}t|}||ks*tdS)N)Zfoobar)CD)nprrr )rrrrrtest_maybe_mangle_lambdas_named)srzorder, expected_reorder)heightz)rmax)weightr)rz)rz _0)rz _1)Zcol2min)col1z)r!z _0)r!z _1)r!z _2)colz)r"z _0)r"z _1)r"z _2cCst|}||kstdS)N)rr )orderZexpected_reorderrrrrtest_make_unique/s)r$)ZnumpyrZpytestZpandas.core.aggregationrrr rrrrmarkZ parametrizer$rrrrs: