U
|
ý°d\ ã @ sT d dl Z d dlmZmZmZmZ d dlZd dlm Z
|
d dl m Z G dd de Z dS )é N)Ú is_bool_dtypeÚis_numeric_dtypeÚis_object_dtypeÚis_string_dtype)ÚBaseExtensionTestsc @ sj e Zd ZdZdd Zej dddg¡dd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd ZdS )ÚBaseGroupbyTestszGroupby-specific tests.c
|
C sh t ddd d ddddg|d¡}| d¡jjd }| d¡jjd }t |j|jj ¡ t
|
|j|¡ d S )NÚBÚAÚC©r r r ) ÚpdÚ DataFrameÚgroupbyZgrouperZ groupingsÚtmZassert_numpy_array_equalZgrouping_vectorr ÚvaluesZ assert_extension_array_equal)ÚselfÚdata_for_groupingÚdfZgr1Zgr2© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/groupby.pyÚtest_grouping_grouper s ÿz&BaseGroupbyTests.test_grouping_grouperÚas_indexTFc
|
C s¢ t ddddddddg|d¡}|jd|dj ¡ }t j|dd \}}|r|t j|dd
|
}t jd d d g|dd}| ||¡ n"t |d d d gd¡}| ||¡ d S )Né é é é r r )r T©Úsort©Únameç @ç ð?ç @r ©Úindexr )r r )
|
r r r r ÚmeanÚ factorizeÚIndexÚSeriesÚassert_series_equalÚassert_frame_equal) r r r r ÚresultÚ_Zuniquesr$ Úexpectedr r r Útest_groupby_extension_agg s z+BaseGroupbyTests.test_groupby_extension_aggc
|
C s t ddddddddg|d¡}|jddddg }| d¡}| d¡ d d
|
i¡}| ||¡ | d¡ d
|
¡}| ||¡ | d¡ ¡ }| ||¡ d S ) Nr r r r r r é r r Úfirst)r r ÚilocZ set_indexr Úaggr* r0 )r r r r- r+ r r r Útest_groupby_agg_extension* s
|
z+BaseGroupbyTests.test_groupby_agg_extensionc C st t dt d¡t d¡t d¡di¡}t t jdddgddddd gd
|
¡}| d ¡}|jd d }| ||¡ d S )NÚtdz0 days 01:00:00z0 days 01:15:00)r r r z0 days 00:15:00ztimedelta64[ns])ÚdtypeÚaÚb)r4 Úgrpsr8 )r4 Zcumsum)r4 )r r Z Timedeltar( r r2 r* )r r- r Úgbr+ r r r ÚBtest_groupby_agg_extension_timedelta_cumsum_with_named_aggregation: s& ýÿÿ þûÿ
|
zSBaseGroupbyTests.test_groupby_agg_extension_timedelta_cumsum_with_named_aggregationc
|
C sz t ddddddddg|d¡}|jdddj ¡ }t j|dd\}}t j|dd }t jd
|
d d g|d d}| ||¡ d S )Nr r r r r r Fr 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_groupby_extension_no_sortR s z/BaseGroupbyTests.test_groupby_extension_no_sortc C sf || ¡ }t ddddddg|d¡}| d¡j t¡}tjddddddgdd}| ||¡ d S ) Nr r r r r r r r ) Zisnar r r r Z transformÚlenr( r) )r r Zvalidr r+ r- r r r Ú test_groupby_extension_transform[ s
|
z1BaseGroupbyTests.test_groupby_extension_transformc
|
C sx t ddddddddg|d¡}|jddd |¡ |jdddj |¡ |jd dd |¡ |jd ddj |¡ d S )
|
Nr r r r r r F)Z
|
group_keysr )r r r Úapplyr r )r r Zgroupby_apply_opr r r r Ú test_groupby_extension_applyd s
|
z-BaseGroupbyTests.test_groupby_extension_applyc
|
C s¨ t ddddddddg|d¡}| d¡j dd ¡}t j|jjd dd
|
g j|jjddg j|jjdd g j|jjd g jgt jddddgdd dd}| ||¡ d S )Nr r r r r r c S s | j S )N)Úarray)Úxr r r Ú<lambda>m ó z>BaseGroupbyTests.test_groupby_apply_identity.<locals>.<lambda>r é é r/ r r r# )
|
r r r r r> r( r1 r@ r' r) )r r r r+ r- r r r Útest_groupby_apply_identityk s üø
|
z,BaseGroupbyTests.test_groupby_apply_identityc C sæ t ddddddddg|ddddddddgd¡}|j}t|slt|sl|jdkslt|slt|sl|jdkrt dd g¡}|
|
d
|
¡ ¡ j }nJt d g¡}t jtd d |
|
d
|
¡ ¡ j W 5 Q R X |
|
d
|
¡j d dj }t ||¡ d S )Nr r r r )r r r
|
ÚdecimalÚmr r
|
r zdoes not support)ÚmatchT)Z numeric_only)r r r5 r r r r r Úkindr' r ÚsumÚcolumnsÚpytestZraisesÚ TypeErrorr Zassert_index_equal)r r r r5 r- r+ r r r Útest_in_numeric_groupbyz s4 ýÿÿþýüûú z(BaseGroupbyTests.test_in_numeric_groupbyN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r rM ÚmarkZ parametrizer. r3 r: r; r= r? rF rO r r r r r s
|
|
r )rM Zpandas.core.dtypes.commonr r r r Zpandasr Zpandas._testingZ_testingr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
|
