Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/methods/__pycache__/test_nlargest.cpython-37.pyc
Назад
B �5�g� � @ s� d Z ddlmZ ddlZddlZddlZddlmZ ddl m Z ddddd d d d d ddddg Z ej dd� �Zej e d�dd� �Zdd� ZG dd� 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. � )�productN)�Series�datetime� datetimetz� timedelta�int8�int16�int32�int64�float32�float64�uint8�uint16�uint32�uint64c C st t �t �dddddg�t �dddddg��d�t �ddddd g�d ��} x&d D ] }td d dd dg|d�| |< qNW | S )z� A DataFrame with many dtypes * datetime * datetimetz * timedelta * [u]int{8,16,32,64} * float{32,64} The columns are the name of the dtype. Z2003Z2002Z2001Z2005z US/EasternZ3dZ2dZ1dZ5d)r r r ) r r r r r r r r r r � � � � )�dtype)�pdZ DataFrame� to_datetimeZ tz_localizeZ to_timedeltar )Zdfr � r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/methods/test_nlargest.py� s_main_dtypes s r )�paramsc C s || j S )z Each series in s_main_dtypes.)�param)�requestr r r r �s_main_dtypes_splitG s r c C sP t | |d�}t||�d�}|dkr,dddgndddg}|j| }t�||� d S )N)r r � nsmallestr r r )r �getattr�loc�tm�assert_series_equal)�valsr �method�ser�resultZ expected_idxr�expectedr r r � assert_check_nselect_boundaryM s r) c @ s e Zd Zej�dedddddgdd�edddddgdd�edddddgd d�eed ��eed �d d�g�d d � �Zdd� Z dd� Z ej�de dd��dd� �Z dd� Z dd� Zej�dddg�d d � �Zd d � Zej�d d!d"gd!gfd!d"d!d!gd!gfg�d#d$� �Zd%d&� Zd'S )(�TestSeriesNLargestNSmallest�rg @r r �5�object)r r Z complex128Zabcde�categoryc C sj |j }d|� �}dt|�ddf}|j|jf}x8t||�D ]*\}}tjt|d�� ||� W d Q R X q8W d S )Nz3Cannot use method 'n(largest|smallest)' with dtype r r ���)�match)r �len�nlargestr r �pytest�raises� TypeError)�selfr+ �dt�msg�args�methodsr% �argr r r �test_nlargest_errorW s z/TestSeriesNLargestNSmallest.test_nlargest_errorc C s" |}t �|�d�|jddg � t �|jddd�|jddg � |jdd� }t �|�d�|� t �|�d�|� t �|�d�|� t �|�d�|� t �|�t|��|�� � t �|�t|�d �|�� � t �|�t|��|jdddddg � t �|�t|�d �|jdddddg � d S ) Nr r �last)�keepr r r/ � )r" r# r �ilocr2 r1 � sort_values)r6 r r& �emptyr r r �test_nsmallest_nlargestl s &z3TestSeriesNLargestNSmallest.test_nsmallest_nlargestc C sP t dtjdddg�}t�|�� |jddddg � t�|�� |jddddg � d}tj t |d �� |jd d � W d Q R X tj t |d �� |jd d � W d Q R X t dgd dddddgd �}t dgd dddgd �}t dgd dddgd �}|�d�}t�||� |jdd d �}t�||� |�d�}t�||� |jdd d �}t�||� d S )Ng @r r r r? r r z#keep must be either "first", "last")r0 �invalid)r> )�indexr= ) r �np�nanr" r# r2 r@ r r3 r4 � ValueError)r6 r& r8 Zexpected_firstZ expected_lastr' r r r �test_nlargest_misc s& z.TestSeriesNLargestNSmallest.test_nlargest_misc�nc C sl t ddddgddddgd�}|�|�}|jdd��|�}t�||� |�|�}|�� �|�}t�||� d S ) Nr r? r r r )rE F)Z ascending)r r2 rA �headr" r# r )r6 rJ r& r' r( r r r �test_nlargest_n� s z+TestSeriesNLargestNSmallest.test_nlargest_nc C s< t �|�}|j|j }}||d |d |g}t|||� d S )Nr )rF �iinfo�min�maxr) )r6 �nselect_methodZ any_int_dtype� dtype_info�min_val�max_valr$ r r r � test_nlargest_boundary_integer� s z:TestSeriesNLargestNSmallest.test_nlargest_boundary_integerc C sL t �|�}|j|j }}t j||gd|d�\}}||||g}t|||� d S )Nr )r )rF ZfinforN rO Z nextafterr) ) r6 rP Z float_dtyperQ rR rS Zmin_2ndZmax_2ndr$ r r r � test_nlargest_boundary_float� s z8TestSeriesNLargestNSmallest.test_nlargest_boundary_floatr zdatetime64[ns]ztimedelta64[ns]c C sB t �d�}|j|j }}|d |d |d ||g}t|||� d S )Nr r r )rF rM rN rO r) )r6 rP r rQ rR rS r$ r r r �#test_nlargest_boundary_datetimelike� s z?TestSeriesNLargestNSmallest.test_nlargest_boundary_datetimelikec C s� t ddddddddg�}|jddd�}t dddddddg�}t�||� |jd dd�}t dddddgdd dd dgd �}t�||� d S ) N� � � � � r? �all)r> r r r )rE )r r2 r" r# r )r6 r& r' r( r r r �%test_nlargest_duplicate_keep_all_ties� s zATestSeriesNLargestNSmallest.test_nlargest_duplicate_keep_all_tiesz data,expectedTFc C s* t |�}|�d�}t |�}t�||� d S )Nr )r r2 r" r# )r6 �datar( r& r' r r r �test_nlargest_boolean� s z1TestSeriesNLargestNSmallest.test_nlargest_booleanc C sz |}t j�d�j|�� dd�}t|�� |d�}tj|d< |� d�}tt � |d�|j � d�d�� d��|�}t � ||� d S )NrW F)�copy)r r r )rE )rF �randomZrandnZastype�lowerr r` r ZNAr2 �deleterE r" r# )r6 Zany_nullable_numeric_dtyper Zarrr& r' r( r r r �test_nlargest_nullable� s z2TestSeriesNLargestNSmallest.test_nlargest_nullableN)�__name__� __module__� __qualname__r3 �markZ parametrizer �listr< rC rI �rangerL rT rU rV r] r_ rd r r r r r* V s" , r* )�__doc__� itertoolsr �numpyrF r3 Zpandasr r Zpandas._testingZ_testingr" Z main_dtypesZfixturer r r) r* r r r r �<module> s. (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка