Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_rank.cpython-37.pyc
Назад
B �5�g9= � @ sp d dl m Z mZ d dlZd dlZd dlmZmZ d dlm m Z d dl m Z mZ d dlmZ G dd� d�ZdS )� )�datetime� timedeltaN)�Infinity� NegInfinity)� DataFrame�Seriesc @ s� e Zd Zeddddejdddejdg �Zeeed��Ze� ddd d ejd dd ejdg �e� ddd dejddd ejdg �e� ddd dejddd ejdg �e� ddd dejddd ejdg �e� ddddejdddejdg �d�Z e j dddddgd�dd� �Z ejdd� �Zdd� Zd d � Zd d � Zejd d!� �Zd"d#� Zejd$d%� �Zeje j�d&d'd(d)g�e j�d*�d+d,� ���Ze j�d-d.dg�e j�d&d/e g�d0d1� ��Z e j�d2dd3d3d3gd3d4d5gd3d4d6ggfdd6d3d3gd6d6d5gd6d6d6ggfdd3d3d3gd3d5d5gd3d5d6ggfdd5d3d3gd5d4d5gd5d4d6ggfdd6d3d3gd5d6d5gd3d5d6ggfg�d7d8� �Z e jj e jj d9d:� ��Z!e j�d;ej" d<d=d>d?d@d.dAdBdCddDej"g dEfej" d<d=d>d?dFd.dAdBdCddDej"g dGfe�#ej$�j%dddHe�#ej$�j&gdIfe�#ej'�j%dJd.ddKdLdMe�#ej'�j&gdNfe(� dOdPdQdRdSe)� gdTfe*dUdd�e*dUdd�e*dUdd�gdVfg�dWdX� �Z+dYdZ� Z,d[d\� Z-e j�d]ddd^gdddgd_�ed`d3dadbgi�fd^ddd^giee.d�dc�fg�ddde� �Z/d/S )f�TestRank� � � � � )�A�Bg �?g @g @g @g @� � � )�average�min�max�first�denser r r r r )�paramsc C s |j S )z5 Fixture for trying all rank methods )�param)�self�request� r �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_rank.py�method" s zTestRank.methodc C sD dd l }ddl m} tj|d d d d�< tj|d d d d�< tj|d d d d�< tj|d d d d �< |�� }|�d �}t�|j�}|�tj�j}t� |d|�}tj||< t� |d |�} tj| |< t � |j|� t � |j| � t tj jdd d d ��d��} | �� } | �t��� } t �| | � | �d �} | �t��d �} t �| | � d S )Nr )�rankdatar r r r �Cr �Dr r �( )�size)� r )� scipy.statsr �np�nan�rank�isnan�values�fillna�inf�apply_along_axis�tm�assert_almost_equalr �random�randintZreshape�astype�float�assert_frame_equal) r � float_frame�scipyr �ranks0�ranks1�mask�fvals�exp0�exp1�df�result�expr r r � test_rank) s. zTestRank.test_rankc C s� t dddgdddgg�}t dddgdddgg�d }|jddd�}t�||� t dddgdddgg�}|�d �d }|jd dd�}t�||� t d d d gd d d gg�}t dddgdddgg�}|jdd d�}t�||� t dddgdddgg�}|jd d d�}t�||� t d tjd gd d d gg�}t dtjdgdddgg�}|jdd d�}t�||� t dtjdgdddgg�}|jd d d�}t�||� tddd�tjtddd�gtddd�tddd�tddd�gg}t |�}t dtjdgdddgg�}|jdd dd�}t�||� t dtjdgdddgg�}|jdd d d�}t�||� t d dddddddgi�}t d ddddd d dgi�}t�|�� |� d S ) Nr r r g �?g @g @T)�pctr �b�c�aF)� numeric_onlyg �?i� r i� )rE � ascendingg#B� ���;���r$ gD�����cLg_��Q�Pg���KH�9g @g @g @g @)r r( r. r4 r& r' r )r r= �expectedr> �datar? r r r � test_rank2M sD $ zTestRank.test_rank2c C s: t tj�dd�dd�}|�� }|�� |}t�||� d S )Nr$ r �float64)�dtype)r r& r0 Zrandn�copyr( r. r4 )r r= rH r> r r r �test_rank_does_not_mutate~ s z"TestRank.test_rank_does_not_mutatec C sD t �� |d< tddd�|d< |�d�}|jddd�}t�||� d S )Nr r )�days�secondsr T)rE )r �nowr r( r. r4 )r Zfloat_string_framer> rH r r r �test_rank_mixed_frame� s z TestRank.test_rank_mixed_framec C s� dd l }ddl m} tj|d d d d�< tj|d d d d�< tj|d d d d�< tj|d d d d �< |jd d �}|jd d d �}|�tj�j}t�|d|�}t�|d |�}t � |j|� t � |j|� |jdd �}|jd dd �}|�|� � d � � �j} |j } | �| � � d � � �j } | �tj�j} t�|d| �}t�|d | �}t � |j|� t � |j|� |jddd�}|jd ddd�}|�tj�j}t�|d| �}t�|d | �}t � |j|� t � |j|� |jd dd�}|jd d dd�}|�|� � d � � �j} |j } | �| � � d � � �j } | �tj�j} t�|d| �}t�|d | �}t �|j|� t �|j|� d} tjt| d�� |jddd� W d Q R X tjt| d�� |jddd� W d Q R X d S )Nr )r r r r r r r r! r �bottom)� na_optionr �topF)rT rF z3na_option must be one of 'keep', 'top', or 'bottom')�match�badT)r% r r& r' r( r+ r, r* r- r. r/ r �to_dict�TZassert_numpy_array_equal�pytestZraises� ValueError) r r5 r6 r r7 r8 r: r; r< Zfval0Zfval1�msgr r r �test_rank_na_option� sZ z TestRank.test_rank_na_optionc C sP t ddgddgg�}t�|jdd�|jdd�� t�|jdd�|jdd�� d S ) Nr r r r r )�axis�index�columns)r r. r4 r( )r r= r r r �test_rank_axis� s zTestRank.test_rank_axisc C s� dd l }ddl m} tj�ddd�}|d d }dd� t|jd �D �}x�||d |d gD ]�}t||d �}xnd D ]f}x`d D ]X}|j||d�} t� ||||dkr�|nd�} | � tj �} t| |d �� d�} t � | | � q|W qrW q\W d S )Nr )r � )�d � g $@c S s g | ]}t td �| ��qS )�z)�chr�ord)�.0�ir r r � <listcomp>� s z4TestRank.test_rank_methods_frame.<locals>.<listcomp>r g ��.Ag���ư>)r` )r r )r r r r r )r^ r r ZordinalrK )r% r r& r0 r1 �range�shaper r( r- r2 rK r. r4 ) r r6 r Zxs�cols�valsr= Zax�mr> ZsprankrH r r r �test_rank_methods_frame� s z TestRank.test_rank_methods_framerL �O�f8�i8z(ignore:.*Select only valid:FutureWarningc C s� d|kr| j �� }n | j �|�}|jdd�}|�� | �� }t�||� |dkr\|dkr\d S |�� | j|d�}|dkr�|j|ddd�}t�||� |j|ddd�}t�||� d S ) Nri F)rF r rq )r T)r rF rE )r= Zdropnar2 r( r r. r4 )r r rL r= �resrH Zres2Zres3r r r �test_rank_descending� s z TestRank.test_rank_descendingr^ r Nc C sZ | j }ddd�}tdfh}||f|kr*d S |d kr6|n|�|�}||| j| ||d� d S )Nr r c S s@ t ||d��}|dkr"| j} |j}| j||d�}t�||� d S )N)r r r )r r^ )r rY r( r. r4 )r= rH r r^ Zexp_dfr> r r r �_check2d s z3TestRank.test_rank_2d_tie_methods.<locals>._check2dr )r r^ )r r )r= �objectr2 �results)r r r^ rL r= rv �disabled�framer r r �test_rank_2d_tie_methods s z!TestRank.test_rank_2d_tie_methodsz method,expg �?g �?gUUUUUU�?gUUUUUU�?c C sF t dddgdddgdddgg�}|j|dd�}t |�}t�||� d S ) Ni� �B r �A r r T)r rA )r r( r. r4 )r r r? r= r> rH r r r �test_rank_pct_true( s ! zTestRank.test_rank_pct_truec C sB t t�d�t�ddd�d��}|jdd��� }|dk�� s>t�d S )Ni r ���)r r T)rA r )r r& Zaranger( r �all�AssertionError)r r= r> r r r �test_pct_max_many_rowsO s z TestRank.test_pct_max_many_rowszcontents,dtypei���r g#B� ��ǻg�}���g ��Jzg�Ww'&l�7g#B� ���;g�����|�=r" rK g�j7�֖��float32rc �uint8i����i' i�� g _�B�int64�1r ZBAZBar rw i� � datetime64c C s� t jt jd t �d�d�}t j||d�}t jtt|��dd�d }||kr�|| }t j�tt|��d�}t �|||�}t �||t j�}t j� t|��} ||| �} ||| dd�} | � � } t � | | � d S )NZnat)rK r� rw r� )rL rK g �?r ) r& r' r� �arrayrk �lenr0 �choice�insertZ permutationr( r. Z assert_equal) r �contentsrL Zframe_or_seriesZ dtype_na_mapr* Z exp_orderZna_valueZ nan_indicesZ random_order�objrH r> r r r �test_rank_inf_and_nanY s > z TestRank.test_rank_inf_and_nanc C s� ddddddddd d g }ddddddddddg }ddt jdddt jt jdt j g }t||d �|d d �}|�� }|�� }|d �� |d< |d �� |d< t�||� d S )Nr r r r r r r r � r$ )�col1�col2rr )rI r_ rL r� r� )r& r' r, r r( rM r. r4 )r r_ r� r� r= Z df_resultZ series_resultr r r �"test_df_series_inf_nan_consistency� s "z+TestRank.test_df_series_inf_nan_consistencyc C sB t dtj dtjgi�}t ddddgi�}|�� }t�||� d S )NrD r g �?g @g @)r r& r, r( r. r4 )r r= rH r> r r r �test_rank_both_inf� s zTestRank.test_rank_both_infz data,expectedrD )rD rB rB g @g @)r_ c C s t |�}|�� }t�||� d S )N)r r( r. r4 )r rI rH r= r> r r r �test_rank_mixed_axis_zero� s z"TestRank.test_rank_mixed_axis_zero)0�__name__� __module__� __qualname__r r& r' �sr r= r� rx rZ Zfixturer �tdZskip_if_no_scipyr@ rJ rN rR r] ra rp Z&skip_array_manager_not_yet_implemented�markZ parametrize�filterwarningsru rw r{ r~ ZsingleZ high_memoryr� r, Ziinfor� r r r� r r r r� r� r� rk r� r r r r r s� & $1 N $ &r )r r �numpyr& rZ Zpandas._libs.algosr r Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r Zpandas._testingZ_testingr. r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.07 |
proxy
|
phpinfo
|
Настройка