Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/categorical/__pycache__/test_equals.cpython-37.pyc
Назад
B �5�g � @ s: d dl Zd dlZd dlmZmZmZmZ G dd� d�ZdS )� N)�Categorical�CategoricalIndex�Index� MultiIndexc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestEqualsc C s" t ddgddgdd�}t ddgdddgdd�}|�|�s<t�|�|�rJt�|�|�t��s^t�|�t��|�srt�||k�� s�t�||k�� r�t�||k�� r�t�||k �� r�t�||k�� s�t�||k�� s�t�|dk�� r�t�|tddg�k�� s�t�||jk�� �st�tj t dd�� |tdddg�k W d Q R X d }tj t|d�� ||k W d Q R X tj t|d�� |t|jd d�k W d Q R X tj t|d�� |t|jt d�d �k W d Q R X t t d�dddgd �}|�t d���r�t�|�t t d����s�t�|�t t d�dd���rt�|�|�� ��s.t�t t d�tjg dddgd �}|�t d���r`t�|�t t d����rxt�|�|�� ��s�t�t t d�tjg dddgd �}|�t d�tjg ��r�t�|�t t d�tjg ���s�t�|�t t d�tjg dd���r t�|�|�� ��st�d S )N�a�bT)� categories�ordered�c� zLengths must match)�matchz>Categoricals can only be compared if 'categories' are the sameF)r �abc)r Zaabca)r �equals�AssertionError�astype�object�allr �values�pytestZraises� ValueError� TypeErrorr �list�copy�np�nan)�selfZci1Zci2�msg�ci� r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/categorical/test_equals.py�test_equals_categorical sL $z"TestEquals.test_equals_categoricalc C sd t dgddgd�}t dgddgd�}t dgddgd�}|�|�sDt�|�|�rRt�|�|�r`t�d S )N�A�B)r �C)r r r )r r r r r r r �!test_equals_categorical_unorderedC s z,TestEquals.test_equals_categorical_unorderedc C s8 t ddtjtjg�}tdddtjg�}|�|�r4t�d S )Nr"