3
���ha
� @ s8 d dl Zd dlZd dlmZmZ d dljZd dl m
Z
m Z ej dd� �Z
ej dd� �Zej dd � �Zej d
d � �Zej d d
� �Zej dd� �Zej dd� �Zej ee
�d�dd� �Zej ee �d�dd� �Zej ee
�ee � d�dd� �Zej ddgd�dd � �Zej ddgd�d d � �Zej ddgd�d d � �ZdS )!� N)� DataFrame�
MultiIndex)�reduction_kernels�transformation_kernelsc
C sn t ddddgdddggdddd d d
d
d d d g
dd d
dd d d
dd d
g
gd d
gd�} ttjjdd �| dddgd�S )N�foo�barZbazZqux�one�two�threer � � � �first�second)Zlevels�codes�names�
�A�B�C)�index�columns)r r �np�random�randn)r � r �?/tmp/pip-build-5_djhm0z/pandas/pandas/tests/groupby/conftest.py�mframe s
. r c
C sB t ddddddddgddddddddgtjjd�tjjd�d��S )Nr r r r r
� )r r r �D)r r r r r r r r �df s
r c C s t j� S )N)�tmZmakeTimeSeriesr r r r �ts s r" c C s t j� S )N)r! ZgetTimeSeriesDatar r r r �tsd$ s r# c C s t | �S )N)r )r# r r r �tsframe) s r$ c
C sL t ddddddddgddddddddgtjjd�tjtjjd�dd�d ��S )
Nr r r r r
r Zfloat32)Zdtype)r r r r )r r r r �arrayr r r r �df_mixed_floats. s
r& c C sp t dddddddddddg dddddddddddg dddddddddddg tjjd�tjjd�tjjd�d��S ) Nr r r r ZdullZshiny� )r r r r �E�F)r r r r r r r r � three_group: sJ
r* )�paramsc C s | j S )zT
yields the string names of all groupby reduction functions, one at a time.
)�param)�requestr r r �reduction_funcl s r. c C s | j S )z@yields the string names of all groupby transformation functions.)r, )r- r r r �transformation_funct s r/ c C s | j S )z5yields both aggregation and transformation functions.)r, )r- r r r � groupby_funcz s r0 TFc C s | j S )z'parallel keyword argument for numba.jit)r, )r- r r r �parallel� s r1 c C s | j S )z$nogil keyword argument for numba.jit)r, )r- r r r �nogil� s r2 c C s | j S )z'nopython keyword argument for numba.jit)r, )r- r r r �nopython� s r3 )Znumpyr ZpytestZpandasr r Zpandas._testingZ_testingr! Zpandas.core.groupby.baser r Zfixturer r r" r# r$ r& r* �sortedr. r/ r0 r1 r2 r3 r r r r �<module> s"
2 |