U
|
ý°dT$ ã @ sò d dl Zd dlZd dlmZ d dlmZmZm Z m
|
Z
|
d dl m Z d dl mZ d dlmZ G dd dZG dd d eZG d
|
d d eZej d d dg¡dd Zdd Zej dddd dg¡dd Zdd Zdd Zd d Z d d Z d d! Z dS )"é N)Úgroupby)Ú group_cumprodÚ group_cumsumÚ
|
group_meanÚ group_var)Úensure_platform_int)Úisnac @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚGroupVarTestMixinc C sÎ t j d¡}t jt d¡ | j¡}t jddd}d| dd¡ | j¡ }t t
|
d¡d ¡ d
|
¡}t |¡j d d d j dddd d d
t jf }|d }| ||||¡ t ||| j¡s¾tt ||¡ d S )NéÒ )é é r Úint64©Údtypeé
|
é r )é Úintp)r r ÚF©Úorder)ÚaxisÚddofé r )ÚnpÚrandomÚ RandomStateÚnanÚonesÚastyper ÚzerosÚrandÚtileÚarangeZsqueezeÚreshapeÚstdZnewaxisÚalgoÚallcloseÚrtolÚAssertionErrorÚtmÚassert_numpy_array_equal©ÚselfÚprngÚoutÚcountsÚvaluesÚlabelsZ expected_outZexpected_counts© r3 ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_libgroupby.pyÚtest_group_var_generic_1d s ÿþz+GroupVarTestMixin.test_group_var_generic_1dc C s® t j d¡}t jt d¡ | j¡}t jddd}d| dd¡ | j¡ }t jddd}t |j
|
dd d
|
gg¡}|d }| ||||¡ t ||| j ¡stt ||¡ d S ) Nr
|
©r r r r r r r r )r r )r r r r r r r r r! Úarrayr% r& r' r( r) r* r+ r, r3 r3 r4 Ú%test_group_var_generic_1d_flat_labels$ s z7GroupVarTestMixin.test_group_var_generic_1d_flat_labelsc C s¼ t j d¡}t jt d¡ | j¡}t jddd}d| dd¡ | j¡ }t t
|
d¡d¡ d ¡}t j | ddd¡d
|
d d d }|d }| ||||¡ t ||| j¡s¬tt ||¡ d S ) Nr
|
©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' r( r) r* r+ r, r3 r3 r4 Ú$test_group_var_generic_2d_all_finite4 s z6GroupVarTestMixin.test_group_var_generic_2d_all_finitec C sö t j d¡}t jt d¡ | j¡}t jddd}d| dd¡ | j¡ }t j|d d
df< t t
|
d¡d ¡ d
|
¡}t |d d
d f j ddd d j dddd t jt d¡ g¡j | j¡}|d }| ||||¡ tj||dd t ||¡ d S )Nr
|
r9 r r r r r r r: r r r r r; gíµ ÷Æ >©r( )r r r r r r r r r! r" r# Zvstackr$ r% ÚTr& r* Úassert_almost_equalr+ r, r3 r3 r4 Ú"test_group_var_generic_2d_some_nanC s" (þÿûz4GroupVarTestMixin.test_group_var_generic_2d_some_nanc C s t jt jgg| jd}t jdgdd}dt jd| jd }t jddd}| ||||¡ |d dksht|d dksxtt |d d ¡ d S )
|
Nr r r gUUUÕª¦ê?)r r r r ©r r g )
|
r r7 r r r r r& r) r* r? )r- r/ r0 r1 r2 r3 r3 r4 Útest_group_var_constantX s z)GroupVarTestMixin.test_group_var_constantN)Ú__name__Ú
|
__module__Ú __qualname__r5 r8 r< r@ rB r3 r3 r3 r4 r s
|
r c @ s* e Zd ZdZeeZejZ dZ
|
dd Z dS )ÚTestGroupVarFloat64Tgñhãµøä>c C s t j d¡}t jt jgg| jd}t jdgdd}| d¡d | j¡}d|_t j ddd}|
|
||||¡ |d dks|t t j |d d
|
d d d S ) Nr
|
r r r é@B l J)£)rG r r rA gUUUUUUµ?gü©ñÒMb@?r= )r r r r7 r r r! r Úshaper r& r) r* r? )r- r. r/ r0 r1 r2 r3 r3 r4 Útest_group_var_large_inputsn s z/TestGroupVarFloat64.test_group_var_large_inputsN) rC rD rE Ú__test__Ú staticmethodr r& r Úfloat64r r( rI r3 r3 r3 r4 rF g s
|
rF c @ s" e Zd ZdZeeZejZ dZ
|
dS )ÚTestGroupVarFloat32Tg{®Gáz?N) rC rD rE rJ rK r r& r Úfloat32r r( r3 r3 r3 r4 rM } s rM r rN rL c C s4 t jt j d¡| d}t dddg¡}t d| ¡}t jt|t jd}tt t d¡t
|
t j d|f ¡¡}t j }||||d d
d f | dd }t ||d d
||dd
||dd
g¡}t ||¡ t |t jddd
|
gt jd¡ t j|d d
< ||||d d
d f | t j|d< t ||¡ d S ) Né r é é )r é r r c S s6 t | ¡ rt tjd¡S | d | ¡ | ¡ | d gS )NrR r éÿÿÿÿ)r Úallr Úrepeatr ÚmaxÚmin)Úgroupr3 r3 r4 Ú_ohlc s z test_group_ohlc.<locals>._ohlcé )r r7 r Zrandnr Úlenr r rU r# ZdiffZr_Ú
|
libgroupbyZ
|
group_ohlcr* r? r+ r ) r ÚobjZbinsr/ r0 r2 ÚfuncrY Úexpectedr3 r3 r4 Útest_group_ohlc
s &4
|
r` c C s| d}t jdgdgdgdgg|d}t |¡}t jddddgt jd}d}| ||||| tj|||dd
df dd dS )
|
a
|
Check a group transform that executes a cumulative function.
|
|
Parameters
|
----------
|
pd_op : callable
|
The pandas cumulative function.
|
np_op : callable
|
The analogous one in NumPy.
|
dtype : type
|
The specified dtype of the data.
|
Fr r r rR r r N)Z check_dtype)r r7 Ú
|
zeros_liker r* r+ )Úpd_opÚnp_opr Úis_datetimelikeÚdataZanswerr2 Úngroupsr3 r3 r4 Ú(_check_cython_group_transform_cumulative¡ s
|
rg Únp_dtyper Zuint64c C s( t | ¡j}tt j }}t||| d S ©N)r r Útyper Zcumsumrg )rh r rb rc r3 r3 r4 Ú"test_cython_group_transform_cumsumº s rk c C s" t j} tt j }}t||| d S ri )r rL r Zcumprodrg )r rb rc r3 r3 r4 Ú#test_cython_group_transform_cumprod s rl c
|
C s¤ d} t jdddddgt jd}d}t jdgdgdgt jgdggdd}t |¡}| t j¡ t||||| t jddd t jd
|
gdd}t |d d
df |¡ t |¡}| t j¡ t ||||| t jddd t jd gdd}t |d d
df |¡ d } t jt
|
dd ¡gd ddd d
d f }t j|dd}t || d¡||| t t
|
dd ¡t
|
dd ¡t
|
dd ¡t
|
dd ¡t
|
dd ¡g¡}t |d d
df d¡|¡ d S )NFr r r r r rR rL rP é r TÚnsr úm8[ns]r ) r r7 r r ra Úfillr r* r+ r Ú timedelta64Úview)rd r2 rf re Úactualr_ r3 r3 r4 Ú!test_cython_group_transform_algosÉ s6 $
|
|
(
|
|
|
|
|
ûÿ rt c C s® t jddd} t jdgdd}t jt dd¡t d d¡t d
|
¡gd dd d
d f d¡ d¡}t jt|t jd}t| |||d d t
|
| d d
df t jdgdd¡ d S )Nr6 rL ©rH r r r r r rn rR ÚNaTro T©rd r ) r r r7 rq rr r r[ r r r* r+ ©rs r0 re r2 r3 r3 r4 Ú#test_cython_group_mean_datetimelikeï s þ
|
ýüûÿry c C sx t jddd} t jddd}t jdddd d
d f }t jdt jd}tjtdd t| |||d d
|
d W 5 Q R X d S ) Nr6 rL ru r r r Ú min_count)ÚmatchTr )rd rz )r r r ÚpytestZraisesr) r rx r3 r3 r4 Ú&test_cython_group_mean_wrong_min_count s r} c C s¸ t jddd} t jdgdd}t jt d¡t d¡gddd d
d f d¡ d¡}t jt|t jd}t| |||d d
|
t
|
| d d
df t jt t |d |d ¡d ¡dd¡ d S ) Nr6 rL ru r r r rv ro Frw r r ) r r r7 rq rr r r[ r r r* r+ ÚdivideÚaddrx r3 r3 r4 Ú:test_cython_group_mean_not_datetimelike_but_has_NaT_values s& þ
|
ýüûÿ $ÿr ) Únumpyr r| Z pandas._libsr r\ Zpandas._libs.groupbyr r r r Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr* r rF rM ÚmarkZ parametrizer` rg rk rl rt ry r} r r3 r3 r3 r4 Ú<module> s& U
|
|
&
|
