Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/__pycache__/test_take.cpython-37.pyc
Назад
B �5�g�. � @ s" d dl m Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m m Z ejddgd�dd� �Zejejdfejdfejdfejdfejdfejdfejdfejdfejdfejdfejdfejdfg d�d d � �Z ejeje�d �ejfeje�d �ejfejd ejfejdejfejdej fejdejfejdejfejd ejfejdejfejdej fejdejfejdejfej d ej fej dej fej dej fej dejfej dejfejd ejfejdejfejdejfejdejfejdejfgd�dd� �Z G dd� d�Z G dd� d�Z dS )� )�datetimeN)�iNaTTF)�paramsc C s | j S )N)�param)�request� r �L/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/test_take.py� writeable s r c C s | j S )N)r )r r r r �dtype_can_hold_na s r � � � g @y @ @� c C s | j S )N)r )r r r r �dtype_fill_out_dtype) s r c @ sv e Zd Ze�d�Zdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� ZdS )�TestTakez Incompatible type for fill_valuec C s� |\}}}t j�ddd��|�}ddddg}tj|||d�}|dddg |dddg k�� s`t�|d |kspt�|j|ks~t�ddddg}tj|||d�}|ddddg || k�� s�t�|j|ks�t�d S )Nr � � r ���)� fill_value� ) �np�random�randint�astype�algos�take_nd�all�AssertionError�dtype)�selfr r r � out_dtype�data�indexer�resultr r r �test_1d_fill_nonnaK s $ zTestTake.test_1d_fill_nonnac C s� |\}}}t j�ddd��|�}ddddg}tj||d|d�}|dddgd d �f |dddgd d �f k�� srt�|dd d �f |k�� s�t�|j|ks�t�tj||d|d�}|d d �dddgf |d d �dddgf k�� s�t�|d d �df |k�� s�t�|j|k�st�ddddg}tj||d|d�}|ddddgd d �f ||d d �f k�� �s^t�|j|k�snt�tj||d|d�}|d d �ddddgf |d d �|f k�� �s�t�|j|k�s�t�d S )Nr r )� r r r )�axisr r ) r r r r r r r r r )r r r r r r! r" r# r r r �test_2d_fill_nonna[ s$ 4 4 22zTestTake.test_2d_fill_nonnac C s� |\}}}t j�ddd��|�}ddddg}tj||d|d�}|dddgd d �d d �f |dddgd d �d d �f k�� s~t�|dd d �d d �f |k�� s�t�|j|ks�t�tj||d|d�}|d d �dddgd d �f |d d �dddgd d �f k�� �st�|d d �dd d �f |k�� �s&t�|j|k�s6t�tj||d|d�}|d d �d d �dddgf |d d �d d �dddgf k�� �s�t�|d d �d d �df |k�� �s�t�|j|k�s�t�ddddg}tj||d|d�}|ddddgd d �d d �f ||d d �d d �f k�� �st�|j|k�s*t�tj||d|d�}|d d �ddddgd d �f |d d �|d d �f k�� �szt�|j|k�s�t�tj||d|d�}|d d �d d �ddddgf |d d �d d �|f k�� �s�t�|j|k�s�t�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 r r �test_3d_fill_nonnas s2 @"B$B$ >>>zTestTake.test_3d_fill_nonnac C sP t j�d��t j�}ddddg}t�||�}|�|�}t j|d< t � ||� d S )N� r r r r ) r r �randnr �float32r r �take�nan�tm�assert_almost_equal)r �arrr" r# �expectedr r r �test_1d_other_dtypes� s z TestTake.test_1d_other_dtypesc C s� t j�dd��t j�}ddddg}tj||dd�}|j|dd�}t j|d< t � ||� tj||dd�}|j|dd�}t j|d d �df< t � ||� d S ) Nr) r% r r r r r )r& ) r r r* r r+ r r r, r- r. r/ )r r0 r" r# r1 r r r �test_2d_other_dtypes� s z TestTake.test_2d_other_dtypesc C sl t jdddgtd�}t�|ddddg�}|�ddddg�}t�||� t�|dddg�}|jt j ksht �d S )Nr r )r r r ) r �array�boolr r r, r. �assert_numpy_array_equalr �object_r )r r0 r# r1 r r r � test_1d_bool� s zTestTake.test_1d_boolc C s� t jdddgdddgdddggtd�}t�|ddddg�}|jddddgdd�}t�||� tj|ddddgdd�}|jddddgdd�}t�||� t�|dddg�}|jt j ks�t �d S )Nr r )r r )r& r ) r r4 r5 r r r, r. r6 r r7 r )r r0 r# r1 r r r � test_2d_bool� s & zTestTake.test_2d_boolc C s� t j�dd��t j�}dddddg}tj||dd�}|j|dd�}t j|ddgd d �f< t � ||� tj||dd�}|j|dd�}t j|d d �ddgf< t � ||� d S )Nr r r r r r )r& ) r r r* r r+ r r r, r- r. r/ )r r0 r" r# r1 r r r �test_2d_float32� s zTestTake.test_2d_float32c C s^ t j�ddd�d }|jdd�}ddd d d g}tj||dd �}|j|dd �}t|�t j�dd gd d �f< t � ||� tj||dt d d d �d�}|j|dd �}t d d d �|dd gd d �f< t � ||� tj||d d �}|j|d d �}t|�t j�d d �dd gf< t � ||� tj||d t d d d �d�}|j|d d �}t d d d �|d d �dd gf< t � ||� d S )Ni �� i�Y� )r% r l h�] zdatetime64[ns])r r r r r )r& r i� )r& r ) r r r �viewr r r, r �int64r. r/ r )r r0 r" r# r1 r r r �test_2d_datetime64� s& zTestTake.test_2d_datetime64c C s� t �d��dd�}t�|ddg�}t �dddgdd d gg�}t�||� tj|ddgd dd �}t �dddgdddgg�}t�||� d S ) N� r r r r r r � r) � T)� allow_fillr )r �arange�reshaper r, r4 r. r6 )r r0 r# r1 r r r �test_take_axis_0� s zTestTake.test_take_axis_0c C s� t �d��dd�}tj|ddgdd�}t �ddgdd gd d gd d gg�}t�||� tj|ddgdddd�}t �ddgddgd dgd dgg�}t�||� tj t dd�� tj|ddgdddd� W d Q R X d S )Nr> r r r r r )r& r r% � � r? r@ T)r& rA r zindices are out-of-bounds)�match) r rB rC r r, r4 r. r6 �pytest�raises� IndexError)r r0 r# r1 r r r �test_take_axis_1� s " " zTestTake.test_take_axis_1c C s� t �dddg�}t �ddg�}tjtdd�� tj||ddgd� W d Q R X t jdddgtd �}tj||ddgd�}t jddggtd �}t� ||� d S ) Nr r r r zfill_value must be a scalar)rG T)rA r )r ) r r4 rH rI � ValueErrorr r, �objectr. r6 )r r0 r" r# r1 r r r �!test_take_non_hashable_fill_value s z*TestTake.test_take_non_hashable_fill_valueN)�__name__� __module__� __qualname__�re�compileZ fill_errorr$ r' r( r2 r3 r8 r9 r: r= rD rK rN r r r r r G s " r c @ sF e Zd Zdd� Zdd� Zej�dddg�dd � �Zd d � Z d d � Z dS )�TestExtensionTakec C sv t �ddg�}d}tjt|d�� tj|ddgdd� W d Q R X d}tjt|d�� tj|ddgd d� W d Q R X d S ) Nr r zindices are out-of-bounds)rG r T)rA z2index 2 is out of bounds for( axis 0 with)? size 2F)r r4 rH rI rJ r r, )r r0 �msgr r r �test_bounds_check_large s z)TestExtensionTake.test_bounds_check_largec C s� t jdddgt jd�}dddg}d}tjt|d �� tj||d d � W d Q R X t�||�}t jdddgt jd�}t� ||� d S ) Nr r r )r r r ���z7'indices' contains values less than allowed \(-2 < -1\))rG T)rA ) r r4 r< rH rI rL r r, r. r6 )r r0 r" rU r# r1 r r r �test_bounds_check_small) s z)TestExtensionTake.test_bounds_check_smallrA TFc C s` t jg t jd�}tj|g |d�}t�||� d}tjt |d�� tj|dg|d� W d Q R X d S )N)r )rA zHcannot do a non-empty take from an empty axes.|indices are out-of-bounds)rG r ) r r4 r< r r, r. r6 rH rI rJ )r rA r0 r# rU r r r �test_take_empty5 s z!TestExtensionTake.test_take_emptyc C s: t jt�g �ddgddd�}t�ddg�}t�||� d S )Nr Tg )rA r )r r, r r4 r. r6 )r r# r1 r r r �test_take_na_emptyC s z$TestExtensionTake.test_take_na_emptyc C s8 dddg}t �|ddg�}t�ddg�}t�||� d S )Nr r r r )r r, r r4 r. r6 )r r0 r# r1 r r r �test_take_coerces_listH s z(TestExtensionTake.test_take_coerces_listN) rO rP rQ rV rX rH �markZ parametrizerY rZ r[ r r r r rT s rT )!r rR �numpyr rH Z pandas._libsr Zpandas._testingZ_testingr. Zpandas.core.algorithms�coreZ algorithmsr Zfixturer Zfloat64r+ Zuint64Zuint32Zuint16Zuint8r< Zint32Zint16Zint8r7 Zbool_r Z complex128r r rT r r r r �<module> s\ U
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка