Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_counting.cpython-37.pyc
Назад
B �5�g!2 � @ s� d dl mZ d dlmZ d dlZd dlZd dlmZm Z m Z m Z m Z m Z mZmZ d dlmZ G dd� d�Zdd� Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� ZdS )� )�product)�ascii_lowercaseN)� DataFrame�Index� MultiIndex�Period�Series� Timedelta� Timestamp� date_rangec @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zej�d#d$d%� ed&d'�D �d(d%� ed&d'�D �d)d%� ed&d'�D �d*d%� ed&d'�D �d+d%� ed&d'�D �g�d,d-� �Zd.d/� Zd0d1� Zd2d3� Zd4S )5� TestCountingc C sh t dgdgdgdgdggdgd�}|�d�}|j}tdddddg�}t�||�� � t�||�� � d S ) N�a�b�A)�columnsr � � � )r �groupbyr r �tm�assert_series_equal�cumcount)�self�df�g�sg�expected� r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_counting.py� test_cumcount s " zTestCounting.test_cumcountc C sN t � jdd�}ttd�jdd�}tdd�}t�||�� � t�||�� � d S )Nr )�level)�dtype�int64)r r r �objectr r r )r �ge�se�er r r �test_cumcount_empty s z TestCounting.test_cumcount_emptyc C sz t dgdgdgdgdggdgdgd d�}|�d�}|j}tddddd gdgd d �}t�||�� � t�||�� � d S ) Nr r r r � )r �indexr r r )r) )r r r r r r r )r r r r r r r r �test_cumcount_dupe_index) s ( z%TestCounting.test_cumcount_dupe_indexc C s� t �ddgddgddgddgddgg�}tdgdgdgdgdggdg|d�}|�d�}|j}tdddddg|d �}t�||�� � t�||�� � d S ) Nr r r r r r )r r) r )r) ) r � from_tuplesr r r r r r r )r �mir r r r r r r �test_cumcount_mi5 s ($ z TestCounting.test_cumcount_mic C s� t dgdgdgdgdggdgdgd d�}|�dddddg�}|j}tddddd gdgd d �}t�||�� � t�||�� � d S ) Nr r r r r( )r r) r r r )r) )r r r r r r r )r r r r r r r r � test_cumcount_groupby_not_col@ s ( z*TestCounting.test_cumcount_groupby_not_colc C sV t dtd�i�}|�d�}|j}tdddddg�}t�||�� � t�||�� � d S )Nr �aaabar r )r �listr r r r r �ngroup)r r r r r r r r � test_ngroupL s zTestCounting.test_ngroupc C sT t dtd�i�}|�d�}|j}ttd�dd�}t�||�� � t�||�� � d S )Nr Zabcder( r" )r! ) r r0 r r r �ranger r r1 )r r r r r r r r �test_ngroup_distinctV s z!TestCounting.test_ngroup_distinctc C sT t ddgd i�}|�d�}|j}tdgd �}t�||�� � t�||�� � d S )Nr r r( )r r r r r r r1 )r r r r r r r r �test_ngroup_one_group` s z"TestCounting.test_ngroup_one_groupc C sN t � jdd�}ttd�jdd�}tdd�}t�||�� � t�||�� � d S )Nr )r )r! r" )r r r r# r r r1 )r r$ r% r& r r r �test_ngroup_emptyj s z TestCounting.test_ngroup_emptyc C s@ t dtd�i�}ttd��}t�|�|��� |�|��� � d S )Nr r/ )r r0 r r r r r1 )r r �sr r r � test_ngroup_series_matches_framet s z-TestCounting.test_ngroup_series_matches_framec C sj t dtd�idgd d�}|�d�}|j}tdddddgdgd d�}t�||�� � t�||�� � d S )Nr r/ r r( )r) r )r r0 r r r r r r1 )r r r r r r r r �test_ngroup_dupe_indexz s z#TestCounting.test_ngroup_dupe_indexc C s� t �ddgddgddgddgddgg�}tdtd�i|d�}|�d�}|j}tdddddg|d�}t�||� � � t�||� � � d S )Nr r r r r/ )r) ) r r+ r r0 r r r r r r1 )r r, r r r r r r r �test_ngroup_mi� s ( zTestCounting.test_ngroup_mic C st t dtd�idgd d�}|�dddddg�}|j}tdddddgdgd d�}t�||�� � t�||�� � d S )Nr r/ r r( )r) r )r r0 r r r r r r1 )r r r r r r r r �test_ngroup_groupby_not_col� s z(TestCounting.test_ngroup_groupby_not_colc C s� t dddddgdgd�}|�dg�}tdddddg�}tdddddg�}t�||jd | � t�||jdd�� t�||jd d�� d S ) Nr r r )r r r T)� ascendingF)r r r r r Zngroupsr1 )r r r r< Z descendingr r r �test_ngroup_descending� s z#TestCounting.test_ngroup_descendingc C s� t ddgddgddgddgddggddgd�}|�ddg�}|�� }|�� }tdd d dd g�}tdddd dg�}t�||� t�||� d S ) Nr �x�yr r �X)r r r r r )r r r1 r r r r )r r r Zg_ngroupZ g_cumcountZexpected_ngroupZexpected_cumcountr r r � test_ngroup_matches_cumcount� s z)TestCounting.test_ngroup_matches_cumcountc s� x�t td�dd�D ]x�td�i�}|�dg�}tt���� � fdd��D �}�fdd�t��D �}t�|� � t |�� t�|� � t |�� qW d S )Nr � )�repeatr c s g | ]}� � |��qS r )r) )�.0�val)�orderr r � <listcomp>� s z:TestCounting.test_ngroup_cumcount_pair.<locals>.<listcomp>c s" g | ]\}}� d |� � |��qS )N)�count)rD �irE )�pr r rG � s ) r r3 r r �sorted�set� enumerater r r1 r r )r r r ZngroupdZ cumcountedr )rF rJ r �test_ngroup_cumcount_pair� s z&TestCounting.test_ngroup_cumcount_pairc C s� t j�d� tdt j�td�d�i�}x�dD ]�}|jdg|d�}d|d< d|d < xNt|�D ]B\}\}}||j|j df< x$t|j �D ]\}}||j|d f< q�W qZW t � t |d j �|�� � t � t |d j �|�� � q,W d S ) Nr r Zabcdef�d )FT)�sort���Zgroup_idZ group_index)�np�random�seedr �choicer0 r rM �locr) r r r �valuesr1 r ) r r Z sort_flagr rI �_�group�j�indr r r �"test_ngroup_respects_groupby_order� s z/TestCounting.test_ngroup_respects_groupby_order� datetimelikec C s g | ]}t d |d�d���qS )z2016-05-�02dz 20:09:25+00:00)r )rD rI r r r rG � s zTestCounting.<listcomp>r rB c C s g | ]}t d |d�d���qS )z2016-05-r^ z 20:09:25)r )rD rI r r r rG � s c C s" g | ]}t d |d�d�dd��qS )z2016-05-r^ z 20:09:25�UTC)�tz)r )rD rI r r r rG � s c C s g | ]}t |d d��qS )�h)�unit)r )rD r> r r r rG � s c C s g | ]}t d d|d��qS )Z2Wi� )�freq�year�month)r )rD r> r r r rG � s c C sR t dddg|d��}|�d��� }t dddgiddgd�}d|j_t�||� d S ) Nr r )r> r? r> r? r r )r) )r r rH r) �namer �assert_frame_equal)r r] r �resr r r r � test_count_with_datetimelike� s z)TestCounting.test_count_with_datetimelikec C sx t tjtjgddgddgd��}|�ddg�j�� }tg ddggg g gddgd�}tg |tjd d �}t j ||d d � d S ) Nr r r r )r �B�Cr rj )Zlevels�codes�namesrk )r) r! rf F)Zcheck_index_type) r rR �nanr rk rH r r r" r r )r r �resultr, r r r r �(test_count_with_only_nans_in_first_group� s z5TestCounting.test_count_with_only_nans_in_first_groupc C sl t dddddgddtjddgd��}|�dg��� }t tdd d d gdd �d ddddgid�}t�||� d S )Nr r( rB r r )r rj rj g g @g @g @)rf r )r) �data)r rR �NaNr rH r r rg )r r rh r r r r �4test_count_groupby_column_with_nan_in_groupby_column� s $&zATestCounting.test_count_groupby_column_with_nan_in_groupby_columnc C s� t dddd�}tt�d�|t�d�gd�}|�dd� �}|�� }tt�d�t�d�|gd�}|�dd� �}|�� }t�||� d S ) Nz1/1/2012Z5min� )�startrc �periods)r) c S s | d d dkS )Nr r r r )r> r r r �<lambda>� � z@TestCounting.test_groupby_count_dateparseerror.<locals>.<lambda>c S s | d d dkS )Nr r r )r> r r r rw � rx )r r rR �aranger rH r r )r �drZser�groupedro r r r r �!test_groupby_count_dateparseerror� s z.TestCounting.test_groupby_count_dateparseerrorN) �__name__� __module__� __qualname__r r' r* r- r. r2 r4 r5 r6 r8 r9 r: r; r= rA rN r\ �pytest�markZ parametrizer3 ri rp rs r| r r r r r s6 r c C s\ t td�t�d��d�d��} tddgtddgdd �d d �}| �d�j� � }t � ||� d S ) N�ababrB ztimedelta64[ns])r �deltr r r r )rf r� )r) rf ) r r0 rR ry �astyper r r r� rH r r )r r ro r r r �#test_groupby_timedelta_cython_count s r� c C sR d} t d| d dd�}ttj�tt�| �tj�dd| �tj�| �� d�tj�d d| �tj�|| �tj�| �� d�tj�| �� d�tj�|| �tj�|d � tj�tt�| �d � �}x8|j � d d dg�D ]"}tj |j tj�| | d �|f< q�W |d �d�|d< xNd d d d ggD ]<}|�|��� }|�|��tj�j |d d�}t�||� �qW d S )Ni � z 2015-08-30rt �T)rv rc r r( r i����r ) �1st�2ndZ3rd�4thZ5thZ6thZ7thZ8th�9thr� r� r� r� �category)Zaxis)r r rR rS rU r0 r �randintZrandn�roundr Zdroprn rV r� r rH �applyr rg )�nrz r �col�key�left�rightr r r � test_count s& " r� c C s� t dddgdtjdgdtjtjggdddgd �} | �d��� }| jdd d ��� }t ddgd d ggddgddgd �}d|j_t�||� � � t�||� | �d�d �� }t� ||d � d S )Nr r Zfoo�barr r rj rk )r F)Zas_indexr )r r) ) r rR rn r rH r) rf r rg Z reset_indexr )r Zcount_asZ count_not_asr Zcount_Br r r �test_count_non_nulls) s " r� c C s� t dgd dgd dgd dgd d��} | �d�j�� }tddgtddgdd�dd�}t�||� t dtj tj gdgd dgd dgd d��} | �d�j�� }td dgtddgdd�dd�}t�||� d S ) Nr r r r )r �cr� )rf )r) rf r ) r r r rH r r r r rR rn )r ro r r r r �test_count_object= s . 2 r� c C s� t �t j�ddd�t j�ddd�f�} t| ddddgd �}t j||dk< |�ddg��� }xNd D ]F}|d �|�|d< |d �|�|d< |�ddg��� }t � ||� q`W d S ) Nr r( )rO r r r r r� �d)r )Zfloat32r# ) rR ZhstackrS r� r rn r rH r� r rg )�valsr r �tro r r r �test_count_cross_typeI s " r� c C s� t t�ddddgtj�t�ddddgtj�t�dddd gtj�td �d ��} | �d ��� }t ddgddgddgd �t td�d d�d�}t � ||� d S )Nr r r rO r � rB r( � r� )r r r� �grpr� )r r r� �ab)rf )r) ) r rR �arrayZint8Zuint32Zint16r0 r rH r r rg )r ro r r r r �test_lower_int_prec_countZ s *r� c s� G dd� dt ��G �fdd�d�� t� fdd�td�D �td�d ��} | �d ��� }td d d gittd �d d�d�}t�||� d S )Nc @ s e Zd ZdS )zAtest_count_uses_size_on_exception.<locals>.RaisingObjectExceptionN)r} r~ r r r r r �RaisingObjectExceptionk s r� c s* e Zd Zd� fdd� Z�fdd�Z� ZS )z8test_count_uses_size_on_exception.<locals>.RaisingObject�I will raise inside Cythonc s t � �� || _d S )N)�super�__init__�msg)r r� )� __class__r r r� o s zAtest_count_uses_size_on_exception.<locals>.RaisingObject.__init__c s � | j ��d S )N)r� )r �other)r� r r �__eq__s s z?test_count_uses_size_on_exception.<locals>.RaisingObject.__eq__)r� )r} r~ r r� r� � __classcell__r )r� )r� r � RaisingObjectn s r� c s g | ] }� � �qS r r )rD rX )r� r r rG w s z5test_count_uses_size_on_exception.<locals>.<listcomp>rB r� )r r� r� r r r� )rf )r) ) � Exceptionr r3 r0 r rH r r rg )r ro r r )r� r� r �!test_count_uses_size_on_exceptionj s $ r� )� itertoolsr �stringr �numpyrR r� Zpandasr r r r r r r r Zpandas._testingZ_testingr r r� r� r� r� r� r� r� r r r r �<module> s ( p
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка