U
|
ý°d ã
|
@ s® d dl m Z d dlZd dlZd dlmZmZmZ d dlm Z
|
d dl m Z ej de
|
de d d¡i¡e
|
je d d¡ddg¡d d
|
Zd d Zej dee
|
jg¡d d ZdS )é )ÚdatetimeN)Ú DataFrameÚIndexÚSeries)Úget_groupby_method_argsÚobjÚAé
|
©Únamec C sÜ t | tr"|dkr"t d| ¡ | t dd¡¡}t | d¡t| sLt t
|
|| }t ||| }|j |f| }dddh}t | t r¢||kr¢t |tjs´t nt |t| s´t t |t rÌt ||¡ n t ||¡ d S )N> Úcorrwithz Not applicable for Series and r r ZngroupZcumcountÚsize)Ú
|
isinstancer ÚpytestÚskipÚgroupbyÚnpÚarangeZ get_groupÚtypeÚAssertionErrorr ÚgetattrÚaggr ÚtmÚSubclassedSeriesZassert_frame_equalÚassert_series_equal)r Z groupby_funcZgroupedÚargsZresult1Zresult2Zslices© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_groupby_subclass.pyÚ test_groupby_preserves_subclass s
|
|
|
|
r c C s t dddgdddgdddgd¡} d| jks2td | _| d
|
¡D ]\}}|jd ksBtqBd d }| d
|
¡ |¡}t jd gd tdddgd
|
d d}t ||¡ dd }t dddg¡}d |_| | d
|
¡ |¡}t ||¡ | | d
|
¡
|
|¡}t ||¡ d S )Né é é é é é )ÚaÚbÚcÚtestattrZhellor' c S s$ t | tjstt| ds t| jS ©Nr( )r r ÚSubclassedDataFramer Úhasattrr( ©Úgroupr r r Úfunc= s z-test_groupby_preserves_metadata.<locals>.funcr
|
)Úindexc S s$ t | tjstt| ds t| jS r) )r r r r r+ r( r, r r r Úfunc2F s z.test_groupby_preserves_metadata.<locals>.func2) r r* Ú _metadatar r( r Úapplyr r r r )Z custom_dfÚ_Zgroup_dfr. ÚresultÚexpectedr0 Z custom_seriesr r r Ú test_groupby_preserves_metadata4 s $" r6 c C s¤ | d ¡ ddddddgtddddd tdddddtdd
|
dd d tdd
|
dd
|
d tdd d d d tddd dd gd}| d¡}| d¡ d¡ ¡ }t|| s td S )Nz Carl Carl Carl Carl Joe Carlé r! é r r$ iÝ é r r é é r é )ÚBuyerZQuantityÚDater> r= Z5D)Úsplitr Z set_indexr ZresampleÚsumr r )r Zdfr4 r r r Ú(test_groupby_resample_preserves_subclassS s úýÿ
|
rA )r Únumpyr r Zpandasr r r Zpandas._testingZ_testingr Zpandas.tests.groupbyr ÚmarkZ parametrizer* r r r r6 rA r r r r Ú<module> s þþ
|
|
