Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/methods/__pycache__/test_reset_index.cpython-37.pyc
Назад
B �5�g� � @ s� d dl m Z d dlZd dlZd dlZd dlmZmZmZm Z m Z m Z d dl m Z G dd� d�Zej�dddgefejd d d d �ejd d �fg�dd� �ZdS )� )�datetimeN)� DataFrame�Index� MultiIndex� RangeIndex�Series� date_rangec @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestResetIndexc C s� t dddd��d �}tdtj�t|��i|d�}|�� }|jd t� d�ksPt �|� d �}t j ||d d � td d d�}t|dggddgd�}|� d�}|jd |ks�t �|�� d d |ks�t �d S )Nz1/1/2001z6/1/2001�D)�start�end�freq�v)�indexr zM8[ns]r F)Z check_namesi� � � g333333(@�Date�Value)�columns)r Z _with_freqr �np�randomZrand�len� reset_index�dtypes�dtype�AssertionError� set_index�tm�assert_frame_equalr r )�selfZdtiZd1Zd2Zd3Zstamp�df� r! �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/methods/test_reset_index.py� test_reset_index_dti_round_trip s z.TestResetIndex.test_reset_index_dti_round_tripc C sF t �� d d� }|�� }ddg|j_d|_|�� }d|ks>t�|jdd�}d|ksVt�|jdd�}|}|jddd �}|d ks�t�t �||� t d gd d d gddggddddddgddddddgddddddggd�}t t j � d�|d�}|jdd�}t|j�dk�st�|jddgdd�}t �|jt|�d��� t|t ��sBt�d S )N� �hash�category�valueZvalue2)�nameT)�drop)r) Zinplace�bar�one�two�threer � � )�levels�codes� )r )�level)r3 r) )r Z makeDataFrame�stackr �namesr( r r �assert_series_equalr r r r Zrandnr r Zassert_index_equalr Zget_level_values� isinstance)r r �ser�s�s2Z return_valuer �rsr! r! r" �test_reset_index# s, 2 z TestResetIndex.test_reset_indexc C sN t dddgttd�dd�d�}|�� jjd ks2t�|jdd�jjd ksJt�d S ) Nr. r/ � �x)r( )r T)r) )r r �ranger r r( r )r r9 r! r! r" �test_reset_index_nameC s z$TestResetIndex.test_reset_index_namec C s� t dddgdddggddd gd �}�x�ddgd dgfD �]r}|�ddg�d }|j|d d �}t�||�d�� |j|d d� d �}t�||�d�� |j|d �}t�||� |�ddg�j|d d�}t�||d g � tjtdd�� |jddgd � W d Q R X |�d�d }|j|d d �}t�||ddg � |j|d d� d �}t�||ddg � |j|d d d�}t�||d � tjt dd�� |jd ddgd � W d Q R X q6W t t d��jg d d�}t t d��}t�||� d S )Nr. r/ r= � r$ r2 �A�B�C)r r )r3 T)r3 r) zLevel E )�match�EzToo many levels)r) ) r r r r r �pytest�raises�KeyErrorr6 � IndexErrorr r? )r r r0 r9 �result�expectedr! r! r" �test_reset_index_levelH s2 " z%TestResetIndex.test_reset_index_levelc C s^ t td�ddd�}|�� }t|jt�s*t�tddgddggddgtdd�d �}t� ||� d S ) Nr/ rB �int64)r( r r r. r )�stop)r r ) r r? r r7 r r r r r r )r r9 Z series_resultZseries_expectedr! r! r" �test_reset_index_ranger s "z%TestResetIndex.test_reset_index_rangec C s� t td��}tjtdd�� |jddd� W d Q R X tjtdd�� |�d� W d Q R X t td�t�ddggd �d �}tjtd d�� |jddd� W d Q R X d S ) NrA zdoes not match index name)rE ZwrongT)r) r. r/ )r z not found)r r? rG rH rI r r � from_product)r r9 r! r! r" � test_reset_index_drop_errors| s z+TestResetIndex.test_reset_index_drop_errorsc C s~ |}|� � }t|t�st�t|j�t|jj�d ks8t�|jj|jjksLt�|j dd�}t|t �sft�|jj|jjkszt�d S )Nr. T)r) ) r r7 r r r r r r0 r( r )r Z series_with_multilevel_indexr8 Z deleveledr! r! r" �test_reset_index_with_drop� s z)TestResetIndex.test_reset_index_with_dropc C st t dddgtdddgdd�d�}d}tjt|d�� |�dd �}W d Q R X tdddgdddgd ��}t�||� d S ) Nr. r/ r= �a)r( )r zwIn a future version of pandas all arguments of Series\.reset_index except for the argument 'level' will be keyword-only)rE F)rT r )r r r Zassert_produces_warning� FutureWarningr r r )r r8 �msgrK rL r! r! r" � test_drop_pos_args_deprecation� s z-TestResetIndex.test_drop_pos_args_deprecationN) �__name__� __module__� __qualname__r# r< r@ rM rP rR rS rW r! r! r! r" r s * r z array, dtyperT �bz 12-1-2000r/ zQ-DEC)Zperiodsr )r c C sX t �ddgddg| g�}tt|d�d d� �� j}ttjtj|td��}t � ||� d S )Nr r. g �?g �?)r r )Zlevel_0Zlevel_1Zlevel_2r ) r rQ r �objectr r r rN Zfloat64r r6 )�arrayr �idxrK rL r! r! r" �7test_reset_index_dtypes_on_empty_series_with_multiindex� s r_ )r �numpyr rG Zpandas�pdr r r r r r Zpandas._testingZ_testingr r �markZ parametrizer\ Z period_rangeZ PeriodDtyper_ r! r! r! r" �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка