Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_rank.cpython-37.pyc
Назад
B �5�g�S � 8 @ sH d dl m Z d dlZd dlZd dlZd dlmZmZmZm Z d dl m Z dd� Z ej�ddgddgg�ejjd d d � d D �e�d �e�d �e�d�e�d �e�d�gejd dd�ejd dd�ejddd�ejd dd�ejddd�ge�d �e�d � e�d �e�d � e�d�e�d � e�d �e�d � e�d�e�d � ge�d ��d�e�d ��d�e�d��d�e�d ��d�e�d��d�gg dd� d�ej�ddddddddd gfdddd d d d d gfdddd d d d dgfdddd d d d d gfd!ddd d dd d gfd!ddd d d d d gfd!ddd"d"d d"dgfd!ddd#d#d d#d gfd$ddd"d"dd"d gfd$ddd#d#d d#d gfd$ddddd ddgfd$ddd d d d d gfd%ddd ddd"d gfd%ddd d d d#d gfd%ddd"d d ddgfd%ddd#d d d d gfd&ddd d d"d dgfd&ddd'd'd d'd(gfd&ddd"d"d d"dgfd&ddd d d'd d(gfg�d)d*� ���Zej�ddgddgg�ej�d ej ej ejd ejejejgg�ej�d+ddd,d-d-ejd.ejd/d/gfddd0d1d1d-dd-d2d2gfddd3d-d-d2d"d2d/d/gfddd,d/d/ejd.ejd-d-gfddd0d2d2d-dd-d1d1gfddd3d/d/d2d"d2d-d-gfd!dd,d d ejd"ejd d gfd!dd0d"d"d dd d4d4gfd!dd3d d d4d"d4d d gfd!dd,d d ejd"ejd d gfd!dd0d4d4d dd d"d"gfd!dd3d d d4d"d4d d gfd$dd,ddejd"ejddgfd$dd0d d dddd5d5gfd$dd3ddd5d"d5ddgfd$dd,ddejd"ejddgfd$dd0d5d5dddd d gfd$dd3ddd5d"d5ddgfd%dd,d dejd"ejd dgfd%dd0d"d d ddd4d5gfd%dd3d dd4d"d5d dgfd%dd,d dejd"ejd dgfd%dd0d4d5d ddd"d gfd%dd3d dd4d"d5d dgfd&dd,d d ejdejd"d"gfd&dd0ddd d"d d d gfd&dd3d d d dd d"d"gfd&dd,d"d"ejdejd d gfd&dd0d d d d"d ddgfd&dd3d"d"d dd d d gfg �d6d7� ���Zej�ddgddgg�ejjd d8d � d9D �e�d �e�d �eje�d�e�d �e�d�ejejgejd dd�ejd dd�ejejddd�ejd dd�ejddd�ejejge�d �e�d � e�d �e�d � eje�d�e�d � e�d �e�d � e�d�e�d � ejejge�d ��d�e�d ��d�eje�d��d�e�d ��d�e�d��d�ejejgg d:d� d�ej�d;ddd,dddejddd ejejgfddd,dd d ejd d d ejejgfddd,dd d ejd d dejejgfddd,dd d ejd d d ejejgfd!dd,dd d ejdd d ejejgfd!dd,dd d ejd d d ejejgfd!dd,dd"d"ejd d"dejejgfd!dd,dd#d#ejd d#d ejejgfd$dd,dd"d"ejdd"d ejejgfd$dd,dd#d#ejd d#d ejejgfd$dd,dddejd ddejejgfd$dd,dd d ejd d d ejejgfd%dd,dd dejdd"d ejejgfd%dd,dd d ejd d#d ejejgfd%dd,dd"d ejd ddejejgfd%dd,dd#d ejd d d ejejgfd&dd,dd d ejd"d dejejgfd&dd,dd'd'ejd d'd(ejejgfd&dd,dd"d"ejd d"dejejgfd&dd,dd d ejd'd d(ejejgfddd3dddd5ddd d5d5gfddd3dd<d<d=d>d<d?d=d=gfddd3dd d d5d d dd5d5gfddd3dd?d?d=d@d?d<d=d=gfd!dd3dd d d4dd d d4d4gfd!dd3dd@d@dAd>d@d?dAdAgfd!dd3dd"d"d4d d"dd4d4gfd!dd3ddBdBdAd@dBd<dAdAgfd$dd3dd"d"dCdd"d dCdCgfd$dd3ddBdBd d>dBd?d d gfd$dd3ddddCd dddCdCgfd$dd3dd>d>d d@d>d<d d gfd%dd3dd dd4dd"d d5dCgfd%dd3dd@d<dAd>dBd?d=d gfd%dd3dd"d d4d ddd5dCgfd%dd3ddBd?dAd@d>d<d=d gfd&dd3dd d d d"d dd d gfd&dd3dd<d<d dAd<d?d d gfd&dd3dd"d"d d d"dd d gfd&dd3ddAdAd d<dAd?d d gfg(�dDdE� ���Zej�dFdd"d"d"d"d"gfdd#d#d#d#d#gfg�dGdH� �Zej�dIdJdKdLdMdNdOg�ej�dPddg�dQdR� ��Zej�dSdd!d$d%d&g�ej�dTddg�ej�dUd,d0d3g�ej�dVddg�ej�d dWdWdXdWdYgdWejdXejdYgg�dZd[� �����Zej�dUdd\d]g�ej�dSdd!d$d%d&g�ej�dTddg�ej�dVddg�ej�d dWdWdXdWdYgdWejdXejdYgd]ejd^ejd.gg�d_d`� �����Zdadb� Zej�dcd]d^gd]d]gd d gfd]d]d^d^gd]d^d]d^gd?d d?d gfd]d]d^d^gd]d^d]ejgd?d d ejgfd]d]d^gd]d^ejgd?d ejgfg�ddde� �Z dfdg� Z ej�dhddg�didj� �Z dkdl� Z dmdn� Z dodp� Z!dS )q� )�datetimeN)� DataFrame�NaT�Series�concatc C s t �dd�} t �dd�}tjjdddd�}tjjdddd�}ttj�d�| �|�|�|�d��}|�dd g�j � � }d d � |�dd g�D �}t |dd �}|� |j �}t �||� |�dd g�j j d d�}dd � |�dd g�D �}t |dd �}|� |j �}t �||� d S )N� �d � r i� )�size)�value�key1�key2r r c S s g | ]\}}|j �� �qS � )r �rank)�.0�key�piecer r �T/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_rank.py� <listcomp> s z#test_rank_apply.<locals>.<listcomp>)�axisT)�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�valsc C s$ g | ] }t jd d dd dg|d��qS )� � � )�dtype)r �array)r r' r r r r 3 s r ) �i8�i4�i2�i1�u8Zu4�u2�u1�f8�f4�f2z 2018-01-02z 2018-01-08z 2018-01-06z US/Pacific)�tz�Dc C s t | d �S )Nr )�type)�xr r r �<lambda>T � r7 )Zidsz ties_method,ascending,pct,expZaverageTFg @g @g @g�������?g �?g�������?g�������?�ming @g333333�?�max�first�densegUUUUUU�?gUUUUUU�?c C s� t �| t|��}|}t|�t| � }t|t j�r@t j||jd�}t||d��}|� d�j |||d�} t|t| � dgd�} t � | | � d S )N)r' )r �valr )�method� ascendingr r= )�columns) r �repeat�len�list� isinstance�ndarrayr( r' r r r r �assert_frame_equal) r" r# � ties_methodr? r �expr � orig_valsr r �exp_dfr r r �test_rank_args/ s A rK z#ties_method,ascending,na_option,expZkeepg �?� g @�topg @g @�bottomg @g @c C sf t �| t|��}|t| � }t||d��}|�d�j|||d�}t|t| � dgd�} t�|| � d S )N)r r= r )r> r? � na_optionr= )r@ )r rA rB r r r r rF ) r" r# rG r? rO rH r r r rJ r r r �test_infs_n_nans~ s ) rP c C s0 g | ](}t jd d t jdd dt jt jg|d��qS )r$ r% r& )r' )r r( �nan)r r' r r r r � s )r0 r1 r2 c C s t | d �S )Nr )r5 )r6 r r r r7 � r8 z'ties_method,ascending,na_option,pct,expg �?g �?g �?g �?g �?g �?g �?g @c C s� t �| t|��}|}t|�t| � }t|t j�r@t j||jd�}t||d��} | � d�j ||||d�} t|t| � dgd�} t � | | � d S )N)r' )r r= r )r> r? rO r r= )r@ ) r rA rB rC rD rE r( r' r r r r rF ) r" r# rG r? rO r rH r rI r r rJ r r r �test_rank_args_missing� s r rR zpct,expc C s\ t ddddddddddg dgd d��}|�d�j| d�}t |d d gd �}t�||� d S ) N�a�b� r )r r= r )r r$ r= )r@ )r r r r rF )r rH r r rJ r r r �test_rank_resets_each_group� s &rV r' �int64Zint32Zuint64Zuint32�float64Zfloat32�upperc C s� |r(| d � � | dd � } | �dd�} tdgd dgd d��}|d �| �|d< |d j| ksft�|�d ��� }td d d d gdgd �}t� ||� d S ) Nr rU ZUiZUIrS � )r r= r= r g @)r@ ) rY �replacer �astyper' �AssertionErrorr r r rF )r' rY r r rJ r r r �test_rank_avg_even_vals� s r^ rG r? rO r �bar�fooZbazc C s� t dgd |d��}|d �� }|�d�}|j| |||d�}|�� rht dgd dtjdtjd gd��} n t dgd ddddd gd��} | �d�} | j| |||d�} t�|| � d S ) Nr` � )r r= r= r )r> r? rO r r r$ rU ) r Zisnar r �anyr rQ r rF ) rG r? rO r r# r �mask�gb�resZdf2Zgb2�altr r r �test_rank_object_dtype� s $ rg �badrU r$ c C sN t dgd |d��}d}tjt|d�� |�d�j| |||d� W d Q R X d S )Nr` ra )r r= z3na_option must be one of 'keep', 'top', or 'bottom')�matchr )r> r? rO r )r �pytestZraises� ValueErrorr r )rG r? rO r r# r �msgr r r �test_rank_naoption_raises� s rm c C s� d} t dddgdtjdgd��}|�| �jjdd�}td tjdgd d �}t�||� |�| �jdd�}t d d tjdgi�}t� ||� d S ) N�Ar rU g �?g @)rn �BT)r g �?ro )�name) r r rQ r ro r r r r rF )�columnr r r r r r �test_rank_empty_group s rr z"input_key,input_value,output_valuec C s> t | |d��}|�d�jddd�}t d|i�}t�||� d S )N)rn ro rn r< T)r> r ro )r r r r rF )Z input_keyZ input_valueZ output_valuer r r r r r �test_rank_zero_div s rs c C s� t dddgt�tj�jt�tj�jt�tj�jgttddd�tgd��} | �d�� � }t dddgtj dtj gd��}t � ||� d S ) NrU r$ i� )�grp�int_col� datetimelikert g �?g @)ru rv ) r r ZiinforW r9 r: r r r r �NaNr rF )r r r r r r �test_rank_min_int" s rx �use_nanc C s� | r t jnd}tddgddgd|gd|ggddgd�}|�dg�d jdd d �}| rntd dt jt jgdd �}ntd dddgdd �}t�||� d S )NrL ���rU r$ �groupr= )r@ r< T)r> r g �?)rp gUUUUUU�?gUUUUUU�?)r rQ r r r r r r )ry Z fill_valuer r r r r r �.test_rank_pct_equal_values_on_group_transition8 s r| c C s� t tddgddgd��tddgdd gd ��d �dd �} | jd dd�}|jdd �}t | d jdd �| d jdd �gdddgd�}t�||� d S )NrL rZ rU r$ )Zcol1Zcol2ra r& � r% )Zcol3Zcol4)rS rT )r r )�levelr rS rT )r �keys)r r r r r rF )r rd r r r r r �test_rank_multiindexQ s r� c C s� t ddddgddddgd d d d gd �ddddgd�} | jddd�}|jdd�}t| jd jdd�| jd jdd�gdd�}t�||� |jdd�}|�|�r�t�d S )NrU rL ra r} r$ rZ r& r% g �?g @g @g @)r rU r$ rS rT )r r )r~ r )r ) r r r r �locr rF �equalsr] )r rd re r rf r r r � test_groupby_axis0_rank_axis1i s " , r� c C s~ t ddddgddddgd d d d gd �ddddgd�} | jddd�}|jdd�}| ddg �tj�}|d |d< t�||� d S )NrU rL ra r} r$ rZ r& r% g �?g @g @g @)r rU r$ rS rT )r r )r~ r )r )r r Zcummaxr\ r rX r rF )r rd Zcmaxr r r r � test_groupby_axis0_cummax_axis1} s " r� )"r �numpyr rj Zpandas�pdr r r r Zpandas._testingZ_testingr r! �markZ parametrize� TimestampZ to_periodrK �infrQ rP rR rV r^ rg rm rr rs rx r| r� r� r� r r r r �<module> sV && * """""" .0 4 & $,
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка