Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/__pycache__/test_pivot.cpython-37.pyc
Назад
B �5�g�- � @ s� d dl mZm Z mZ d dlmZ d dlZd dlZd dlZ d dlm Z m Z m Z m Z mZmZmZmZ d dlmZ d dlmZ d dlmZ ejddgd �d d � �Zejd gd d gd fe d d�e d d �fgd �dd� �Z G dd� d�Z G dd� d�Z dS )� )�date�datetime� timedelta)�productN)� Categorical� DataFrame�Grouper�Index� MultiIndex�Series�concat� date_range)�CategoricalDtype)� pivot_tableTF)�paramsc C s | j S )N)�param)�request� r �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/test_pivot.py�dropna s r � � � c C s | j \}}ttj�|||��S )N)r r �pd� IntervalIndex� from_arrays)r �closed�left�rightr r r �interval_values s r c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zej�d d d d gfd d d d gfd!d d d gfg�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zej�d*d+d,g�d-d.� �Zej�d/�ej�d*d+d,g�d0d1� ��Zd2d3� Zej�d*d+d,g�d4d5� �Z d6d7� Z ej�d8d9d:ge � d9d:g�e d9d:g�e!d9d:g�g�ej�d*d+d,g�d;d<� ��Z"ej�d8d=d9ge � d=d9g�e d=d9g�e!d=d9g�g�ej�d*d+d,g�d>d?� ��Z#d@dA� Z$ejj%dBdC�ej�d*d+d,g�dDdE� ��Z&ej�d*d+d,g�dFdG� �Z'dHdI� Z(ej�dJdKe j)dLdLdMdMgdNdNdOdOgge!d=dPdQdPgdKdR�fdKdSgdTdUdVdWdXdYdZgd[d\d]dZdZdWgge*j+d^d_d`dadbdcgdKdSgdd�fg�dedf� �Z,dgdh� Z-didj� Z.ej�dkdldmdndog�dpdq� �Z/drds� Z0dtdu� Z1dvdw� Z2dxdy� Z3dzd{� Z4d|d}� Z5d~d� Z6d�d�� Z7d�d�� Z8ej�d�dQd�d�d�d�d�gg�d�d�� �Z9d�d�� Z:d�d�� Z;d�d�� Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�� Z@d�d�� ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZIej�d�dTe jJfd�e j)fd�e jKfdTd�ge jJe j)gfdTd�ge jJe jKgfd�d�ge jKe j)gfg�d�d�� �ZLejjMd�d�� �ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�S )��TestPivotTablec C sv t dddddddddddg dddddddddddg dddddddddddg tj�d�tj�d�tj�d�d��| _d S ) N�foo�bar�one�two�dull�shiny� )�A�B�C�D�E�F)r �np�random�randn�data)�self�methodr r r � setup_method' sJ zTestPivotTable.setup_methodc C s� ddg}d}t | jd|||d�}| jj d|||d�}t�||� t | jd||d� t|�dkrt|jjt|�ks�t�n|jj |d ks�t�t|�dkr�|j j|ks�t�n|j j |d ks�t�| j� ||g �d � t j��� }t�||� d S ) Nr( r) r* r+ )�values�index�columns�observed)r5 r6 r8 r r )r r1 �tm�assert_frame_equal�lenr6 �names�tuple�AssertionError�namer7 �groupby�aggr. �mean�unstack)r2 r8 r6 r7 �table�table2�expectedr r r �test_pivot_tableW s "z TestPivotTable.test_pivot_tablec C s� t td�td�dddddgd��}|jd d d tjd d �}|j�d�|_|j�d�|_|j�d�|_|j �d�|_ |jd d d tjd |d�}t � ||� d S )NZabcdeZfghijr � r r � )�col1�col2�col3rJ rL rK r )r6 r5 r7 �aggfunc� fill_value�category)r6 r5 r7 rM rN r8 ) r �listr r. �sumr6 �astyper7 rJ rK r9 r: )r2 r8 �dfrF �resultr r r �+test_pivot_table_categorical_observed_equals s z:TestPivotTable.test_pivot_table_categorical_observed_equalc C s� t dddgdddgddd gd ��}|jd tjd �}|jd tjd �j}t�||� |jd ddid �}|jd ddid �j}t�||� d S )N�a�b�c�x�y�zr rH r )�rows�colsr5 r] )r7 rM )r6 rM r5 rB )r r r. rQ �Tr9 r: )r2 rS �rsZxpr r r �test_pivot_table_nocols� s z&TestPivotTable.test_pivot_table_nocolsc C s� t ddddd�ddddd�d d d d d�d dddd�ddddd�d��}|jddddgdd�}|jdddgddd�}tjdd d d d d d!d"d#d$d%d&g ddgd'�}t�|j|� t�|j|� d S )(Ni`� i�� iP� i0u )r r rH r r( r) r* i[ i] i\ i^ rV rW rX �di�� i � i@B )�amount�customer�monthr �quantityre rd rc r F)r )r( rV )r( rW )r( rX )r( ra )r) rV )r) rW )r) rX )r) ra )r* rV )r* rW )r* rX )r* ra )r< )r r r � from_tuplesr9 �assert_index_equalr7 r6 )r2 rS Zpv_colZpv_ind�mr r r �test_pivot_table_dropna� s4 z&TestPivotTable.test_pivot_table_dropnac C s� t ddddgdddgdd�}t ddddgdddgdd�}t||d d d d gd ��}t|dddgdd�}tj||gddgd�}tdd d d d gi|d�}t�||� d S )NrV rW r[ T)� categories�orderedrX ra rZ r rH r r )r( r) r5 r5 r( r) )r5 r6 r )r< )r6 )r r r r r r9 r: )r2 Zcat1�cat2rS rT Z exp_indexrF r r r � test_pivot_table_categorical� s z+TestPivotTable.test_pivot_table_categoricalc C s� ddddg}t dddddddddg dddddddddg tdd �d ��}|d �t|d d ��|d < |jdd d|d�}tdddgd d�}|�t|d d ��}tdddgdd�}t dddgdddgdddgg||d�}|s�|jt|�d��d�}t� ||� d S )NrV rW rX ra r rH r r � )r( r) r* r( F)rk r) r* )r6 r7 r5 r )r? � r � rI � )r6 r7 )r7 �float) r �rangerR �CDTr r Zreindexr r9 r: )r2 r rj rS rT �expected_columns�expected_indexrF r r r �$test_pivot_table_dropna_categoricals� s" z3TestPivotTable.test_pivot_table_dropna_categoricalsc C s t ttjddddgddgdd�ddddd gd ��}|jd d |d �}t d ddgittjddgddgdd�d d�d�}t�||� t tdddddgdddgdd�t d�d ��}|jd d |d �}t d dddgittjdddgdddgdd�d d�d�}|�s|d � t �|d < t�||� d S )N�low�highT)rj rk g g �?g @g @g @)r( r) r( r) )r6 r5 r r r )r? )r6 r rI rH r ) r r r. �nanr r Z from_codesr9 r: rs rR rr )r2 r rS rT rF r r r �%test_pivot_with_non_observable_dropna� s< z4TestPivotTable.test_pivot_with_non_observable_dropnac C sX t |dd��}|jdd|d�}t ddit|�� dd�d�}|sH|�t�}t�||� d S )Nr )r( r) r( r) )r6 r5 r )r? )r6 )r r r �uniquerR rr r9 r: )r2 r r rS rT rF r r r � test_pivot_with_interval_index s z-TestPivotTable.test_pivot_with_interval_indexc C s� t j�ddddgddddg�}ttjdddtjd�ddddgt|d d �jd d �d ��}t |ddddd d�}|d }t dddgt t � dd�t � dd�dgdd�dtjd�}t �||� d S )Nr r rH r ���)�dtyperV rW T)rk F)Z ascending)r( r) r* r* r) r( rQ )r6 r7 r5 rM �margins�Allr rp � )r? )r6 r? r )r r r r r. �arangeZintpr Z sort_valuesr r r ZIntervalr9 �assert_series_equal)r2 Z ordered_catrS Z pivot_tabrT rF r r r �&test_pivot_with_interval_index_margins( s z5TestPivotTable.test_pivot_with_interval_index_marginsc C s< | j jd| j j| j jd�}| j jdddd�}t�||� d S )Nr+ )r6 r7 r( r* )r1 r r( r* r9 r: )r2 rT rF r r r �test_pass_arrayB s z TestPivotTable.test_pass_arrayc C sD | j jddd� | j jd�}| j jd| j jd dd�}t�||� d S )Nr+ c S s | d S )NrI r )rY r r r �<lambda>H � z3TestPivotTable.test_pass_function.<locals>.<lambda>)r6 r7 rI r* )r1 r r* r6 r9 r: )r2 rT rF r r r �test_pass_functionG s z!TestPivotTable.test_pass_functionc C sJ ddg}d}t | j||d�}| j�||g ��tj��� }t�||� d S )Nr( r) r* )r6 r7 ) r r1 r@ rA r. rB rC r9 r: )r2 r6 r7 rD rF r r r �test_pivot_table_multipleL s z(TestPivotTable.test_pivot_table_multiplec C s t ddddgddddgddddgd ��}|jd d ks:t�t|d dgd gd tjd�}|j}tt�d �gd tt d�d d�d�}t � ||� t ddddgddddgddddgd ��}|jd dks�t�t|d dgd gd tj d�}|j}tt�d�gd tt d�d d�d�}t � ||� d S )N�cat�batr rH r r rV rW )rV �v�ir� �int64r� r )r5 r6 r7 rN rM �ab)r? )r6 g �?g @g @g @�float64) r �dtypesr> r r. rQ r r r rP r9 r� rB )r2 �fr[ rT rF r r r �test_pivot_dtypesS s( $ $z TestPivotTable.test_pivot_dtypeszcolumns,values�bool1�float1�float2�bool2c C sj t jdt jd�}t||d |dk|dkd��}|�� jd||d�}t|j�}d d � |D �}||ksft�d S ) NrI )r g @rH r )r� r� r� r� r6 )r6 r7 r5 c S s i | ]}t �d �|�qS )r� )r. r )�.0�colr r r � <dictcomp>� s z=TestPivotTable.test_pivot_preserve_dtypes.<locals>.<dictcomp>) r. r� r� r � reset_indexr �dictr� r> )r2 r7 r5 r� rS Zdf_resrT rF r r r �test_pivot_preserve_dtypesw s z)TestPivotTable.test_pivot_preserve_dtypesc C sH t �dddddg�}tdddddd gi|d �}|j|jj|jjd �}t�d d g�}tddgdt j ggddg|d �}t � ||� tddddd gt ddd d�d�|d �}|j|jjtddd�d �}t�dt �d�fg�}d dg|_tddgddg|d �}t � ||� |jtdd�tddd�d �}tdgt jdgdd�|d �}t � ||� d S )Nz 2011-01-01z 2011-02-01z 2011-01-02r( r rH r r rI )r6 )r6 r7 )r( r )r( rH g @g @g @r+ )�freq�periods)r( �dtr� �M)�keyr� z 2011-01-31g @)r� z 2011-12-31)r � DatetimeIndexr r r6 rd �dayr rf r. rz r9 r: r r � Timestampr< )r2 �idxrS �resZ exp_columns�expr r r �test_pivot_no_values� s* z#TestPivotTable.test_pivot_no_valuesc C sN t | jddgdddgdd�}t | jjdgd d �dddgdd �}t�||� d S ) Nr+ r, r( r) r* r )r5 r6 r7 rN r- r )�axis)r6 r7 rN )r r1 Zdropr9 r: )r2 rT rF r r r �test_pivot_multi_values� s z&TestPivotTable.test_pivot_multi_valuesc s� � fdd�}|t jt jg�}|t j�}|t j�}t||gddgdd�}t�||� � fdd�}|t jt jg�}|t j�}|t j�}t||gddgdd�}t�||� d S )Nc s t � jddgddgd| d�S )Nr+ r, r( r) r* )r5 r6 r7 rM )r r1 )�func)r2 r r r� � s z;TestPivotTable.test_pivot_multi_functions.<locals>.<lambda>rB �stdr )�keysr� c s t � jddgddgd| dd�S )Nr+ r, r( r) r* T)r5 r6 r7 rM r� )r r1 )r� )r2 r r r� � s )r. rB r� r r9 r: )r2 r� rT ZmeansZstdsrF r )r2 r �test_pivot_multi_functions� s z)TestPivotTable.test_pivot_multi_functionsr3 TFc C s� t j}tdd|dgddddgdd d d gd ��}|rB|�d dd�}nt�|d dd�}t||d |gd|||g|d ||g|||d ggt|dddgd d�tddddgdd�d�}t�||� t�|�dd d�|j� tt dddd�dt � d� d��}|d t� d� |d< | |j d< |j d< | |j d< |j d < |�rB|�d dd�}nt�|d dd�}|� � j�� t|�k�snt�x:|�� D ].\}}|j |d |d f |d k�sxt��qxW |�r�|�dd d�}nt�|dd d�}t�||j� d S ) NZR1ZR2ZR4�C1�C2ZC3ZC4r� � � � )rV rW rX rV rW rX )r? )r6 r7 z 2014-02-01ro r+ )r� r� �d )rV rX z 2014-02-02)r rV )r rV )r rW )r rW )r. rz r �pivotr r r9 r: r^ r r� r� �locZnotnar5 rQ r; r> Ziterrows) r2 r3 rz rS rT rF �pv�_�rowr r r �test_pivot_index_with_nan� sB *z(TestPivotTable.test_pivot_index_with_nanz1ignore:Timestamp.freq is deprecated:FutureWarningc C s� t tddddd�tddddd�tddddd�tddddd�gtddddd�tddddd�tddddd�tddddd�gtjddd �tjddd �d ��}|d �d d � �|d < |d �dd � �|d< tddddg�}tjddgd ddd�}t� ||g�}t ddddgddddggtjddgd dd�|d�}|�r@|j d dd�}ntj |d dd�}t � ||� t ddgddggtjddgd dd�tjddgddd�d�}|�r�|j d ddd�}ntj |d ddd�}t � ||� d S ) Ni� r rn r rH i� r r� )r )�dt1�dt2�data1�data2r� c S s t j| dd�S )Nz US/Pacific)�tz)r r� )ra r r r r� r� z3TestPivotTable.test_pivot_with_tz.<locals>.<lambda>r� c S s t j| dd�S )Nz Asia/Tokyo)r� )r r� )ra r r r r� r� r� r� z2014/01/01 09:00z2014/01/02 09:00z Asia/Tokyo)r? r� r z2013/01/01 09:00z2013/01/02 09:00z US/Pacific)r6 r7 )r6 r7 r5 ) r r r. r� �applyr r r� r r r� r9 r: )r2 r3 rS �exp_col1�exp_col2�exp_colrF r� r r r �test_pivot_with_tz sH z!TestPivotTable.test_pivot_with_tzc C s t dtjddd�d�dtjddd�d�dtjddd�d�dtjddd�d�dtjd dd�d�g�}|�d ��� }|j�d d � �}t|�d ��� d d g|gtj d�}t tjddd�tjddd�ggt dgd d�tj tjddd�tjddd�gd d�d�}t � ||� d S )NZaaz2016-08-12 13:00:00-0700z US/Pacific)r� )�uid�tsz2016-08-12 08:00:00-0700z2016-08-12 14:00:00-0700z2016-08-25 11:00:00-0700z2016-08-25 13:00:00-0700r� c S s | j ddddd�S )Nr )�hour�minute�second� microsecond)�replace)rY r r r r� ` r� z8TestPivotTable.test_pivot_tz_in_values.<locals>.<lambda>r� )r5 r6 r7 rM )r? z2016-08-12 00:00:00z2016-08-25 00:00:00)r6 r7 )r r r� � set_indexr� r� �mapr r. �minr r� r9 r: )r2 rS ZminsrT rF r r r �test_pivot_tz_in_valuesD s8 z&TestPivotTable.test_pivot_tz_in_valuesc C s� t t�dd�t�dd�t�dd�t�dd�gt�dd�t�dd�t�dd�t�dd�gtjddd �tjddd �d ��}td d d d g�}tjddgd ddd�}t�||g�}t dd dd gddddggtjddgddd�|d�}|r�|j ddd�}ntj |ddd�}t � ||� t dd gddggtjddgddd�tjddgddd�d�}|�r`|j ddd d�}ntj |ddd d�}t � ||� d S )Nz 2013-01-01r+ z 2013-01-02z2013-01r� z2013-02r r� )r )�p1�p2r� r� r� r� rH r� )r? r� r r r r� )r6 r7 )r6 r7 r5 ) r r �Periodr. r� r Z PeriodIndexr r r� r9 r: )r2 r3 rS r� r� r� rF r� r r r �test_pivot_periods{ s< z!TestPivotTable.test_pivot_periodsc C s� t ddddgt�d�t�d�t�d�t�d�gdd��}t dtdddgdd �tt�d�t�d�dgd d �d �}|jdd d d d�}t�||� d S )Nr rH Z2019Q1Z2019Q2g �?)rV rW rY r� rV )r? rW )r1 r6 r7 rY T)r6 r7 r5 r� )r r r� r r r9 r: )r2 rS rF rT r r r � test_pivot_periods_with_margins� s "z.TestPivotTable.test_pivot_periods_with_marginsr5 �baz�zooc C s� t ddddddgddddddgdddd d d gd d ddddgd��}|rX|jdd|d�}ntj|dd|d�}dddd d dgd d d dddgg}tddgdd�}tddgdddggddddddgddddddggd dgd�}t |||dd �}t�||� d S ) Nr# r$ r( r) r* r rH r r rI ro rY rZ r[ �q�w�t)r! r" r� r� r! r" )r6 r7 r5 )r1 r? r� r� r )�levels�codesr< �object)r1 r6 r7 r )r r� r r r r9 r: ) r2 r5 r3 rS rT r1 r6 r7 rF r r r � test_pivot_with_list_like_values� s z/TestPivotTable.test_pivot_with_list_like_valuesr" c C s, t ddddddgddddddgdddd d d gd d ddddgd��}|rX|jdd|d�}ntj|dd|d�}tjdtjd gtjdtjd gtjdtjd gdtjdtjgdtjdtjgdtjdtjgg}tdddd d dgdd�}tddgddggddddgddddggd dgd�}t |||dd �}t�||� d S ) Nr# r$ r( r) r* r rH r r rI ro rY rZ r[ r� r� r� )r! r" r� r� r� r! )r6 r7 r5 )r1 r? r" r� r )r� r� r< r� )r1 r6 r7 r ) r r� r r. rz r r r9 r: ) r2 r5 r3 rS rT r1 r6 r7 rF r r r �%test_pivot_with_list_like_values_nans� s* z4TestPivotTable.test_pivot_with_list_like_values_nansc C sR t dddgdddgdddgd��}d}tjt|d �� |jd d d � W d Q R X d S ) NrV rW rX r rH r )rJ rK rL z0pivot\(\) missing 1 required argument: 'columns')�matchrJ rL )r6 r5 )r �pytest�raises� TypeErrorr� )r2 rS �msgr r r �#test_pivot_columns_none_raise_error s "z2TestPivotTable.test_pivot_columns_none_raise_errorzBMultiIndexed unstack with tuple names fails with KeyError GH#19966)�reasonc C s t ddddddgd�}dd dd gdd dd gdd ddgdd ddgdd ddgdd ddgg}tddgddggddddgddddggd�}t|||dd�}|r�|jdd d d �}ntj|dd d d �}tddgddgd �tddgddgd �tddgddgd �d �}t|�}t�||� d S )!Nr r rH r r rI )r1 r# r( rY r) rZ r* r[ r$ r� r� ro r� r" r� �firstr� )r� r� r� )r1 r6 r7 r )r" r� )r" r� )r� r� )r6 r7 r5 )r6 )r( r) r* )r r r r� r r r9 r: )r2 r3 r6 r1 r7 rS rT rF r r r �test_pivot_with_multiindex s4 z)TestPivotTable.test_pivot_with_multiindexc C s� t ddddddgddddddgdddd d d gd d ddddgd��}tjtdd��. |rh|jdddd� ntj|dddd� W d Q R X d S )Nr# r$ r( r) r* r rH r r rI ro rY rZ r[ r� r� r� )r! r" r� r� z^\('bar', 'baz'\)$)r� r� r! )r" r� )r6 r7 r5 )r r� r� �KeyErrorr� r )r2 r3 rS r r r � test_pivot_with_tuple_of_valuesA s z.TestPivotTable.test_pivot_with_tuple_of_valuesc s\ ddgdgdf� fdd� }� j jdddgddtjd �}||d� � j jdddgddtjd d �}||dd d � � j jddgddtjd �}x"|jjd D ]}||| |� q�W dd� � j jD �� j _� j jddgdtjd�}x2|jD ](}|jd|f }|� j | �� ks�t�q�W � j jddgddd�}x4dD ],}|jd|f }|� j | �� k�s(t��q(W d S )Nr( r) r* r� c s� | j | jd d� |f }� j�|�| �� }tj||dd� |j|ksJt�| � � } | j |df j d d� }� j�|�| �� }tj||dd� |j|dfks�t�| j |df|f } � j| �� } | | ks�t�d S )Nr~ F)� check_names� ) r� r6 r1 r@ rB r9 r� r? r> Z sort_indexZiloc) rT Z values_colr6 r7 � margins_colZ col_marginsZexpected_col_marginsZ index_marginsZexpected_ix_marginsZgrand_total_marginsZexpected_total_margins)r2 r r � _check_outputT s z2TestPivotTable.test_margins.<locals>._check_outputr+ T)r5 r6 r7 r� rM ZTotals)r5 r6 r7 r� rM � margins_name)r� )r6 r7 r� rM r c S s g | ] }|d �qS )rH r )r� �kr r r � <listcomp>� s z/TestPivotTable.test_margins.<locals>.<listcomp>ZAA�BB)r6 r� rM )r� r� rB )ZDDZEEZFF)r1 r r. rB r7 r� r� r> )r2 r� rT rD Z value_colZtotals�itemr )r2 r � test_marginsS s2 zTestPivotTable.test_marginsz*columns, aggfunc, values, expected_columnsr( g @g������@g @g������@r� r! )r? r) rQ rn � � rI ro r' � � � )r" r# )r" r$ )r" r� )r! r# )r! r$ )r! r� )r< c C s� t dddddddddg dddddddddg dddddddddg dddd d d d d d g dd d d d d dddg d��}|j|d|d�}t |tddg�|d�}t�||� d S )Nr! r" r# r$ �small�larger rH r r rI ro rp rq rn )r( r) r* r+ r, T)r7 r� rM r+ r, )r6 r7 )r r r r9 r: )r2 r7 rM r5 ru rS rT rF r r r �%test_margin_with_only_columns_defined� s" z4TestPivotTable.test_margin_with_only_columns_definedc C s� | j �� }t�t|�d ��t|�d��d�|dddg< ttddgdd g��d g }t j |d d �}t d ddddgdddddgd�|d�j ddd�}|d |d |d < |j dd d gdd tjdd!�}t�||� d S )"Nr �i8r+ r, r- r" r! r# r$ )r� r� )r( r) )r< � � rn �- �! r �$ �3 �x )r% r& )r6 r* r )r� r% r& r� r( r) T)r5 r6 r7 r� rM rN )r1 �copyr. r� r; �reshaperR rP r r rf r � rename_axisr rQ r9 r: )r2 rS �mi_val�mirF rT r r r �test_margins_dtype� s . $ z!TestPivotTable.test_margins_dtypec C s� t tddgddg��dg }tj|dd�}tddd dd gd d d d d gd �|d�jddd�}|d |d |d< | jjdddgddtd d�}t � ||� d S )Nr" r! r# r$ )r� r� )r( r) )r< r rH rI r ro )r% r& )r6 r* )r� r% r& r� r+ r( r) T)r5 r6 r7 r� rM rN ) rP r r rf r r r1 r r; r9 r: )r2 r r rF rT r r r �test_margins_dtype_len� s $ z%TestPivotTable.test_margins_dtype_lenr] )r rH )rV rW )r rW )rV r c C sx t |d dddg|d dddgddddgi�}|jd|d �}t dddggtjd d d g|d �tdg�d�}t�||� d S )Nr r rH r r� r rI ro )r5 r7 )r r )rH rH )r r )r< )r7 r6 )r r r rf r r9 r: )r2 r] �df2rT rF r r r � test_pivot_table_multiindex_only� s . z/TestPivotTable.test_pivot_table_multiindex_onlyc C sZ t dddd�}ttj�d�tj�d�|d��}|jddgdd �}|jjd �|�sVt �d S ) Nz 2016-01-01r zEurope/Amsterdam)r� r� )r( r) r* r) r* F)r6 r r ) r r r. r/ r0 r r6 r� �equalsr> )r2 ZdtirS rT r r r �test_pivot_table_retains_tz s z*TestPivotTable.test_pivot_table_retains_tzc s� t j� ttddgdddgddg� fdd �td �D �d g��}t|�}|jd d ddgdgd�}|jtd�}|jddddgdgd�}t j ||dd� d S )Nr! r" r( r) r* �x1Zx2c s g | ]}� t |� �qS r )r )r� r� )ra r r r� s z=TestPivotTable.test_pivot_integer_columns.<locals>.<listcomp>r� g �?r r r r rH )r5 r6 r7 )r7 �4�0�1�3�2F)r� ) r r� rP r rs r r �rename�strr9 r: )r2 r1 rS rD r rE r )ra r �test_pivot_integer_columns s z)TestPivotTable.test_pivot_integer_columnsc C s� t ddddddddgd ddddddddgd dgd dgd d tj�d �d ��}|jd ddd gd �}|�ddd g�d �� }|�d��d �jddd�}t � ||� d S )NrV rW rH r r r! r r" � )rV rW rX �valuer rX )r6 r7 �all)r� �how) r r. r/ r0 r r@ rB rC r r9 r: )r2 r1 rD �groupedrF r r r �test_pivot_no_level_overlap s z*TestPivotTable.test_pivot_no_level_overlapc C s@ d}t �dtfdtfdtfdtfdtfdtfdtfg�}t jd d d d d ddddg dtfdtfgd�}t j||d�}t j�dt |�|�}|d | |d< |d | |d< t t ddd�t ddd��}|t j�dt |�|� }|j |d< |j |d< |j|d< t j�dd|�|d< t|�}|jdddgdddgdd �} | jj�s<t�d S ) Ni' r �Symbol�Year�MonthZDay�QuantityZPrice)�SP500ZADBE)r$ ZNVDA)r$ ZORCL)�NDQ100ZAAPL)r% ZMSFT)r% ZGOOG)�FTSEzDGE.L)r& zTSCO.L)r& zGSK.L)r r i� r i� r� � g @g @rB )r6 r7 rM )r. r r� �intrr �array�emptyr/ �randintr; r r �yearrd r� Z lognormalr r r7 Z is_monotonicr> ) r2 �nr Zproducts�itemsZiproductZdr�datesrS �pivotedr r r � test_pivot_columns_lexsorted/ sJ z+TestPivotTable.test_pivot_columns_lexsortedc C sL dgdgd�}| j �ddg��|��d�}| j jdd|d�}t�||� d S )Nr� rQ )r+ r, r( r) )r6 r7 rM )r1 r@ rA rC r r9 r: )r2 r� rF rT r r r �test_pivot_complex_aggfuncc s z)TestPivotTable.test_pivot_complex_aggfuncc C sF | j ddg jddgtdd�}|�� }t|d d� �|d ksBt�d S )Nr( r) T)r6 rM r� r~ )r1 r r; �tolistrQ r> )r2 rT Z result_listr r r � test_margins_no_values_no_colsj s z-TestPivotTable.test_margins_no_values_no_colsc C sB | j dddg jddgdtdd�}|j�� ddddd gks>t�d S ) Nr( r) r* T)r6 r7 rM r� g @g �?g @g &@)r1 r r; r� r3 r> )r2 rT r r r � test_margins_no_values_two_rowsr s z.TestPivotTable.test_margins_no_values_two_rowsc C s8 | j ddg jddtdd�}|j�� dddgks4t�d S )Nr( r) T)r6 r7 rM r� g @g @g &@)r1 r r; r� r3 r> )r2 rT r r r �&test_margins_no_values_one_row_one_colz s z5TestPivotTable.test_margins_no_values_one_row_one_colc C sh ddddddddd d d g | j d < | j d ddd g jd dgdd gtdd�}|j�� dddddgksdt�d S )NrV rW rX ra �er� �g�hr� �jr� r+ r( r) r* T)r6 r7 rM r� g @g �?g @g &@)r1 r r; r� r3 r> )r2 rT r r r �'test_margins_no_values_two_row_two_cols� s z6TestPivotTable.test_margins_no_values_two_row_two_cols� margin_namer# i� NrV rW c C s� d|� d�}t jt|d��" t| jdddgdgd|d � W d Q R X t jt|d��" t| jddgddgd|d � W d Q R X t jt|d�� t| jddgdgd|d � W d Q R X d S ) NzConflicting name "z3" in margins|margins_name argument must be a string)r� r+ r( r) r* T)r5 r6 r7 r� r� )r� r� � ValueErrorr r1 )r2 r< r� r r r �-test_pivot_table_with_margins_set_margin_name� s2 z<TestPivotTable.test_pivot_table_with_margins_set_margin_namec C s t d�� d�� ddddddddgtddd�tddd�tdd d�tdd d �tdd d�tdd d �tdd d �tdd d �gd ���d �}t tjd ddgdd��dd�tjtdd d�gdd�d�� d�}d |j _ d|j _ t |t dd�ddtjd�}t�||� t |dt dd�dtjd�}t�||j� t t�dtjdddtjg��d d�tjtddd�tddd�gdd�d�� d�}d |j _ d|j _ |�r�|d �d�|d< t |t dd�ddtjd�}t�||� t |dt dd�dtjd�}t�||j� |�� }t |t dd d �ddtjd�}t�||� t |dt dd d �dtjd�}t�||j� d }tjt|d ��" t |t dd d �ddtjd� W d Q R X tjt|d ��" t |dt dd d �dtjd� W d Q R X |�d �}t |t dd d �ddtjd�}t�||� t |dt dd d �dtjd�}t�||j� d!}tjt|d ��" t |t dd d �ddtjd� W d Q R X tjt|d ��" t |dt dd d �dtjd� W d Q R X t d�� d�� ddddddddgtdd"dd#d$�tdddd#d�tdd dd%d$�tdd d d d$�tdd"dd%d$�tdd d d d$�tdd d d d$�tdd dd&d$�gtdd d'd$d$�tdd d(d#d�tdddd%d$�tdd"d d d$�tdd dd%d$�tdddd d$�tdd d)d d$�tdd"d%d&d$�gd*��}t |t d+d d �t d+d,d �dtjd�}t t�tjdtjtjd-tjddtjdtjtjtjtjdtjg��d'd'�tjtddd)�tdd d�tdd"d)�tdd d�gd+d�tjtddd)�tdd d�tdd"d)�tdd d�gd+d�d�}d |j _ d,|j _ t�||� t |t d+d,d �t d+d d �dtjd�}t�||j� tddd)�tdd d�ftdd d�tddd)�ftdd d�tdd"d)�ftdd d�tdd d�ftdd"d)�tdd d�ftdd d�tdd"d)�fg}tj|d d,gd.�}t t�dtjd-tjdtjdtjdtjtjdg ��d-d �|dd/gd�}d0|j _ t |t d+d d �t d+d,d �gd0gdtjd�}t�||� t |d0gt d+d d �t d+d,d �gdtjd�}t�||j� d S )1NzA A A A A A A Bz$Carl Mark Carl Carl Joe Joe Joe Carlr r rI rq rn i� r� rH r� )�Branch�Buyerr# �DaterA r� r� )r r' r( )r� z Carl Joe Mark)r6 r7 r@ r# )r6 r7 r5 rM rp Z6MSZCarl)r� r� z#'The grouper name foo is not found')r� r! )r� �levelzThe level foo is not validr' r� r r� r� r r� � )r? r@ r# rA �PayDayr� rD ro )r< r) r? )r �splitr r� r. r) r r r� r6 r? r7 r r rQ r9 r: r^ rz rR r� r� r� r� r= r rf )r2 Zusing_array_managerrS rF rT r� Ztuplesr� r r r �test_pivot_timegrouper� s� * z%TestPivotTable.test_pivot_timegrouperc C s� ddddddg}ddddddg}t ddddddg||tjdd d �d d gd d��}|d �dd� �|d< |d �dd� �|d< tjdddgddd�}tddg�}tddgdd�}t�||g�}t dd gd dgd dgg||d �}t |dgdgdgd �} t � | |� td d d d d d d d g�}tddd d gd �}tjddgd d!dd�} t�||| g�}t tj dd d d dd d d gd dd d d dd d gd dd d d dd d ggd d �||d �}t |dgdgdd gtj tjgd"�} t � | |� d S )#Nz2011-07-19 07:00:00z2011-07-19 08:00:00z2011-07-19 09:00:00z2013-01-01 15:00:00z2013-02-01 15:00:00rV rW ro r� )r r rH r )�labelr� r� �value1�value2r� c S s t j| dd�S )Nz US/Pacific)r� )r r� )ra r r r r� � r� z7TestPivotTable.test_pivot_datetime_tz.<locals>.<lambda>r� c S s t j| dd�S )Nz Asia/Tokyo)r� )r r� )ra r r r r� � r� z US/Pacific)r� r? rH rG )r? r r rI )r6 r7 )r6 r7 r5 rQ rB rI z Asia/Tokyo)r6 r7 r5 rM )r r. r� r� r r� r r r r r9 r: r) rQ rB ) r2 �dates1�dates2rS �exp_idxr� r� r� rF rT Zexp_col3r r r �test_pivot_datetime_tz� sh z%TestPivotTable.test_pivot_datetime_tzc C s\ ddddddg}ddddddg}t ddddddg||tjdd d �d d gd d��}|d �dd� �|d< |d �dd� �|d< t|d|d jjdd�}tddgdd�}t dd gd dgd dgd�|td d d gdd�d �}t� ||� t||d jj |d jjdd�}t dd gd dgd dgd�td d gdd�td d d gdd�d �}t� ||� t||d jj j |d jj|d jj gdd�}t jd d d d d d gd d gd gddgd �}t tjdd d dd dggd d �d!g|d �}t� ||� t|t�d"d"d"d"d#d#g�|d jj|d jj gdd�}t t�dd d tjd tjgtjtjtjdtjdgg�d"d#g|d �}t� ||� d S )$Nz2011-07-19 07:00:00z2011-07-19 08:00:00z2011-07-19 09:00:00z2013-01-01 15:00:00z2013-02-01 15:00:00rV rW ro r� )r r rH r )rG r� r� rH rI r� c S s t �| �S )N)r r� )ra r r r r� ) r� z6TestPivotTable.test_pivot_dtaccessor.<locals>.<lambda>r� c S s t �| �S )N)r r� )ra r r r r� * r� rG rH )r6 r7 r5 )r? r r rI )rp rq rn rp rq rn )r6 r7 )r< i� �X�Y)r r. r� r� r r� r� r r9 r: rd r, r5 r r r) rz )r2 rJ rK rS rT rL rF r� r r r �test_pivot_dtaccessor sp && ,z$TestPivotTable.test_pivot_dtaccessorc C s� t dddd�}ttj�t|��|d�}tt|�|jj |jj d�}|j � d�|_ t� |jj �}xRtdd �D ]D}|||k }|jj |_|| �� }tj||d d � |j|ksht�qhW d S ) Nz1/1/2000z 12/31/2004r+ )r� )r6 )r6 r7 r r io F)r� )r r r. r/ r0 r; r r r6 r, Z dayofyearr7 � droplevelZasarrayrs r r9 r� r? r> )r2 �rngr� �annualZdoyr� �subsetrT r r r � test_dailya s zTestPivotTable.test_dailyc C s� t dddd�}ttj�t|��|d�}tt|�|jj |jj d�}|j � d�|_ |jj }xRt dd �D ]D}|||k }|jj |_|| �� }tj||d d � |j|ksbt�qbW d S ) Nz1/1/2000z 12/31/2004r� )r� )r6 )r6 r7 r r r� F)r� )r r r. r/ r0 r; r r r6 r, rd r7 rQ rs r r9 r� r? r> )r2 rR r� rS rd r� rT rT r r r � test_monthlyt s zTestPivotTable.test_monthlyc C s� ddd�}t | jdgt|�� �|d�}t | jdg|�� |d�}t�||� dd� |�� D �}t | jdg||d�}t�||� d S )NrQ rB )r+ r, r( )r6 r5 rM c s s | ] }|V qd S )Nr )r� r r r r � <genexpr>� s zGTestPivotTable.test_pivot_table_with_iterator_values.<locals>.<genexpr>)r r1 rP r� r9 r: )r2 ZaggsZpivot_values_listZpivot_values_keysZagg_values_genZpivot_values_genr r r �%test_pivot_table_with_iterator_values� s z4TestPivotTable.test_pivot_table_with_iterator_valuesc C s� d}t ddddgddddgddd d gd ��}|jd d d |tjtgd�}tdd|gdd d�}dddd|fdddd|fg}tj|d d d gd�}t |j||d�}t � ||� d S )NZWeekly�baconZcheeseg @g @g������ @gffffff @r� r^ )r� �costr� r� r� T)r6 r7 r� r� rM r� )r r? )rB rZ r� )rB rZ r^ rB rZ )�maxrZ r� )r[ rZ r^ r[ )r< )r6 r7 ) r r r. rB r[ r r rf r5 r9 r: )r2 r� ZcostsrD ZixZtupsr] rF r r r �/test_pivot_table_margins_name_with_aggfunc_list� s* z>TestPivotTable.test_pivot_table_margins_name_with_aggfunc_listc C s� |r|j �tjjdd�� tt�d�t�d�d t�d�d d��}tddd gd d d gd ddgg�}tdddgdd�|_ tdddgdd�|_ |j ddd|dd�}t � ||� d S )Nz1GH#17035 (np.mean of ints is casted back to ints))r� rq r rH )rY rZ r[ g �?g @g �?rI ro g @r g @r r r� rZ )r? r[ rY T)r r� )�node� add_markerr� �mark�xfailr r. r� r r6 r7 r r9 r: )r2 r8 r rS rF rD r r r �test_categorical_margins� s ( z'TestPivotTable.test_categorical_marginsc C s� |r|j �tjjdd�� tt�d�t�d�d t�d�d d��}tddd gd d d gd ddgg�}tdddgdd�|_ tdddgdd�|_ |j � d�|_ |j � d�|_ |jddd|dd�}t�||� d S )Nz1GH#17035 (np.mean of ints is casted back to ints))r� rq r rH )rY rZ r[ g �?g @g �?rI ro g @r g @r r r� rZ )r? r[ rO rY T)r r� )r] r^ r� r_ r` r r. r� r r6 r7 rZ rR r[ r r9 r: )r2 r8 r rS rF rD r r r �!test_categorical_margins_category� s ( z0TestPivotTable.test_categorical_margins_categoryc C s� t ddddgddddgddddgddd d gd ��}t|d d d �}t dddgdddgdddgd�tdd dgd d�d�}t�||� d S )NrH r ro rq r rI r rN rO )r( r) r* r+ r+ T)r6 r� rp g @g @g @g @)r( r) r* r� )r? )r6 )r r r r9 r: )r2 r8 rS rT rF r r r � test_margins_casted_to_float� s z+TestPivotTable.test_margins_casted_to_floatc C s� t jdddt jg}t jddt jdg}tt|ddg|d�t|ddg|d�tdd�d��}|jd d d |d �}tjddg|d d �}tdt jgt jdgg|d�}ttddgddg|d�d d�|_ t � ||� |jd d |d�}tddgg|td g�d�}t � ||� d S )Nrx ry r( r) )rj rk r ro )�In�Col�Valrd re rf )r6 r7 r5 r8 )rk r? g @g @)r1 r7 )r? )r7 r5 r8 g @)r1 r7 r6 ) r. rz r r rs r r �CategoricalIndexr r6 r9 r: ) r2 r8 rk r� r� rS rT Z expected_colsrF r r r �test_pivot_with_categorical� s$ z*TestPivotTable.test_pivot_with_categoricalc C s� t ddddgddddgdddd gd ��}|d �d �|d < |jd d d|dd�}tjdddgdddgdd d�}tddgdd�}tjddgddgddggtjd�}t |||d�}t � ||� d S )Nr( r) r* rV rW r rH r r )r� r� �Vr� rO ri r� �count)r6 r7 r rM F)rj rk r? )r? r )r )r6 r7 ) r rR r r rg r r. r) r� r9 r: )r2 r8 rS rT rv ru � expected_datarF r r r �test_categorical_aggfunc s &" z'TestPivotTable.test_categorical_aggfuncc C s� t dddgdddgdddgd��}ddd d d d d dddddg }|d �d�j�|�|d< |jddd|dd�}tjddgdd�}tj||ddd�}d dggd d ggd }t |||d �}|r�|jdg }t � ||� d S ) Nr� r �� �Januaryi� i� )�Salesr" r! �FebruaryZMarchZApril�MayZJuneZJulyZAugustZ SeptemberZOctoberZNovemberZDecemberr" rO ro r! rQ )r5 r6 r7 r8 rM )r? F)rj rk r? i@ r r' )r6 r7 ) r rR r� Zset_categoriesr r Z Int64Indexrg r� r9 r: ) r2 r8 rS �monthsrT ru rv rk rF r r r �%test_categorical_pivot_index_ordering$ s@ z4TestPivotTable.test_categorical_pivot_index_orderingc C sB t dddgdddgdddgd ��}|jd d d gtjd �}tjdddgdddggd d gd�}t dddg|d gd�}t�||� |jd d d tjd�}t dtjtjgtjdtjgtjtjdggt dddgd d�t dddgd d�d�}t�||� |jd d tjgd �}t�dgd gg�}t dddgt dddgd d�|d�}t�||� d S )Nr r rI r* r+ r, r rn )rJ rK rL rJ rL rK )r6 rM )r< )r6 r7 )r6 r7 rM )r? rQ ) r r r. rQ r r r9 r: �NaNr )r2 rS rT rh rF r r r �test_pivot_table_not_seriesP s "" & "z*TestPivotTable.test_pivot_table_not_seriesc C sZ d}t ddddgi�}t|dgtd|d�}tddd|gddd �}t |d �}t�||� d S ) Nu Δοκιμήr! r rH r T)r6 rM r� r� r� )r r? )r6 )r r r; r r9 r: )r2 �greek�framerD r6 rF r r r � test_pivot_margins_name_unicodel s z.TestPivotTable.test_pivot_margins_name_unicodec C sB t dddddddddddg dddddddddddg td�d��}t|ddd d �}td gddggd d gd d ggd dgd�}t ddd�ddd�d�|d��d�}t�||� t|ddd dgd �}td dgd gddggd d d d gd d d d gd d d d ggd d dgd�}t ddd�ddd�ddd�ddd�d�|d��d�}t�||� d S ) Nr! r" r# r$ r' )r( r) r* r( r) rQ )r6 r7 rM r* r r )r� r� r< r� r� )r" r! rp r� ))r* r# )r* r$ )r7 rB g @g @g @g������@))rB r* r# )rB r* r$ )rQ r* r# )rQ r* r$ )r rs r r r r9 r: )r2 r1 rT r rF r r r �test_pivot_string_as_funcw sZ & z(TestPivotTable.test_pivot_string_as_funcz f, f_numpyrB r� c C s4 t | jdd|d�}t | jdd|d�}t�||� d S )Nr( r) )r6 r7 rM )r r1 r9 r: )r2 r� Zf_numpyrT rF r r r � test_pivot_string_func_vs_func� s z-TestPivotTable.test_pivot_string_func_vs_funcc C sP t t�d�t�d�dd��}d}tjt|d�� |jddddd � W d Q R X d S ) Ni r )�ind1�ind2rj z6Unstacked DataFrame is too big, causing int32 overflow)r� r{ r| rj )r6 r7 r5 rM )r r. r� r� r� r= r )r2 rS r� r r r �-test_pivot_number_of_levels_larger_than_int32� s z<TestPivotTable.test_pivot_number_of_levels_larger_than_int32c C s� t dddgdddgdddgd��}dd � }d d � }d d � }t|d|||g|d�}ddtjtjddgddtjtjddgg}tjdddgddggd dgd�}t |ddg|d�} |r�| jdd�} t�|| � d S )N�apple�peachr rH rp ro )�fruit�size�tastec S s dS )Nr r )rY r r r �ret_one� s z?TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_onec S s t | �S )N)rQ )rY r r r �ret_sum� s z?TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_sumc S s t jS )N)r. rz )rY r r r �ret_none� s z@TestPivotTable.test_pivot_table_aggfunc_dropna.<locals>.ret_noner� )r7 rM r r r� r� r� r� )r< r� r� )r6 r7 r7 )r� ) r r r. rz r Z from_productr r9 r: ) r2 r rS r� r� r� rT r1 r� rF r r r � test_pivot_table_aggfunc_dropna� s" ( z.TestPivotTable.test_pivot_table_aggfunc_dropnac C s� t dddgdtjdgdtjtjgd��}t|dtj|d�}d tjgdtjgg}tddgdd �}t |d d g|d �}|r�|jdd�}t�||� d S )Nr# r$ r rH r )r( rY rZ r( )r7 rM r g @)r? rY rZ )r6 r7 r7 )r� ) r r. rz r rB r r r9 r: )r2 r rS rT r1 r� rF r r r �&test_pivot_table_aggfunc_scalar_dropna� s & z5TestPivotTable.test_pivot_table_aggfunc_scalar_dropnac C sz t dddddgdddddgdd d dd gd d d d d gd ��}|jd d d tjd�}t tg dd d�d�}d |j_t�||� d S )NrH r rI ro rp rq rn �pr� )r( �idr* r+ r( r+ r� )r6 r7 r5 rM r� )r r? )r6 ) r r r. r� r r7 r? r9 r: )r2 rS rT rF r r r � test_pivot_table_empty_aggfunc s z-TestPivotTable.test_pivot_table_empty_aggfuncc C s` dd� }t ddddgddddgddddgd ��}tjtd d �� |jd d d |d� W d Q R X d S )Nc S s t �| �S )N)r. rB )Zarrr r r rA s z=TestPivotTable.test_pivot_table_no_column_raises.<locals>.aggr r r� r� rC �( )rN rO �ZZ notpresent)r� rN rO )rM )r r� r� r� r )r2 rA r! r r r �!test_pivot_table_no_column_raises s (z0TestPivotTable.test_pivot_table_no_column_raisesc C s t dddddddddg dddddddddg dddddddddg dddd d d d d d g dd d d d d dddg d��}t|ddgddgtjtttjgd�d�}t�ddddg�}tjdd d d gddgd �}t�d d!d"d#gd d!d$d%gd&d'd(d)gd*d#d+d&gg�}t |||d,�}t � ||� d S )-Nr! r" r# r$ r� r� r rH r r rI ro rp rq rn )r( r) r* r+ r, r+ r, r( r* )r+ r, )r5 r6 rM )r+ rB )r, r[ )r, rB )r, r� )r" r� )r" r� )r! r� )r! r� )r< g @g "@g @g @g !@g @g @g @g @g @g["8���@g. UUU@)r7 r6 ) r r r. rB r� r[ r rf r) r9 r: )r2 rS rD r] r6 �valsrF r r r � test_pivot_table_doctest_case s> z,TestPivotTable.test_pivot_table_doctest_casec C s� t dddgdddgddd gd d d gd ��}|jdd gddddd�}t dtjgdtjgtjd ggtd d gdd�tjdddgdddggdd gd�d�}t�||� d S )NZd1Zd4Zd3rV rW rX � r �" Z2018Z2019)rV r� �numr, r� r, r� rQ F)r6 r7 r5 rM �sort)r? )r< )r7 r6 ) r r r. rz r r r r9 r: )r2 rS rT rF r r r �test_pivot_table_sort_falseQ s "z*TestPivotTable.test_pivot_table_sort_false)U�__name__� __module__� __qualname__r4 rG rU r` ri rm rw r{ r} r� r� r� r� r� r� r_ Z parametrizer� r� r� r� r� �filterwarningsr� r� r� r� r. r) r r r� r� r� r` r� r� r� rB r rf r� r r r r r r r1 r2 r4 r5 r6 r; r> rF rM rP rU rV rX r\ ra rb rc rh rl rs ru rx ry rQ r� rz Zslowr} r� r� r� r� r� r� r r r r r & s� 0 % 5 $ ! 6 <7. " '< " 4 $& GS #, > # 1r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestPivotc C s� ddddddgddddddgddddddgd �}t |�}|jd d d d �}t dddd �dddd �d��}d\|j_|j_t�||� |jjd ks�t�|jjd ks�t�|jd d d�}|jjd ks�t�|jjdks�t�d S )Nr( r) r* �One�Twog �?g @g @)r6 r7 r5 r6 r7 r5 )r( r) r* )r� r� )r6 r7 )Nr7 ) r r� r6 r? r7 r9 r: r> r< )r2 r1 rw r0 rF r r r � test_pivoti s zTestPivot.test_pivotc C sZ t dddddgdddddgdddddgd ��}tjtd d �� |�d d d� W d Q R X d S )Nr" r! r# r$ g �?g @g @g @)rV rW rX zduplicate entries)r� rV rW rX )r r� r� r= r� )r2 r1 r r r �test_pivot_duplicates� s z TestPivot.test_pivot_duplicatesc C s8 t dddgd�}|�ddd�}t � }tj||dd� d S )NrV rW rX )r7 F)r� )r r� r9 r: )r2 rS rT rF r r r �test_pivot_empty� s zTestPivot.test_pivot_emptyc C sD t ddgd�}|jdddd�}t|� t�|jtdd gdd �� d S ) N)r( r �A1)r) r ZB2)r1 r r rH )r6 r7 r5 r( r) )r? )r r� �reprr9 rg r7 r )r2 rS rT r r r �test_pivot_integer_bug� s z TestPivot.test_pivot_integer_bugc C s ddddddgddddddgddddddgd �}t |��d �}|jd d d �}t dddd�dddd�d��}d\|j_|j_t�||� |jd d�}tj ddgd d gd�|_d |j_tj||dd� |jjd ks�t �|jj dks�t �|j� d�|_|jd d d �}d |j_t�||� d S )Nr( r) r* r� r� g �?g @g @)r6 r7 r5 r6 r7 r5 )r7 r5 )r( r) r* )r� r� )r6 r7 )r7 )r5 r� )r5 r� )r< F)r� )Nr7 r ) r r� r� r6 r? r7 r9 r: r rf r> r< rQ )r2 r1 rw rT rF r r r �test_pivot_index_none� s* z TestPivot.test_pivot_index_nonec C s t ddddddgddddddgddddddgddddddgddddddgd��}d d g}d g}|j||d d �}t t�d ddd gdtjdtjgddddgtjdtjdgg�tjddgd d gd�tjddgd d gd�d�}t�||� |d d gks�t �|d gk�st �d S )Nr rH r r rI ro r )�lev1�lev2�lev3�lev4r5 r� r� r� )r6 r7 r5 g �?g @g g @g @g @g @)r r rH rH )r rH r rH )r< )r� r� r5 r5 )r6 r7 ) r r� r. r) rz r r r9 r: r> )r2 rS r6 r7 rT rF r r r �0test_pivot_index_list_values_none_immutable_args� s. z:TestPivot.test_pivot_index_list_values_none_immutable_argsN) r� r� r� r� r� r� r� r� r� r r r r r� h s $r� ) r r r � itertoolsr �numpyr. r� Zpandasr r r r r r r r r Zpandas._testingZ_testingr9 Zpandas.api.typesr rt Zpandas.core.reshape.pivotr Zfixturer rs r r r� r r r r �<module> s8 ( 8 R
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка