Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/__pycache__/test_cython.cpython-37.pyc
Назад
B �5�g�+ � @ s� d Z ddlZddlZddlmZ ddlZddlmZm Z m Z m Z m Z m Z mZ ddlmZ ej�ddddd d d ejd ej�d �gd�dddg �dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Z ej�d d ej fd ej fd ej fd!ej fdej!fdej"fdej#fd"d#d$� fd%d&d$� fg �d'd(� �Z$ej�d d ej fd d)d$� fd d*d$� fdej"fdej#fg�d+d,� �Z%d-d.� Z&ej�d/d"d%ddg�ej�d0e d1�e d2�g�d3d4� ��Z'ej�d5dddddd d d d6d7d8d9d:d;d<d"d%d=d>d?g�d@dA� �Z(ej�ddddd d d d dddg �dBdC� �Z)ej�dDdEdFg�ej�dGdHdIdJdKdLdMdNdOdPg �ej�d0ej*dQdRdSdTgdUdV�ej*dQdRdSdTgdWdV�ej*dXdYdZd[gd\dV�ej*dXdYdZd[gd]dV�ej*dEdEdFdFgd^dV�g�d_d`� ���Z+dS )az test cython .agg behavior � N)�is_float_dtype)� DataFrame�Index�NaT�Series� Timedelta� Timestamp� bdate_range�op_name�count�sumZstd�varZsem�mean�medianzignore::RuntimeWarning)Zmarks�prod�min�maxc s< ddddddddddt jt jg ddgd t j�d�d�}t|�}t j|jd d d �d f< �fd d �� |jdgdd��d�}� fdd�|D �}td |i�}d|j_ � |�}t � ||� |�ddg�}i }x,|D ]$\\}}} � | d �|� |i �|< q�W t|�j jdd�}ddg|j_d |_ � |�d }�dk�r8t �||� d S )Nr � g �?�A�B� � )r r �C� � r c s t | � �� S )N)�getattr)�x)r � �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/test_cython.py�<lambda>4 � z(test_cythonized_aggers.<locals>.<lambda>)�axisc s i | ]\}}� |d �|�qS )r r )�.0�cat�group)�opr r � <dictcomp>8 s z*test_cythonized_aggers.<locals>.<dictcomp>F)Zdropna)r r )�np�nan�random�randnr �locZdrop�groupby�index�name�tm�assert_frame_equal� setdefault�T�stack�names�assert_series_equal) r �data�df�grouped�exp�resultZexpdZcat1�cat2r$ r )r% r r �test_cythonized_aggers s, r< c C sd t tj�ddd�tj�ddd��d�d��} | �d�d �� }| �d�d �tj�}t� ||� d S ) Nr � �2 r �bool)�a�br@ rA ) r r' r) �randint�astyper, r �aggr/ r5 )�framer: �expectedr r r �test_cython_agg_booleanL s rG c C s� t tj�ddd�ddgd d��} tjtdd �� | �d �d jd d � W d Q R X tjt dd �� | �d �d �� W d Q R X t tj�ddd�ddgd d��} t j t dd�� | d g �| d ��� }W d Q R X t g | d � � �� d�}t �||� d S )Nr r= r> �foo�bar� )r@ rA zdoes not implement)�matchr@ rA T)� numeric_onlyz Could not convert (foo|bar)*F)Zcheck_stacklevel)r- )r r' r) rB �pytest�raises�NotImplementedErrorr, r � TypeErrorr/ Zassert_produces_warning� FutureWarningZ sort_valuesZdrop_duplicatesr0 )rE r: rF r r r � test_cython_agg_nothing_to_aggY s " ""rR c C sb t tj�ddd�ddgd tjdddd �d ��} tjtd d �� | � d �j j dd� W d Q R X d S )Nr r= r> rH rI rJ �nowr2 )�periods�freq)r@ rA �dateszdoes not implement)rK rA T)rL ) r r' r) rB �pdZ date_rangerM rN rO r, rV r )rE r r r �)test_cython_agg_nothing_to_agg_with_datesj s rX c C sf t dddgdddgd��} | jddd ��� | jddd ��� | jddd ��� | jddd ��� d S ) Nr r � � r= )r �yr �columns)�levelr! )r r, r )r7 r r r � test_cython_agg_frame_columnsv s r^ c C s� t ddddddddgddddddddgtj�d�tj�d�d��} | �d�d �d d � �}td d d d �ddd d �gtddgdd�d d�}t� ||� d S )NrH rI �one�two�three� )r r r �Dr r c S s | � � �� S )N)Z value_counts�to_dict)r r r r r � r z-test_cython_agg_return_dict.<locals>.<lambda>r )r` r_ ra r )r. )r- r. ) r r' r) r* r, rD r r r/ r5 )r7 �tsrF r r r �test_cython_agg_return_dict� s rf c C sX t ddd�} tdddddgd | d �}|�d d � �}|�� }|�tj�}t�||� d S ) Nz1/1/2000r> )rT r r r rc �Er )r- c S s | j S )N)�month)r r r r r � r z&test_cython_fail_agg.<locals>.<lambda>)r r r, r rD r' r/ r5 )Zdrre r8 ZsummedrF r r r �test_cython_fail_agg� s ri z op, targop�add�firstc C s | j d S )Nr )�iloc)r r r r r � r r �lastc C s | j d S )N���)rl )r r r r r � r c C s^ t tj�d��}tjjdddd��t�}|�|�j| d dd�}|�|�� |�}t � ||� d S )Ni� r r> )�sizeT)�altrL ) r r' r) r* rB rC �floatr, �_cython_agg_generalrD r/ r0 )r% �targopr7 �labelsr: rF r r r �test__cython_agg_general� s ru c C s t | �dkrt�| �S tjS )Nr )�lenr' r r( )r r r r r � r c C s t j| dd�S )Nr )Zddof)r' r )r r r r r � r c s� t dddg�}tddd�}|jt�|d |�|d�}|j| d dd �}|jt�|d |�|d�}|�� fd d ��}t�||� d S ) N� r � r �7 r= )�observedT)rp rL c s � | �S )Nr )r )rs r r r � r z/test_cython_agg_empty_buckets.<locals>.<lambda>) r �ranger, rW �cutrr rD r/ r0 )r% rs rz r7 �grps�gr: rF r )rs r � test_cython_agg_empty_buckets� s r c C s t dddgdgd�}tddd�}|jt�|d |�| d �jd d d d �}tjdd dd�}t dddddgitj|dd d�d�}| r�||jdk }t � ||� |jt�|d |�| d �jdd d d �}t dddddgitj|dd d�d�}| r�||jdk }t � ||� d S )Nrw r rx r@ )r\ r rJ r= )rz rj T)rp rL � )rU �$ )r. Zordered)r- r r i� ) r r{ r, rW r| rr Zinterval_rangeZCategoricalIndexr@ r/ r0 )rz r7 r} r: Z intervalsrF r r r �$test_cython_agg_empty_buckets_nanops� s& r� r% r6 z2016-10-14 21:00:44.557z17088 days 21:00:44.557c C sZ t ddg|tgd��}tddgdd�}t d|tgi|d�}|�d��| �}t�||� d S )Nr r )r@ rA r@ )r. rA )r- )r r r r, � aggregater/ r0 )r% r6 r7 r- rF r: r r r �"test_cython_with_timestamp_and_nat� s r� rD ZohlcZcumprodZcumsum�shift�any�allZquantileZrankZcumminZcummaxc C sr t dddddgdddddgd��}d|jjd j_|�d g��d | i�}|�� �d g��d | i�}t� ||� d S ) Ngffffff@g������@g������@gffffff@g @Zsetosa)� sepal_length�speciesFr r� r� ) r Z_mgrZarrays�flagsZ writeabler, rD �copyr/ Z assert_equal)rD r7 r: rF r r r � test_read_only_buffer_source_agg� s r� c C s� t ddgd tjddddddd d d tjg d d �d��}t|�d�d | �� }|j|d �d�d�}t|�d�d | �� }| dkr�|�d�}t� ||� d S )Nr r r= r r rY rZ r � rb � �Int64)�dtype)r r Zfloat64)r r ) r rW �array�NAr r, ZassignrC r/ r5 )r r7 r: Zdf2rF r r r � test_cython_agg_nullable_int' s * r� �with_naTFzop_name, action)r � large_int)r � always_float)r r� )r r� )r r� )r �preserve)r r� )rk r� )rm r� r r rY rZ r� )r� ZInt8g�������?g�������?g333333�?g�������?ZFloat32ZFloat64�booleanc C s |rt j| d< tddddg| d��}|�d�}|dkr@t �� }nV|dkrdt| j�rZ| j}q�t �� }n2|dkr�t| j�r~| j}q�t �� }n|d kr�| j}t||�� }|d j|ks�t �|� |�}|d j|ks�t �t|d |�� }|j|ks�t �|d � |�}|j|k�s t �d S ) NrY r@ rA )�key�colr� Z always_intr� r� r� r� ) rW r� r r, Z Int64Dtyper r� Z Float64Dtyper �AssertionErrorr� )r6 r �actionr� r7 r8 Zexpected_dtyper: r r r � test_cython_agg_EA_known_dtypesJ s0 r� ),�__doc__�numpyr' rM Zpandas.core.dtypes.commonr ZpandasrW r r r r r r r Zpandas._testingZ_testingr/ �markZ parametrize�param�filterwarningsr< rG rR rX r^ rf ri r r r r r r r ru r r� r� r� r� r� r� r r r r �<module> s� $ $ !
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка