Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/__pycache__/test_aggregate.cpython-37.pyc
Назад
B �5�g�� � @ s� d Z ddlZddlZddlmZ ddlZddlZddlZddlm Z ddl m Z ddl Z ddl mZmZmZmZmZmZ ddlmZ ddlmZ ddlmZ d d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"dd� Z#ej$�%dd d � d d � d d � gg�d d � �Z&d!d"� Z'ej$�%d#d$d%d&d'gd(i fd)d*gd+ e(e)e)e)d,�fd-d.gd+ e(e)e)e)d,�fd$d%d&d'gd/d0d(ifd)d*gd+ d/e)e)e)d,�fd-d.gd+ d/d1d1d1d,�fg�d2d3� �Z*ej$�%d4d$d5d6gd7d8gd9d:ggd(d(d;�fd)d<gd5 gd+ d1fd-d5gd5 gd+ d1d1d;�fg�d=d>� �Z+d?d@� Z,dAdB� Z-dCdD� Z.dEdF� Z/dGdH� Z0dIdJ� Z1dKdL� Z2dMdN� Z3dOdP� Z4ej$�%dQdRd � dSd � dTd � dUd � dVd � dWd � g�dXdY� �Z5ej$�%dZd[ged\gd[d]�fd[d^ged\gd5ggdgdggd[d^gd_�fg�ej$�%d`dadbd(dcd1g�ej$�%dddadbd(dcd1g�ej$�%dedfdgdhg�didj� ����Z6ej$�%dZd[ged\gd[d]�fd[d^ged\gd5ggdgdggd[d^gd_�fg�ej$�%dkdld\dmg�ej$�%dne7e(e)g�ej$�%dedfdgdhg�dodp� ����Z8dqdr� Z9ej$�%dnej:ej;g�ej$�%dsdtdudvdwdxdyg�dzd{� ��Z<d|d}� Z=ej$�%d~e �>e?d��e �@dd+�e jAd�d+d�d��e�Bd�d�d�g�g�d�d�� �ZCd�d�� ZDd�d�� ZEG d�d�� d��ZFG d�d�� d��ZGej$�%d�d�d�ejHfd�d\d+gdd5gd�d�gfd�d�d � fd�d�d � fd�d\d+gd\d\gd�d�gfe �Id�dw�e �Id�ejJ�e �Id�d�d � �d\d+gd�d�gd\d\gfg�d�d�� �ZKd�d�� ZLd�d�� ZMej$�%d�d�dvgig dvgd��g�d�d�� �ZNd�d�� ZOd�d�� ZPd�d�� ZQej$�%d�d�d � d�d � d�d � g�d�d�� �ZRd�d�� ZSej$�%d�d�d � eSg�d�d�� �ZTd�d�� ZUej$jVd�d��d�d�� �ZWG d�d�� d��ZXd�d�� ZYej$�%d�dvdvd��d\d%gd[d�gd��fd�dvid�d[d�gifd�dvid�d\d%gifg�d�dń �ZZej$�%d�dvdwgdvd��d�d�gfdvdvdwgd��d�d�gfd�dvdwgid�d�gfg�d�d̈́ �Z[d�dτ Z\d�dф Z]d�dӄ Z^d�dՄ Z_d�dׄ Z`d�dل Zaej$�%d�d�d�d�gd�gfd�d�d�gd�gfg�d�d� �ZbdS )�zN test .agg behavior / note that .apply is tested generally in test_groupby.py � N)�partial)�PerformanceWarning)�is_integer_dtype)� DataFrame�Index� MultiIndex�Series�concat� to_datetime)�SpecificationError)�Groupingc C sB t ddddgddddgd��} | �d �d }d d � }|�|� d S ) N�a�b�c� � � � )�key�valuer r c S s t | �dkst�| �� S )Nr )�len�AssertionError�sum)�x� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.py� dummy_func# s z3test_groupby_agg_no_extra_calls.<locals>.dummy_func)r �groupby�agg)�df�gbr r r r � test_groupby_agg_no_extra_calls s r! c C s: | � dd� dd� g�}|�tj�}|�� }t�||� d S )Nc S s | j S )N)�year)r r r r �<lambda>+ � z&test_agg_regression1.<locals>.<lambda>c S s | j S )N)�month)r r r r r# + r$ )r r �np�mean�tm�assert_frame_equal)�tsframe�grouped�result�expectedr r r �test_agg_regression1* s r. c C sf | � d�d }d}tjt|d�� |�dd� � W d Q R X tjt|d�� |�dd� � W d Q R X d S )N�A�Cz Must produce aggregated value)�matchc S s | � � S )N)Zdescribe)r r r r r# 6 r$ z#test_agg_must_agg.<locals>.<lambda>c S s | j d d� S )Nr )�index)r r r r r# 8 r$ )r �pytest�raises� Exceptionr )r r+ �msgr r r �test_agg_must_agg1 s r7 c C sN | j }dd� }| j �| j| jg��|�}| �ddg��� d }t�||� d S )Nc S s | � � S )N)r )r r r r r# ? r$ z(test_agg_ser_multi_key.<locals>.<lambda>r/ �Br0 )r0 r r/ r8 � aggregater r( �assert_series_equal)r �ser�f�resultsr- r r r �test_agg_ser_multi_key; s r> c C s t dddtjddddgdddtjddd d gd �tjd d ddddddgddgd�d�} t dddddddtjddddg d ddddddtjd dddg d d dd tjd dd d dtjd g d!d"dd#tjd$d#dd dtjtjg d%��}|�ddg�}|d&d'g �� }t�|| � d S )(N� � r r r �7 �M �! �, � )�v1�v2)r �_ )r �c )r rH )r rI )�big�damp)�blue�dry)�redrN )rN �wet�by1�by2)�names)r2 � � �X rI rN rL r rJ � rO rM rH rK )rF rG rP rQ rF rG ) r r& �nanr � from_tuplesr r' r( r) )r- r �gr, r r r �$test_groupby_aggregation_mixed_dtypeE s@ rZ c C s� ddddgddt jdgddt jdgddt jdgg} t| t�ddddg�d�}|jddd �}|jdd �}td dddgdd ddgd ��}t�||� d S )NTF)r/ r )r/ r )r8 r )r8 r )�data�columnsr )�level�axis)Z numeric_onlyg @r )r r ) r& rW r r rX r r r( r) )�lstr r r, r- r r r �+test_groupby_aggregation_multi_level_columnv s r` c C s� | � | tj �}| jtjks t�tg tjtg tjd�d�}t� |� � |� t� |� tj �|� tj |� tj �|dd� |� |d tj �}t |jttg dtjd�d�}t�|� � |� t�|� tj �|� t�|� tj �|� d S )N)�dtype)ra r2 F)Zcheck_index_typer/ )�namera )r\ ra r2 )r r& rW ra �float64r r r r( r: r r �applyr r\ �floatr) )�tsr* r+ �expZexp_dfr r r �test_agg_apply_corner� s rh c C s� t �� }|�dd� �}|jjd j}t| jt|��|jjd< |� t j �}|� � }t � ||� t| jt |��|jjd< |� t j �}|� � }t � ||� d S )Nc S s | j S )N)r" )r r r r r# � r$ z1test_agg_grouping_is_list_tuple.<locals>.<lambda>r )r( ZmakeTimeDataFramer �grouperZ groupingsZgrouping_vectorr r2 �listr r& r' r) �tuple)rf r r+ ri r, r- r r r � test_agg_grouping_is_list_tuple� s rl c C s2 | � ddg�}|�tj�}|�� }t�||� d S )Nr/ r8 )r r r& r' r( r) )Zmframer+ r, r- r r r �test_agg_python_multiindex� s rm � groupbyfuncc C s | � � S )N)�weekday)r r r r r# � r$ r# c C s | j S )N)r% )r r r r r# � r$ c C s� | � |�}|d �d�}|d �� }t�||� |�d�}|�� }t�||� |�ddddd��}t|d �� |d �� |d � � |d � � d��}t�||� d S ) Nr/ �std�varr' �sem)r/ r8 r0 �Dr8 r0 rs ) r r rp r( r: r9 rq r) r r' rr )r* rn r+ r, r- r r r �test_aggregate_str_func� s rt c C sZ | j dd�}|dkr t}d}nt}d|� d�}tj||d�� |j|dd � W d Q R X d S ) Nr )r] )ZidxmaxZidxminz3reduction operation '.*' not allowed for this dtypez Operation z does not support axis=1)r1 r )r^ )r � TypeError� ValueErrorr3 r4 r )r Zreduction_funcr �errorr6 r r r �%test_agg_str_with_kwarg_axis_1_raises� s rx z(func, expected, dtype, result_dtype_dictr r? r@ rT �int64rp g�l��v� @r )�i�j�krq g @�Int64r{ rc c C s� t ddddddggd t�ddgd d d gg�d ��||d ��}|jddd��| �}t |gd d d d gd ��|�}t�||� d S )Nr r r r r? � r r rz r{ r| )r\ ))r r{ )r r{ )r] r^ )r r � from_product�astyper r r( r) )�funcr- ra �result_dtype_dictr r, r r r �(test_multiindex_groupby_mixed_cols_axis1� s r� z&func, expected_data, result_dtype_dictr r � rV � � )r� r� g�;f���?c C s� t t�d��dd�tdddgdd�tdd dd gd d�d d ��dd i�}|jd dd��| �}t |tdddgdd�tdd gd d�d��|�}t� ||� d S )NrV r r r r �y)rb r� r� r ry )r2 r\ ra r} )r^ )r[ r2 r\ ) r r& Zarange�reshaper r� r r r( r) )r� � expected_datar� r r, r- r r r � test_groupby_mixed_cols_axis1� s r� c C s� | � d�}dd� }|�|�}| jdk�� }| jdk�� }t|j�}tt�|g| �t d�dd�}t � |� d�|� tt�|g| �t d�dd�}t � |� d�|� dd � }t� � | j��|�}t|t�s�t�t|�d ks�t�d S ) Nr/ c S s | j S )N)�size)r; r r r r# r$ z-test_aggregate_item_by_item.<locals>.<lambda>�foo�barZBCD)r2 rb c S s | j S )N)r� )r; r r r �aggfun& s z+test_aggregate_item_by_item.<locals>.aggfunr )r r r/ r r r\ r r& �arrayrj r( r: ZxsZassert_almost_equalr � isinstancer )r r+ r� r, r� r� �Krg r r r �test_aggregate_item_by_item s r� c C sv | � ddg�}dd� }tjtdd�� |�|�}W d Q R X | jd d �| jdkf }|� ddg��|�}t�||� d S )Nr/ r8 c S s | j tkrt�n| �� S d S )N)ra �objectru r )r; r r r r� 1 s z test_wrap_agg_out.<locals>.funczDropping invalid columns)r1 r0 )r r( �assert_produces_warning� FutureWarningr9 �locr\ r) )Z three_groupr+ r� r, Z exp_groupedr- r r r �test_wrap_agg_out. s r� c C sP dt jfdt jfdt jfg}| �d�d �|�}tdddg�}t�|j |� d S )Nr' �max�minr/ r0 ) r& r' r� r� r r r r( �assert_index_equalr\ )r �funcsr, Zexp_colsr r r �*test_agg_multiple_functions_maintain_order> s r� c s� t tj�dd�tjdddd�dddgd �� � �d ��dttj d d �ttj d d �gi�} tjdd dd�}t � ddg�}t� � fdd�dD ��j }t |||d�}t�| |� d S )Ni� r z1/1/2012�S)�freq�periodsr/ r8 r0 )r2 r\ �3Tg ��H.��?)�qg�镲 q�?r~ )r/ �quantilec s" g | ]}� � d �jj|d�j�qS )r� )r� )�resampler/ r� �values)�.0r� )r r r � <listcomp>T s z9test_agg_multiple_functions_same_name.<locals>.<listcomp>)g ��H.��?g�镲 q�?)r\ r2 )r r& �random�randn�pd� date_ranger� r r r� r rX r� �Tr( r) )r, �expected_index�expected_columns�expected_valuesr- r )r r �%test_agg_multiple_functions_same_nameG s " r� c s� t tj�dd�tjddddd�tddd gd d �d �� � �d ��ddt tj dd�t tj dd�gi�} tjdd ddd�}t j ddddddgd d d gd�}t� � fdd�dD ��j}t�� �d �j�� |g�}t |||d �}t�t�� t�| |� W d Q R X d S ) Ni� r z1/1/2012r� Zdti)r� r� rb r/ r8 r0 �alpha)rb )r2 r\ r� �ohlcg ��H.��?)r� g�镲 q�?r~ )r/ r� �open)r/ r� �high)r/ r� �low)r/ r� �close)r/ r� r/ )rR c s" g | ]}� � d �jj|d�j�qS )r� )r� )r� r/ r� r� )r� r� )r r r r� t s zKtest_agg_multiple_functions_same_name_with_ohlc_present.<locals>.<listcomp>)g ��H.��?g�镲 q�?)r\ r2 )r r& r� r� r� r� r r� r r r� r rX r� r� Zhstackr/ r� r( r� r r) )r, r� r� Znon_ohlc_expected_valuesr� r- r )r r �7test_agg_multiple_functions_same_name_with_ohlc_present\ s, $ r� c C st ddg}ddg}| � d�d �|�}| � d�d �|�}t�||� | � d��|�}| � d��|�}t�||� d S )N)r� r' rp )rp rp r/ r0 )r r r( r) )r r� Zex_funcsr, r- r r r �-test_multiple_functions_tuples_and_non_tuples s r� c C s@ | � d�}|�tjtjd��}|�tjtjd��}t||gddgdd�}|jdddd�jddd �}tjtjgtjtjgd�}|�|�}t � ||� |�tjtjtjgd��}|�tjtjtjgd��}t � ||� d d � }d d � }d} t j t | d��& tjtjtjd�d�}|�|� W d Q R X tjg||gd�}|�|� d S )Nr/ )r0 rs r' rp r )�keysr^ r )r^ )r] r^ c S s t �| �S )N)r& r' )r r r r r� � s z4test_more_flexible_frame_multi_function.<locals>.fooc S s t j| dd�S )Nr )Zddof)r& rp )r r r r r� � s z4test_more_flexible_frame_multi_function.<locals>.barz nested renamer is not supported)r1 )r� r� )r r r& r' rp r Z swaplevelZ sort_indexr9 r( r) r3 r4 r ) r r+ ZexmeanZexstdr- �dr, r� r� r6 r r r �'test_more_flexible_frame_multi_function� s& r� c C s� | � d�}ddd�ddid�}d}tjt|d�� |�|� W d Q R X ddd�dd�}tjt|d�� |�|� W d Q R X ddd�dd�}tjt|d�� |�|� W d Q R X d S ) Nr/ r' rp )r� r� r )r0 rs z nested renamer is not supported)r1 )r r3 r4 r r9 )r r+ r� r6 r r r � test_multi_function_flexible_mix� s r� c C s� t ddddgddddgd d ddgd��} | �d�}tddgdd�}|d �d d � �}td d g|dd �}t�||� |d �dd � �}td d g|dd �}t�||� d S )Nr r r r )r r r r )rb r c S s | dk� � S )Nr )�all)r r r r r# � r$ z1test_groupby_agg_coercing_bools.<locals>.<lambda>FT)r2 rb r c S s | � � �� S )N)Zisnullr� )r r r r r# � r$ )r r r r9 r r( r: )Zdat�gpr2 r, r- r r r � test_groupby_agg_coercing_bools� s ( r� �opc C s | � � S )N)r )r r r r r# � r$ c C s | � � S )N)�cumsum)r r r r r# � r$ c C s | � d�S )Nr )� transform)r r r r r# � r$ c C s | � d�S )Nr� )r� )r r r r r# � r$ c C s | � d�S )Nr )r )r r r r r# � r$ c C s | � d�S )Nr� )r )r r r r r# � r$ c C sd t ddgddgd��}|�d�d }| |�d��d j}t|�sDt�| |�d��j}t|�s`t�d S )Nr FT)r r r r )r � set_indexr ra r r )r� r �sr, r r r �test_bool_agg_dtype� s r� zkeys, agg_indexr r )rb r )rR � input_dtype�boolZint32Zfloat32� result_dtype�methodrd r9 r� c s� t dgdgdgd��}|d �|�|d< t|�| �dg |�}|� fdd��}|dkrbt�d d�n|}t d|d jd gi|d ��� �} |d kr�d g| j_t � || � d S ) Nr r T)r r r r c s | � � �jd S )Nr )r� �iloc)r )r� r r r# r$ z2test_callable_result_dtype_frame.<locals>.<lambda>r� r )r2 rd ) r r� �getattrr r� � RangeIndexr� r\ rR r( r) ) r� � agg_indexr� r� r� r r� r, r� r- r )r� r � test_callable_result_dtype_frame� s r� �inputTg �?ra c s� t dgdg|gd��}t|�| �d |�}|� fdd��}|dkrNt�dd�n|}t|d jd g|dd ��� �} t� || � d S ) Nr r )r r r r c s | � � �jd S )Nr )r� r� )r )ra r r r# r$ z3test_callable_result_dtype_series.<locals>.<lambda>r� r )r2 rb ) r r� r r� r� r r� r� r( r: ) r� r� r� ra r� r r� r, r� r- r )ra r �!test_callable_result_dtype_series s r� c C sf t ddddgddddgd��} | �d��ddd d d g�}|jjd }tddd d d g�}t�||� d S ) Nr r r r )r/ r8 r/ r r� r' r� r� )r r r r\ �levelsr r( r� )r �resr, r- r r r �#test_order_aggregate_multiple_funcs$ s r� �how�first�lastr� r� r' �medianc C st t dddgd��}|�d��d|i�}|j�| �|_|�d��d|i�}|dkr`|j�tj�|_tj||dd � d S ) Nl &tLW Hd~ r r )r r� r� r )r' r� T)Z check_exact) r r r r r� r& ry r( r) )ra r� r r- r, r r r �test_uint64_type_handling0 s r� c C sT d} t ddddgddddgd��}tjt| d�� |�d ��d d g� W d Q R X d S ) NzFunction namesr r r r r )r/ r8 )r1 r/ r� )r r3 r4 r r r )r6 r r r r �test_func_duplicates_raises> s r� r2 �abcZ2020rs )r� r� )r r )r r )r r c C s` t dddgdddgd�| d�}|�d��dtji�}t ddgddgd���d�}t�||� d S )Nr r r )�groupr )r2 r� r )r r r r Znuniquer� r( r) )r2 r r, r- r r r �$test_agg_index_has_complex_internalsF s r� c C sr t dddddgdddddgdddddgd��} | �d��� }t ddgddgd �tddgdd �d �}t�||� d S ) Nr r �one�two�three�six)�key1�key2�key3r� )r� r� )rb )r2 )r r r� r r( r) )r r, r- r r r �test_agg_split_blockW s r� c C s� t tjddd�ddddgdd d dgdddd gtjddd�dd d dgd ���t�} | �d d d d g��� }t t�d�gdgdgdgt�d�gdgd ��}t� ||� d S )N�2000r )r� r r r r� r r r �e)r/ r8 r0 rs �E�Fr ) r r� r� r� r� r r� � Timestampr( r) )r r, r- r r r �#test_agg_split_object_part_datetimeh s" r� c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zej� d e j d d d �d d d gg�dd� �Z dS )�TestNamedAggregationSeriesc C s� t ddddg�}|�ddddg�}|jddd�}tdd gddgd�d d gddgd �}t�||� |jddd �}|d d g }t�||� d S )Nr r r r r r r� )r r r@ r r )r\ r2 )r r )r r r r r( r) )�selfr �grr, r- r r r �test_series_named_agg� s " z0TestNamedAggregationSeries.test_series_named_aggc C sX t ddg��ddg�}tjtdd�� |�� W d Q R X |�g �}t� }t�||� d S )Nr r r z Must provide)r1 ) r r r3 r4 ru r r r( r) )r� r� r, r- r r r �test_no_args_raises� s z.TestNamedAggregationSeries.test_no_args_raisesc C sN t dddg��dddg�}|jddd�}tddgddgd��}t�||� d S )Nr r r r r )r r )r r r r r( r) )r� r� r+ r- r r r �*test_series_named_agg_duplicates_no_raises� s zETestNamedAggregationSeries.test_series_named_agg_duplicates_no_raisesc C sV t dddg��dddg�}|jdd� dd� d�}tddgddgd��}t�||� d S ) Nr r r r c S s dS )Nr r )r r r r r# � r$ z9TestNamedAggregationSeries.test_mangled.<locals>.<lambda>c S s dS )Nr r )r r r r r# � r$ )r r )r r r r r( r) )r� r� r, r- r r r � test_mangled� s z'TestNamedAggregationSeries.test_mangled�inp�anythingr� )�column�aggfunc)r� r� c C sZ t ddddddddg�}dt|�j� �}tjt|d�� |�|j�j|d� W d Q R X d S ) Nr r r r r? z func is expected but received )r1 )r ) r �type�__name__r3 r4 ru r r� r )r� r� r� r6 r r r �test_named_agg_nametuple� s z3TestNamedAggregationSeries.test_named_agg_nametupleN) r� � __module__� __qualname__r� r� r� r� r3 �mark� parametrizer� �NamedAggr� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestNamedAggregationDataFramec C s t ddddgddddgddd d gd ��}|�d �jd dd�}t ddgdd gd�tddgd d�ddgd�}t�||� tjtj dd�}|�d �jddt fdtj fd dd|fd�}t dd gddgddgddgdd gdd gd�tddgd d�d d d ddd gd�}t�||� d S )!Nr r r r r r r? r~ r@ rS )r� r/ r8 r� )r/ r� )r8 r� )�a_max�b_max)rb r� r� )r2 r\ �b )r� )r8 r� r/ )�b_min�a_min�a_meanr� r� �a_98g �?g @g\��(\�?gףp= �@r r r r ) r r r r r( r) � functoolsr r& � percentiler� r' )r� r r, r- Zp98r r r �test_agg_relabel� s4 & z.TestNamedAggregationDataFrame.test_agg_relabelc C sn t ddddgddddgddd d gd ��}|�d �jf d di�}t d ddgitddgd d�d�}t�||� d S )Nr r r r r r r? r~ r@ rS )r� r/ r8 r� zmy col)r/ r� )rb )r2 )r r r r r( r) )r� r r, r- r r r � test_agg_relabel_non_identifier� s & z=TestNamedAggregationDataFrame.test_agg_relabel_non_identifierc C s t ddddgddddgd��}|�d�jddd �}t ddgddgd �tddgdd �d �}t�||� tjtj d d �}tjtj dd �}d|_ d|_ t dddddgdddddgd��}|�d�jd|fd|fd�}t ddgddgd�tddgdd �d �}t�||� d S ) Nr r r r r )r/ r8 r/ )r8 r� )r r )rb )r2 �2 )r� �F �quant50�quant70r r r? )�col1�col2r r )Z quantile_50Z quantile_70g �?g @g333333�?g������@) r r r r r( r) r r r& r r� )r� r r+ r- r r �testr r r �test_duplicate_no_raises� s & " z6TestNamedAggregationDataFrame.test_duplicate_no_raisesc C s� t ddddgddddgd�t�ddgd d gg�d �}|jdd �jd ddd�}t ddgddgddgd�ddgd �}t�||� d S )Nr r r r r )r/ r8 r/ r8 r r )r2 )r] )r/ r� )r/ r� )r8 r' )ZaaZbb�ccg �?g @)r r r r r r( r) )r� r r, r- r r r �test_agg_relabel_with_level� s "z9TestNamedAggregationDataFrame.test_agg_relabel_with_levelc C s� t dddgdddgd��}|�d�}d}tjt|d�� |jdd � W d Q R X tjt|d�� |�� W d Q R X tjt|d�� |jd d d � W d Q R X d S ) Nr r r r )r/ r8 r/ z Must provide)r1 )r� )r8 r� )r r r )r r )r r r3 r4 ru r )r� r r+ r1 r r r � test_agg_relabel_other_raises s z;TestNamedAggregationDataFrame.test_agg_relabel_other_raisesc C sP t ddgddgd��}t�d�}tjt|d�� |�d�jdd � W d Q R X d S ) Nr r r )r/ r8 z Column(s) ['C'] do not exist)r1 r/ )r0 r )r )r �re�escaper3 r4 �KeyErrorr r )r� r r1 r r r �test_missing_raises s z1TestNamedAggregationDataFrame.test_missing_raisesc C s` t ddgddgd��}|�d�jt�dd�tjddd �d �}|�d�jd d d �}t�||� d S ) Nr r r )r/ r8 r/ r8 r �count)r� r� )r r )r8 r )r8 r )r r r r� r� r( r) )r� r r, r- r r r �test_agg_namedtuple s z1TestNamedAggregationDataFrame.test_agg_namedtuplec C sv t ddgddgddgd��}|�d�jdd d � fd d d � fd �}t ddgddgd �tddgdd�d�}t�||� d S )Nr r r r r )r/ r8 r0 r/ r8 c S s dS )Nr r )r r r r r# ( r$ z<TestNamedAggregationDataFrame.test_mangled.<locals>.<lambda>r0 c S s dS )Nr r )r r r r r# ( r$ )r r )rb )r2 )r r r r r( r) )r� r r, r- r r r r� & s $&z*TestNamedAggregationDataFrame.test_mangledN) r� r� r� r r r r r r r r� r r r r r� � s $ r� zCagg_col1, agg_col2, agg_col3, agg_result1, agg_result2, agg_result3))r� r/ r� )r� r/ ))r� r8 r' g @g @c C s t | �S )N)r� )r r r r r# 9 r$ c C s dS )Nr r )r r r r r# : r$ )r� r8 c C s dS )Nr r )r r r r r# C r$ c C s� t ddddgddddgddd d gd ��}t�d d dg�|_tddgd d�}|�d �jdd�}t dddgi|d�} t�|| � |�d �j| ||d�}t |||d�|d�} t�|| � d S )Nr r r r r r r? r~ r@ rS )r� r/ r8 )r r� )r� r/ )r� r8 )rb ))r� r/ r� )r� r� )r2 )Zcol_1Zcol_2Zcol_3) r r rX r\ r r r r( r) ) Zagg_col1Zagg_col2Zagg_col3Z agg_result1Z agg_result2Z agg_result3r �idxr, r- r r r �"test_agg_relabel_multiindex_column- s !& r c C sj t ddddgddddgddd d gd ��} t�d d dg�| _tjtdd�� | �d �jdd� W d Q R X d S )Nr r r r r r r? r~ r@ rS )r� r/ r8 )r r� )r� r/ )r� r8 z do not exist)r1 ))�Yr r� )r ) r r rX r\ r3 r4 r r r )r r r r �,test_agg_relabel_multiindex_raises_not_exista s &r c C s� t ddddgddddgddd d gd ��} t�d d dg�| _| �d �jddd�}tddgd d�}t ddgddgd�|d�}t�||� d S )Nr r r r r r r? r~ r@ rS )r� r/ r8 )r r� )r� r/ )r� r8 ))r� r/ r� )r r )rb )r2 ) r r rX r\ r r r r( r) )r r, r r- r r r �&test_agg_relabel_multiindex_duplicatesl s & r �kwargsr )r r c C sn t dddgdddgdddgd��}|�d��| �}t ddgtddgddd�t�d d gg�d �}t�||� d S ) Nr r r r )r r r r ry )ra rb r r� )r2 r\ )r r r r r rX r( r) )r r r, r- r r r � test_groupby_aggregate_empty_key} s "r c C sb t dddgdddgdddgd��} | �d��dg i�}t tdgg gg g gd�d �}t�||� d S ) Nr r r r )r r r r r )r� �codes)r\ )r r r r r( r) )r r, r- r r r �-test_groupby_aggregate_empty_key_empty_return� s " r" c C st t ddddddgdddddd gd ��} | jd d d �}|jtjddd�d�}t dddgddd gd��}t�||� d S )Nr r� �zg �?g�������?g @g @g������ @g �?)r �valr F)�as_indexr$ r� )r� r� )�min_val)r r& )r r r r� r� r( r) )r r+ r, r- r r r �9test_grouby_agg_loses_results_with_as_index_false_relabel� s $r' c C s� t ddddddgddddddgdddd d d gd ��} | jd dgdd�}|jtjddd�d�}t dddgdddgdd dgd��}t�||� d S )Nr r� r r r g �?g�������?g @g @g������ @g �?)r r� r$ r r� F)r% r$ r� )r� r� )r& )r r� r& )r r r r� r� r( r) )r r+ r, r- r r r �Dtest_grouby_agg_loses_results_with_as_index_false_relabel_multiindex� s r( r� c C s | � � S )N)r' )r� r r r r# � r$ c C s t �| �S )N)r& r' )r� r r r r# � r$ c C s t �| �S )N)r& Znanmean)r� r r r r# � r$ c C s� dddgdddgg}t |tjdddgdddggddgd �d �}|�t�d dg���| �}d d d�ddd�dd d�d�}t |�}|j|_t� ||� d S )Nr r r r? r@ r ZSiskoZJaneway)rR )r\ r g �?g @)r r g @g @g @))r r )r r )r r ) r r Z from_arraysr r& r� r r\ r( r) )r� r[ r r, � expected_dictr- r r r �test_multiindex_custom_func� s "r* c C s t j| dd�S )Ng�������?)r� )r& r )r� r r r �myfunc� s r+ c C s t j| dd�S )Ng�������?)r� )r& r )r� r r r r# � r$ c C sz t ddddgddddgddd d gd ��}|�d �jd d| fd�}t ddgddggddgtddgd d�d�}t�||� d S )N�cat�dogg333333"@g @g #@g A@g������ @g @g������#@g �h@)�kind�height�weightr. )r/ r' r/ )� mean_height�perc90g������"@g0�' 5"@g 4@g5^�I @r1 r2 )rb )r\ r2 )r r r r r( r) )r� Zanimalsr, r- r r r �test_lambda_named_agg� s r3 c C s� t t�dgd ��dd�td�td�d�} dddg| d < | �d ��d d � �}dgdgdggddgddgddggg}t |tddgd d d �tdddgd d�d�}t� ||� d S )Nr rT r ZXYZr� )r[ r\ r2 zgroup 1r �groupingc S s | � � S )N)�tolist)r r r r r# � r$ z,test_aggregate_mixed_types.<locals>.<lambda>r� )ra rb �Xr �Z)ra )r2 r\ ) r r& r� r� rj r r9 r r( r) )r r, r� r- r r r �test_aggregate_mixed_types� s (&r8 z Not implemented;see GH 31256)�reasonc C sj dd� } t dt�dddg�i�}|�dddg��| �}t dtjdtjgdd�iddgd �}t�||� d S ) Nc S s t | dk�rdS tjS d S )Nr r )r� r� �NA)r r r r r� s z5test_aggregate_udf_na_extension_type.<locals>.aggfuncr/ r r r r} )ra )r2 )r r� r� r r r: r( r) )r� r r, r- r r r �$test_aggregate_udf_na_extension_type� s $r; c @ sN e Zd Zdd� Zdd� Zejjdd�ej�d�dd � ��Z d d � Z d d � Z dS )�TestLambdaManglingc C st t ddddgddddgd��}|�d��dd d � d d � gi�}t ddgddgd �tddgdd �d�}t�||� d S )Nr r r r r )r/ r8 r/ r8 c S s dS )Nr r )r r r r r# r$ z/TestLambdaMangling.test_basic.<locals>.<lambda>c S s dS )Nr r )r r r r r# r$ ))r8 z <lambda_0>)r8 z <lambda_1>)rb )r2 )r r r r r( r) )r� r r, r- r r r � test_basic s z TestLambdaMangling.test_basicc C sZ t ddddg��ddddg�}|�dd� dd� g�}tddgddgd ��}t�||� d S ) Nr r r r r c S s dS )Nr r )r r r r r# r$ z?TestLambdaMangling.test_mangle_series_groupby.<locals>.<lambda>c S s dS )Nr r )r r r r r# r$ )z <lambda_0>z <lambda_1>)r r r r r( r) )r� r� r, r- r r r �test_mangle_series_groupby s z-TestLambdaMangling.test_mangle_series_groupbyz GH-26611. kwargs for multi-agg.)r9 z-ignore:Dropping invalid columns:FutureWarningc C s� ddd�}ddd�}t ddg��ddg��||gd�}tdgdgd ��}t�||� t ddg��ddg�j||gdd d �}td gd gd ��}t�||� d S )Nr c S s | � � | | S )N)r )r r� r r r r r# r$ z5TestLambdaMangling.test_with_kwargs.<locals>.<lambda>r c S s | � � || S )N)r )r r� r r r r r# ! r$ r r r~ )z <lambda_0>z <lambda_1>r� )r � � )r )r )r r r r r( r) )r� �f1�f2r, r- r r r �test_with_kwargs s " &z#TestLambdaMangling.test_with_kwargsc C s� t ddddgddddgddd d gd ��}d d dg}t ddgddgd d gd�tddgdd�|d�}|jdd�jtjddd� d�tjddd�tjddd�d�}t�||� |jdd�jdd d� fd d d�}t�||� d S ) Nr, r- g333333"@g @g #@g A@g������ @g @g������#@g �h@)r. r/ r0 �height_sqr_min� height_max� weight_maxg�p= ׳T@g B@)rD rE rF r. )rb )r2 r\ )�byr/ c S s t �| d �S )Nr )r& r� )r r r r r# B r$ z=TestLambdaMangling.test_agg_with_one_lambda.<locals>.<lambda>)r� r� r� r0 c S s t �| d �S )Nr )r& r� )r r r r r# K r$ )r/ r� )r0 r� )r r r r r� r� r( r) )r� r r\ r- �result1�result2r r r �test_agg_with_one_lambda* s, z+TestLambdaMangling.test_agg_with_one_lambdac C s t ddddgddddgddd d gd ��}d d dddg}t ddgddgd d gddgddgd�tddgdd�|d�}|jdd�jddd� fdd dd d� fd d d� fd�}t�||� |jdd�jtjdd d� d!�tjdd"d!�tjd d"d!�tjdd#d� d!�tjd d$d� d!�d�}t�||� d S )%Nr, r- g333333"@g @g #@g A@g������ @g @g������#@g �h@)r. r/ r0 rD rE rF � height_max_2� weight_ming�p= ׳T@g B@)rD rE rF rK rL r. )rb )r2 r\ )rG r/ c S s t �| d �S )Nr )r& r� )r r r r r# p r$ z=TestLambdaMangling.test_agg_multiple_lambda.<locals>.<lambda>)r/ r� )r0 r� c S s t �| �S )N)r& r� )r r r r r# s r$ r0 c S s t �| �S )N)r& r� )r r r r r# t r$ c S s t �| d �S )Nr )r& r� )r r r r r# { r$ )r� r� r� c S s t �| �S )N)r& r� )r r r r r# r$ c S s t �| �S )N)r& r� )r r r r r# � r$ )r r r r r( r) r� r� )r� r r\ r- rH rI r r r �test_agg_multiple_lambdaQ s@ z+TestLambdaMangling.test_agg_multiple_lambdaN) r� r� r� r= r> r3 r� �xfail�filterwarningsrC rJ rM r r r r r< s 'r< c C s^ t dddgdddgd��} | �d��ddd � i�}t ddgddgd���d�}t�||� d S ) Nr� �Wg �?g @)r/ r8 r/ r8 c S s | � | jd �S )N���)�getr2 )r r r r r# � r$ z+test_groupby_get_by_index.<locals>.<lambda>)r r r r� r( r) )r r� r- r r r �test_groupby_get_by_index� s rS zgrp_col_dict, exp_data)�nr�cat_ordrU rT c C s� t ddddddddgtd �td �d ��}|�d d d ��}|d j�� |d< |�d��| �}tjddgddgddd d�}t ||d�}d|kr�|d j }|d �|�|d< t � ||� d S )Nr r r r r? r~ r@ rS �aabbccdd�aaaabbbb)rT rU r, �category)r, rU rU r, r r F)� categories�orderedrb ra )r[ r2 ) r rj r� r, � as_orderedr r r� �CategoricalIndexra r( r) )� grp_col_dict�exp_data�input_df� result_df� cat_index� expected_dfra r r r � test_groupby_single_agg_cat_cols� s rc )r r r )r? rS r )r r r )r? r r� )r r )r r� c C s, t ddddddddgtd �td �d ��}|�d d d ��}|d j�� |d< |�d��| �}tjddgddgddd d�}g }xJ| � � D ]>\}}t |t�r�x*|D ]}|� ||g� q�W q�|� ||g� q�W t � t|��} t || |d�} x8| jD ].} t | t�r�d| kr�| | �|d j�| | < q�W t�|| � d S )Nr r r r r? r~ r@ rS rV rW )rT rU r, rX )r, rU rU r, r r F)rY rZ rb ra )r[ r\ r2 )r rj r� r, r[ r r r� r\ �itemsr� �appendr rX rk r\ ra r( r) ) r] r^ r_ r` ra Zmulti_index_listr| �vr Z multi_indexrb �colr r r �#test_groupby_combined_aggs_cat_cols� s* rh c C s\ t ddddgddddgd��} | �d�}|�dg�}|j�d�|_|�d�}t�||� d S )Nr r )r r r r� rQ )r r r r\ Z droplevelr( r) )r rY r, r- r r r �test_nonagg_agg� s ri c C s� t ddggd ddgd�} | �ddd � dd � g�}t d d d gd d d gd �dd d gd�}t�||� | d �ddd � dd � g�}td d d gdd d gdd�}t�||� d S )Nr rT r r/ r8 )r\ r c S s | � � S )N)r )r r r r r# � r$ z*test_agg_no_suffix_index.<locals>.<lambda>rV � )r/ r8 z<lambda>)r2 c S s | � � S )N)r )r r r r r# � r$ )r2 rb )r r r( r) r r: )r r, r- r r r �test_agg_no_suffix_index� s " rk c C sh t ddgt�ddddddd�t�d dddddd�gd ��} | �d �j�� }| �d �d }t�||� d S ) Nr6 r i� r r� r@ � i`= i� )r/ r8 r/ r8 )r �datetimer r8 r� r� r( r: )r r, r- r r r � test_aggregate_datetime_objects s rn c C s� t dddgdddgdddgd��} | j�d�| _| �dd g�}|j�d d � �}tjd d gdd�}tddg|dd�}t � ||� d S )Nr r� r r r )�c0�c1�p�Oro rp c S s t | dk�S )Nr )r� )r r r r r# r$ z1test_groupby_index_object_dtype.<locals>.<lambda>)r r )r r� )ro rp )rR FTrq )r2 rb ) r r2 r� r rq r r rX r r( r: )r r+ r� r� r- r r r � test_groupby_index_object_dtype s "rs c C sT dd� } t dgt�d�gd�}|�dd� ��| �}t dggdgd�}t�||� d S ) Nc S s | � � �� rd S t�| �S )N)Zisnar� r& r )r; r r r r� % s z)test_timeseries_groupby_agg.<locals>.funcg �?z2018-01-16 00:00:00+00:00)r2 c S s dS )Nr r )r r r r r# + r$ z-test_timeseries_groupby_agg.<locals>.<lambda>r )r r� r� r r r( r) )r� r r� r- r r r �test_timeseries_groupby_agg" s rt c C sD t dddgdd�} t dgdd�}| �dddg��� }t�||� d S )Nz1 dayz3 days�NaTztimedelta64[ns])ra z2 daysr )r r r' r( r: )r[ r- r, r r r � test_group_mean_timedelta_nat1 s rv zinput_data, expected_outputz2021-01-01T00:00ru z2021-01-01T02:00z2021-01-01T01:00z2021-01-01T00:00-0100z2021-01-01T02:00-0100z2021-01-01T01:00-0100c C s< t t| ��}t t|��}|�dddg��� }t�||� d S )Nr )r r r r' r( r: )Z input_dataZexpected_outputr[ r- r, r r r � test_group_mean_datetime64_nat; s rw )c�__doc__rm r r r �numpyr& r3 Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandasr� r r r r r r Zpandas._testingZ_testingr( Zpandas.core.baser Zpandas.core.groupby.grouperr r! r. r7 r> rZ r` rh rl rm r� r� rt rx �intre r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ry Zuint64r� r� r\ rj Zinterval_rangeZ period_rangerX r� r� r� r� r� r� r� r' r r r r r" r' r( r* r+ r3 r8 rN r; r< rS rc rh ri rk rn rs rt rv rw r r r r �<module> s 1$ & ( #$,"," $ 5v ( " y " ,
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка