Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/base/__pycache__/getitem.cpython-37.pyc
Назад
B �5�g: � @ s8 d dl Zd dlZd dlZd dlmZ G dd� de�ZdS )� N)�BaseExtensionTestsc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zejjd d d d gejd d d gd d!�e�d d d g�gd"d#d$gd%�d&d'� �Zejjd d d d ejgejd d d ejgd d!�gd"d#gd%�d(d)� �Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"ej�d<d=d>g�d?d@� �Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMS )N�BaseGetitemTestsz%Tests for ExtensionArray.__getitem__.c C sX t �|�}|jd d� }t �|d d� �}| �||� |jddddg }| �||� d S )N� r � � � )�pd�Series�iloc�assert_series_equal)�self�data�ser�result�expected� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/base/getitem.py�test_iloc_series s z!BaseGetitemTests.test_iloc_seriesc C s* t �|tjt|�dd�d��}t �d|d d� i�}|jd d�dgf }| �||� |jdddd gdgf }| �||� t j|d d� dd �}|jd d�df }| �||� |jd d�df }| �||� |jd d �d d d�f }| �||dg � |d dg jd d �d d d�f }| �||d g � d S ) N�int64)�dtype)�A�Br r r r r r )�namer ) r � DataFrame�np�arange�lenr �assert_frame_equalr r )r r �dfr r r r r �test_iloc_frame s z BaseGetitemTests.test_iloc_framec C s� t �d|i�}|jd d �d d �f }| �||� |jd d �d d�f }| �||� |jd d �d d�f }| �||� |jd d �d d d�f }| �||� |jd d �dd�f }| �||jd d �d d�f � |jd d �dd �f }| �||� d S )Nr r r r ���)r r r r )r r r r r r r � test_iloc_frame_single_block0 s z-BaseGetitemTests.test_iloc_frame_single_blockc C sX t �|�}|jd d� }t �|d d� �}| �||� |jddddg }| �||� d S )Nr r r r r )r r �locr )r r r r r r r r �test_loc_seriesF s z BaseGetitemTests.test_loc_seriesc C s� t �|tjt|�dd�d��}t �d|d d� i�}|jd d�dgf }| �||� |jddd dgdgf }| �||� t j|d d� dd �}|jd d�df }| �||� |jd d�df }| �||� d S ) Nr )r )r r r r r r r r )r ) r r r r r r" r r r )r r r r r r r r �test_loc_frameO s z BaseGetitemTests.test_loc_framec C s~ t �d|i�}t j|d gdgd|jd�}|jd }| �||� t j|d gdgt|�d |jd�}|jd }| �||� d S )Nr r )�indexr r r r )r r r r r" r r r )r r r r r r r r � test_loc_iloc_frame_single_dtypee s z1BaseGetitemTests.test_loc_iloc_frame_single_dtypec C s> |d }t ||jj�st�t�|�d }t ||jj�s:t�d S )Nr )� isinstancer �type�AssertionErrorr r )r r r r r r �test_getitem_scalart s z$BaseGetitemTests.test_getitem_scalarc C s |d }|||�st �d S )Nr )r) )r � data_missing�na_cmp�na_valuer r r r �test_getitem_scalar_na{ s z'BaseGetitemTests.test_getitem_scalar_nac C sL |g }t |�dkst�t|t|��s*t�|tjg dd� }| �||� d S )Nr r )r )r r) r' r( r �array� assert_extension_array_equal)r r r r r r r �test_getitem_empty s z#BaseGetitemTests.test_getitem_emptyc C s� t jt|�td�}|| }t|�dks*t�t|t|��s<t�t jt|�td�}t�|�| }t|�dkslt�|j |j ks|t�d|d< || }t|�dks�t�t|t|��s�t�t�|�| }t|�dks�t�|j |j ks�t�d S )N)r r Tr ) r �zerosr �boolr) r' r( r r r )r r �maskr r r r �test_getitem_mask� s z"BaseGetitemTests.test_getitem_maskc C sr t �ddg�}dt|�� �}tjt|d�� || W d Q R X tj|dd�}tjt|d�� || W d Q R X d S )NTFz-Boolean index has wrong length: 2 instead of )�match�boolean)r )r r/ r �pytest�raises� IndexErrorr )r r r4 �msgr r r �test_getitem_mask_raises� s z)BaseGetitemTests.test_getitem_mask_raisesc C s� t jtj|jdd�dd�}|| }t|�dks2t�t|t|��sDt�t � |�| }t|�dksbt�|j |j ksrt�d|d d�< |� dddd d g�}|| }| � ||� t � |�}t � |�| }| � ||� d S ) Nr3 )r r7 r T� r r r r )r r/ r r2 �shaper r) r' r( r r �taker0 r )r r r4 r r r r r � test_getitem_boolean_array_mask� s z0BaseGetitemTests.test_getitem_boolean_array_maskc C s� t jtj|jdd�dd�}t j|d d�< d|dd�< || }||�d� }| �||� t �|�}|| }||�d� }| � ||� d S )Nr3 )r r7 r Tr F) r r/ r r2 r> �NAZfillnar0 r r )r r r4 r r �sr r r �(test_getitem_boolean_na_treated_as_false� s z9BaseGetitemTests.test_getitem_boolean_na_treated_as_false�idxr r r ZInt64)r �listz integer-arrayz numpy-array)Zidsc C sn || }t |�dkst�t|t|��s*t�|�dddg�}| �||� t�|�}t�|�| }| �||� d S )Nr r r r ) r r) r' r( r? r0 r r r )r r rD r r r r r �test_getitem_integer_array� s z+BaseGetitemTests.test_getitem_integer_arrayc C s* d}t jt|d�� || W d Q R X d S )Nz9Cannot index with an integer indexer containing NA values)r6 )r8 r9 � ValueError)r r rD r; r r r �(test_getitem_integer_with_missing_raises� s z9BaseGetitemTests.test_getitem_integer_with_missing_raisesc C s@ |t d� }t|t|��s t�|t d� }t|t|��s<t�d S )Nr r )�slicer' r( r) )r r r r r r �test_getitem_slice� s z#BaseGetitemTests.test_getitem_slicec C s� |dd d �f }| � ||� |d d �df }| � ||� |dd d�f }| � ||d d� � |d d�df }| � ||d d� � |dd d d�f }| � ||d d d� � |d d d�df }| � ||d d d� � d S )N.r r )r0 )r r r r r r � test_getitem_ellipsis_and_slice� s z0BaseGetitemTests.test_getitem_ellipsis_and_slicec C s� t j|dd� tt|��D �d�}|�d�|jd ks8t�|�ddg�}|jddg }| �||� |�td��}|jdd g }| �||� |�d �d ks�t�|�|j � � d �d ks�t�t j|d d� t d �d�}|�d �|jd ks�t�|�td d��}|jd ddg }| �||� |�d�}|d k�s,t�|�d�|jd k�sFt�|�d �|jd k�s`t�|�t|��d k�sxt�t �|�}|d d d� }|�d �d k�s�t�d S )Nc S s g | ] }d | �qS )r r )�.0�ir r r � <listcomp> s z-BaseGetitemTests.test_get.<locals>.<listcomp>)r% r r � r r r r Zabcdef�c�b�d�Z) r r �ranger �getr r) r rI r% �maxrE )r r rB r r �s2r r r �test_get s. zBaseGetitemTests.test_getc C sZ t �|�dddg }|jd |d ks*t�|jd |d ks@t�|jd |d ksVt�d S )Nr r r r )r r r r) )r r r r r r �test_take_sequence/ s z#BaseGetitemTests.test_take_sequencec C s� |� ddg�}|j|jks t�|d |d ks2t�|d |d ksFt�|j ddgd|d�}|d |d ksnt�||d |�s�t�tjtdd�� |� t|�d g� W d Q R X d S )Nr r r T)� allow_fill� fill_valuez out of bounds)r6 )r? r r) r8 r9 r: r )r r r- r, r r r r � test_take5 s zBaseGetitemTests.test_takec C s� |d d� }|j dgdd�}||d |�s.t�d}tjt|d�� |� dg� W d Q R X tjtdd�� |� ddg� W d Q R X d S ) Nr r T)rZ z;cannot do a non-empty take from an empty axes|out of bounds)r6 zcannot do a non-empty taker )r? r) r8 r9 r: )r r r- r, �emptyr r; r r r �test_take_emptyB s z BaseGetitemTests.test_take_emptyc C sJ t |�}|�d| |d dg�}|�dd|d |d g�}| �||� d S )Nr r r )r r? r0 )r r �nr r r r r �test_take_negativeP s z#BaseGetitemTests.test_take_negativec C sX |d }|d }|j |||g|jd�}|jddg|dd�}|�ddg�}| �||� d S )Nr r )r r T)r[ rZ )�_from_sequencer r? r0 )r r+ r[ �na�arrr r r r r �test_take_non_na_fill_valueW s z,BaseGetitemTests.test_take_non_na_fill_valuec C s2 t jtdd�� |jddg|dd� W d Q R X d S )N� )r6 r ���T)r[ rZ )r8 r9 rG r? )r r r- r r r �&test_take_pandas_style_negative_raisesb s z7BaseGetitemTests.test_take_pandas_style_negative_raisesrZ TFc C sB |d d� }t jtdd�� |jt�ddg�|d� W d Q R X d S )Nr zout of bounds|out-of-bounds)r6 r )rZ )r8 r9 r: r? r Zasarray)r r rZ rc r r r � test_take_out_of_bounds_raisesf s z/BaseGetitemTests.test_take_out_of_bounds_raisesc C sb t �|�}|�ddg�}t j|j|d |t|�d g|jd�dt|�d gd�}| �||� d S )Nr r r )r )r% )r r r? ra r r r )r r rB r r r r r �test_take_seriesm s "z!BaseGetitemTests.test_take_seriesc C s� t �|�}|�dddg�}t j|�dddg�dddgd�}| �||� t|�}|�dd|g�}t j|j||d |g|jd�dd|gd�}| �||� |�||d g�}t j|j||g|jd�||d gd�}| �||� d S )Nr r r )r% r )r )r r �reindexr? r r ra r )r r r- rB r r r_ r r r � test_reindexv s "z BaseGetitemTests.test_reindexc C sn |d }|d }|j ||g|jd�}t�|�}|jdddg|d�}t�|j |||g|jd��}| �||� d S )Nr r )r r )r[ )ra r r r rj r )r r+ �validrb rc r r r r r r � test_reindex_non_na_fill_value� s z/BaseGetitemTests.test_reindex_non_na_fill_valuec C sd t �d|i�}|jdgdf }|jdks,t�|jjd jdksBt�t|jd�r`|jjjdks`t�d S )Nr r r �blocks) r r r" �ndimr) Z_mgr�arrays�hasattr�_block)r r r �resr r r � test_loc_len1� s z BaseGetitemTests.test_loc_len1c C s~ t �|�}|d d� �� }||d ks*t�d}tjt|d�� |d d� �� W d Q R X tjt|d�� |�� W d Q R X d S )Nr r z6can only convert an array of size 1 to a Python scalar)r6 )r r �itemr) r8 r9 rG )r r rB r r; r r r � test_item� s zBaseGetitemTests.test_itemc C s^ G dd� dt jj�}t �d|tjddgtd��i�}|jd d� }| �t |d jj �d� d S ) Nc s e Zd ZdZ� fdd�Z� ZS )zBBaseGetitemTests.test_ellipsis_index.<locals>.CapturingStringArrayz6Extend StringArray to capture arguments to __getitem__c s || _ t� �|�S )N)� last_item_arg�super� __getitem__)r ru )� __class__r r ry � s zNBaseGetitemTests.test_ellipsis_index.<locals>.CapturingStringArray.__getitem__)�__name__� __module__� __qualname__�__doc__ry � __classcell__r r )rz r �CapturingStringArray� s r� Zcol1ZhelloZworld)r r zslice(None, 1, None)) r rp Z StringArrayr r r/ �objectr Z assert_equal�strrw )r r� r �_r r r �test_ellipsis_index� s z$BaseGetitemTests.test_ellipsis_indexN)*r{ r| r} r~ r r r! r# r$ r& r* r. r1 r5 r< r@ rC r8 �markZ parametrizer r/ r rF rA rH rJ rK rX rY r\ r^ r` rd rg rh ri rk rm rt rv r� r r r r r sL * $# r )�numpyr r8 Zpandasr Z pandas.tests.extension.base.baser r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка