Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_rename_axis.cpython-37.pyc
Назад
B �5�g� � @ sB d dl Zd dlZd dlmZmZmZ d dlmZ G dd� d�Z dS )� N)� DataFrame�Index� MultiIndexc @ s� e Zd Zdd� Zdd� Zdd� Zej�ddd d �ddfdd d �ddfddiddfddiddfddd�ddfi ddfg�dd� �Z dS )�TestDataFrameRenameAxisc C s� |� d�}|�� }|j ddd� }}|d ks0t�|d ks<t�t�||� |j ddd�}|�� }|j dddd� }}|d ks~t�|d ks�t�t�||� d S )N�fooT)�inplace�bar� )�axis)r r )�rename_axis�copy�AssertionError�tm�assert_frame_equal)�selfZfloat_frame�expected�resultZreturn_valueZ no_return� r �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_rename_axis.py�test_rename_axis_inplace s z0TestDataFrameRenameAxis.test_rename_axis_inplacec C s� t ddgddgd��}tjtdd�� |jtdd� W d Q R X tjtdd�� |jdd d �dd� W d Q R X tjtdd�� |jtdd� W d Q R X tjtdd�� |d �t� W d Q R X d S )Nr � )�A�Bz Use `.rename`)�matchr )r � � )r r r )r �pytest�raises� ValueErrorr �id)r �dfr r r �test_rename_axis_raises s z/TestDataFrameRenameAxis.test_rename_axis_raisesc C s� t jdddgddggddgd�}tttt|���d d � tt|��D �d�|d�}|jd dd�}t�|j t ddgd d�� |jd didd�}t�|j t ddgdd�� |jddid�}|jjddgks�t �|jtjdd�}|jjddgks�t �|jddgd�}|jjddgk�st �|�� �d�jddgd�}|jddd�}|jjdk�sLt �|j jdk�s^t �tjtdd�� |jdd� W d Q R X tjtd d�� |jdgd� W d Q R X tjtd!d�� |jd d"� W d Q R X d S )#N�a�b�cr r �ll�nn)�namesc S s g | ]}|d �qS )r r )�.0�ir r r � <listcomp>2 s zCTestDataFrameRenameAxis.test_rename_axis_mapper.<locals>.<listcomp>)�x�y)�index�cols)r r+ r, )�name�new)�columnsr r r )r- r ZLLZNNZgoo)r1 Zmeh)r- r1 z Must pass)r ZwrongzLength of names�bogus)r2 )r Zfrom_productr �list�range�lenr r Zassert_index_equalr1 r r- r'