Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/series/accessors/__pycache__/test_cat_accessor.cpython-37.pyc
Назад
B �5�g�* � @ s� d dl Z d dlZd dlZd dlmZmZmZmZm Z m Z m Z m Z d dl mZ d dlmZmZmZ d dlmZ d dlmZ G dd� d�ZdS )� N)� Categorical� DataFrame�Index�Series� Timestamp� date_range� period_range�timedelta_range)� DatetimeArray� PeriodArray�TimedeltaArray)�CategoricalAccessor)� Propertiesc @ s� e Zd Zej�ddd� dd� dd� dd� dd� dd� d d� d d� g�d d � �Zd d� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd � Zd S ) �TestCatAccessor�methodc C s | j �dddg�S )N� � � )�cat�set_categories)�x� r �e/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/series/accessors/test_cat_accessor.py�<lambda> � zTestCatAccessor.<lambda>c C s | j jdddgdd�S )Nr r r T)�ordered)r �reorder_categories)r r r r r r c C s | j �dddg�S )Nr r r )r �rename_categories)r r r r r r c C s | j �� S )N)r �remove_unused_categories)r r r r r ! r c C s | j �dg�S )Nr )r Zremove_categories)r r r r r " r c C s | j �dg�S )N� )r Zadd_categories)r r r r r # r c C s | j �� S )N)r � as_ordered)r r r r r $ r c C s | j �� S )N)r � as_unordered)r r r r r % r c C s6 t dddgdd��d�}d}||�j}||ks2t�d S )Nr r r �A)�name�category)r �astyper# �AssertionError)�selfr �ser�expected�resultr r r �!test_getname_categorical_accessor s z1TestCatAccessor.test_getname_categorical_accessorc C s� t tddtjdg��}t�|jjtddg�� |jj r>t d��tddtjdgddgd�}tj t dd�� |jj ddgdd�}W d Q R X |d ks�t �t�|j|� |j� ddg�}t�|j|� d|d d �< |j�� }t�|jjtdg�� d S )N�a�bF)� categories)�check_stacklevelT)�inplace)r r �np�nan�tm�assert_index_equalr r. r r r& �assert_produces_warning� FutureWarningr Zassert_categorical_equal�valuesr )r' r( �exp� return_value�resr r r �test_cat_accessor/ s z!TestCatAccessor.test_cat_accessorc C sl t jtkst�t td���d�}t|jt�s0t�t dg�}tjt dd�� |j W d Q R X t |d�rht�d S )N�aabbcder$ r zonly use .cat accessor)�matchr ) r r r r& �listr% � isinstance�pytest�raises�AttributeError�hasattr)r' r( �invalidr r r �test_cat_accessor_apiD s z%TestCatAccessor.test_cat_accessor_apic C s8 t td���d�}tjtdd�� d|j_W d Q R X d S )Nr< r$ z You cannot add any new attribute)r= r, )r r> r% r@ rA rB r Zxlabel)r' r r r r �#test_cat_accessor_no_new_attributesP s z3TestCatAccessor.test_cat_accessor_no_new_attributesc C sx t td���d�}|jddd�}|d ks,t�tjtdd�� |jj dd�}W d Q R X |d ks`t�t |jj �dkstt�d S ) N�abcr$ r T)r0 F)r/ r ) r r> r% Zdropr& r3 r5 r6 r r �lenr. )r' r( r9 r r r �$test_cat_accessor_updates_on_inplaceV s z4TestCatAccessor.test_cat_accessor_updates_on_inplacec C s6 d}t jt|d�� tdddg�j W d Q R X t jt|d�� tdddg��� W d Q R X t jt|d�� tdddg�j W d Q R X t jt|d�� tt�d ��j W d Q R X t jt|d�� ttd �g�j W d Q R X ttddddgd d ��}t dddg�}t � |jj |� dddg|j_ t dddg�}t � |jj |� td ddd gdd�}t � |jj|� |jj�slt�|j�� }|jj�r�t�|jjd d�}|d k�s�t�|jj�s�t�ttddddgd d ��}t dddg�}tjddddgtjd�}|j�dddg�}t � |jj |� t �|j�� |� t �|�� |� ttddddgdddgd��}t ddg�}tjddddgtjd�}|j�� }t � |jj |� t �|j�� |� t �|�� |� d}t jt|d�� |�ddddg� W d Q R X ttddddgd d ��}|j�dd� �}ttddddgdddgd d��}t � ||� d S )Nz3Can only use \.cat accessor with a 'category' dtype)r= r r r r, r- �cg @Z20130101T)r r Zint8)�dtype)r0 )r. z1'Series' object has no attribute 'set_categories'r c S s | � � S )N)�upper)r r r r r � r z>TestCatAccessor.test_categorical_delegations.<locals>.<lambda>r" �B�C)r. r )r@ rA rB r r r1 Zaranger r r r3 r4 r. �assert_series_equal�codesr r& r! r �arrayZobject_r Zassert_numpy_array_equalr7 Z __array__r r ) r' �msgr( Zexp_categoriesZ exp_codesr9 Z exp_valuesr* r) r r r � test_categorical_delegationsa s^ z,TestCatAccessor.test_categorical_delegationsc s� t tdddd��}|�d�}t tdddd��}|�d�}t tdd ��}|�d�}d d � }d |t�||fd |t�||fd|t�||fg}t|j t �s�t �ddi fddi fddi fddi fddi fddi fg} dd� | D ��ddg� �xT|D �]J\} �} } � ��fd d�t | j �D �} d d� | D �}x,| D ]$}|d t | j �k�r&|� |� �q&W xn|D ]f\}}}t�� �@ |d k�r~t�d t� t| j |�||�}t| j |�||�}W d Q R X t�||� �qTW x4�D ],}|d!k�rq�t| j |�}t| j |�}�q�W t|t��rt�||� q�t|t ��r(t�||� q�t�||� q�W t d"d#d$g��d�}d%}tjt|d&�� |j W d Q R X t |d'��r�t �d S )(Nz1/1/2015� ZMET)�periods�tzr$ �D)�freqrU z1 daysz10 daysc S s | j S )N)Z_datetimelike_ops)r r r r r � r zFTestCatAccessor.test_dt_accessor_api_for_categorical.<locals>.<lambda>ZDatetimeZPeriodZ Timedelta�strftime)z%Y-%m-%dZ tz_convert)�EST�round)rW �floor�ceilZasfreqc S s g | ] }|d �qS )r r )�.0�fr r r � <listcomp>� s zHTestCatAccessor.test_dt_accessor_api_for_categorical.<locals>.<listcomp>Z tz_localize� componentsc s2 g | ]*}|� d �s|�ks|�ks|� ks|�qS )�_)� startswith)r^ r_ )� _ignore_names�_special_func_names� attr_namesr r r` � s c S s g | ]}|d i f�qS )r r )r^ r_ r r r r` � s r Z to_period�ignore)�weekZ weekofyearr r r z+Can only use .dt accessor with datetimelike)r= �str) r r r% r r r r r r? �dtr r& �dir�append�warnings�catch_warnings� simplefilter� UserWarning�getattrr3 Z assert_equalr Zassert_frame_equalrO Zassert_almost_equalr@ rA rB rC )r' Zs_drZc_drZs_prZc_prZs_tdrZc_tdrZget_opsZ test_dataZspecial_func_defsr# �srJ Z func_namesZ func_defsZf_def�func�args�kwargsr: r8 �attrrD rR r )rd re rf r �$test_dt_accessor_api_for_categorical� s` z4TestCatAccessor.test_dt_accessor_api_for_categoricalc C s� t dddgdd�}|j}|j�dddg�}tjtdd�� |jjdddgdd � W d Q R X |j�|jj�rnt �|j�|jj�r�t �|j|jks�t �|jj�|jj�s�t �t� ||� d S ) Nr, r- rJ r$ )rK z`inplace` parameter)r= T)r0 ) r rK r r r3 r5 r6 r. �equalsr& rO )r' r( Z orig_dtyper) r r r �%test_reorder_categories_updates_dtype� s z5TestCatAccessor.test_reorder_categories_updates_dtypec C s� t dddgdddgd�dd�}ddg|d j_d d g|d j_t|d �d d d gksXt�t|d �dddgksrt�t|d d d gd d �|d < t|d ddgd d �|d< t|d �d d d gks�t�t|d �dddgks�t�d S )Nr r )�Survived�Sexr$ )rK ZNoZYesrz ZfemaleZmaler{ F)r. r )r r r. r> r& r )r' Zdfr r r �test_set_categories_setitem s z+TestCatAccessor.test_set_categories_setitemN)�__name__� __module__� __qualname__r@ �markZ parametrizer+ r; rE rF rI rS rw ry r| r r r r r s$ EWr )rm �numpyr1 r@ Zpandasr r r r r r r r Zpandas._testingZ_testingr3 Zpandas.core.arraysr r r Z pandas.core.arrays.categoricalr Z pandas.core.indexes.accessorsr r r r r r �<module> s (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка