Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/__pycache__/test_replace.cpython-37.pyc
Назад
B �5�g� � @ s� d dl Zd dlZd dlZd dlmZ d dlmZ ej � ddddddgdfdddddgdfdddddgdfd d dddgdfdgddddgdfddgddddgdfddgddddgdfdd d d dgdfdddddgdfdgdgdddgd fddgd dgd ddgd fdddddgdfdddgddddgd fg �dd� �Zej � dddddgdfddddgdfddddgdfddddgdfg�dd� �ZdS )� N)�Categoricalz)to_replace,value,expected,flip_categories� � � F� � � )r r r )r r T�4�3�5c C s� t |t� ptt�|��dk}tjdddgdd�}|�| |�}tj|dd�}|j| |dd� |rz|j�|jj d d d� �}|s�t�t �|��}tj ||d d � tj ||d d � d S )Nr r r �category)ZdtypeT)�inplace���F)Zcheck_category_order)� isinstance�list�len�pd�uniqueZSeries�replace�catZset_categories� categories�npZasarray�tmZassert_series_equal)� to_replace�value�expectedZflip_categoriesZstays_categorical�s�result� r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/test_replace.py�test_replace s r z-to_replace, value, result, expected_error_msg�b�c�az$Categorical.categories are different�dz+Categorical.categories length are differentc C sz t ddg�}t |�}|�| |�}t�||� | dkrZtjt|d�� t�||� W d Q R X |j| |dd� t�||� d S )Nr# r! )�matchT)r )r r r Zassert_categorical_equal�pytestZraises�AssertionError)r r r Zexpected_error_msgr r r r r � test_replace24 s r( ) �numpyr r&