Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_nlargest.cpython-37.pyc
Назад
B �5�gK � @ sp d Z ddlmZ ddlZddlZddlZddlm Z ej dd� �Z ej dd� �Z ej dd � �ZG d d � d �ZdS ) z� Note: for naming purposes, most tests are title with as e.g. "test_nlargest_foo" but are implicitly also testing nsmallest_foo. � )�ascii_lowercaseNc C s> t jdddddgdddddgdddddgd�dddddgd�S ) N� � � � r � )�a�b�c)�index)�pd� DataFrame� r r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_nlargest.py� df_duplicates s (r c C s4 t �tj�d�ttd d� �tj�d��d�d��S )N� Zfloat64)r r r )r r �np�randomZ permutation�listr �astyper r r r � df_strings s r c C s� t jdddgdddgdddgtd�t �td���d�d d d gt jd dd �t jd ddd�t jdddd�d� ddddddddd g d �S ) Nr r r g @g @g @�abc�category� � � Z20130101)�periodsz US/Eastern)r �tzz1 s�s)r �freq) �group�int�float�string�category_string� category_int�datetime� datetimetz� timedeltar r! r" r# r$ r% r& r' r( )�columns)r r r ZSeriesr Z date_rangeZtimedelta_ranger r r r �df_main_dtypes! s& r* c @ s e Zd Zej�ddgdgddgddgddgddgdddgdddgdddgdddgdddgdddgg �ej�dedd��dd � ��Zej�d d d gd d gg�dd� �Zdd� Z dd� Z dd� Z ej�ddddgdddgdgdgddgddgg�ej�dedd��dd� ��Z dd� Z dd � Zd S ) �TestNLargestNSmallest�orderr r r �nr � c C s� |}d|krDd|� d�}t jt|d�� t||�||� W d Q R X n8|dk}t||�||�}|j||d��|�} t�|| � d S )Nr z0Column 'b' has dtype object, cannot use method 'z' with this dtype)�match� nsmallest)� ascending)�pytest�raises� TypeError�getattr� sort_values�head�tm�assert_frame_equal) �selfr �nselect_methodr- r, �df� error_msgr1 �result�expectedr r r �test_nlargest_nA s z%TestNLargestNSmallest.test_nlargest_nr) r r$ r# c C s| |}|d }d|� d|| j � d|� d�}|�dd��dd ��d d ��d d �}tjt|d�� t||�d|� W d Q R X d S )Nr zColumn 'z ' has dtype z, cannot use method 'z' with this dtype�(z\(�)z\)�[z\[�]z\])r/ r )Zdtype�replacer2 r3 r4 r5 )r: r* r; r) r< �colr= r r r �test_nlargest_errore s z)TestNLargestNSmallest.test_nlargest_errorc C s@ |}|� dtt|�ddh �� |�dtt|�ddh �� d S )Nr r$ r# )r0 r �set�nlargest)r: r* r< r r r �test_nlargest_all_dtypesy s z.TestNLargestNSmallest.test_nlargest_all_dtypesc C s� t �ddddddgddddddgd��}|jdddgd�}t jddddgddddgd�ddd dgd �}t�||� |jdddgd�}t jddddgddddgd�d ddd gd �}t�||� d S ) Nr r r )r r r r r )r) r )r r )r r rI r8 r9 r0 )r: r< r> r? r r r �+test_nlargest_duplicates_on_starter_columns~ s (( (zATestNLargestNSmallest.test_nlargest_duplicates_on_starter_columnsc C s� t �dgd dddddgd��}|�dd�}t jdgd dddgd�dddgd �}t�||� |�dd�}t �dgd dddgd��}t�||� d S ) Nr r r r r )r r r r )r )r r rI r8 r9 r0 )r: r< r> r? r r r � test_nlargest_n_identical_values� s & z6TestNLargestNSmallest.test_nlargest_n_identical_values� c C s\ |}|� ||�}|�|��|�}t�||� |�||�}|j|dd��|�}t�||� d S )NF)r1 )r0 r6 r7 r8 r9 rI )r: r r- r, r< r> r? r r r � test_nlargest_n_duplicate_index� s z5TestNLargestNSmallest.test_nlargest_n_duplicate_indexc C s� t �ddddddddgdddddd dd gd ��}|jdd d d�}t �dddddddd�ddddd dd d�d ��}t�||� |jdd d d�}t �dddddd�ddd dd d�d ��}t�||� d S )Nr r r r r r r r �2 � )r r r �all)Zkeep)r r r r r rM r )r r r rM r )r r rI r8 r9 r0 )r: r< r> r? r r r �%test_nlargest_duplicate_keep_all_ties� s , z;TestNLargestNSmallest.test_nlargest_duplicate_keep_all_tiesc C s� t jt j�dgddgg�ddgddgddgd d ggd �}|�d d �}|jddd g }t�||� |�d d�}|jd ddg }t�||� d S )N�xr r g ��Q� �?g�p= ף�?g��Q� ��?g �?gffffff�?g333333�?g ��Q� �?)r) �datar )rS r r r )rS r r ) r r Z MultiIndexZ from_productr0 Zilocr8 r9 rI )r: r< r> r? r r r �&test_nlargest_multiindex_column_lookup� s z<TestNLargestNSmallest.test_nlargest_multiindex_column_lookupN)�__name__� __module__� __qualname__r2 �markZ parametrize�ranger@ rG rJ rK rL rN rR rU r r r r r+ = s4 ( r+ )�__doc__r# r �numpyr r2 Zpandasr Zpandas._testingZ_testingr8 Zfixturer r r* r+ r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка