Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/__pycache__/test_union_categoricals.cpython-37.pyc
Назад
B �5�gy8 � @ sV d dl Zd dlZd dlmZ d dlZd dlmZmZm Z d dl mZ G dd� d�Z dS )� N)�union_categoricals)�Categorical�CategoricalIndex�Seriesc @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zdd� Zd d� Z dd� Z dd� Zdd� ZdS )�TestUnionCategoricalsc C sf t d�t d�t d�fdddgdddgddddddgfdd dgdd dgdd ddd dgfddtjdgdtjd gddtjddtjd gft�dd�t�dd�t�dd�ftjdddd�tjdddd�tjdddd�ft�dd�t�dd�t�dd�fg}xX|D ]P\}}}xDtttgD ]6}t |t|��|t|��g�}t|�}t �||� �qW q�W tdddg�}tddd g�} t || g�}tdddddd gdddddd gd�}t �||� tdd dgdd�}tdd dgdd�} t || g�}tdd ddd dgdd�}t �||� tdd dg�}tdddg�} d} tj t| d�� t || g� W d Q R X d} tj t| d�� t g � W d Q R X d S )N�abcZabdZabcabdr � � � � g333333�?g333333@�b�a�cz 2014-01-01z 2014-01-05z 2014-01-06z 2014-01-07z US/Central)�tz�x�y�z)� categoriesT)�orderedz$dtype of categories must be the same)�matchzNo Categoricals to union)�list�np�nan�pdZ date_rangeZperiod_ranger r r r �tm�assert_categorical_equal�pytest�raises� TypeError� ValueError)�self�datar r �combined�box�result�expected�s�s2�msg� r) �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/test_union_categoricals.py�test_union_categorical sN "z,TestUnionCategoricals.test_union_categoricalc C s� t tddtjg�tddtjg�g�}tddtjddtjg�}t�||� t tddg�tddtjg�g�}tddddtjg�}t�||� t�d�t�d�tjg}tjt�d�t�d�g}t t|�t|�g�}t|| t�d�t�d�t�d�gd �}t�||� t ttj tjtjgt d ��tdg�g�}ttjtjdg�}t�||� t ttjtjg�ttjtjg�g�}ttjtjtjtjg�}t�||� d S )Nr r r �A�Bz 2011-01-01z 2011-03-01z 2011-02-01)r )�dtype�X)r r r r r r r � TimestampZNaT�array�object)r �res�expZval1Zval2r) r) r* �test_union_categoricals_nanL s4 ""z1TestUnionCategoricals.test_union_categoricals_nanc C sX t tg �tg �g�}tg �}t�||� t tg �tdg�g�}tdg�}t�||� d S )N�1)r r r r )r r3 r4 r) r) r* �test_union_categoricals_emptyx s z3TestUnionCategoricals.test_union_categoricals_emptyc C s� t ddddgddddgd�}t dddtjgddddgd�}t||g�}t dddddddtjgddddgd�}t�||� t dddgdddgd�}t dddgdddgd�}t||g�}t ddddddgdddgd�}t�||� d S ) Nr r r r )r r r r )r r r r r r )r �c1�c2r3 r4 r) r) r* �$test_union_categorical_same_category� s &z:TestUnionCategoricals.test_union_categorical_same_categoryc C sj t dddgdddgd�}t dddgdddgd�}t||g�}t ddddddgdddgd�}t�||� d S )Nr r r )r )r r r r )r r8 r9 r$ r% r) r) r* �6test_union_categorical_same_categories_different_order� s zLTestUnionCategoricals.test_union_categorical_same_categories_different_orderc C s8 t dddgdd�}t dddgdd�}d}tjt|d�� t||g� W d Q R X t||g�}t ddddddgdd�}t�||� t dddtjgdd�}t ddgdddgdd �}t||g�}t dddtjddgdd�}t�||� t dddgdd�}t dddgdddgdd �}d }tjt|d�� t||g� W d Q R X d S )Nr r r T)r Fz$Categorical.ordered must be the same)r )r r z>to union ordered Categoricals, all categories must be the same) r r r r r r r r r )r r8 r9 r( r3 r4 r) r) r* �test_union_categoricals_ordered� s$ z5TestUnionCategoricals.test_union_categoricals_orderedc C s� t dddgdd�}t dddgdd�}t||gdd�}t ddddddg�}t�||� d}tjt|d �� t||gdd� W d Q R X t||gdd�}t ddddddg�}t�||� t||gdd�}t ddddddgdddgdd �}t�||� t dddtjgdd�}t ddgdddgdd �}t||gdd�}t dddtjddg�}t�||� t dddgdd�}t dddgdddgdd �}t||gdd�}t ddddddg�}t�||� t||gddd�}t ddddddgdddgd�}t�||� t dddgdd�}t d ddgdd�}t||gdd�}t dddd ddg�}t�||� d}tjt|d �� t||gdd� W d Q R X tjt|d �� t||g� W d Q R X d S )Nr r r T)r F)�ignore_orderz$Categorical.ordered must be the same)r )r r )r= �sort_categories)r r � � z>to union ordered Categoricals, all categories must be the same) r r r r r r r r r )r r8 r9 r3 r4 r( r$ r% r) r) r* �$test_union_categoricals_ignore_order� sJ z:TestUnionCategoricals.test_union_categoricals_ignore_orderc C s� t dddg�}t dddg�}t||gdd�}t ddddddgddddddgd �}t�||� t ddgdddgd �}t ddgdddgd �}t||gdd�}t ddddgdddgd �}t�||� t ddgdddgd �}t ddgdddgd �}t||gdd�}t ddddgdddgd �}t�||� t ddgdddgd �}t ddgdddgd �}t||gdd�}t ddddgdddgd �}t�||� t dtjg�}t tjdg�}t||gdd�}t dtjtjdgddgd �}t�||� t tjg�}t tjg�}t||gdd�}t tjtjg�}t�||� t g �}t g �}t||gdd�}t g �}t�||� t ddgdddgdd �}t ddgdddgdd �}d}tjt|d�� t||gdd� W d Q R X d S ) Nr r r r r r T)r>