Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/__pycache__/test_other.cpython-37.pyc
Назад
B �5�gSO � @ s� d Z ddlZddlmZ ddlZddlZddlm m Z ddl Z ddl mZmZmZmZmZmZmZ ddlmZ ddlmZ ddlmZ dd� Z d d � Z d d � Z d d� Z dd� Z dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%dd � Z&d d � Z'd d � Z(d!d"� Z)d#d$� Z*e j+d%d&� �Z,ej-�.d'd(d)g�d*d+� �Z/d,d-� Z0d.d/� Z1ej-�.d0e2ed1d2d3d4�i�fe3ed1d5d5d5gd6d7d7gd4�i�fd8d9� ed1d2d3d4�i�fd:d9� ed1d5d5d5gd6d7d7gd4�i�fg�d;d<� �Z4ej-�.d0e2ed2d3gd5d6gd1d=�fe3ed5d5d5gd6d7d7ggd5d6gd1d=�fd>d9� ed2d3gd5d6gd1d=�fd?d9� ed5d5d5gd6d7d7ggd5d6gd1d=�fg�d@dA� �Z5dBdC� Z6dDdE� Z7dFdG� Z8ej-�.dHe9e:e;e<e=e>e?e@g�dIdJ� �ZAdS )Kz test all other .agg behavior � N)�partial)� DataFrame�Index� MultiIndex� PeriodIndex�Series� date_range� period_range)�SpecificationError)� pprint_thingc C s� t tj�d�tj�d�dddddgdddddgd��} | �d�}dd � }tjtd d d �� |�|g�}W d Q R X d dg|_ tjtd d d �� |�|�}W d Q R X t� ||� d S )N� �a�b�one�two)�data1�data2�key1�key2r c S s | � � | �� S )N)�max�min)Zarr� r �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/test_other.py� peak_to_peak, s z"test_agg_api.<locals>.peak_to_peakzDropping invalidF)�matchZcheck_stacklevelr r ) r �np�random�randn�groupby�tm�assert_produces_warning� FutureWarning�agg�columns�assert_frame_equal)�df�groupedr �expected�resultr r r � test_agg_api s r) c C s� dddgdddgdd dgg} t d d � | D �d d � | D �d d � | D �d ��}dd � | D �} t dd � | D �dd � | D �dd � | D �d ��}|d |d �� |d< |�d��tj�}|d |d �� |d< |�d��tj�}t|�t|�ks�t�d S )N� z 2012-01-01g �?� z 2012-01-02g @� g @c S s g | ] }|d �qS )r r )�.0�xr r r � <listcomp>A s z,test_agg_datetimes_mixed.<locals>.<listcomp>c S s g | ] }|d �qS )r* r )r- r. r r r r/ B s c S s g | ] }|d �qS )r+ r )r- r. r r r r/ C s )�key�date�valuec S s< g | ]4}|d |d r,t j�|d d��� nd|d g�qS )r r* z%Y-%m-%dNr+ )�dt�datetime�strptimer1 )r- �rowr r r r/ H s c S s g | ] }|d �qS )r r )r- r. r r r r/ R s c S s g | ] }|d �qS )r* r )r- r. r r r r/ S s c S s g | ] }|d �qS )r+ r )r- r. r r r r/ T s r2 �weightsr1 )r �sumr � aggregater �len�AssertionError)�dataZdf1Zdf2Zgb1Zgb2r r r �test_agg_datetimes_mixed< s r= c C s� t dddd�} ttj�dd�| d�}|jdd��� }t|jt �sDt �t d d dd �}t tj� t |��|d�}t tj� t |��|d�}t�||d ��}|�|jj�}t|� d S ) Nz2012-1-1�Mr, )�freq�periodsr+ )�indexr )�levelz1999-01r )�startr@ r? )�s1�s2)r r r r r r r8 � isinstancerA r r; r Zrandr: � from_dict�month�list)Zprngr% �rsrA rD rE r& r r r �test_agg_period_indexa s rK c C s� t ddddddddgtdddd�d ��} d | jd d d d gdf< | jd dddg �d�}| �d�}t�|�� |� t�|�d�|� t�|�ddi�|� t� |j �� |d � t� |j �d�|d � | jd dddg �d�}| �d�}t�|� � |� t�|�d�|� t�|�ddi�|� t� |j � � |d � t� |j �d�|d � t d d d d gt td�dd�dd�}t� |j �t�|� t� |j �� |� t d d d d gt td�dd�dd�}t� |j �� |� d S )N�A�B�C�Dz1/1/2011� �H)r@ r? )�class�timer r* r+ r rS r, � � rR �first� �lastZABCD)�name)rA rY )r r �locZ set_indexr r r$ rV r" �assert_series_equalrS rX r r rI r: �size�count)r% �expr& r r r �*test_agg_dict_parameter_cast_result_dtypesp s. ""r_ c C s^ dd� t d�D �} td�}t|| d��}|�d�d �t�}|�d�d �� }t�||� d S )Nc S s g | ]}t �d |d d��qS )i� r* )r3 r4 )r- r. r r r r/ � s z0test_agg_cast_results_dtypes.<locals>.<listcomp>� Z aaabbbbbbccd)�X�Yra rb ) �rangerI r r r" r: r] r r[ )�u�vr% r( r' r r r � test_agg_cast_results_dtypes� s rf c C s� t dddddgdddddgdddddgd��} t dddddgiddddgd �}d |j_| �d �dg �� }t�||� t ddddgddddgd �ddddgd �}d |j_| �d �dd g �� }t�||� d S ) Nr* r+ r, rT r )r r �cr g @)rA r )r rg rg )r rA rY r �meanr r$ )r% r' r( r r r � test_aggregate_float64_no_int64� s . *ri c C s t ddddddddgddddddddgtj�d�d t�d�d��} | �dd g�}|d �� }|d �� }|d �� }|d �� }|d �d d g�}t j ||gdd�}d d g|_ t j ||dd� |�tjtjg�}t j ||||gdd�}t�d d gd d gg�|_ t j ||dd� |d d g �tjtjg�}t j ||||gdd�}t�d d gd d gg�|_ t j ||dd� |�d d d��}t j ||gdd�}t j ||dd� |�d d gd d gd��}t j ||||gdd�}t�d d gd d gg�|_ d}tjt|d��" |d d g �tjtjd�� W d Q R X d S )N�foo�barr r rP g �?)rL rM rN rO rL rM rN rO r8 rh r* )ZaxisT)Z check_like)rN rO z&Column\(s\) \['r', 'r2'\] do not exist)r )�r�r2)r r r r �aranger rh r8 r" �pd�concatr# r r$ r Z from_product�pytest�raises�KeyError) r% r& Zc_meanZc_sumZd_meanZd_sumr( r' �msgr r r � test_aggregate_api_consistency� s> ru c C s� t dddddgtd�td�d��} d}tjt|d��, | �d��dd d gid d d gid�� W d Q R X d}tjt|d��" | �d�ddg �dd i� W d Q R X d}tjt|d�� | �d�j�dd i� W d Q R X d S )Nr* r+ r )rL rM rN z nested renamer is not supported)r rL rj r8 r rk r] r )rM rN z!Column\(s\) \['ma'\] do not existrM rN �ma) r rc rq rr r r r" rs rM )r% rt r r r �"test_agg_dict_renaming_deprecation� s " &&rw c C s� t ddddddddgddddddddgtj�d�d t�d�d��} | �dd g�}d }tjt|d �� |d � d ddgi� W d Q R X tjt|d �� |d � ddd�� W d Q R X d S )Nrj rk r r rP g �?)rL rM rN rO rL rM z nested renamer is not supported)r rO rN r8 �std)rN rO ) r r r r rn r rq rr r r" )r% �grt r r r �test_agg_compat� s rz c C s2 t ddddddddgddddddddgtj�d�d t�d�d��} | �dd g�}d }tjt|d ��& |� d d dgidd dgid�� W d Q R X tjt|d ��& |� dd dgidd dgid�� W d Q R X tjt|d �� |d � tj tj d�� W d Q R X tjt|d �� |d � tj tj d�� W d Q R X d S )Nrj rk r r rP g �?)rL rM rN rO rL rM z nested renamer is not supported)r rN rh r8 rO )�r1rm �rarx �rb)rN rO )�result1�result2)rO r ) r r r r rn r rq rr r r9 r" r8 rh )r% ry rt r r r �test_agg_nested_dicts s **"r� c C sd t tjjddd��} dd� }tjtdd��0 tjt dd�� | � d �� |� W d Q R X W d Q R X d S ) N� )� r� )r\ c S s t d� t | �� � td��d S )Nz(----------------------------------------�test)r Z to_string� TypeError)r% r r r �raiseException* s z=test_agg_item_by_item_raise_typeerror.<locals>.raiseExceptionr� )r zDropping invalidr ) r r r �randintrq rr r� r r r! r r" )r% r� r r r �%test_agg_item_by_item_raise_typeerror' s r� c C sB t �� } | �dd� dd� g�}|�tj�}|�� }t �||� d S )Nc S s | j S )N)�year)r. r r r �<lambda>6 � z*test_series_agg_multikey.<locals>.<lambda>c S s | j S )N)rH )r. r r r r� 6 r� )r ZmakeTimeSeriesr r" r r8 r[ )�tsr& r( r' r r r �test_series_agg_multikey4 s r� c C s� t dddddddddddg dddddddddddg dddddddddddg tj�d�tj�d�tj�d�d��} d d � }| �d d g��|�}| �d d g��d d� �}t�||� d S )Nrj rk r r ZdullZshiny� )rL rM rN rO �E�Fc S s t | jj�dkst�dS )Nr rj )r: �values�baser; )r. r r r �badm s z.test_series_agg_multi_pure_python.<locals>.badrL rM c S s dS )Nrj r )r. r r r r� r r� z3test_series_agg_multi_pure_python.<locals>.<lambda>)r r r r r r" r r$ )r< r� r( r' r r r �!test_series_agg_multi_pure_python= sR r� c C s� dd� } t ddddgddd d gt�d dd�t�d dd�t�d dd �t�d dd �gd ��}|�d�}|�| g�}|jjd |_|�| �}t�||� d S )Nc S s t j| �� dd�S )Nr* )�q)r Z percentileZdropna)r r r r �P1y s z test_agg_consistency.<locals>.P1r* r+ r, rT r� � � � i� r� )Zcol1Zcol2r1 r1 r ) r r3 r1 r r" r# Zlevelsr r$ )r� r% ry r' r( r r r �test_agg_consistencyv s r� c C s� t ddgddgd���tj�} G dd� d�}ttjdd � d d � tt�|� g}| �d ��t�}x(|D ] }| �d ��|�}t� ||� qbW d S ) Nr* r+ r, rT )rj rk c @ s e Zd Zdd� ZdS )z$test_agg_callables.<locals>.fn_classc S s t |�S )N)r8 )�selfr. r r r �__call__� s z-test_agg_callables.<locals>.fn_class.__call__N)�__name__� __module__� __qualname__r� r r r r �fn_class� s r� c S s t | �S )N)r8 )r. r r r r� � r� z$test_agg_callables.<locals>.<lambda>c S s | � � S )N)r8 )r. r r r r� � r� rj ) r Zastyper Zint64r8 r r r" r r$ )r% r� Zequiv_callablesr' Zecallr( r r r �test_agg_callables� s r� c C s� t dt�dddg�gdt�dddg�gdt�dddg�ggd d gd �} | �d �}t�dd d g�gt�dddg�gg}tddgd d�}d g}t |||d�}|jdd�}t�||� |jddd�}t�||� d S )Nr* r� r� � �( �2 �< r+ �categoryZ arraydata)r# �F �Z )rY )rA r# F)� numeric_onlyr8 ) r r �arrayr r r8 r r$ r" )r% �gbZ expected_dataZexpected_indexZexpected_columnr' Zaltr( r r r �test_agg_over_numpy_arrays� s $ r� � as_periodTFc C s, t dddd�}| r"|�d ��d�}tddddg|d��}|�d �}|d �d d � �}t|d d d � d d�}d |j_t � ||� |d �dd � �}tddgd d�}d |j_t � ||� |d �dd � �}tt j dd�t j dd�gd d�}d |j_| �r tt j �d�t j �d�gd d�}d |j_t � ||� d S )Nz 2012-01-01rT �UTC)r@ �tzrO r r* )r r r r c S s | j d S )Nr )�iloc)r. r r r r� � r� z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>r+ )rY c S s | j d jS )Nr )r� r� )r. r r r r� � r� i� c S s | j d | j d S )N���r )r� )r. r r r r� � r� )�days)r Z tz_localizeZ to_periodr r r" r rA rY r r[ ro Z Timedelta�offsetsZDay)r� Zdtir% r� r( r' r r r �$test_agg_tzaware_non_datetime_result� s( r� c s� t jddd�� td� fdd�td�D �d��} | �d �d �tj�jd }| �d �d �d d � �jd }| �d �d �� jd }|� ks�t �|� ks�t �|� ks�t �dd� tdd�D �}td d gd |d��} | �d�}| d jd � � |� d �d jd k�st �� |� d�d jd k�s"t �� |� � d jd k�s>t �� |� dd � �jd k�s\t �| d jd � � |�� d jd k�s�t �� |� dd � �jd k�s�t �d S )Nz2016-01-01 12:00:00z US/Pacific)r� r* c s g | ]}� t j|d � �qS ))�minutes)r3 � timedelta)r- �nn)r� r r r/ � s z0test_agg_timezone_round_trip.<locals>.<listcomp>r� )r r r r r c S s t �| �S )N)r r )r. r r r r� � r� z.test_agg_timezone_round_trip.<locals>.<lambda>c S s$ g | ] }t jd |d�d�dd��qS )z 2016-01-0�dz 12:00:00z US/Pacific)r� )ro � Timestamp)r- �ir r r r/ � s r r+ )rL rM rL rM c S s | j d S )Nr )r� )r. r r r r� � r� )r r* c S s | j d S )Nr� )r� )r. r r r r� r� )ro r� r rc r r" r r r� r; Znth�headrV �applyrX )r% r~ r Zresult3�datesr& r )r� r � test_agg_timezone_round_trip� s& r� c C s� t ddgddgddggtd�} | d } td d d gtjd�}t dd d dgi|d�}d|j_| �d�jdd�}t � ||� | �d�jdd�}|g }t � ||� d S )Nr* r+ r, rT r rU )Zdtypel ���� l l l l l l )rA r F)r� T) r �objectr r Zuint64rA rY r r8 r r$ )r% rA r' r( r Z expected2r r r �test_sum_uint64_overflow s r� zstructure, expectedrN )r* r* r* )r, rT rT ))r* r* )r, rT r* r, rT c C s t | �S )N)�tuple)r. r r r r� % r� r� c C s t | �S )N)rI )r. r r r r� ) r� c C sd t ddddddgddddddgddddddgd��}|�ddg��| �}ddg|j_t�||� d S )Nr* r, rT )rL rM rN rL rM )r r r9 rA �namesr r$ )� structurer' r% r( r r r �test_agg_structs_dataframe s 2 r� )rA rY c C s t | �S )N)r� )r. r r r r� = r� c C s t | �S )N)rI )r. r r r r� > r� c C s` t ddddddgddddddgddddddgd��}|�d�d �| �}d|j_t�||� d S )Nr* r, rT )rL rM rN rL rN )r r r9 rA rY r r[ )r� r' r% r( r r r �test_agg_structs_series8 s 2r� c C s� dddg}t tjdddg|d�dddgd��}|jd | d �j�tj�}tddd gtj dddg|d d �d d�}| r|||d k }t � ||� d S )Nr r rg )� categoriesr* r+ r, )rL rM rL )�observedr )r� rY rM )rA rY ) r ro Z Categoricalr rM r" r Znansumr ZCategoricalIndexr r[ )r� r� r% r( r' r r r �test_agg_category_nansumL s " r� c C s� t dd� td�D �dd� td�D �d��} | jdddd�}|�dd d � i�}t d d� td�D �d d� td�D �d��}t�||� d S ) Nc S s g | ] }t |��qS r )�str)r- r. r r r r/ ^ s z+test_agg_list_like_func.<locals>.<listcomp>r, )rL rM rL F)Zas_index�sortrM c S s t | �S )N)rI )r. r r r r� ` r� z)test_agg_list_like_func.<locals>.<lambda>c S s g | ] }t |��qS r )r� )r- r. r r r r/ b s c S s g | ]}t |�g�qS r )r� )r- r. r r r r/ b s )r rc r r" r r$ )r% r& r( r' r r r �test_agg_list_like_func\ s *(r� c C sx t ddgtjddd�tjddd�gd��} | �d��dd d � i�}t tjddd�gtdgdd �dgd �}t�||� d S ) Nr* z 2018-01-01r� )r� z 2018-01-02)�tagr1 r� r1 c S s | � d�S )Nr* )r� )�er r r r� r r� z/test_agg_lambda_with_timezone.<locals>.<lambda>)rY )rA r# )r ro r� r r" r r r$ )r% r( r' r r r � test_agg_lambda_with_timezoneg s r� �err_clsc s� ddl m}m}m} |� d d� }tdddddgdddddg||�d��}t||d |d g��}� fdd�}|d �|d ��|�}tj ||d d � d S ) Nr )� DecimalArray� make_data� to_decimalr r* )�id1Zid2�decimalsr, c s t | �dkr� �| jd S )Nr )r: r� )r. )r� r r � weird_func� s z1test_groupby_agg_err_catching.<locals>.weird_funcr� r� F)Z check_names) Z$pandas.tests.extension.decimal.arrayr� r� r� r r r r" r r[ ) r� r� r� r� r< r% r' r� r( r )r� r � test_groupby_agg_err_catching{ s & r� )B�__doc__r4 r3 � functoolsr �numpyr rq Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasro r r r r r r r Zpandas._testingZ_testingr Zpandas.core.baser Zpandas.io.formats.printingr r) r= rK r_ rf ri ru rw rz r� r� r� r� r� r� Z&skip_array_manager_not_yet_implementedr� �markZ parametrizer� r� r� r� rI r� r� r� r� r� �NotImplementedError� RuntimeErrorrs � IndexError�OSError� ValueError�ArithmeticError�AttributeErrorr� r r r r �<module> sp $ %& / 9 " ,$ 6
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка