Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/__pycache__/test_api.cpython-37.pyc
Назад
B �5�g�U � @ s� d dl Z d dlZd dlZd dlmZmZmZmZm Z d dl m Z d dl mZ d dlmZ G dd� d�ZG dd� de�ZG d d � d �ZdS ) � N)� Categorical�CategoricalIndex� DataFrame�Index�Series)�recode_for_categories)�TestCategoricalc @ s6 e Zd Zdd� Zdd� Zdd� Zej�ddd d d gdd gg�d d � �Z dd� Z dd� Z dd� Z ej�ddgdddgddddgg�dd� �Z dd� Zd d � Zd d � Zd d!� Zej�d"dddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgdgfdddgddgdgfdddgddgdgfdddgddgdgfdddgddgdd#gfg �ej�d$d%d&g�d'd(� ��Zd)d*� Zd+d,� Zd-d.� Zej�d/dgdejgdddgg�d0d1� �Zd2d3� Zd4S )5�TestCategoricalAPIc C s� t td�dd�}t�|jtdddg�� |jr2t�t td�td�dd�}t�|jtdddg�� |jrjt�t td�d d�}t�|jtdddg�� |js�t�t td�td�d d�}t�|jtdddg�� |js�t�d S ) NZacbF)�ordered�a�b�cZbca)� categoriesr T)r �list�tm�assert_index_equalr r r �AssertionError)�selfZcat1�cat2Zcat3Zcat4� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/test_api.py�test_ordered_api s z#TestCategoricalAPI.test_ordered_apic C s� t ddddgdd�}|�� }|jr&t�|�� }|js8t�|jdd� |jrNt�|jdd� |jsdt�|�d�jstt�|�d�jr�t�|jddd� |js�t�|jddd� |jr�t�d}tjt|d �� d|_W d Q R X tjt|d �� d|_W d Q R X d S ) Nr r r T)r )�inplaceFzcan't set attribute)�match) r Z as_unorderedr r Z as_ordered� set_ordered�pytest�raises�AttributeError)r �catr �msgr r r �test_set_ordered% s( z#TestCategoricalAPI.test_set_orderedc C sB t ddddg�}|�dddg�}t�|�� tjddddgtjd�� t�|j t dddg�� tjddddgtj d�}t�|�� |� t dddg�}t�|j |� |�dd � �}t d d d d g�}t� ||� t� t�� |jdddgd d�}W d Q R X |d k�st�t�|�� tjddddgtjd�� t�|j t dddg�� d S )Nr r r � � � )�dtypec S s | � � S )N)�upper)�xr r r �<lambda>P � z;TestCategoricalAPI.test_rename_categories.<locals>.<lambda>�A�B�CT)r )r �rename_categoriesr �assert_numpy_array_equal� __array__�np�array�int64r r r �object_�assert_categorical_equal�assert_produces_warning� FutureWarningr )r r �resZexp_cat�result�expectedr r r �test_rename_categories? s$ z)TestCategoricalAPI.test_rename_categories�new_categoriesr! r" r# � c C s< t ddddg�}d}tjt|d�� |�|� W d Q R X d S )Nr r r zKnew categories need to have the same number of items as the old categories!)r )r r r � ValueErrorr, )r r: r r r r r �*test_rename_categories_wrong_length_raises_ s z=TestCategoricalAPI.test_rename_categories_wrong_length_raisesc C sB t ddg�}|�tddgddgd��}t ddg�}t�||� d S )Nr r r r! )�index)r r, r r r3 )r r r7 r8 r r r � test_rename_categories_seriesi s z0TestCategoricalAPI.test_rename_categories_seriesc C sP t ddddg�}|�ddddd ��}tddddg�}t�|j|� t�t�� |jddddd �d d �}W d Q R X |d ks|t�t�|j|� t ddddg�}|�ddd ��}tddddg�}t�|j|� t ddddg�}|�ddddd dd��}tddddg�}t�|j|� t ddddg�}|�ddd��}tddddg�}t�|j|� d S )Nr r r �dr; r# r" r! )r r r r@ T)r )r r � � )r r r r@ �e�f)rD �g) r r, r r r r r4 r5 r )r r r6 r8 r r r �test_rename_categories_dictp s( " z.TestCategoricalAPI.test_rename_categories_dictc C s� t ddddgdd�}|�� }t ddddgdddgdd�}|�dddg�}t�||� t�||� t�t�� |jdddgdd�}W d Q R X |d ks�t�t�||� d S )Nr r r T)r )r r )r )r �copy�reorder_categoriesr r3 r4 r5 r )r r �old�newr6 r r r �test_reorder_categories� s z*TestCategoricalAPI.test_reorder_categoriesr r r@ r c C s@ t ddddgdd�}d}tjt|d�� |�|� W d Q R X d S )Nr r r T)r z=items in new_categories are not the same as in old categories)r )r r r r< rH )r r: r r r r r � test_reorder_categories_raises� s z1TestCategoricalAPI.test_reorder_categories_raisesc C s\ t ddddgdd�}|�� }t ddddgddddgdd�}|�d�}t�||� t�||� |�dg�}t�||� t�||� t�t�� |jddd�}W d Q R X t�||� |d ks�t�t td �dd�}t td �td �dd�}|�t dd g��}t�||� |�t � dd g��}t�||� |�t dd g��}t�||� |�dd g�}t�||� d S ) Nr r r T)r r@ )r r )r �abcZabcderC ) r rG �add_categoriesr r3 r4 r5 r r r r/ r0 r )r r rI rJ r6 r8 r r r �test_add_categories� s0 z&TestCategoricalAPI.test_add_categoriesc C sH t ddddgdd�}t�d�}tjt|d�� |�dg� W d Q R X d S ) Nr r r r@ T)r z5new categories must not include old categories: {'d'})r )r �re�escaper r r< rN )r r r r r r �#test_add_categories_existing_raises� s z6TestCategoricalAPI.test_add_categories_existing_raisesc C sH t ddddgdd�}tdddg�}tjddddgtjd�}t�t�� |jdddgdd�}W d Q R X t� |j |� t� |� � |� |d ks�t �|�dddg�}t� |j |� t� |� � |� tdddg�}t� |j |� t� |� � |� t ddddgdd�}|�dg�}t� |jtjdd d dgtjd�� |�ddd g�}t� |jtjdd d dgtjd�� t� |j tddd g�� |�dddd g�}tdddd g�}t� |j |� t d d d dd gd d d dgdd�}t� |jtjdd d d dgtjd�� t� |j td d d dg�� tjd d d dd gtjd�}t� t�|�|� |�dd d d g�}t� |jtjd d d dd gtjd�� t� |j tdd d d g�� tjd d d dd gtjd�}t� t�|�|� |�� dk�s�t �|�� d k�s�t �|jdd d d gdd�}|j�r�t �t� t�|�t�|�� |�d��dd d d g�}|j�r,t �t� t�|�t�|�� d S )Nr r r T)r )r$ )r r ���r@ r! r" r# r; )r r F)r r r/ r0 r2 r r4 r5 �set_categoriesr r r- r. r �codes�int8�_codesr1 Zasarray�min�maxr r ) r r Zexp_categoriesZ exp_valuesr6 Zexp_categories_backr �exp�c2r r r �test_set_categories� sR "" $$ z&TestCategoricalAPI.test_set_categoriesc C s6 t ddddgdd�}t�t�� |�� W d Q R X d S )Nr r r T)r )r r r4 r5 Zto_dense)r r r r r �test_to_dense_deprecated, s z+TestCategoricalAPI.test_to_dense_deprecatedz"values, categories, new_categoriesrC r TFc C s4 t ||�}t |||�}|j||d�}t�||� d S )N)r )r rT r r3 )r �valuesr r: r r r8 r7 r r r �test_set_categories_many2 s z+TestCategoricalAPI.test_set_categories_manyc C s: t ddg�}|jdgdd�}t dtjg�}t�||� d S )Nr) r* T)�rename)r rT r/ �nanr r3 )r r r7 r8 r r r � test_set_categories_rename_lessR s z2TestCategoricalAPI.test_set_categories_rename_lessc C s� t dddgddddgd�}|�ddddg� t dddgtd�d�}t�||� t dddgddddgd�}|jddddgdd � t dddgtd�d�}t�||� d S ) Nr r r r@ )r rC ZacdeT)Zfastpath)r Z_set_categoriesr r r3 )r r r8 r r r �test_set_categories_privateY s z.TestCategoricalAPI.test_set_categories_privatec C s� t ddddgdd�}|�� }t ddtjdgddgdd�}|�d�}t�||� t�||� |�dg�}t�||� t�||� t�t�� |jddd�}W d Q R X t�||� |d ks�t �d S )Nr r r T)r )r r )r ) r rG r/ ra �remove_categoriesr r3 r4 r5 r )r r rI rJ r6 r r r �test_remove_categoriese s z)TestCategoricalAPI.test_remove_categories�removalsc C s@ t dddg�}t�d�}tjt|d�� |�|� W d Q R X d S )Nr r z-removals must all be in old categories: {'c'})r )r rP rQ r r r< rd )r rf r �messager r r � test_remove_categories_raises{ s z0TestCategoricalAPI.test_remove_categories_raisesc C s t dddddgdddddgd�}tdddddg�}tddddg�}t�|j|� |�� }t�|j|� t�|j|� t�t�� |jdd�}W d Q R X t�|j|� |d ks�t�t dddt j gdddddgd�}|�� }t�|jtt � dddg��� t j d d d d gt j d �}t� |j|� t�|j|� dt j ddddt j g}t |td�d�}|�� }t�|jtdddg�� t j d d d d d d d gt j d �}t� |j|� |�� |k�s�t�td�} t j�| d d d � d��d�}t j |t j�t|�d�< t || d�}|�� }|�� |�� k�st�d S )Nr r r r@ rC )r T)r r r! r" rS )r$ �F�Dr* ZABCDEFG)r^ r �abcdefghijklmnopqrstuvwxyzi' �object�d )r r r r r Zremove_unused_categoriesr4 r5 r r/ ra r0 rV r- rU r �tolist�random�choiceZastype�len) r r Zexp_categories_allZexp_categories_droppedr6 Z exp_codes�valr �out�alphar r r � test_remove_unused_categories� s< z0TestCategoricalAPI.test_remove_unused_categoriesN)�__name__� __module__� __qualname__r r r9 r �mark� parametrizer= r? rF rK rL rO rR r\ r] r_ rb rc re r/ ra rh ru r r r r r sJ & # %I *r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestCategoricalAPIWithFactorc C s� | j �� }| j jst�tdddgd| j jd�}tdddgdd dgd �|d �}t�||� | j �� }t� t �� |j dddd gd d� W d Q R X |�� }tt d�| j jdd�}tddddgdd ddgd �|d �}t�||� t dddddddddddg �}|�� }tdddg|jdd�}tdddgdddgd �|d �}t�||� t tjdddg�}|�� }tdddgd dd gd �tddtjgddgdd�d �}t�||� d S )Nr r r r )�namer r# r" g �?g �?)�countsZfreqs)r> r@ T)r �abcd)r r| r r! rA g]t�E�?gt�E]t�?g �?)r r| )�factorZdescriber r r r r Zassert_frame_equalrG r4 r5 rT r r r/ ra )r �descZ exp_indexr8 r r r r � test_describe� s> z*TestCategoricalAPIWithFactor.test_describec C sT | j �� }t�t�� |jddddgdd� W d Q R X t�|jtddddg�� d S )Nr r r r@ T)r ) r rG r r4 r5 rT r r r )r r r r r �test_set_categories_inplace� s z8TestCategoricalAPIWithFactor.test_set_categories_inplacec C sV t dddddddddddg �}|jd }t�t�� ||_W d Q R X |j|ksRt�d S )Nr! r" r# )r rW r r4 r5 r )r r Z new_codesr r r � test_codes_setter_deprecated� s z9TestCategoricalAPIWithFactor.test_codes_setter_deprecatedN)rv rw rx r� r� r� r r r r r{ � s 4 r{ c @ s� e Zd Zdd� Zej�dddgddgddgddgfddgddgddgddgfddgddgddgddgfddgddgddgddgfddddgddgdddgddddgfddd d gdddgddgddd d gfddd gdddgdddgddd gfddd gdddgdgd dd gfddd gdddgd gd d d gfddd gdddgg d d d gfd d gg ddgd d gfddgddgddgddgfg �d d � �Zdd� ZdS )�TestPrivateCategoricalAPIc C s� t ddddtjg�}tjdddddgdd �}t�|j|� tjt d d �� tjdddddgdd �|_W d Q R X |j}tjt d d �� d|d < W d Q R X d|d < tjdddddgdd �}t�|j|� d|j d < tjdddddgdd �}t�|j|� d S )Nr r r r r! r" rS rV )r$ zcan't set attribute)r z#assignment destination is read-onlyr; ) r r/ ra r0 r r- rU r r r r< rW )r r rZ rU r r r �test_codes_immutable� s $ z.TestPrivateCategoricalAPI.test_codes_immutablezcodes, old, new, expectedr r! r r r r" rS r@ c C sL t j|t jd�}t j|t jd�}t|�}t|�}t|||�}t�||� d S )N)r$ )r/ Z asanyarrayrV r r r r- )r rU rI rJ r8 r7 r r r �test_recode_to_categories s z3TestPrivateCategoricalAPI.test_recode_to_categoriesc C sR d}t �|�}t|�}t j|d ddt jd�}t|�}t|||�}t�||� d S )Ni� r! rS )r$ )r/ Zaranger Zint16r r r- )r �NrU rI r8 rJ r7 r r r � test_recode_to_categories_large* s z9TestPrivateCategoricalAPI.test_recode_to_categories_largeN) rv rw rx r� r ry rz r� r� r r r r r� � s $$" ( r� )rP �numpyr/ r Zpandasr r r r r Zpandas._testingZ_testingr Z pandas.core.arrays.categoricalr Z&pandas.tests.arrays.categorical.commonr r r{ r� r r r r �<module> s H
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка