Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/__pycache__/test_categorical.cpython-37.pyc
Назад
B �5�g � @ sN d dl Zd dlmZ d dlZd dlmZmZmZ d dl m Z G dd� d�Z dS )� N)�CategoricalDtype)� Categorical� DataFrame�Seriesc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestCategoricalConcatc C s t tjddd��dd�dddgd �}t tjd dd��d d �ddgd �}d d dd ddd g}tt|��|d< tj||fdd|d�}t dddddddd dddddg ddd dddtjtjtjtjtjtjtjg d d dd d d ddd d dd dg d gd | d ��}|d � |d j �|d< t � ||� d S ) N� �int64)�dtype� � �a�b�c)�columns� � � �one�two�hr T)�axis� ignore_index�sort� � � � � � � � � � � � )r r r r ) r �np�arangeZreshaper r �pd�concat�nan�astyper �tm�assert_frame_equal)�selfr �df1�df2Z cat_values�res�exp� r2 �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/concat/test_categorical.py�test_categorical_concat s2 "" z-TestCategoricalConcat.test_categorical_concatc C s� dddg}t dddg�}tdddg�}tddd g�}tjt|�||gd|d �}|jd k}td d d g|d�}t�||� |jdk}td d d g|d�}t�||� |jdk}td d d g|d�}t�||� d S )N�cat�obj�numr r r r r r )r �keys�objectFT)�indexr �category)r r r' r( Zdtypesr+ �assert_series_equal)r- r: r5 r6 r7 �df�result�expectedr2 r2 r3 � test_categorical_concat_dtypes4 s z4TestCategoricalConcat.test_categorical_concat_dtypesc C s� dddddg}t dtjddg|d�d�}t dtjddg|d�d�}t dtjddg|d�d�}tj|||gdd�}tjddddg|d�}tddtjtjgtjddtjgtjtjddgd �dddg|d �}t�||� d S ) Nr r r r r )� categories)r: )r )r r r )r r: ) r r' ZCategoricalIndexr( r r% r) r+ r, )r- rA r r r r> Zexp_idxr1 r2 r2 r3 � test_concat_categoricalindexI s z2TestCategoricalConcat.test_concat_categoricalindexc C s8 t td�dd�}t td�dd�}t td��}tj||gdd�}t�||� t td�dd�}tj||gdd�}t�||� t td�d d d d d d gdd �}t�||g�}t�||� t tjd dd��}t td��}t||� t td���d��}t�||g�}tt�||g�t�||g�� t td���d��}t� ||� d S )N�abcr; )r ZabdZabcabdT)r Zabcabcr r r )r: r r r �aabbca�cab)�A�B) r �listr' r( r+ r< r% r& r r* r r, )r- �s�s2r1 r0 r r r/ r2 r2 r3 � test_categorical_concat_preserve_ s& "z6TestCategoricalConcat.test_categorical_concat_preservec C s t tjddd��}t td��}t||�ttd���d���d�}t� ||g�}tt� ||g�t� ||g��ttd���d���d�}t � ||� t|t |td�d �d���d�}t� ||g�}t� |j |j�t�d d �|j |j�t�d d �g�}t � ||� d S ) Nr r )r rD rE )rF rG rG Zabe)rA r )r )r r% r& rH r r* r Z set_indexr' r( r+ r, r Zset_axisr: r9 )r- r r r/ r> r? Zdf3r2 r2 r3 � test_categorical_index_preserver s& " z6TestCategoricalConcat.test_categorical_index_preserverc C sj t tjdddd��}t ddgdd�}tj||gd d �}t tjddd �tjd dd �ddg�}t�||� d S ) Nz 2017-01-01r z US/Pacific)Zperiods�tzr r r; )r T)r )rM z 2017-01-02)r r' Z date_ranger( � Timestampr+ r< )r- r r r> r? r2 r2 r3 �test_concat_categorical_tz� s z0TestCategoricalConcat.test_concat_categorical_tzc C s� t tdddgddd��}tddd gddd gd d �}tj||gdd �}t tdddtjgdd�tddtjd gdd�d��}t�||� d S )Nr r r r; rF )r �namer r r r rG )r: rP )r )r �float)rF rG )r r r' r( r% r) r+ Z assert_equal)r- r= Zserr> r? r2 r2 r3 �!test_concat_categorical_unchanged� s z7TestCategoricalConcat.test_concat_categorical_unchangedc C s� t ddddddgtd�d��}t|d �|d < |d j�d d d g� |dd� }|dd � }t�|d jj|d jj� t�|d jj|d jj� t� ||g�}t�|d jj|d jj� |� |�}t�|d jj|d jj� d S )Nr r r r r" r Zabbaae)�id� raw_graderT Zgrade�er r r ) r rH r r5 Zset_categoriesr+ Zassert_index_equalrA r' r( �append)r- r= r. r/ ZdfxZdfar2 r2 r3 � test_categorical_concat_gh7864� s z4TestCategoricalConcat.test_categorical_concat_gh7864N) �__name__� __module__� __qualname__r4 r@ rB rK rL rO rR rW r2 r2 r2 r3 r s % r ) �numpyr% Zpandas.core.dtypes.dtypesr Zpandasr' r r r Zpandas._testingZ_testingr+ r r2 r2 r2 r3 �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка