U
|
ý°do ã @ s@ d dl Zd dlZd dlmZmZ d dlmZ dd Z dd Z
|
dS )é N)Ú DataFrameÚIndexc C s t j d¡} tddddddddg| d¡| d¡d}dd }dd }| d
|
¡ |¡ |¡}tddgd d
|
d }tj d dgd|d}t
|
||¡ d S )NiÒIZfooÚbaré )ÚAÚBÚCc S s | j ¡ | j ¡ ¡ S ©N)r Úmaxr Úmin)Údfgb© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_pipe.pyÚf s ztest_pipe.<locals>.fc S s | d S )Né r )Zsrsr r r Úsquare s ztest_pipe.<locals>.squarer Úobject)ÚdtypeÚnamegÂÖ®vqû!@g Ö¯Y @r )r Úindex) ÚnpÚrandomZ RandomStater ZrandnÚgroupbyÚpiper ÚpdÚSeriesÚtmÚassert_series_equal)Z random_stateÚdfr r Úresultr Úexpectedr r r Ú test_pipe s ýÿr! c C sô t dddddgdddddgdd d
|
d d gd } dd }dd }dd }| d¡ |d¡ |d¡ |d¡}tddgdd}tjddg|d}t ||¡ t d d d d d d g¡}| |¡ d d ¡}tjd!d"d#gtd d d gtj d$d}t ||¡ d S )%Nr r r g ð?g @g @g @g $@g Y@g @@g YÀg @À)ÚgroupÚxÚyc s | j fdddd}| d¡S )Nc s | j ¡ kS r )r$ Zmean©Úgrp©Úarg1r r Ú<lambda>9 ó z+test_pipe_args.<locals>.f.<locals>.<lambda>F)Zdropnar" )Úfilterr )r r( Úfilteredr r' r r 8 s ztest_pipe_args.<locals>.fc S s | ¡ | ¡ ¡ | S r )Úsum)r Zarg2r r r Úg< s ztest_pipe_args.<locals>.gc S s | j | j | S r )r# r$ )r Zarg3r r r Úh? s ztest_pipe_args.<locals>.hr" r é
|
éd )r g#áSÀgùÜbeøSÀ)r é r é c S s | ¡ | ¡ S r )r- Úcountr% r r r r) L r* z test_pipe_args.<locals>.<lambda>é r é )r )
|
r r r r r r r r r Úint64)r r r. r/ r r r Zserr r r Útest_pipe_args, s" ýÿ" $r8 ) Únumpyr Zpandasr r r Zpandas._testingZ_testingr r! r8 r r r r Ú<module> s
|
!
|
