Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_stack_unstack.cpython-37.pyc
Назад
B �5�g � @ s d dl m Z d dlmZ d dlZd dlZd dlZd dlm m Z d dl Z d dl m Z mZmZmZmZmZmZ d dlmZ G dd� d�Zdd� Zd d � Zd d � Zej� d ddg�dd� �Z ej� d ddg�ej� ddd g�dd� ��Z dd� Z dd� Z dd� Z!G dd � d �Z"dS ) � )�datetime)�StringION)� DataFrame�Index� MultiIndex�Period�Series� Timedelta� date_rangec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e j�ddddddgddddgddddgddddggejdddd gd d gd �ejd d!d"d#gd$dgd �fd%ddddddddgddddddddgged&d'gd d(�ejd)d*d+d,d-d.d/d0gd$dd gd �fg�d1d2� �Zd3d4� Zd5d6� Zd7d8� Zd9d:� Zd;d<� Zd=d>� Zd?d@� ZdAdB� ZdCdD� Z dEdF� Z e j�dGdHd ge d$�g�dIdJ� �Z dKdL� Z dMdN� Z!dOdP� Z"dQdR� Z#dSdT� Z$dUdV� Z%dWdX� Z&e'j(dYdZ� �Z)d[d\� Z*d]d^� Z+e j�d_d`dag�e j�dbe,dc�e,dd�g�dedf� ��Z-e j�d_d`dag�e j�dge,dh�didjdkdldmdngfe,do�dmdndkdldidjgfg�dpdq� ��Z.drds� Z/e j�dtdududdge�0ddgdvdwgg�fdududd&ge�0ddgdvdwgg�fduddd&ge�0ddgdvdwgg�fg�dxdy� �Z1e j�dzdudg�d{d|� �Z2e j�dzdud}g�d~d� �Z3d$S )��TestDataFrameReshapec C s� |� � }t�t�|j���|j�|d d �< |�� }t||d��}|�� }|�� }t � ||� t � |d |� |�d�}|�d�}t � |j |� t � |d j |� d S )N)�foo�barr r ) �copy�np�arange�prod�shape�reshape�stackr �unstack�tm�assert_frame_equal�T) �self� float_frame�df�stackedZ stacked_df� unstackedZ unstacked_dfZunstacked_colsZunstacked_cols_df� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_stack_unstack.py�test_stack_unstack s " z'TestDataFrameReshape.test_stack_unstackc C s� t d�dddgddgg}td|d |d d�}|�� }tdt�|d d� �d�}t�||� td|d t�|dd � �d�}|�d�}tdt�|d |d g�|d d�}t�||� |ddg �d�}|ddg }t�||� d S ) N� �a�b� � r )�index�columns)r& ) �ranger r r r � from_productr �assert_series_equalr )r �levelsr �result�expectedr r r �test_stack_mixed_level* s " z+TestDataFrameReshape.test_stack_mixed_levelc C sl t ddtjgddtjgd��}|dg }|d |d< |sLt|jj�dksLt�|�� }|�� }t� ||� d S )Nr$ r% g @� )�x�yr0 r1 ) r r �NaN�len�_mgr�blocks�AssertionErrorr r r* )r �using_array_managerr �df2�resr- r r r � test_unstack_not_consolidatedA s z2TestDataFrameReshape.test_unstack_not_consolidatedc C s� t ddddgtjd�}t�dddd g�|_|jd d �}tdd dgddd gd �d ddgtjd�}t� ||� |jdd �}tdddgdddgd �d ddgt d�}t� ||� tdddgdddgdddgdddgd��� d ddg�}|jd dgdd �}d}|| }t dddg|j|d�}t� ||� |� d dg�}|j�|jj�|_|�tj�}|j|j }t� ||� |d }|jd dgdd �}|d }t� ||� d S )Nr$ r% r/ � )�dtype)r0 r" )r0 r# )r1 r# )�zr" ���)� fill_value)r" r# r0 r1 r= )r& r< g �?r" r# �j�kr )r0 r1 r= �w)rB r# r@ )r& �namerB )r r �int16r � from_tuplesr& r r r r �float� set_indexr* r Zreorder_levels�names�astype�int64�loc) r �datar, r- r r �keyr �sr r r �test_unstack_fillM s8 & $ ( z&TestDataFrameReshape.test_unstack_fillc C sr ddgddgddgddgg}t |td �tjd �}t�d d d dg�|_|jdd�}ddddgddddgddddgg}t |td�tjd�}t�ddddg�|_t � ||� |d � tj �|d< |d � tj �|d< |jdd�}|d � tj �|d< |d � tj �|d< t � ||� |jdd�}ddddgddddgddddgg}t |td�td�}t�ddddg�|_t � ||� d S )Nr$ r% r! r/ r; � � � ZAB)r' r< )r0 r" )r0 r# )r1 r# )r= r" r> )r? �xyz)r& r< )�Ar" )rT r# )�Br" )rU r# rT rU g �?)r �listr �int32r rE r& r r' r r rI rD �float64rF )r �rowsr r, r- r r r �test_unstack_fill_frame| s, " "z,TestDataFrameReshape.test_unstack_fill_framec C s� t ddd�j}t|�}t�ddddg�|_|�� }t|d tj |d g|d |d tj gd �d ddgd�}t � ||� |j|d d�}t|d |d |d g|d |d |d gd �d ddgd�}t � ||� d S )Nz 2012-01-01r/ )�periods)r0 r" )r0 r# )r1 r# )r= r" r r! r$ r% )r" r# r0 r1 r= )r& )r? ) r �valuesr r rE r& r r �pd�NaTr r )r ZdvrL r, r- r r r � test_unstack_fill_frame_datetime� s ( ,z5TestDataFrameReshape.test_unstack_fill_frame_datetimec C s� dd� t d�D �}t|�}t�ddddg�|_|�� }t|d tj|d g|d |d tjgd �d ddgd�}t � ||� |j|d d�}t|d |d |d g|d |d |d gd �d ddgd�}t � ||� d S )Nc S s g | ]}t |d ��qS ))�days)r )�.0�ir r r � <listcomp>� s zJTestDataFrameReshape.test_unstack_fill_frame_timedelta.<locals>.<listcomp>r/ )r0 r" )r0 r# )r1 r# )r= r" r r! r$ r% )r" r# r0 r1 r= )r& )r? ) r( r r rE r& r r r] r^ r r )r �tdrL r, r- r r r �!test_unstack_fill_frame_timedelta� s ( ,z6TestDataFrameReshape.test_unstack_fill_frame_timedeltac C s� t d�t d�t d�t d�g}t|�}t�ddddg�|_|�� }t|d d |d g|d |d d gd �dddgd�}t�||� |j|d d�}t|d |d |d g|d |d |d gd �dddgd�}t�||� d S )Nz2012-01z2012-02z2012-03z2012-04)r0 r" )r0 r# )r1 r# )r= r" r r! r$ r% )r" r# r0 r1 r= )r& )r? ) r r r rE r& r r r r )r r[ rL r, r- r r r � test_unstack_fill_frame_period� s$ $ z3TestDataFrameReshape.test_unstack_fill_frame_periodc C s� t ddddgdd�}t�dddd g�|_|�� }ttjtd �td �d �tjtd �td �d �d�td�d�}t � ||� d}t j t |d�� |jdd� W d Q R X |jdd�}ttjtd�td �d �tjtd�td �d �d�td�d�}t � ||� d S )Nr" r# �c�category)r< )r0 r" )r0 r# )r1 r# )r= r" Zaxa�abc)� categoriesZbcx)r" r# rS )r& z 'fill_value=d' is not present in)�match�d)r? ZacaZbcc)r r rE r& r r r] � CategoricalrV r r �pytest�raises� TypeError)r rL r, r- �msgr r r �#test_unstack_fill_frame_categorical� s$ z8TestDataFrameReshape.test_unstack_fill_frame_categoricalc C s� t jdddgdddggddgd �}tdgd dgd d �|d �}|�d�}tddddddgddddddgddddddggt jd dddddgd dgd �tdddgdd�d�}t�||� d S )Nr" r# rg r$ r% r! )rT r" )rU r# )rH � )rl �e)r& )rl r" )rl r# )rl rg )rt r" )rt r# )rt rg )rC )r' r& )r r) r r rE r r r )r �idxr r, r- r r r �$test_unstack_tuplename_in_multiindex s , z9TestDataFrameReshape.test_unstack_tuplename_in_multiindexz>unstack_idx, expected_values, expected_index, expected_columns)rT r" r$ r% )r$ r! )r$ r/ )r% r! )r% r/ rU �C)rH )rl r" )rl r# )rt r" )rt r# N))rT r" rU r! r/ )rC )rl r" r$ )rl r" r% )rl r# r$ )rl r# r% )rt r" r$ )rt r" r% )rt r# r$ )rt r# r% c C sl t jddgddgddggddd gd �}tdgd dgd d �|d �}|�|�}t|||d�}t�||� d S )Nr" r# r$ r% r! r/ )rT r" rU rw )rH rR )rl rt )r& )r' r& )r r) r r r r ) r Z unstack_idxZexpected_valuesZexpected_indexZexpected_columnsru r r, r- r r r �*test_unstack_mixed_type_name_in_multiindex* s &" z?TestDataFrameReshape.test_unstack_mixed_type_name_in_multiindexc C s t dddgdddgtdddg��d�tj�d�d d t�d �td d dg��d�tdddg��d�dtd ddg��d�d� �}dd� }|�dd g�}||d � |�dd g�}||d � |�d d g�}||d � |�d d!dg�}||d!� |�d"d#g�}||d"� |�dd g�}|d# }||d � d S )$NZILZMIZNCr" r# rg rh r! r$ r Z20010102g �?g I@g Y@Zfloat32g @g @g @rX F�� i2 Zint8) �stater& �some_categoriesrT rU rw �D�E�F�G�Hc S s& | � |g�}| � |�}t�||� d S )N)r r r )r Z column_nameZ unstacked1Z unstacked2r r r �unstack_and_comparem s zNTestDataFrameReshape.test_unstack_preserve_dtypes.<locals>.unstack_and_comparerz r& r{ r~ rw r rU r} rT ) r r rI r �randomZrandr] � TimestamprG )r r r� �df1rN r r r � test_unstack_preserve_dtypes[ s4 z1TestDataFrameReshape.test_unstack_preserve_dtypesc C s� t �ttjtd�dd���}ttj� dd�|d�}t � |j ddgd�|j dd�j dd�� t � |j d d gd�|j dd�j dd�� |� � }|jjtd�d d �}|d ks�t�t � |j ddgd�|j dd�j dd�� d S ) Nr! )�repeat� � )r' r$ r% )�level���r> T)Zinplace)r rE rV � itertools�productr( r r r� �randnr r r r r' Z set_namesr6 )r r' r Zdf_namedZ return_valuer r r �test_stack_ints� s ($ z$TestDataFrameReshape.test_stack_intsc C s t jddddgdddgd�}ttj�d d �|d �}|jddgd �}|jddgd �}|�� }ddd g|j_ t j |jdd gd �|d d� t j |jdd gd �|d d� d}t j t|d�� |jddgd � W d Q R X |�� }dddg|j_ t j |jddgd �|d d� d S )N)rT �cat�long)rU r� r� )rT �dog�short)rU r� r� �exp�animal� hair_length)rH r/ )r' )r� r$ F)� check_nameszSlevel should contain all level names or all level numbers, not a mixture of the two)rk r )r rE r r r� r� r r r' rH r r rn ro � ValueError)r r' r �animal_hair_stacked�exp_hair_stackedr8 rq �df3r r r �test_stack_mixed_levels� s, z,TestDataFrameReshape.test_stack_mixed_levelsc C s: t jddddgdddgd�}ttj�d d �|d �}|jddgd �}|jddgd �}|jddgd �}|�� }d d dg|j_ t j |jd dgd �|dd� t j |jd d gd �|dd� t j |jd dgd �|dd� |�� }dd d g|j_ t j |jd d gd �|dd� t j |jdd gd �|dd� t j |jdd gd �|dd� d S )N)rT r� r� )rU r� r� )rT r� r� )rU r� r� r� r� r� )rH r/ )r' )r� r r$ r% F)r� ) r rE r r r� r� r r r' rH r r )r r' r Zexp_animal_stackedr� r� r8 r� r r r �test_stack_int_level_names� s4 z/TestDataFrameReshape.test_stack_int_level_namesc C s| t ddgt�ddgddgg�dgd�}|�� }t tjdtjgtjdggtd�ddgt�ddgddgg�d�}t� ||� d S ) NFr" r# rg �l�col)r& r' )r< ) r r � from_arraysr r �array�nan�objectr r )r r �rs�xpr r r �test_unstack_bool� s z&TestDataFrameReshape.test_unstack_boolc C s� t ddgddgddggddddgddddgddddggd d d gd �}td|d �}|�ddg��d�}t ddgddggddddgddddggd d gd �}ttjtjdgdtjgtjdgdtjggtjd�|t ddgd d�d�}t � ||� d S )Nr r �one�twor" r# r r$ �first�second�third)r+ �codesrH )r& r% )r< )rC )r& r' ) r r r r r r r� r� rX r r r )r �mirN r, Z expected_mir- r r r �test_unstack_level_binding� s *z/TestDataFrameReshape.test_unstack_level_bindingc C s� |� � }t|t�st�|� � j}t�||� tddtj gddtj gd��}t dddg�|_ |� � }t d d gdddggd d d dddgd ddd ddggd �}tddtj d dtj g|d�}t� ||� |�� }xtd�D ] }|� � }q�W t�||� d S )Nr$ r% g @r/ )r0 r1 r" r# rg r0 r1 r )r+ r� r! )r& )r � isinstancer r6 r r r r r r2 r r& r r* r r( ) r r rL Zundor, �midxr- Zold_data�_r r r �test_unstack_to_series s $ z+TestDataFrameReshape.test_unstack_to_seriesc C s� ddddgddddgddddgddddgg}t |td�d�}|j}tt�d�gd td�d�}t�||� |�d d g�}|� d �}|j}tt�d�gd t j d d d d gddddggd d�d�}t�||� |�d d g�}d|d < |� d �}|j}tt�d�gd t�d�gd t j d d d d gddddggd d�d�}t�||� d|d < |� d �}|j}tt�d�gd t�d�gd t j d d d d gddddggd d�d�}t�||� x�t� d�t� d�ftj ddd�tj dddd�ffD ]�\}}t dgd ||tddd�d��}|jd d� jdd �} |�d d g�}|d �d�|d < |jd d� � d �} | �d d g�� d �} | d �d�| d < | jd k�s�t�t�| | � �q�W d S ) Nr$ r! r/ r% ZABCD)r' rJ )r& rT rU rw r| )NrU )rH g @rX r r� r; Zf8)r< � r" z 2012-01-01)r[ )rT rw r| rU T)�deepr )r| r" )r! r% )r rV �dtypesr r r< r r* rG r r r� �zerosr r �ilocr rI r r6 r ) r rY r r, r- r8 r� rg rl �right�leftr r r �test_unstack_dtypes. s\ , " " " (z(TestDataFrameReshape.test_unstack_dtypesc C sx t jddgddgd�}tddg|d�}d}tjt|d �� |�d� W d Q R X tjt|d �� |j�d� W d Q R X d S ) N)r" r# )rg rl �c1)rH r$ r% )r& z5The name c1 occurs multiple times, use a level number)rk ) r rE r rn ro r� r r r )r ru r rq r r r �#test_unstack_non_unique_index_namesv s z8TestDataFrameReshape.test_unstack_non_unique_index_namesc C s~ t �dgddddgg�d d� }tddggd |d �}|�� }t �ddgdddgg�}tddddddggdg|d �}t�||� |jjd |jd k�� s�t �ddd gddd d gg}ddddgdd dd gg}t ||�}t � d�� d d �}tt � ||d g�|d �}|�� }tt j |d |d d gdd�|d�}t�||� |jjd |jd k�� �s\t �dd dgdd dd gg}ddddgdd dd gg}t ||�}t � d�} t| � dd �|d �}dddddd dddgt jdd gt jddgfdddddddddgt jddgt jdd gff} xn| D ]f\} } } }|j| d�}t �d �t j }| || < t �ddg| g�}t|� d d�||d �}t�||� �qW d S ) Nr" rT rU rw r| r> r$ r r! )r& )r& r' rQ r% r/ )�axis)r' rg r; rR � � rP rs � � � )r� � )r r) r r r r r' r+ �allr6 r r r Z concatenater� r� )r ru r r, Zexp_colr- r+ r� �blockrL Zcasesr� ZidcesZ col_levelZ idx_levelZexp_data�colsr r r �test_unstack_unused_levels� s> $ *. z/TestDataFrameReshape.test_unstack_unused_levelsr� rT c C s� t dddgdddggddd gd �}|jddd gd d �}|jtd �td �df|f }|�� }|jd g | }tj|jdggd d gd�|_|j � d �|_ t � ||� d S )Ni� r" �Ii� r# ZIIrT rU rw )r' F)Zdropr )rH ) r rG rK �slicer r� r r) r' r& Z droplevelr r )r r� r �indZ selectionr, r- r r r �test_unstack_unused_level� s " z.TestDataFrameReshape.test_unstack_unused_levelc C s� t dggtjdggdgd�tjdddddddggddddd d d gd�d �}|�dddd d d g�}t dggtjdddddddggddddd d d gd�tdgdd �d �}t�||� d S )Nr$ r r� )rH �i1�i2�i3�i4Zi5�i6�i7)r' r& )rC )r r rE r r r r )r r r, r- r r r �test_unstack_long_index� s z,TestDataFrameReshape.test_unstack_long_indexc C s~ t ddgddggtjddgddggddgd�tjdd d gdd d ggd d dgd�d�}|�d d g�jjdd � d d gkszt�d S )Ng rU rw r| r� �c2)rH r� � r� �( r� r� r� )r' r& r� )r r rE r r' rH r6 )r r r r r � test_unstack_multi_level_cols� s $z2TestDataFrameReshape.test_unstack_multi_level_colsc C s� t ddgddgddgddggt�d d d gd d dgg�tjdddgdddgdddgdddggdddgd�d�}|�ddg�}|�dg��dg�}t�||� d S )Nr$ r% r! r/ r> r� ������r" r# rg rl rt �f�m1ZP3�� ZA5�o �m2r� r� r� )rH )r' r& )r r rE r r r )r r r, r- r r r �&test_unstack_multi_level_rows_and_cols� s z;TestDataFrameReshape.test_unstack_multi_level_rows_and_colsc sb dd� � � fdd�}t ddtjdgdd d d gd d ddgd��}|�ddg��� d }|�ddg��� d j}t�||� xnt� |j d d� �D ]V}|�t |��}xBt d�D ]6}|j|d�}|� � j�� t|�ks�t�||d � q�W q�W t dgd tjgd dgd dgd dgd dgd d gd dgd tjgd d gd tjgd d gd tjgd dd d d d d d!d"d#d$d%d&d'd%d(d)d*d d+d,gd-��}|j� fd.d/�d0d1�|j� fd2d/�d0d1� |d3< |d4< x�t� d5d6d7g�D ]p}|�t |��}xZt d�D ]N}|j|d�}|� � j�� dt|� k�s6t�xd8D ]} ||| � �q<W �qW �q�W d S )9Nc S s | | kr dn| }|d�S )N� �1r )�valZval_strr r r �cast� s z:TestDataFrameReshape.test_unstack_nan_index1.<locals>.castc s� dd� }| � � j�� \}}xdt||�D ]V\}}t| j||f �d��}|| j| �|| j| � }tt � |��}||ks&t �q&W d S )Nc S s t | t�rt| �S | gS )N)r� �tuplerV )r" r r r �<lambda>� � zNTestDataFrameReshape.test_unstack_nan_index1.<locals>.verify.<locals>.<lambda>�.) �notnar\ Znonzero�zip�sortedr� �splitr& r' �mapr6 )r Zmk_listrY r� rb r@ r� r� )r� r r �verify� s z<TestDataFrameReshape.test_unstack_nan_index1.<locals>.verifyr" r# rl rB r0 r1 r= za.wzb.xz .yzd.z)�jim�joe�jolier� r� r� r% )r� r! r; rg rt r/ �C �' �5 �H �9 �P � r� r� r� �; �2 �> �L �4 � �< �3 )�1st�2nd�3rdc s d� t� | ��S )Nr� )�joinr� )�r)r� r r r� < r� z>TestDataFrameReshape.test_unstack_nan_index1.<locals>.<lambda>r$ )r� c s d� t� | jd d d� ��S )Nr� r> )r r� r� )r )r� r r r� = r� �4th�5thr� r� r )r r )r r r� rG r r r r r� � permutationsr' rV r( r� r\ �sumr3 r6 �apply) r r� r r� r� ru r� ZlevZudfr� r )r� r �test_unstack_nan_index1� s^ <J " z,TestDataFrameReshape.test_unstack_nan_index1c C sr t td�td�td�d��}tj|jd< |�ddg��d�}ddd d tjtjtjtjgtjtjtjtjd d d dgg}tt tt |� ��}t tjdd d d d d dgdd�}t dgddggddgdd ggd dgd�}t |||d�}t �||� t td�ttd ��d td�d��}tj|jd< |�ddg��d�}d tjgdd gd d gtjd gddgg}t dgddggddgdd ggd dgd�}t tjdd d dgdd�}t |||d�}t �||� t td�ttd ��d td�d��}tj|jd< |�ddg��d�}dtjgdd gd d gd d gtjdgg}t dgddggddgdd ggd dgd�}t tjdd d dgdd�}t |||d�}t �||� d S )NZaaaabbbbrR )rT rU rw )r! r$ rT rU r r! r$ r% r/ r; rP rQ )rC rw r" r# )r+ r� rH )r' r& )r% r$ )r rV r( r r2 r� rG r r� r� r� r r r r )r r r� �valsru r� r� r r r �test_unstack_nan_index2H s: & $ && $ &&z,TestDataFrameReshape.test_unstack_nan_index2c C s t td�tddd��� d t�d�d��}tj|jd< |�d d g�� � }t� d d d dtj dgtj dddddgg�}t ddgd d�}t dgtddd�gd d d d d d gdd d dd dggd d gd�}t |||d�}|r�|jd dd dg }|| �|d j�||< t�||� d S )NZ aaaaabbbbbz 2012-01-01r; )r[ r% r� )rT rU rw )r! r$ rT rU r! r r$ r/ rP rQ rR rs r" r# )rC rw r> )r+ r� rH )r' r& )r rV r �tolistr r r2 r� rG r r� r� r r r' rI r< r r )r r7 r r� r ru r� r� r r r �test_unstack_nan_index3s s" * z,TestDataFrameReshape.test_unstack_nan_index3c C sV dt jt jdgddt jdgddt jdgddd d gd dd d gdd d d gg}t|d dddgddddddgd�}|�� �ddd g��� }t jt jdt jdgdd t jdt jgg}tdd gd ggddgddggddgd�}tdgd ddddggdddddgddd d d ggd d gd�}t|||d�}t�||� |j dd � �� �ddd g�}t�|�� |� d S ) NZHgi��(�Ug ZPbg� V�Z��>ZSng�@���>g�]K�=�?i)=�(ZAgga2U0*�#�ZagentZchangeZdosageZs_idioC ipC iqC irC isC itC )r' r& r r$ r> )r+ r� rH r% r! r/ ) r r� r r rG r r r r rK )r r r r� ru r� r� r r r �test_unstack_nan_index4� s2 z,TestDataFrameReshape.test_unstack_nan_index4c C s t ddddddgtdddd�dt�d� tj�d�d �d�d ��}|d t�d � |d < tj |j d < |j d< tj |j d< |j d< |� dd d g�� d d g�}|� � j�� dt|� ks�t�xRdD ]J}xD|�� D ]8\}}|d ||d |d ff}|| |j | ks�t�q�W q�W d S )Nr$ r% z 2014-02-01rP r| )r[ �freq�d r� )r� r� r� r� r� z 2014-02-02r )r$ r� )r! r� )r$ r )r/ r r� )r� r� )r r r r r� r� �roundr] r� r� rK rG r r� r\ r r3 r6 Ziterrows)r r r� r� r� r rM r r r �test_unstack_nan_index5� s z,TestDataFrameReshape.test_unstack_nan_index5c C sz t ddd�}tddddgt�|ddfg�d�}|�� }t�d d g�}t�|dfg�}tddddg||d �}t�||� d S ) Ni� r$ r% r! r/ rT rU )r' )r r$ r% r! )rU )r& r' )r r r rE r r) r r )r �tr r, ZeidxZecolsr- r r r �%test_stack_datetime_column_multiIndex� s "z:TestDataFrameReshape.test_stack_datetime_column_multiIndexc C sd dd� }t jdddddgdd gd �}x�d d d ddgd d d dgd d d dgd d d gd d dgd ddgd d gd d gd dgd gd gdgf D ]0}||| � t|�d kr�|�� ||| � q�W tt�d��d d�|d d dg d�}|jdd�}td d gd tj gddgdtj ggt d d gddddggd d d d gd dd dggd d gd�t ddgdd �|j d d �}t � ||� d S ) Nc S s� t t�dt| � ��dt| ��| d�}x�dddddgddgfD ]�}|j|dd�}t|t�r�|j|dd�}t|t�r�t � ||� n t � ||� t j |j�� |jjd �|_|j|dd�}t|t�r�t � ||� q>t � ||� q>W d S ) Nr! )r' r> r r$ F)r� �dropnaT)rH )r r r r3 r r r� �intr r r* r r rE r' �to_numpyrH )Z multiindexr r� r, r- r r r �_test_stack_with_multiindex� s zWTestDataFrameReshape.test_stack_partial_multiIndex.<locals>._test_stack_with_multiindex)rU r0 )rU r= )rT r1 )rw r0 )rw �u�Upper�Lower)rH r r$ r% r! r/ rP )r' F)r r; r r0 r1 r= )r+ r� rH rU rw )rC )r& r' r< )r rE r3 �reverser r r r r r� r r� r r )r r Zfull_multiindexZmultiindex_columnsr r, r- r r r � test_stack_partial_multiIndex� s@ $ z2TestDataFrameReshape.test_stack_partial_multiIndex�orderedFT�labelsZyxzZyxyc C sb t j|td�|d�}tdddgg|d�}|�� }t�|j|g�}tdddg|d�}t � ||� d S )NrS )rj r r� r� r� )r' )r& ) r] �CategoricalIndexrV r r r r) r& r r r* )r r r �cidxr r, r� r- r r r �%test_stack_preserve_categorical_dtype s z:TestDataFrameReshape.test_stack_preserve_categorical_dtypez labels,datarS r� r� r� r� r� r� Zzyxc C s� t j|t|�|d�}t jddg|d�}t�||g�}tt|�g|d�}|�ddg�}t jt|�|d�} t|t�dg| |g�d�} t� || � d S ) N)rj r r �v)r )r' r r$ )r& ) r] r r� r r) r r r r r* ) r r r rL r! Zcidx2r� r r, Zs_cidxr- r r r �+test_stack_multi_preserve_categorical_dtype, s z@TestDataFrameReshape.test_stack_multi_preserve_categorical_dtypec C st t �ddddg�}t||d��}|�� }t�ddddgd d gg�}tt �ddddddddg�|d �}t�||� d S ) Nr" r# rg )rT rU r r$ r% r! rT rU )r& ) r] rm r r r r) r r r* )r r� r r, r& r- r r r �,test_stack_preserve_categorical_dtype_valuesA s zATestDataFrameReshape.test_stack_preserve_categorical_dtype_valueszindex, columnsr r" r# c C st t ||d��d�}|�� }t�|j�� �}t |�� ||jd�}t� ||� t � |jj �}t � |j �}t� ||� d S )N)r& r' r$ )r �fillnar r rE r& r r' r r r Zasarrayr� Zassert_numpy_array_equal) r r& r' r r � new_indexr- Z stacked_codesZexpected_codesr r r �)test_stack_multi_columns_non_unique_indexL s z>TestDataFrameReshape.test_stack_multi_columns_non_unique_indexr� c C s� t jdddgddgd�}ttjddd gd d �t�dddg�d �|d �}|j|d �}|�t�j|d �}t |j j gd |j j gd |j d �}t�|j|� t�|�t�|� d S )N)rT r )rT r$ )rU r$ r" r# )rH r r$ ZInt64)r< )rT rU )r& )r� r% )r rE r r] r� rm r rI r� r rT r< rU r' r r* r� r )r r� r& r r, r- Zexpected_dtypesr r r �"test_unstack_mixed_extension_typesa s $z7TestDataFrameReshape.test_unstack_mixed_extension_types�bazc C s� t jdgddggddgd�}tddgdd gg|d d gd �}d |j_td dddggt jddddgdd gd�d�}d|j_|�� jdd�j d|d�}t � ||� d S )Nr rl rg r r* )rH r% r$ r! rU rT )r& r' r )rg rT )rg rU )rl rT )rl rU )r' )r� )r� r� ) r r) r r' rC rE r& r � swaplevel� sort_indexr r )r r� r� r r- r, r r r � test_unstack_swaplevel_sortlevelu s z5TestDataFrameReshape.test_unstack_swaplevel_sortlevel)4�__name__� __module__� __qualname__r r. r: rO rZ r_ re rf rr rv rn �mark� parametrizer rE r rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rd Z&skip_array_manager_not_yet_implementedr r r rV r" r$ r% r) r( r) r- r r r r r s� /% #* & *,)H +" W+ ) B$& .r c C s� t ddddgdd�} t�dddd g�| _| �� }tdtjdgddtjgd �td �d �}t � ||� | jd d�}tdd dgddd gd �td �d �}t � ||� d S )Nr" r# rg r� )r< )r0 r" )r0 r# )r1 r# )r= r" )r" r# rS )r& rl )r? ) r r rE r& r r r r� rV r r )rL r, r- r r r � test_unstack_fill_frame_object� s $ r3 c C s� t tjddd�gdgdgdgd�ddddgd �} | �ddg��� }t tjddd�dggtdgdd �tddgdggd d gd d ggd dgd �d�}t�||� d S )Nz 2017-08-27 01:00:00.709949+0000�UTC)�tzr" r# rg )� timestampr" r# rg r6 )r' )rC r r$ )r+ r� rH )r& r' ) r r] r� rG r r r r r )r r, r- r r r �"test_unstack_timezone_aware_values� s r7 c C sn t ddddd�} td| iddd gd �}|�� }t| tddd gdggd d d gd d d ggd�d �}t�||� d S )Nr| Z20180101Z20180103zAmerica/New_York)r �start�endr5 rT r" r# rg )r& r r$ r% )r+ r� )r r r r r r r* )�tsr r, r- r r r � test_stack_timezone_aware_values� s ,r; r TFc C s@ t tg g gg g g�tjd�}ttjd�j| d�}t�||� d S )N)r& r< )r< )r )r r r rX r r r r* )r r- r, r r r �test_stack_empty_frame� s r< r? c C s8 t tjd�j| d�j|d�}t tjd�}t�||� d S )N)r< )r )r? )r r rJ r r r r )r r? r, r- r r r � test_stack_unstack_empty_frame� s r= c C s2 d} t jt| d�� ttjd��� W d Q R X d S )Nz'index must be a MultiIndex to unstack.*)rk )r< )rn ro r� r r rJ r )rq r r r � test_unstack_single_index_series� s r> c C s� t ddgddgddgddgdd gd ��} | �d d d dg�} | jdgdd�} | jd dd�jd dd�}| jd d gdd�}t ddddgddddggtddgd d�tjddddgd dd d gd�d�}t�||� d S )NZAliceZBobg #@rR FTr �female�male)rC �score�employed�kids�genderrC rB rC rD )r? g g @)rC )rA r? Fr )rA r? Tr )rA r@ Fr )rA r@ Tr )rH )r& r' )r rG r r r rE r r )r r- r, r r r � test_unstacking_multi_index_df� s( rE c C s� t jddgddgd�} tddddgg| d�}|�d�}tdd gdd �}t jd d gd dgd�}tddgddgg||d �}t�||� d S )N)r0 r1 )r1 r= r" )rH r$ )r' r r1 r= )rC )r r0 )r r1 )r& r' )r r) r r r rE r r )r' r r, Z new_columnsr' r- r r r �2test_stack_positional_level_duplicate_column_names� s rF c @ s� e Zd Zdd� Zej�dddddddgddddddggdd d d d d gdddddgfdddddgdddddggdd d d d gddddgfdddddgdddddggdd d d d gddddgfg�dd� �Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z ej�d d d g�d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Zd7d8� Zejjd9d:� �Z d;d<� Z d=d>� Z d?d@� Z dAdB� Z dCdD� Z!dEdF� Z"dGdH� Z#dIdJ� Z$dKdL� Z%dMdN� Z&dS )O�TestStackUnstackMultiLevelc C s6 |}|� � }|� � |�t�� � |�tj�� � d S )N)r rI r r rW )r �/multiindex_year_month_day_dataframe_random_data�ymdr r r r � test_unstack s z'TestStackUnstackMultiLevel.test_unstackz5result_rows,result_columns,index_product,expected_rowr$ Ng >@r% �ix1�ix2Zcol1Zcol2Zcol3Zcol4c C sp t ||d��ddg�}|jdd� �d�}t |gtj|dd � |ggd dgd�tdgdd�d�}t�||� d S ) N)r' rK rL r$ r% )rH )rC )r' r& ) r rG r� r r r) r r r )r Z result_rowsZresult_columnsZ index_productZ expected_rowr, r- r r r �test_unstack_partial s z/TestStackUnstackMultiLevel.test_unstack_partialc C sR t �ddddg�}ttj�d�|d�}|�ddg�}|jdd d �}t� ||� d S ) N)r r r )r r r )r$ r* r$ )r$ Zquxr$ r/ )r& r$ r% r� )r� �how) r rE r r r� r� r r r r )r r& rN r r- r r r �&test_unstack_multiple_no_empty_columnsA s zATestStackUnstackMultiLevel.test_unstack_multiple_no_empty_columnsc C sv |}|� � }|�� }t�||� |jdd�}|� d�}|�� }t�|jdd�|� |d d d� }|� d�}|�� �dd�}t�|jdd�|� |�dd�}|� d�jdddd�}|�d��dd�}t�|jdd�|� |� � }|jddd�}|�� }t�||� |� d�� d�}|�d�}|� � }t�||� |�d�}|� d�}t�||� |�d�}|�� � d�� d�}t�||� |� d�jd d �d d d �f }|�� �� }|�� } t�|| �|j �� |� d��d �}|� d��d�}d d � } t t � d �� dd �td�dddgd�} tddgdddggt �t � d��d �d�t �t � d �d�gd�} | �� tt � d �| d� } }| | |� dddg| _tddgddggt �t � d��d �d�t �dddgd�gd�} | �� tt � d �| d� } }| | |� d}t�|�| _ tddgddgddggt �t � d��d �d�t �dddgd dd g�t �dddgd�gd�} | �� tt � d �| d� } }| | |� d S )Nr% )r� r r> r$ )r� F)r� Z ascendingr! r� c S s: t �| |� | jjdks t�| j|j }}t �||� d S )NF)r r* r& Z is_uniquer6 �assert_index_equal)r� r� Zli�rir r r �check� s z4TestStackUnstackMultiLevel.test_stack.<locals>.checkr� r/ �ababr� r� r )r& r' r" r# )r+ r� )r& ))r" r% )r# r$ )r" r$ )r# r% rP )r r r r r, r+ rK r* �reindexr& r r r r rV r �tiler� r r' rE )r rH rI r � restackedZ unlexsortedr, r- r Z ymd_stackedrR r r� r� r� Ztplsr r r � test_stackL sz . . z%TestStackUnstackMultiLevel.test_stackc C s@ d}t �t|���dddg�}|�d�}|�� }t�||� d S )Na day,time,smoker,sum,len Fri,Dinner,No,8.25,3. Fri,Dinner,Yes,27.03,9 Fri,Lunch,No,3.0,1 Fri,Lunch,Yes,13.68,6 Sat,Dinner,No,139.63,45 Sat,Dinner,Yes,120.77,42 Sun,Dinner,No,180.57,57 Sun,Dinner,Yes,66.82,19 Thu,Dinner,No,3.0,1 Thu,Lunch,No,117.32,44 Thu,Lunch,Yes,51.51,17�day�timeZsmokerr% )r] Zread_csvr rG r r r r )r rL r r, �reconsr r r �test_unstack_odd_failure� s z3TestStackUnstackMultiLevel.test_unstack_odd_failurec C sr |}|j }d|d< |jddd�}|�� }|d �� �� }tj|d |dd� |jd ksZt�|d jtj ksnt�d S )Nr )r �fourr$ )r� r� F)r� r ) r r, r r r* rC r6 r< r Zfloat_)r � multiindex_dataframe_random_data�framer r r, r r r �test_stack_mixed_dtype� s z1TestStackUnstackMultiLevel.test_stack_mixed_dtypec C s� t ddddddgddddddgddddddgd d d d d d gt�d �d ��}|�ddddg��t�}|�� }|�� }t� ||� |j �� t �� d S )NZnaive�activer" r# r$ r% r! r/ �hiZbyeZpeaceg @)rz r� �barcoder# �extrarz r� rb r# )r r r �groupbyr r3 r r r r* rT r& rI rF )r r r, r rV r r r �test_unstack_bug� s z+TestStackUnstackMultiLevel.test_unstack_bugc C sP |}|� � }|jjdks t�|jjddgks0t�|�� }|jj|jjksLt�d S )Nr� r� r� )r r& rC r6 r' rH r )r r] r^ r rV r r r �!test_stack_unstack_preserve_names� s z<TestStackUnstackMultiLevel.test_stack_unstack_preserve_names�methodr r c C s| |}|j d }tjtdd�� t||�d� W d Q R X |dkrx|jd d �df }tjtdd�� t||�d� W d Q R X d S )Nr zdoes not match index name)rk Zmistaker r )rK rn ro �KeyError�getattrr� )r rg r] r^ r rN r r r �#test_stack_unstack_wrong_level_name� s z>TestStackUnstackMultiLevel.test_stack_unstack_wrong_level_namec C s* |}|� d�}|j dd�}t�||� d S )Nr� r$ )r� )r r r )r r] r^ r, r- r r r �test_unstack_level_name s z2TestStackUnstackMultiLevel.test_unstack_level_namec C sT |}|� d�}|�d�}|� � �d�}t�||� |�d�}|�� }t�||� d S )Nr� r� r )r r r r r* )r r] r^ r r, r- r r r �test_stack_level_name s z0TestStackUnstackMultiLevel.test_stack_level_namec C s2 |}|� ddg�}|� d�� d�}t�||� |jj|jjksBt�|d }|� ddg�}t�||d � |�ddg�}|�dd��dd�}|jdd�}t�||� |j j|j jks�t�|� ddg�}|� d�� d�j ddd �}t�||� |� ddg�}|� d�� d�j ddd �}t�||j d d �|jf � d S ) N�year�monthrT r r$ r% )r� r� )r� rN ) r r r r' rH r6 r r+ r, r& r rK )r rH rI r r- rN Z s_unstackedrV r r r �test_stack_unstack_multiple s& z6TestStackUnstackMultiLevel.test_stack_unstack_multiplec C s> |}|� ddg�}tjtdd�� |�ddg� W d Q R X d S )Nrm rn zlevel should contain)rk r )r rn ro r� r )r rH rI r r r r � test_stack_names_and_numbers2 s z7TestStackUnstackMultiLevel.test_stack_names_and_numbersc C sf |}|� ddg�}tjtdd�� |�ddg� W d Q R X tjtdd�� |�dd g� W d Q R X d S ) Nrm rn zToo many levels)rk r% r! znot a valid level numberr� r� )r rn ro � IndexErrorr )r rH rI r r r r �!test_stack_multiple_out_of_bounds= s z<TestStackUnstackMultiLevel.test_stack_multiple_out_of_boundsc C s t jddddddgddd�}tddgd d d �}d d d dddg}t�||g�}t||d�}|�� }|jd d�}|jdd�}t jdddgddd�} td d dgd ddgd�| ddgd�} d | j_ t � || � t � || � t � || j � t jddddddgddd�}t jdddddd gdd d�}t�||g�}t||d�}|�� }|jd d�}|jdd�}t jdddgddd�} t jd dddddgdd d�} tt jt jt jt jd d gt jt jdd t jt jgddt jt jt jt jgg| | d�} t � || � t � || � t � || j � d S ) Nz2013-01z2013-02z2013-03�MZperiod)r rC rT rU r! �str)rC r$ r% r/ r; rP )r& )r� r )rT rU )r& r' �period1z2013-12z2013-11z2013-10z2013-09z2013-08z2013-07�period2)r] � PeriodIndexr r r� r r r r' rC r r r r r� ) r �idx1�idx2�valueru rN �result1�result2�result3Ze_idxr- �e_colsr r r �test_unstack_period_seriesJ s^ " z5TestStackUnstackMultiLevel.test_unstack_period_seriesc C s� t jddddddgddd�}t jddddddgddd�}d d d d d dgdd d d d d gd�}t�||g�}t||d�}|�� }|jd d�}|jdd�}t jddgddd�} t jddddddgddd�} t�d�� | g�} td d dd dd gd d d d d d gg| | d�} t�|| � t�|| � t jddddgddd�} t jdddgddd�} t�d�� | g�} td d d d gd d dd gdd d d gg| | d�} t�|| � d S )Nz2014-01z2014-02rs ru )r rC z2013-12z2013-10rv r$ r% r! r/ r; rP )rT rU )r& )r� r z A A A B B B)r& r' zA A B B) r] rw r r� r r r� r r ) r rx ry rz ru r r{ r| r} Ze_1Ze_2r~ r- r r r �test_unstack_period_frame� s@ " ( *z4TestStackUnstackMultiLevel.test_unstack_period_framec C s� dgd dgd }dgd dgd }t �dddgd �}tj�d d d �}t||||d ��}|�d dg�}d|j_|� d�}|� d�� � }|� d�} |j d d �dgf � d�� � � d�} d| j_t�| | � d S )Nr$ r! r% r" r# z 2013-01-03z 2013-01-04z 2013-01-05r r rP )�ID�NAME�DATE�VAR1r� r� ZParamszW-THUr� )r] � to_datetimer r� �randintr rG r' rC r ZresampleZmeanr rK r r ) r Zid_colrC �dateZvar1r �multiZunstZdownr� r� r r r �test_stack_multiple_bug� s $z2TestStackUnstackMultiLevel.test_stack_multiple_bugc C sv t ddgddgddgd��}|�ddg�}|�� jd d �}t|�t|�� �ksRt�|�� jd d �}t�||�� � d S ) N�a1�a2�b1�b2r$ )rT rU rw rT rU F)r T) r rG r r r3 r r6 r r )r r r r r r �test_stack_dropna� s z,TestStackUnstackMultiLevel.test_stack_dropnac C s� t ddddddddgddddddddgddddddddggddddgddddggd�}dddg|j_ddg|j_|�ddg� d S ) Nr r$ )r& r' r" r# rg rl rt )r r& rH r' r )r r r r r �"test_unstack_multiple_hierarchical� s z=TestStackUnstackMultiLevel.test_unstack_multiple_hierarchicalc C s� d}t tjjd|d�tjjd|d�tjjdd|d�tjjdd|d�tjjd |d�tj�|�d ��}|�d d d ddg�}|�d� d S )Ni� r )�sizei, i����rQ i��� i� )rT rU rw r| r} r~ rT rU rw r| r} )r r r� r� r� rG r )r ZNUM_ROWSr Zidfr r r � test_unstack_sparse_keyspace� s z7TestStackUnstackMultiLevel.test_unstack_sparse_keyspacec C s~ ddgddddgg}ddddgddddgg}t ||�}ttj�dd�|d�}|�� }t|j�dksft�|� � }t � ||� d S )Nr r$ r% r! r/ )r& ) r r r r� r� r r3 r' r6 r r r )r r+ r� r& r r, rZ r r r � test_unstack_unobserved_keys s z7TestStackUnstackMultiLevel.test_unstack_unobserved_keysc C sL t tj�dd�t�d�t�d�gd�}tjtdd�� |�� W d Q R X d S )Ni r% )r& zint32 overflow)rk ) r r r� r� r rn ro r� r )r r r r r �/test_unstack_number_of_levels_larger_than_int32 s $zJTestStackUnstackMultiLevel.test_unstack_number_of_levels_larger_than_int32c C s dd� }x�dD ]�}t jt �dddg|�dd�}xl|D ]d}t|ddddgddddggd�}t|td �gd �}x,td�D ] }|�|�}||||d| � qtW q4W qW td d d gd d d ggt�td�d�t� td�d�gd�} t| td�t� dt | � �� dd�d�}|||�d�dd� d S )Nc s& t � ���fdd�� jD ��s"t�d S )Nc 3 sB | ]:}� j D ].}� j||f �j||� f|� f kV q qd S )N)r' rK )ra �rowr� )r � df_stacked�lev0�lev1r r � <genexpr> s zsTestStackUnstackMultiLevel.test_stack_order_with_unsorted_levels.<locals>.manual_compare_stacked.<locals>.<genexpr>)r� r& r6 )r r� r� r� r )r r� r� r� r �manual_compare_stacked s z`TestStackUnstackMultiLevel.test_stack_order_with_unsorted_levels.<locals>.manual_compare_stacked)r% r! r r$ r% )r� )r+ r� r/ )r' rL rT rw rU r! r; r> )r' r& rL ) r� r� r r r r( r r r� rU r r3 r ) r r� �widthZ levels_possr+ r' r Z stack_levr� r� r r r �%test_stack_order_with_unsorted_levels s $&z@TestStackUnstackMultiLevel.test_stack_order_with_unsorted_levelsc C s� t �d�}t �dd� |D �dd� |D �g�}t|jddgd�}d|j_d |i}tj|d d �}d dg|j_ |j t |j�d d �}|� dd g�� dd g�}td dgddgddgddgddggdd dddgtjdd gdd gd �d �}t�||� d S ) Nr; c S s g | ]}d |� ��qS )r# r )ra r0 r r r rc > s zVTestStackUnstackMultiLevel.test_stack_unstack_unordered_multiindex.<locals>.<listcomp>c S s g | ]}d |� ��qS )r" r )ra r0 r r r rc ? s r# r" )r' r� r0 r$ )r� r� Za0Zb0r� r� r� r� �a3Zb3Za4Zb4r r% r! r/ )r" r0 )r# r0 )rH )r& r' )r r Zvstackr r r' rC r] �concatrH rT r� r r r rE r r )r r\ rL r Zsecond_level_dictZmulti_level_dfr, r- r r r �'test_stack_unstack_unordered_multiindex9 s" zBTestStackUnstackMultiLevel.test_stack_unstack_unordered_multiindexc C s^ |}d|d< d|d< |� d�}|d jtjks2t�|d jtjksFt�|d jtjksZt�d S ) Nr r} r% r~ rn )rT r$ )r} r$ )r~ r$ )r r< r rX r6 Zobject_)r rH rI r r r r �test_unstack_preserve_typesR s z6TestStackUnstackMultiLevel.test_unstack_preserve_typesc C sj t �t �d�d�}t �d�}t|gd ddgg |gd t �d��d�g d�}tt �d�|d�}|�� }|jd ksxt�|� � }t � ||� |j �� tddgg|gd t �d��d�g|gd d�}tt �d�|d�}|�d�}|jd ks�t�t|gd ddgg |gd |gd t �d��d�g |gd d�}tt �d�|d�}|�d �}|jd k�sft�d S ) Ni� r% rR r r$ )r+ r� i� )r& )i� r% r/ )r rU r r r� r r r r6 r r r* rT r& )r r� r� r& rN r, r r r r �!test_unstack_group_index_overflow_ s, * z<TestStackUnstackMultiLevel.test_unstack_group_index_overflowc C s� t dddgdddgdgd d���dd g�}d |d < |sLt|jj�d ksLt�|�d �}|d �d�|d < t ddddgtj dddggt ddgddd�t j ddd dgd d gd�d�}|r�|d � d�|d< |d � d�|d< t�||� d S )NrT rU �ca�cbr� r! )r" r# r# r" r# r$ �is_r% )r� r� r g $@g �?g r� )r< rC )r# r� )r# r� )r� r� )rH )r& r' rJ )r rG r3 r4 r5 r6 r r& r r� r r rE rI r r )r r7 r r, r- r r r �+test_unstack_with_missing_int_cast_to_float� s( zFTestStackUnstackMultiLevel.test_unstack_with_missing_int_cast_to_floatc C s� t ddddgddddgddddgddddgd��}|�dddg�}d d d d g}|jj|dd �|_|jj|dd �|_|�d�d �� j}td d d d gd d d d ggddddgddddggddgd�}t�||� d S )Nr$ r% r! r/ )�L1�L2�L3r0 r� r� r� Zn1Zn2Zn3)r+ r� )r0 r$ r )r+ r� rH ) r rG r& Z set_levelsr r, r r rP )r r� Z new_levelsr, r- r r r � test_unstack_with_level_has_nan� s z:TestStackUnstackMultiLevel.test_unstack_with_level_has_nanc C s� t t�ddg�t�dddddg�d�}|�d �}t d tjtjgtjd d gtjd d ggtd d d g�td ddg�d�}t� ||� d S )Nr$ r; )r NN)r r% r )r r% r$ )r r! r )r r! r$ )r' r% g )r N)r r )r r$ )r r% )r r! )r& r' ) r r r� r rE r r� r r r )r r r, r- r r r �$test_stack_nan_in_multiindex_columns� s " z?TestStackUnstackMultiLevel.test_stack_nan_in_multiindex_columnsc C s� t �dgd dgd t�td��t�td��g�}tt�d��dd�|d�}|� d dg�}td tj gtj dgd tj gtj d gdtj gtj d gd tj gtj dggddgt �d gd d gd t�td��t�td��g�d�}t � ||� d S )NrT r% rU rS ZccddrR r/ )r' r$ r r! rP r; rQ ZaabbaabbZcdcdcdcd)r' r& ) r r� r] rm rV r r r r r r� r r )r r� r r, r- r r r �"test_multi_level_stack_categorical� s* z=TestStackUnstackMultiLevel.test_multi_level_stack_categoricalc C s� t t�d��dd�tjdtjfdgddgd�tdd gd d �tjd �}|� � }t d tjgtjd gdtjgtjdggtddgdd �tjdtjfdd tjfdgd dgd�d�}t � ||� d S )Nr/ r% rT )rU r# r r )rH r r$ �Num)rC )r' r& r< g g @g @rU )r r# )r$ r# )r' r& ) r r r r r rE r� r rX r r r )r Zdf_nanr, r- r r r �test_stack_nan_level� s "&z/TestStackUnstackMultiLevel.test_stack_nan_levelc C s� t �dgddgg�}tdt�ddg�i|d�}|�� }ttjdgddgd�tjdgddgd�d �dgd�}t �d d g�|_t� ||� d S ) NrT r r$ r� r" r# )r& )rj )r r$ )r� r )r� r$ ) r r) r r] rm r rE r' r r )r ru r r, r- r r r � test_unstack_categorical_columns s z;TestStackUnstackMultiLevel.test_unstack_categorical_columns)'r. r/ r0 rJ rn r1 r2 rM rO rW r[ r_ re rf rj rk rl ro rp rr r r� r� r� r� r� r� Zslowr� r� r� r� r� r� r� r� r� r� r� r r r r rG sZ c E/ ! % !rG )#r �ior r� �numpyr rn Z pandas.util._test_decorators�utilZ_test_decoratorsrd Zpandasr] r r r r r r r Zpandas._testingZ_testingr r r3 r7 r; r1 r2 r< r= r> rE rF rG r r r r �<module> s6 $ y
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка