Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_reset_index.cpython-37.pyc
Назад
B �5�g�_ � @ s� d dl m Z d dlmZ d dlZd dlZd dlmZmZ d dl Z d dl m Z m Z m Z mZmZmZmZmZmZmZ d dlmZ G dd� d�Zej�dd d gefe j d d d d�e j d d�fg�dd� �Z dd� Z dd� Z dd� Z!dd� Z"dS )� )�datetime)�productN)�is_float_dtype�is_integer_dtype) � Categorical�CategoricalIndex� DataFrame�Index� IntervalIndex� MultiIndex� RangeIndex�Series� Timestamp� date_rangec @ s� e Zd Zdd� Zdd� Zdd� Zej�ddd g�d d � �Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zej�ddd d d e�d �ed d!d"�d#g�d$d%� �Zd&d'� Zd(d)� Zej�d*�d+d,� �Zd-d.� Zd/d0� Zd1d2� Zej�d3ej d4fej d fgej ej gd4d gd5d6gd7�fej d4fed8�d fgej ed8�gd4d gd5d6gd7�fej d4fe�d9d:�d fgej e�d9d:�gd4d gd5d6gd7�fg�d;d<� �Z ej�d=d>d>d4d4gd>d4d>d4ggd>d>d?d4gd>d4d>d4ggf�d@dA� �Z dS )B�TestResetIndexc C sd t dddgdd�}tdddd gi|d �}|�� }|d jt�d �ksHt�|�d�}t�|j |� d S ) Nl l l �foo)�name�Ar � � )�indexZuint64) r r � reset_index�dtype�np�AssertionError� set_index�tm�assert_index_equalr )�self�idx�df�result� r" �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_reset_index.py�test_set_reset s z TestResetIndex.test_set_resetc C sb t tdddd�dd�}tddd d gi|d �}|�� }|d jd ksFt�|�d�}t�|j |� d S ) N�20130101� z US/Eastern)�periods�tzr )r r r r r )r zdatetime64[ns, US/Eastern]) r r r r r r r r r r )r r r r! r" r" r# �!test_set_index_reset_index_dt64tz) s z0TestResetIndex.test_set_index_reset_index_dt64tzc s� |� t ddd� dd�}ttd�ddddd gd �|d �}ttd d d �td d d�td d d�td d d�td d d�gtd�ddddd gd�dddgd�}|d �� fdd��|d< t�|�� |� d S )Nz1/1/2011� �Dr )r' �freqr( r r �B�C�E)�a�b)r i� r r r& � )r r0 r1 r0 r1 )�columnsc s t | � d�S )N)r( )r )�d)r( r" r# �<lambda>J � z4TestResetIndex.test_reset_index_tz.<locals>.<lambda>)r r �ranger �applyr �assert_frame_equalr )r Ztz_aware_fixturer r �expectedr" )r( r# �test_reset_index_tz5 s z"TestResetIndex.test_reset_index_tzr( z US/Easternzdateutil/US/Easternc C sR t dd|d�}ttj�t|��|�}|�� �d�}|jj }|jj }||ksNt �d S )Nz 2012-06-02� )r' r( r ) r r r �random�randn�lenr r r r( r )r r( Zdrr Z roundtripped�xp�rsr" r" r# �$test_frame_reset_index_tzaware_indexM s z3TestResetIndex.test_frame_reset_index_tzaware_indexc C st t jt�d�dd�}t|t�d�d��ddg }|�d�}tdt�d�i|d�}t�||� |�� }t�||� d S )N� �x)r r< )rD �yrE )r ) r Z from_breaksr �aranger r r r9 r )r r �originalr! r: Zresult2r" r" r# � test_reset_index_with_intervalsV s z.TestResetIndex.test_reset_index_with_intervalsc C s� |� � d d d� }t||d��}ddg}||j_|�� }xLtt|jj|jj��D ]2\}\}}|� |�}|| } t � |t || �� qPW d d g|j_|�� } t j |d | d dd� t j |d | d dd� |�� } t|jjd d �} t � | d | � |�� } d | d < | �� } t|jjdd �} t � | d | � d |j_|�� }t � |d t|j�� t j |jt tt|���d d � d|j_|�� }|jjdk�s�t�|�� �d ddg�} | �ddg�}t j||dd� | �d ddg�}t j||�� dd� | �d ddg�}t j||�� dd� | �d�}|�� �d dg�}t j||dd� |�� } |�� }| jd d�}|d k�s`t�t j| |dd� |�� �d ddg�} | jdd d�}|�� }|d= |jdgd d�}t j||dd� d S )Nr )r �bar�first�second�level_0F)Z check_names�level_1r )r r T)�exactr3 r r- )�inplace)�drop)�append)�stackr r �namesr � enumerate�zip�levels�codesZtaker r r �assert_series_equalr �values�copyr r7 r? r3 r r r9 )r Z float_frameZstackedrS � deleveled�iZlevZ level_codesrY r Z deleveled2�rdf�expr �resetrA r@ � return_valuer" r" r# �test_reset_indexa sh z TestResetIndex.test_reset_indexc C s� t ddddgddddggd d d d gttd�d d�d�}|�� jjd ksJt�|jdd�jjd ksbt�|jdd�}|d kszt�|jjd ks�t�d S )Nr r r& r2 r* � � � r r- r. r+ rD )r )r3 r T)rP )rO )r r r7 r r r r )r r r` r" r" r# �test_reset_index_name� s z$TestResetIndex.test_reset_index_namec C s� t ddddgddddggd d d d gd �}�x<d d gddgfD �]&}|�d d g�j|d d�}t�||�d �� |�d d g�j|d d� d�}t�||�d �� |�d d g�j|d�}t�||� |�d d g�j|dd�}t�||d d g � |�d �j|d d�}t�||� |�d �j|d d� d�}t�||� |�d g�j|d dd�}t�||d d d g � q<W xzd d gd gfD ]h}tjtdd�� |�|�jd dgd� W d Q R X tjtdd�� |�|�jdddgd� W d Q R X �qxW d S )Nr r r& r2 r* rb rc rd r r- r. r+ )r3 r )�levelT)rf rP z(L|l)evel \(?E\)?)�matchr/ zToo many levels) r r r r r9 �pytest�raises�KeyError� IndexError)r r rV r! Zidx_levr" r" r# �test_reset_index_level� s* ( z%TestResetIndex.test_reset_index_levelc C s~ t �ddt �d�d �}td|d d t|dd�dd�}t|�}|�� }|d jt jks^t �|�� }|d jt jkszt �d S ) Ng r< r g ��Q��#@�time)r �speed)r r ) r rF �sqrtr r r r r �float64r )r rm �s1r r_ r" r" r# � test_reset_index_right_dtype� s z+TestResetIndex.test_reset_index_right_dtypec C s t j�dd��t�}dddg}t �dd� |D �|f�}t|t|dd�d d d gd d d ggd �}|�� }t|dd d d gdd d d ggd �}t � ||� |jd d�}t|dd d d gdd d d ggd �}t � ||� |jddd�}t|dd d d gdd d d ggd �}t � ||� t|t j dddgdddggddgd�d d d gd d d ggd �}|�d�}t|tdddgdd�dd d d gdd d d ggd �}t � ||� |jdd d�}t|tt d�dd�dd d d gdd d d ggd �}t � ||� |jdddd�}t|tt d�dd�dd d d gdd d d ggd �}t � ||� d S )Nr& rD rE �zc S s g | ] }|g�qS r" r" )�.0rD r" r" r# � <listcomp>� s zBTestResetIndex.test_reset_index_multiindex_col.<locals>.<listcomp>r0 )r r1 �cZmeanZmedian)r3 � )�col_fillr Zblah)� col_levelrx r r r4 )rS )rx ry )r r= r> �astype�objectZhstackr r r r r9 r � from_arraysr7 )r �valsr �fullr rA r@ r" r" r# � test_reset_index_multiindex_col� sR z.TestResetIndex.test_reset_index_multiindex_colc C s t dddgddtjgtj�d�d��}|�dd g��� }t�||� t tjddgddd gtj�d�d��}|�dd g��� }t�||� t dddgddd gtjd d gd��}|�dd g��� }t�||� t dddgtjtjtjgtj�d�d��}|�dd g��� }t�||� d S ) Nr0 r1 rv r r r& )r r- r. r r- r g�������?g������@) r r �nanr= Zrandr r r r9 )r r rA r" r" r# � test_reset_index_multiindex_nan! s" $ $ $ z.TestResetIndex.test_reset_index_multiindex_nanr Nr r g @rb z 2012-12-30�UTC)r( z 2012-12-31c C s� d }t |t�r |jd k r t}tddgddggtdd�ddgd �}||j_t� |�� |� � }W d Q R X |d k rr|nd }t |t d dd�t d dd�g�}t |t �r�|d kr�|�d �}n|jtks�t�tdddgdddgg|d�}t�||� d S )Nr r r& r2 z1/1/2013z1/2/2013r r- )r3 r r i� z 2012-12-31zdatetime64[ns])r3 )� isinstancer r( � FutureWarningr r r r r �assert_produces_warningr r r �strrz r r{ r r9 )r r �warnr r! �itemr3 r: r" r" r# �(test_reset_index_with_datetimeindex_cols> s& z7TestResetIndex.test_reset_index_with_datetimeindex_colsc C sv t ddgddggddgtdd�d�}|�� }t|jt�s<t�t dddgdddggdddgtdd�d�}t�||� d S ) Nr r r r- r )�stop)r3 r r )r r r r� r r r r9 )r r r! r: r" r" r# �test_reset_index_rangeh s $z%TestResetIndex.test_reset_index_rangec C s� ddgddgg}t ddgddggt�|�d �}|dg �d��� }t�||� d }tjt |d �� |�d��� W d Q R X |� d g��� }t�||� t dgdggt�d g�d �}t j ||dd g gdd�}|j dgdd��� }t�||� d}tjt |d �� |�dg��� W d Q R X dddgdddgg}t ddgddggt�|�d �}t dgdggt�dg�d �}t j ||gdd�}|�dg�jdd�}t�||� tjt dd �� |�dg�jd d� W d Q R X |�dg�jdd d �}t�||� d S ) Nr rw r- r1 r r r r& )r3 z)cannot insert \('A', ''\), already exists)rg )r rw )rL rw )r- r1 )ZaxisT)rQ z0Item must have length equal to number of levels.)r. rv r\ r0 r\ )r. rv �ii)r. rv r� )rx zFcol_fill=None is incompatible with incomplete column name \('C', 'c'\))rv r� r. )ry rx ) r r � from_tuplesZ rename_axisr r r9 rh ri � ValueErrorr �pd�concat)r rV r r! �msgZidx_colr: Zdf2r" r" r# �#test_reset_index_multiindex_columnst s< z2TestResetIndex.test_reset_index_multiindex_columnsz1ignore:Timestamp.freq is deprecated:FutureWarningc s� |� t ddd� dd�}ttd�ddd�}t�||g�}ttjddd �d d d dd gd�|d�}ttddd�tddd�tddd�tddd�tddd�gtjddd �tjddd �d d d dd gd�ddddgd�}|d � � fdd��|d< t � |� � |� t ddd d d d�}t�|||g�}ttjddd �d d d dd gd�|d�}ttddd�tddd�tddd�tddd�tddd�gtjddd �td dd�td dd�td dd�td dd�td dd�gtjddd �d d d dd gd �ddd ddgd�}|d � � fd!d��|d< |d � d"d� �|d < t � |� � |� t� ddgt d#d� d$�g�}ttjd%dd ��d%d�dg|d&�}td'�� td(dd�td(dd�td(dd�gd tjd%dd �d)�d*d+dgd�}|d+ � � fd,d��|d+< |� � }t � ||� d S )-Nz1/1/2011r* r+ �idx1)r' r, r( r �idx2�int64)r r )r r r- r. r/ )r0 r1 )r i� r r r& r2 )r� r� r0 r1 r0 r1 )r3 c s t | � d�S )N)r( )r )r4 )r( r" r# r5 � r6 z:TestResetIndex.test_reset_index_datetime.<locals>.<lambda>z1/1/2012ZMSz Europe/Paris�idx3i� )r� r� r� r0 r1 c s t | � d�S )N)r( )r )r4 )r( r" r# r5 � r6 c S s t | dd�S )Nz Europe/Paris)r( )r )r4 r" r" r# r5 � r6 r% )r' r( rb )r3 r z a a a b b bi� )rL rM r0 rL rM c s t | � d�S )N)r( )r )r4 )r( r" r# r5 r6 )r r r7 r r| r r rF r r8 r r9 r � from_product�reshape�split) r Ztz_naive_fixturer� r� r r r: r� r! r" )r( r# �test_reset_index_datetime� sv z(TestResetIndex.test_reset_index_datetimec C s� t jtjdddd�td�gddgd�}ttjd d d ��d d �|dgd�}ttj ddd�gd tj ddd�gd tj ddd�gd dddgd tjd d d �d�dddgd�}|� � }t � ||� d S )Nr% r& �M)r' r, �abc�month�feature)rS � r� )r ���r r0 )r r3 z2013-01)r, z2013-02z2013-03r1 rv )r� r� r0 )r3 ) r r� r� � period_range�listr r rF r� ZPeriodr r r9 )r r r r: r! r" r" r# �test_reset_index_period s : z&TestResetIndex.test_reset_index_periodc C sz t tddgddgddg��}tj|ddd gd �}ttj�d d �d ddg|d�}|�� }t |d �sft �t |d �svt �d S )Nr rI r< � g �?g�������?Zprm0Zprm1Zprm2)rS rd r& r r- r. )r3 r ) r� r r r� r r r= r> r r r r )r Ztuplesr r r[ r" r" r# �$test_reset_index_delevel_infer_dtype s z3TestResetIndex.test_reset_index_delevel_infer_dtypec C s@ |}|j dd�}t|j�t|j�ks(t�|jj|jjks<t�d S )NT)rP )r r? r3 r r r )r Z/multiindex_year_month_day_dataframe_random_dataZymdr[ r" r" r# �test_reset_index_with_drop& s z)TestResetIndex.test_reset_index_with_dropzix_data, exp_datar rC � )r0 r1 rD z 2020-01-01�{ r4 c C sF t j|ddgd�}tdddgi|d�}|�� }t|�}t�||� d S )Nr0 r1 )rS rD rC r� )r )r r� r r r r9 )r Zix_dataZexp_dataZixr! r: r" r" r# � test_reset_index_nat_multiindex/ s z.TestResetIndex.test_reset_index_nat_multiindexrW r r� c C s� t tddg�tddg�g|�}dtt|��i}t||d�}ttj|d ddgd�tj|d ddgd�td �d ��}|�� }t� ||� |� d d g��� }t� ||� d S )Nr r- r0 r1 �col)�datar r )� categoriesr r2 )rL rM r� rL rM ) r r r7 r? r r Z from_codesr r r9 r )r rW r r� r r: �resr" r" r# �:test_rest_index_multiindex_categorical_with_missing_valuesI s zITestResetIndex.test_rest_index_multiindex_categorical_with_missing_values) �__name__� __module__� __qualname__r$ r) r; rh �mark� parametrizerB rH ra re rl rr r r� r� Z Timedeltar r� r� r� �filterwarningsr� r� r� r� �NaTr� r� r" r" r" r# r sH M " 8 ! 4] 0 r z array, dtyper0 r1 z 12-1-2000r zQ-DEC)r' r, )r, c C sT t �ddgddg| g�}t|d�d d� �� j}ttjtj|d��}t � ||� d S )Nr r g �?g �?)r )rL rM Zlevel_2) r r� r r Zdtypesr r r� rp r rX )�arrayr r r! r: r" r" r# �6test_reset_index_dtypes_on_empty_frame_with_multiindexe s r� c C s� t td�gddggg g gddgd�} t| ddgd �}|�� }ttd �td d d d �d�}|d �d�|d< |d �d�|d< t�||� d S )Nz2020-07-20 00:00:00r& r2 r0 r1 )rV rW rS rv r4 )r r3 �abcdr r )�startr� �step)r3 r zdatetime64[ns]r� ) r r r r r� r rz r r9 )r r r! r: r" r" r# �7test_reset_index_empty_frame_with_datetime64_multiindexw s r� c C s� t dgdgt�d�d��} | �d��ddg�dg �� } | �� }t dddgtddd d �d �}|d �d �|d< |d �d �|d< t � ||� d S )Ng $@r0 z 2020-01-01)�c1�c2�c3r r� r� r� r )r� r� r� )r3 r zdatetime64[ns]rp ) r r� � to_datetime�head�groupby�sumr r rz r r9 )r r! r: r" r" r# �Dtest_reset_index_empty_frame_with_datetime64_multiindex_from_groupby� s r� c C s� t d�} tdddd�}t| |td�d��}tj|jd< |�dd g��d�}tt d�td�d �tj dd d gd d �d�}t � ||� d S )Nr& z 2015-07-01r+ )r, r' r� )�id�tstampr0 )r r� r� r� )r� r0 z 2015-07-02r� )r )r ) r7 r r r� r� r� �locr r Z DatetimeIndexr r9 )r r� r r! r: r" r" r# � test_reset_index_multiindex_nat� s r� c C sd t ddddgi��d�} d}tjt|d�� | �dd�}W d Q R X t ddddgi�}t�||� d S )Nr0 r r r& zzIn a future version of pandas all arguments of DataFrame\.reset_index except for the argument 'level' will be keyword-only)rg F)r r r r� r� r r9 )r r� r! r: r" r" r# � test_drop_pos_args_deprecation� s r� )#r � itertoolsr �numpyr rh Zpandas.core.dtypes.commonr r Zpandasr� r r r r r r r r r r Zpandas._testingZ_testingr r r� r� r{ r� Z PeriodDtyper� r� r� r� r� r" r" r" r# �<module> s* 0 M
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка