Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/__pycache__/test_categorical.cpython-37.pyc
Назад
B �5�g�J � @ s~ d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m Z m Z m Z m Z mZmZ d dlmZ d dlmZ G dd� d�ZdS )� N)�is_categorical_dtype)� Categorical�CategoricalIndex� DataFrame�Index�Interval�Series� Timedelta� Timestamp)�CategoricalDtypec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zej�d)d*d+d,gd-d.d/gd0d1d2gd3d4d5gfd6d7� ejD �d8d7� ej D �e j d*d9d2ge d:�e d*d;�e d;d<�e d<d=�ge!d>d*d*�e!d>d+d*�e!d>d,d*�ge"d*d?�e"d+d?�e"d,d@�gfdAd7� ej#D �e$j%�&d*d;d<d=g�j e$j'dBd,dC�j e$j(dDd,dE�j f��dFdG� �Z)dHdI� Z*dJS )K�TestCategoricalIndexc C sh t dtjddd�ittd�ttd��dd�d �| _t dtjddd�ittd�ttd ��dd�d �| _d S ) N�A� �int64)�dtype�aabbca�cab�B)r �name)�index�cabe)r �np�aranger �list�CDT�df�df2)�self�method� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/test_categorical.py� setup_method s z!TestCategoricalIndex.setup_methodc C s, t td��}| jjd }ttd�dd��|�}|j|ks<t�tdddd git |�d �}t � ||� | j� � }d |jd< ttd �dd��|�}|j|ks�t�tdd d d ddd git |�d �}t � ||� t jtdd�� |jd W d Q R X |� � }|� � }|j�t�|_d|jd< d|jd< t � ||� d S )Nr �aZaaar )r r r � � )r � r � � � z^'d'$)�match�d� )r r r �locr �astyper �AssertionErrorr r �tm�assert_frame_equal�copy�pytest�raises�KeyErrorr �object)r r �resultZbidx�expectedr Zbidx2r r r r �test_loc_scalar' s, z$TestCategoricalIndex.test_loc_scalarc C s� | j }d|jd< |�� }d|jd< ttd�dd�}tdddd d d dd git|�d �}t�||� |�� }d|jd< tddd d d dtj gtj tj tj tj tj tj dgd�t|�d �}t�||� d S )Nr% r" r+ )r* r Zaabbcadr )r r r& r' r( g $@)r )r* �C)r r9 ) r r, r1 r r r r/ r0 r �nan)r r Zdf3Zbidx3Z expected3Zdf4r r r �,test_loc_setitem_with_expansion_non_categoryG s zATestCategoricalIndex.test_loc_setitem_with_expansion_non_categoryc C s* t jtdd�� | jjd W d Q R X d S )Nz^1$)r) r# )r2 r3 r4 r r, )r r r r �$test_loc_getitem_scalar_non_categoryd s z9TestCategoricalIndex.test_loc_getitem_scalar_non_categoryc C sP t tddddg��}|d d d� }tjddddgtjd�}t�|�� |� tdt� d�d � d �i�}t j |j d d d d dgd�|d< t dtd d �gddgdd�}|jd }t�||� tdt� dd�� d �it� dd�� d �d�}t j |j d d d d dgd�|d< |jdd� }t�||� t dtd d �gddgdd�}|jd }t�||� d S )Nr# r& r' r( ���)r �value�d r r � �2 �K )Zbins�D� r+ )r r � r% )r � � )r r r �arrayr r/ Zassert_numpy_array_equalZ __array__r r r- �pd�cutr> r �iloc�assert_series_equalr0 r, )r �cat�reverse�expr r7 r6 r r r � test_slicingh s$ z!TestCategoricalIndex.test_slicingc C s� t dddddddgdddgd�}tddddd d d g�}d d dddddg}t||d�|d�}t ddgdddgd�}tddg�}ddg}t||d�|d�}t||dd�} tddgddgddd�} d} |jd d�d d �f } t�| |� t| d j�s�t �|jd d d �f } t� | | � t | d t ��s&t �|jd d �df }t� || � t|j��sTt �|jd }|| k�slt �|j dd�d d �f } t�| |� t| d j��s�t �|j dd d �f } t� | | � t | d t ��s�t �|j d d �df }t� || � t|j��st �|j d }|| k�st �|j d|jd f }|| k�s<t �|jd }|| k�sTt �|jd }|| k�slt �|jd g }||d dk }t�||� ||d dk }t�||� |jd }|| k�s�t �|jd } t� | | � t | d t ��s�t �|jtd d� } t�| |� t| d j��s"t �|jd dg } t�| |� t| d j��sPt �|jd d �df }t� || � t|j��s~t �|jd d �tdd �f } t�| |� t| d j��s�t �|jd d �dd gf } t�| |� t| d j��s�t �d S ) Nr" �c�b)� categories�h�i�j�k�l�m�nr# r& r' r( r$ r � )�cats�values)r r\ )r r r] r5 )r r r r )r& r )rV r\ )r r r r rK r/ r0 r r r. rL � isinstance�strr, �columnsZiat�at�slice)r r\ �idxr] r Zcats2Zidx2Zvalues2Zexp_dfZexp_colZexp_rowZexp_valZres_dfZres_rowZres_colZres_valZ exp_fancyZ res_fancyr r r � test_slicing_and_getting_ops� s� z1TestCategoricalIndex.test_slicing_and_getting_opsc C sz t dddddddgdddgd�}tddddd d d g�}d d d d dddg}t||d�|d�}|jd d�d d �f }tt ddgdddgd�d d gd�ddgd�}t�||� |jd d�d d �f j}tddgddg�}t�||� |j dd�df }tt dddgdddgd�dddgdd�}t�||� |j dd�|j dd � f }tdt dddgdddgd�idddgd�}t�||� d S )Nr" rR rQ )rS rT rU rV rW rX rY rZ r# r& r' r( r$ )r\ r] )r �categoryr r\ r] )r r r ) r r r rK r/ r0 Zdtypesr rL r, r` )r r\ rc r] r r6 r7 r r r �test_slicing_doc_examples� s2 z.TestCategoricalIndex.test_slicing_doc_examplesc C s8 | j jddg }| j jddddg }tj||dd� d S ) NrQ r" r( r r# r$ T)�check_index_type)r r, rK r/ r0 )r r6 r7 r r r � test_loc_getitem_listlike_labels" s z5TestCategoricalIndex.test_loc_getitem_listlike_labelsc C s6 t jtt�d�d�� | jjdddg W d Q R X d S )Nz['e'] not in index)r) r" rR �e)r2 r3 r4 �re�escaper r, )r r r r �)test_loc_getitem_listlike_unused_category( s z>TestCategoricalIndex.test_loc_getitem_listlike_unused_categoryc C s* t jtdd�� | jjd W d Q R X d S )Nz^'e'$)r) ri )r2 r3 r4 r r, )r r r r �&test_loc_getitem_label_unused_category. s z;TestCategoricalIndex.test_loc_getitem_label_unused_categoryc C s4 t jtt�d�d�� | jjddg W d Q R X d S )Nz['d'] not in index)r) r" r* )r2 r3 r4 rj rk r r, )r r r r � test_loc_getitem_non_category3 s z2TestCategoricalIndex.test_loc_getitem_non_categoryc C sf | j �� }d|jd< |jdddg }ttd�td�dd�}td d d d d ddgi|d�}t�||� d S )Nr% ri r" rR Zaaabber r )rS r r r r# r$ r& r' )r )r r1 r, r r r r/ r0 )r r r6 � exp_indexr7 r r r �0test_loc_setitem_expansion_label_unused_category8 s zETestCategoricalIndex.test_loc_setitem_expansion_label_unused_categoryc C s� t dddg�}tdddgddd gd �|d �}|jddg }t ddg|jd �}tddgddgd �|d �}tj||d d� |jdddg }t dddg|jd �}tdddgdddgd �|d �}tj||d d� tjtt � d�d�� |jddg W d Q R X d S )Nr" rR rQ r# r& r' r( r$ r )r r )r )rS T)rg z['x'] not in index)r) �x) r r r, rS r/ r0 r2 r3 r4 rj rk )r r r �resro rO r r r �test_loc_listlike_dtypesA s z-TestCategoricalIndex.test_loc_listlike_dtypesc C s� t dddg�}tdddgdddgd �|d �}|jddg }tdddgdddgd �t dddg�d �}tj||d d � |jdddg }tdddddgdddddgd �t dddddg�d �}tj||d d � tjtt� d �d�� |jddg W d Q R X d S )Nr" rR r# r& r' r( r$ r )r r )r T)rg z['x'] not in index)r) rq ) r r r, r/ r0 r2 r3 r4 rj rk )r r r rr rO r r r �8test_loc_listlike_dtypes_duplicated_categories_and_codesX s & zMTestCategoricalIndex.test_loc_listlike_dtypes_duplicated_categories_and_codesc C s t ddddgtd�d�}tdddd gd d d d gd�|d�}|jddg }tdddgd d d gd�t dddgtd�d�d�}tj||dd� |jdddg }tdddddgd d d d d gd�t dddddgtd�d�d�}tj||dd� tjtt � d�d�� |jddg W d Q R X d S )Nr" rR rQ �abcde)rS r# r& r' r( r$ r r[ rG )r r )r T)rg z['x'] not in index)r) rq ) r r r r, r/ r0 r2 r3 r4 rj rk )r r r rr rO r r r �(test_loc_listlike_dtypes_unused_categoryo s " z=TestCategoricalIndex.test_loc_listlike_dtypes_unused_categoryc C s� t ddddgtd�d�}tdddd gd d d d gd�|d�}tjtdd�� |jd W d Q R X tjtt�d�d�� |jddg W d Q R X d S )Nr" rR rQ ru )rS r# r& r' r( r$ r r[ rG )r r )r ri )r) z['e'] not in index) r r r r2 r3 r4 r, rj rk )r r r r r r �9test_loc_getitem_listlike_unused_category_raises_keyerror� s "zNTestCategoricalIndex.test_loc_getitem_listlike_unused_category_raises_keyerrorc C s` t tj�dd�td�td�d�}|�� }t|j�|_t|j�|_t |j dd d �f |jdd�}t � |j dd d �f |� t |j d d �df |jdd�}t � |j d d �df |� ttd�dd d gd �}t |j dd gd d �f |j|d �}t � |j dd gd d �f |� ttd �dddgd �}t |j d d �ddgf |j|d�}t � |j d d �ddgf |� d S )Nr' �ABCZXYZ)r r` r )r r �XZABr r9 )rS )r` r ZXY�Y�Z)r r �random�randnr r1 r r r` r r, r/ rL r0 )r r �cdf�expectro � exp_columnsr r r �test_ix_categorical_index� s " "z.TestCategoricalIndex.test_ix_categorical_indexc C sl t tj�dd�td�td�d�}|�� }t|j�|_t|j�|_ttd�ddgd�}t |j dd d �f |j|d �}t � |j dd d �f |� ttd �d d gd�}t |j d d �d f |j|d�}t � |j d d �d f |� t |j ddgd d �f |jttd ��d �}t � |j ddgd d �f |� t |j d d �d d gf |jttd��d�}t � |j d d �d d gf |� d S )Nr' ZABAZXYX)r r` ZAAr r )rS )r` r ZXXry rz ZAABZXXY) r r r| r} r r1 r r r` r, r/ r0 )r r r~ ro r r� r r r �$test_ix_categorical_index_non_unique� s( z9TestCategoricalIndex.test_ix_categorical_index_non_uniquec C s` d}t jt|d�� | jjdd� W d Q R X | jjdd� }| jjddd g }t�||� d S ) NzRcannot do slice indexing on CategoricalIndex with these indexers \[1\] of type int)r) r# r$ rR rQ r& r' r( )r2 r3 � TypeErrorr r, rK r/ r0 )r �msgr6 r7 r r r �test_loc_slice� s z#TestCategoricalIndex.test_loc_slicec C s| t ddgddgddggtddd g�d �}|d }|jd dksBt�|jd dksTt�|jd dksft�|jd dksxt�d S ) Nr# r& r' r( r$ r r r r9 )r r )r r# )r r r, r. ra )r r �sr r r �&test_loc_and_at_with_categorical_index� s &z;TestCategoricalIndex.test_loc_and_at_with_categorical_index� idx_valuesr# r& r' r= ������g �?g @g @g ��g �g �c C s g | ]}t jd ddg|d��qS )r# r& r' )r )r rH )�.0r r r r � <listcomp>� s z TestCategoricalIndex.<listcomp>c C s g | ]}t jd ddg|d��qS )g �?g @g @)r )r rH )r� Zdtypr r r r� � s rR )r r( r rF i� r* rC c C s g | ]}t jd ddg|d��qS )r# r& r' )r )rI rH )r� r r r r r� � s z 2019-01-01)�periodsZ1d)�startr� c C s� t ||d�}tddddgi|d�}t|d |d �}|j|d }tdgdg|d d �}t�||� |j|d d � }tddg|d d � dgd �}t�||� |j| }tddg|d d � dgd �}t�||� |�� }d |j|d < tdd ddgi|d�}t�||� |�� }d d g|j|d d � df< tdd d dgi|d�}t�||� |�� }d d g|j|df< tdd d dgi|d�}t�||� d S )N)�orderedr Zfoo�barZbaz)r r r# )r r r& )r r` ZquxZqux2) r r rb r, r r/ rL r0 r1 )r r� r� Zcat_idxr �slr6 r7 r r r �+test_loc_getitem_with_non_string_categories� s0 z@TestCategoricalIndex.test_loc_getitem_with_non_string_categoriesc C s� t ddtjg�}ttd�|d�}|tj dks2t�|jtj dksFt�t|�}|jtjdf dksft�|jtj d dks~t�d S )Nr r r' )r r& r )r r r: r �ranger. r, r )r �ciZserr r r r �!test_getitem_categorical_with_nan s z6TestCategoricalIndex.test_getitem_categorical_with_nanN)+�__name__� __module__� __qualname__r! r8 r; r< rP rd rf rh rl rm rn rp rs rt rv rw r� r� r� r� r2 �markZ parametrizer/ ZALL_INT_DTYPESZ FLOAT_DTYPESr rH r5 r r r ZALL_EA_INT_DTYPESrI Z IntervalIndexZ from_breaksZ date_rangeZtimedelta_ranger� r� r r r r r sJ z' *r )rj �numpyr r2 Zpandas.core.dtypes.commonr ZpandasrI r r r r r r r r Zpandas._testingZ_testingr/ Zpandas.api.typesr r r r r r r �<module> s (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка