3
���hW< � 7 @ s� d dl Zd dlZd dlZd dlmZmZmZ d dlj Z
d dl m Z dd� Z
ejjddgddgg�ejjd d
d
d d
d gejd
�ejd
�ejd�ejd
�ejd�gg�ejjdddddddddgfddddddddgfddddddddgfddddddddgfddddddddgfddddddddgfdddd d dd dgfdddd d dd dgfd ddd d dd dgfd ddd d dd dgfd dddddddgfd dddddddgfd dddddd dgfd dddddd dgfd ddd ddddgfd ddd ddddgfd ddddd ddgfd dddRdSdTdUdVgfd ddd d dd dgfd dddWdXdYdZd[gfg�d!d"� ���Zejjddgddgg�ejjd ej ej ejdejejejgg�ejjd#ddd$d%d%ejd&ejd'd'gfddd(d)d)d%dd%d*d*gfddd+d%d%d*d d*d'd'gfddd$d'd'ejd&ejd%d%gfddd(d*d*d%dd%d)d)gfddd+d'd'd*d d*d%d%gfddd$ddejd ejddgfddd(d d dddd,d,gfddd+ddd,d d,ddgfddd$ddejd ejddgfddd(d,d,dddd d gfddd+ddd,d d,ddgfd dd$ddejd ejddgfd dd(dddddd-d-gfd dd+ddd-d d-ddgfd dd$ddejd ejddgfd dd(d-d-dddddgfd dd+ddd-d d-ddgfd dd$ddejd ejddgfd dd(d ddddd,d-gfd dd+ddd,d d-ddgfd dd$ddejd ejddgfd dd(d,d-dddd dgfd dd+ddd,d d-ddgfd dd$ddejdejd d gfd dd(dddd dddgfd dd+dddddd d gfd dd$d d ejdejddgfd dd(dddd dddgfd dd+d d dddddgfg �d.d/� ���Zejjddgddgg�ejjd d
d
ejd d
d ejejgejd
�ejd
�ejejd�ejd
�ejd�ejejgg�ejjd0ddd$dddejdddejejgfddd$dddejdddejejgfddd$dddejdddejejgfddd$dddejdddejejgfddd$dddejdddejejgfddd$dddejdddejejgfddd$dd d ejdd dejejgfddd$dd d ejdd dejejgfd dd$dd d ejdd dejejgfd dd$dd d ejdd dejejgfd dd$dddejdddejejgfd dd$dddejdddejejgfd dd$dddejdd dejejgfd dd$dddejdd dejejgfd dd$dd dejdddejejgfd dd$dd dejdddejejgfd dd$dddejd ddejejgfd dd$dd\d]ejd^d_d`ejejgfd dd$dd d ejdd dejejgfd dd$ddadbejdcdddeejejgfddd+dddd-dddd-d-gfddd+dd1d1d2d3d1d4d2d2gfddd+dddd-dddd-d-gfddd+dd4d4d2d5d4d1d2d2gfddd+dddd,dddd,d,gfddd+dd5d5d6d3d5d4d6d6gfddd+dd d d,dd dd,d,gfddd+dd7d7d6d5d7d1d6d6gfd dd+dd d d8dd dd8d8gfd dd+dd7d7dd3d7d4ddgfd dd+dddd8dddd8d8gfd dd+dd3d3dd5d3d1ddgfd dd+dddd,dd dd-d8gfd dd+dd5d1d6d3d7d4d2dgfd dd+dd dd,dddd-d8gfd dd+dd7d4d6d5d3d1d2dgfd dd+ddddd ddddgfd dd+dd1d1dd6d1d4ddgfd dd+dd d ddd dddgfd dd+dd6d6dd1d6d4ddgfg(�d9d:� ���Zejjd;dd d d d d gfdd d d d d gfg�d<d=� �Zd>d?� Zejjd@ddd d d g�ejjdAddg�ejjdBd$d(d+g�ejjdCddg�ejjd dDdDdEdDdFgdDejdEejdFgg�dGdH� �����ZejjdBddIdJg�ejjd@ddd d d g�ejjdAddg�ejjdCddg�ejjd dDdDdEdDdFgdDejdEejdFgdJejd
ejd&gg�dKdL� �����ZdMdN� ZejjdOdJd
gdJdJgddgfdJdJd
d
gdJd
dJd
gd4dd4dgfdJdJd
d
gdJd
dJejgd4ddejgfdJdJd
gdJd
ejgd4dejgfg�dPdQ� �ZdS )f� N)� DataFrame�Series�concat)� DataErrorc C s t jdd�} t jdd�}tjjdddd�}tjjdddd�}ttjjd�| j|�|j|�d��}|jdd g�j j
� }d
d � |jdd g�D �}t |dd �}|j |j
�}t j||� |jdd g�j j
d
d�}dd � |jdd g�D �}t |dd �}|j |j
�}t j||� d S )N�
�d � r i� )�size)�value�key1�key2r r c S s g | ]\}}|j j� �qS � )r
�rank)�.0�key�piecer
r
�@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/groupby/test_rank.py�
<listcomp> s z#test_rank_apply.<locals>.<listcomp>)ZaxisT)�pctc S s g | ]\}}|j jd d��qS )T)r )r
r )r r r r
r
r r " s )�tmZ rands_array�np�random�randintr ZrandnZtake�groupbyr
r r Zreindex�index�assert_series_equal)Zlev1Zlev2Zlab1Zlab2�df�result�expectedr
r
r �test_rank_apply
s$
r �grpsZquxZquux�vals� � � z
2018-01-02z
2018-01-08z
2018-01-06z ties_method,ascending,pct,expZaverageTFg @g @g @g�������?g �?g�������?g�������?�ming @g333333�?�max�first�densec
C sf t j| t|��}|t| � }t||d��}|jd�j|||d�}t|t| � dgd�} tj|| � d S )N)r �valr )�method� ascendingr r) )�columns)r �repeat�lenr r r r �assert_frame_equal)
r r! � ties_methodr+ r �expr r r �exp_dfr
r
r �test_rank_args) s ( r3 z#ties_method,ascending,na_option,expZkeepg �?� g @�topg @g @Zbottomg @g @c
C sf t j| t|��}|t| � }t||d��}|jd�j|||d�}t|t| � dgd�} tj|| � d S )N)r r) r )r* r+ � na_optionr) )r, )r r- r. r r r r r/ )
r r! r0 r+ r6 r1 r r r r2 r
r
r �test_infs_n_nansZ s )
r7 z'ties_method,ascending,na_option,pct,expg �?g �?g �?g �?g �?g �?g �?g @c C sh t j| t|��}|t| � }t||d��}|jd�j||||d�} t|t| � dgd�}
tj| |
� d S )N)r r) r )r* r+ r6 r r) )r, )r r- r. r r r r r/ ) r r! r0 r+ r6 r r1 r r r r2 r
r
r �test_rank_args_missing� s P
r8 zpct,expc C s\ t ddddddddddg
dgd d��}|jd�j| d�}t |d d gd
�}tj||� d S ) N�a�b� r )r r) r )r r" r) )r, )r r r r r/ )r r1 r r r2 r
r
r �test_rank_resets_each_groupg s
&r< c C sN t dgd dgd d��} | jd�j� }t ddddgdgd�}tj||� d S ) Nr9 � r; )r r) r g @r) )r, )r r r r r/ )r r r2 r
r
r �test_rank_avg_even_valss s r> r0 r+ r6 r �bar�fooZbazc
C sJ t dgd |d��}tjtdd�� |jd�j| |||d� W d Q R X d S )Nr@ � )r r) z No numeric types to aggregate)�matchr )r* r+ r6 r )r �pytest�raisesr r r )r0 r+ r6 r r! r r
r
r �test_rank_object_raisesz s
rE �badr; c
C sN t dgd |d��}d}tjt|d�� |jd�j| |||d� W d Q R X d S )Nr@ rA )r r) z3na_option must be one of 'keep', 'top', or 'bottom')rB r )r* r+ r6 r )r rC rD �
ValueErrorr r )r0 r+ r6 r r! r �msgr
r
r �test_rank_naoption_raises� s
rI c C s� d} t dddgdtjdgd��}|j| �jjdd�}td tjdgd
d �}tj||� |j| �jdd�}t d
d tjdgi�}tj ||� d S ) N�Ar r; g �?g @)rJ �BT)r g �?rK )�name)
r r �nanr rK r r r r r/ )�columnr r r r
r
r �test_rank_empty_group� s rO z"input_key,input_value,output_valuec C s> t | |d��}|jd�jddd�}t d|i�}tj||� d S )N)rJ rK rJ r( T)r* r rK )r r r r r/ )Z input_keyZ input_valueZ output_valuer r r r
r
r �test_rank_zero_div� s rP gUUUUUU�?gUUUUUU�?g �?gUUUUUU�?gUUUUUU�?g �?g �?gUUUUUU�?g �?gUUUUUU�?gUUUUUU�?gUUUUUU�?g �?gUUUUUU�?gUUUUUU�?g �?g �?gUUUUUU�?g �?gUUUUUU�?) Znumpyr rC Zpandas�pdr r r Zpandas._testingZ_testingr Zpandas.core.baser r �markZ parametrizeZ Timestampr3 �infrM r7 r8 r< r> rE rI rO rP r
r
r
r �<module> s�
&
& *
"""""" .0 4 &
$ |