Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/methods/__pycache__/test_rank.cpython-37.pyc
Назад
B �5�gKD � @ s� d dl mZmZ d dlZd dlZd dlmZmZ d dl m m Z d dl mZmZmZmZ d dlmZ d dlmZ G dd� d�Zej�dd d d g�ej�d d gdgfd dgddgfddgddgfd ddgdddgfd ddgdddgfddd gdddgfd d dddgdddddgfd d ddddgddddddgfdddddgdd d d dgfg �d d � ��Zej�dd d d g�ej�d d gdgfd dgddgfddgddgfd ddgdddgfd ddgdddgfddd gdddgfd d dddgddd d d gfd d ddddgd!d!ddd"d"gfdddddgdd d d dgfg �d#d$� ��Zej�dd d d g�ej�d d gdgfd dgddgfddgddgfd ddgdddgfd ddgdddgfddd gdddgfd d dddgd d ddd gfd d ddddgddddddgfdddddgdd d d dgfg �d%d&� ��Z ej�dd d d g�ej�d d gdgfd dgddgfddgd'd'gfd ddgdddgfd ddgdd"d"gfddd gdddgfd d dddgd(d(d)d)d gfd d ddddgd*d*d+d+d,d,gfdddddgdd d d dgfg �d-d.� ��Z ej�dd d g�ej�d d gdgfd dgddgfddgddgfd ddgdddgfd ddgdddgfddd gdddgfd d dddgdd d dd gfd d ddddgd!dddd"dgfdddddgdd d d dgfg �d/d0� ��Z ejj ejj d1d2� ��Z!dS )3� )�chain�productN)�Infinity� NegInfinity)�NaT�Series� Timestamp� date_range)�CategoricalDtypec @ sv e Zd Zeddddejdddejdg �Ze�dddd ejd dd ejdg �e�ddd dejddd ejdg �e�dd d dejddd ejd g �e�ddd dejddd ejd g �e�ddddejdddejdg �d�Zdd� Z dd� Z dd� Z dd� Z e jej�dddg�ej�ddd d d d g�ej�d d!d"d#g�d$d%� ����Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4S )5�TestSeriesRank� � � � � g �?g @g @g @g @� � � )�average�min�max�first�densec C sL t �d� t �d�}tj|d d d�< d|d d� d d d�< |�� }|�d��� }t�||� t�|�}|� tj �}t ||�|j dd �}tj||< t�||� t t� d ��d��}|�� } |�t��� }t�| |� t t� d ��d }|d }|jd d�} t�| |� t t�dd��}t t�dd��}|jd d�} t�| |� tj|d< t t�dd��}tj|d< |jd d�} t�| |� t t� d ��d }tj|d< |d }|jd d�} t�| |� t t�tjd��}|�� }|jd d�} t�| |� t t� d ��d }tj|d< |d }|jd d�} t�| |� tdd d�} t t� d �| �d }tj|jd< |d }|jd d�} t�| |� t ddddddg�}t dddd dd g�}|�� } t�| |� t d d d tgd d!�}t dddtjg�}|�� } t�| |� tjd"d#d$d%d&d'd(dd)dd*g d+d!�} tj�t| ��} t | | �}t | d d+d!�}|�� } t�| |� d S ),Nzscipy.stats.specialzscipy.stats.rankdatar g @� r �O�ts)�index�namer g �?g @T)�pctr �d g)\��(�?g�@�_)�?r z1/1/1990)Zperiodsg ��Jz�5g0��.�++g#B� ���;g{�G�z�?gIf� ���;g�������?g @z3 dayz 1 day 10mz-2 dayzm8[ns])�dtypei������g#B� ��ǻg�}���g ��Jzr g�Ww'&l�7g�����|�=�( �float64)�pytest� importorskip�np�nan�rank�astype�tm�assert_series_equal�isnanZfillna�infr r �arange�repeat�float�copyr Zilocr �array�randomZ permutation�len) �selfZdatetime_series�rankdata�ranksZoranks�maskZfilled�exp�iseriesZiranks�rng�valuesZ random_order� r= �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/methods/test_rank.py� test_rank$ s� zTestSeriesRank.test_rankc C sV t ddddddg�}t ddddddg�}t ddd d d d g��tddd d d d gd d��}t�|�� |� t�|jdd�|� t ddd d d d g��tddd d d d gdd��}t ddddddg�}|�� }t�||� t ddddddg��tddddddgd��}t ddddddg�}|�� } t�| |� t ddd d d d tjg��tddd d d d dgd ��} t dddddddg�} t dddddddg�} t ddddddtjg�} t�| jdd�| � t�| jdd�| � t�| jd d�| � t dddddddg�} t dddddddg�} t ddddddtjg�} t�| jddd �| � t�| jddd �| � t�| jd dd �| � d }tj t |d �� | jd dd � W d Q R X tj t |d �� | jd dd � W d Q R X t ddd d tjg��tddd d gd ��} t d!d"d#dd$g�} t d$d!d"d#dg�} t d%d&d'dtjg�} t�| jdd d(�| � t�| jdd d(�| � t�| jd d d(�| � d S ))Ng �?g @g @g @g @g @r �second�thirdZfourthZfifthZsixthT)� categories�orderedF)� ascendingr r r r r r Zseventhg @�top)� na_option�bottom�keep)rF rD z3na_option must be one of 'keep', 'top', or 'bottom')�match�badg�������?g333333�?g�������?g�������?g �?g �?g �?)rF r ) r r) r r* r+ r( r& �NaNr$ �raises� ValueError)r5 r9 Zexp_descrC � unorderedZ exp_unordered�resZ unordered1Zexp_unordered1Zres1Zna_serZexp_topZexp_botZexp_keep�msgr= r= r> �test_rank_categorical| sj z$TestSeriesRank.test_rank_categoricalc C sD t ddg�}|jdd� d}tjt|d�� |�d� W d Q R X d S )Nr r r )�methodz,No axis named average for object type Series)rI )r r( r$ rL rM )r5 �srP r= r= r> �test_rank_signature� s z"TestSeriesRank.test_rank_signaturec C sz | j }ddd�}d tg}tdfh}| j}xLt||�D ]>\}}||f|krJq4|d krV|n|�|�}|||| |d� q4W d S )Nr c S s | j |d�}t�|t|�� d S )N)rR )r( r* r+ r )rS �expectedrR �resultr= r= r> �_check� s z4TestSeriesRank.test_rank_tie_methods.<locals>._checkr )rR )r )rS �object�resultsr r) ) r5 rS rW �dtypes�disabledrY rR r Zseriesr= r= r> �test_rank_tie_methods� s z$TestSeriesRank.test_rank_tie_methodsrD TFrR r r r r r rF rE rG rH c s� dd t � t� fdtjtjtj fg}d� dh}� fdd�}xZ|D ]R\}}} } | g� |g� | g� } t| |d�} ||f|kr�q@|| |||� q@W d S )NrX r# r )rX r c s6 dddgdddgdddgfdddgdddgdddgfdddgdddgd d d gfdddgdddgddd gfdddgdddgdddgfd �}|| }|d kr�|d |d |d g}n:|d kr�|d |d |d g}n|d t jg� |d g}|r�|n |d d d� }tt�|��}| j|||d�}t�|t|dd�� d S )Nr r r r r r r r � )r r r r r rE r rG r! )rR rF rD r# )r ) r& r' �listr � from_iterabler( r* r+ r ) rS rR rF rD Z exp_ranksr7 �orderrU rV )�chunkr= r> rW � s zATestSeriesRank.test_rank_tie_methods_on_infs_nans.<locals>._check)r )r r r& r' r- r ) r5 rR rF rD rZ r[ rW r Zna_valueZpos_infZneg_infZin_arrr: r= )ra r> �"test_rank_tie_methods_on_infs_nans� s z1TestSeriesRank.test_rank_tie_methods_on_infs_nansc C sN t dtjtjtj dg�}|jdd�}t dtjdddgdd �}t�||� d S ) Nr � F)rD r r r r# )r )r r& r' r- r( r* r+ )r5 r: rV r9 r= r= r> � test_rank_desc_mix_nans_infs s z+TestSeriesRank.test_rank_desc_mix_nans_infsc s� t �d� t �d�}tj�d�� t�� fdd�tddd�D ��� tj�� � dd� tt� ��D �}xt� � d � d gD ]^}t ||d �}xLd D ]D}|j |d �}|||dkr�|nd�}t ||d �� d�}t � ||� q�W qvW d S )Nzscipy.stats.specialzscipy.stats.rankdatar] c s g | ]}� |d � �qS )Nr= )�.0�i)�xsr= r> � <listcomp> s z;TestSeriesRank.test_rank_methods_series.<locals>.<listcomp>r r c S s g | ]}t td �| ��qS )�a)�chr�ord)re rf r= r= r> rh # s g ��.Ag���ư>)r )r r r r r )rR r Zordinalr# )r$ r% r& r3 ZrandnZ concatenate�range�shuffler4 r r( r) r* r+ ) r5 r6 r �valsr �mrV ZsprankrU r= )rg r> �test_rank_methods_series s z'TestSeriesRank.test_rank_methods_seriesc C s� dddg}dgdgfdgdgfdgdgfddgddgfdddgdddgfdddgdddgfddd d dgdddddgfd d d d dgddddd gfg}xT|D ]L\}}xB|D ]:}t |��|�}|jdd�}t |��|j�}t�||� q�W q�W d S )Nr �f8�i8r r r r r r ������������r! r )rR )r r) r( r r* r+ ) r5 rZ Zin_out�serr9 r rS rV rU r= r= r> �test_rank_dense_method. s z%TestSeriesRank.test_rank_dense_methodc C s� dddg}x�t || j�D ]�\}}d|kr4| j�� }n | j�|�}|jdd�}|�� | �� }t�||� |dkrz|dkrzq|�� | j|d�}|j|dd �}t�||� qW d S ) Nr rq rr rf F)rD r )rR )rR rD ) r rY rS �dropnar) r( r r* r+ )r5 rZ r rR rS rO rU Zres2r= r= r> �test_rank_descendingB s z#TestSeriesRank.test_rank_descendingc C sX | j �� �d�}xB| j�� D ]4\}}|j|d�}t|��� }|j|_t� ||� q W d S )Nrr )rR ) rS ry r) rY �itemsr( r r r* r+ )r5 rS rR rO rV rU r= r= r> � test_rank_intV s z TestSeriesRank.test_rank_intc C s@ t tjgd ��t�jdd� t tjgd ��t�jdd� d S )N� T)rD F)r r& r' r) rX r( )r5 r= r= r> �test_rank_object_bug_ s z#TestSeriesRank.test_rank_object_bugc C s4 t td�tg�}|�� }|�� |}t�||� d S )Nz2017-01-05 10:20:27.569000)r r r r1 r( r* r+ )r5 rS rU rV r= r= r> �test_rank_modify_inplacef s z'TestSeriesRank.test_rank_modify_inplaceN)�__name__� __module__� __qualname__r r& r' rS r2 rY r? rQ rT r\ �tdZskip_if_no_scipyr$ �mark� parametrizerb rd rp rx rz r| r~ r r= r= r= r> r s* &XX"$ r r r rq rr zser, expr g �?r g �?r gUUUUUU�?gUUUUUU�?r r rs rt ru rv r! g�������?g�������?g333333�?g�������?c C s< t |��| �}|jddd�}t |��|j�}t�||� d S )Nr T)rR r )r r) r( r r* r+ )r rw r9 rS rV rU r= r= r> �test_rank_dense_pctt s r� gUUUUUU�?g�������?c C s< t |��| �}|jddd�}t |��|j�}t�||� d S )Nr T)rR r )r r) r( r r* r+ )r rw r9 rS rV rU r= r= r> �test_rank_min_pct� s r� c C s< t |��| �}|jddd�}t |��|j�}t�||� d S )Nr T)rR r )r r) r( r r* r+ )r rw r9 rS rV rU r= r= r> �test_rank_max_pct� s r� g �?g333333�?g�������?g �?g�������?gUUUUUU�?c C s< t |��| �}|jddd�}t |��|j�}t�||� d S )Nr T)rR r )r r) r( r r* r+ )r rw r9 rS rV rU r= r= r> �test_rank_average_pct� s r� c C s< t |��| �}|jddd�}t |��|j�}t�||� d S )Nr T)rR r )r r) r( r r* r+ )r rw r9 rS rV rU r= r= r> �test_rank_first_pct� s r� c C s. t t�d��} | jdd��� }|dks*t�d S )Ni T)r r )r r& r. r( r �AssertionError)rS rV r= r= r> �test_pct_max_many_rows� s r� )"� itertoolsr r �numpyr& r$ Zpandas._libs.algosr r Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r r r Zpandas._testingZ_testingr* Zpandas.api.typesr r r� r� r� r� r� r� r� ZsingleZ high_memoryr� r= r= r= r> �<module> s� ] * * * * *
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка