Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexes/categorical/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g�8 � @ s� d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m Z d dl m Z G dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZdS )� N)�InvalidIndexError)�CategoricalIndex�Index� IntervalIndex� Timestampc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestTakec C s� t dddgdd�}|�t�dddg��}t dddgdd�}t�||� t�|j|j� |jt�dddg�dd �}t ddtjgdddgdd �}t�||� t�|j|j� |jt�dddg�d dd �}t dddgdd�}t�||� t�|j|j� t t d �t d�ddd�}|�t�dddg��}t t d�t d�ddd�}t�||� t�|j|j� |jt�dddg�dd �}t ddtjgt d�ddd�}t�||� t�|j|j� |jt�dddg�d dd �}t t d�t d�ddd�}t�||� t�|j|j� d}t j t |d�� |jt�dddg�dd � W d Q R X t j t |d�� |jt�dddg�dd � W d Q R X d}t j t |d�� |�t�ddg�� W d Q R X d S )N� � � �xxx)�namer ���T)� fill_value)� categoriesr F)� allow_fillr ZCBA�ABC)r �orderedr ZBCA�B�CzJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)�match������z3index -5 is out of bounds for (axis 0 with )?size 3)r �take�np�array�tm�assert_index_equalZassert_categorical_equal�values�nan�list�pytest�raises� ValueError� IndexError)�self�idx�result�expected�msg� r) �d/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexes/categorical/test_indexing.py�test_take_fill_value sL $$z TestTake.test_take_fill_valuec C s� t jdddgdd�}t|�}|�t�dddg��}t jdddgdd�}t|�}t�||� |jt�dddg�d d �}t jddd gdd�}t �dddg�}t||d �}t�||� |jt�dddg�d d d�}t jdddgdd�}t|�}t�||� d}tj t |d�� |jt�dddg�d d � W d Q R X tj t |d�� |jt�dddg�d d � W d Q R X d}tj t |d�� |�t�ddg�� W d Q R X d S )Nz 2011-01-01z 2011-02-01z 2011-03-01r )r r r r T)r �NaT)r F)r r zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1)r r r z3index -5 is out of bounds for (axis 0 with )?size 3) �pdZ DatetimeIndexr r r r r r r r! r" r# )r$ r% r& r' Zexp_catsr( r) r) r* � test_take_fill_value_datetimeO s2 $$z&TestTake.test_take_fill_value_datetimec C s� t dddgdd�}dddg}d}tjt|d �� |j|dd � W d Q R X d }tjt|d �� |j||d � W d Q R X d }tjt|d �� |j|dd� W d Q R X d S )Nr r r �foo)r r r z1take\(\) got an unexpected keyword argument 'foo')r )r/ z$the 'out' parameter is not supported)�outz%the 'mode' parameter is not supportedZclip)�mode)r r r! � TypeErrorr r" )r$ r% �indicesr( r) r) r* �test_take_invalid_kwargsw s z!TestTake.test_take_invalid_kwargsN)�__name__� __module__� __qualname__r+ r. r4 r) r) r) r* r s >(r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestGetLocc C s t td�td�d�}ttd��}|�d�|�d�ks8t�|�d�|�d�ksPt�x4||gD ](}tjtdd�� |�d� W d Q R X qZW t td �td�d�}ttd ��}|�d �}t� ||�d �� t� |t � d d d d d d g�� |�d�}||�d�ks�t�|d k�s t�x6||gD ]*}tjtdd�� |�d� W d Q R X �qW t td�td�d�}ttd��}|�d�}||�d�k�s�t�|t ddd �k�s�t�|�d�}||�d�k�s�t�|t ddd �k�s�t�x6||gD ]*}tjtdd�� |�d� W d Q R X �q�W d S )NZabcde�edabc)r �a�ez 'NOT-EXIST')r z NOT-EXISTZaacded�dFT� Zaabbb�abcr r �b� z'c'�c) r r r �get_loc�AssertionErrorr r! �KeyErrorr �assert_numpy_array_equalr r �slice) r$ Zcidx1�idx1�iZcidx2�idx2�resZcidx3Zidx3r) r) r* � test_get_loc� s< zTestGetLoc.test_get_locc C s& t td��}|�d�}|dks"t�d S )Nr> r? r )r r rB rC )r$ �cidxr& r) r) r* �test_get_loc_unique� s z TestGetLoc.test_get_loc_uniquec C s2 t td��}|�d�}tddd �}||ks.t�d S )NZabbcr? r r )r r rB rF rC )r$ rL r& r' r) r) r* � test_get_loc_monotonic_nonunique� s z+TestGetLoc.test_get_loc_monotonic_nonuniquec C s< t td��}|�d�}tjddddgtd�}t�||� d S )NZabcbr? FT)�dtype)r r rB r r �boolr rE )r$ rL r& r' r) r) r* �#test_get_loc_nonmonotonic_nonunique� s z.TestGetLoc.test_get_loc_nonmonotonic_nonuniquec C s, t ddtjg�}|�tj�}|dks(t�d S )N�Ar r )r r r rB rC )r$ �cirJ r) r) r* �test_get_loc_nan� s zTestGetLoc.test_get_loc_nanN)r5 r6 r7 rK rM rN rQ rT r) r) r) r* r8 � s /r8 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestGetIndexerc C sj t td�td�d�}tjt|�tjd�}|�|�}t�||� t j t dd�� |j|dd� W d Q R X d S )N�cab)r )rO zInvalid fill method)r �invalid)�method) r r r Zarange�len�intp� get_indexerr rE r r! r" )r$ r% r' �actualr) r) r* �test_get_indexer_base� s z$TestGetIndexer.test_get_indexer_basec C s� t j�d� ttd�td�dd�}tt �|��}d}xVddd t|�gD ]B}|t jjd t|�|d � }t j t |d �� |� |� W d Q R X qFW x<td�td �gD ](}t j t |d �� |� |� W d Q R X q�W d S )Ni�[�aabbcarV F)r r z8Reindexing only valid with uniquely valued Index objectsr r r@ r )�size)r Zaababca) r �random�seedr r r r rY �randintr r! r r[ )r$ rS Zoidxr( �n�finderr) r) r* � test_get_indexer_requires_unique� s z/TestGetIndexer.test_get_indexer_requires_uniquec C s� t td�td�d�}t td��}xx|td�ttd��gD ]^}d}tjt|d�� |�|� W d Q R X |�|�\}}tj ddd d gtj d �}t � ||� q8W d S ) N�aabcder9 )r �abfz8Reindexing only valid with uniquely valued Index objects)r r r r r )rO ) r r r r r! r r[ Zget_indexer_non_uniquer r rZ r Zassert_almost_equal)r$ rG rI Zindexerr( �r1�_r' r) r) r* �test_get_indexer_non_unique� s z*TestGetIndexer.test_get_indexer_non_uniquec C s� t td�td�d�}t td��}d}tjt|d�� |j|dd� W d Q R X d }tjt|d�� |j|d d� W d Q R X d }tjt|d�� |j|d d� W d Q R X d S ) Nrf r9 )r rg z3method pad not yet implemented for CategoricalIndex)r �pad)rX z8method backfill not yet implemented for CategoricalIndexZbackfillz7method nearest not yet implemented for CategoricalIndexZnearest)r r r r! �NotImplementedErrorr[ )r$ rG rI r( r) r) r* �test_get_indexer_method s z&TestGetIndexer.test_get_indexer_methodc C sf t jtd�td�gtd�}td�td�g}t||ddd�}|�|�}t jddgd d�}t�||� d S ) Nz1999-12-31 00:00:00z2000-12-31 00:00:00)rO F�category)r r rO r r rZ )r r r �objectr r[ r rE )r$ ZarrZcatsrS r& r' r) r) r* �test_get_indexer_array s z%TestGetIndexer.test_get_indexer_arrayc C sP t ddgddgd�}|�t ddgddgd��}tjddgdd�}t�||� d S )Nr: r? )r r rZ )rO )r r[ r r r rE )r$ rS r& r' r) r) r* �+test_get_indexer_same_categories_same_order s z:TestGetIndexer.test_get_indexer_same_categories_same_orderc C sP t ddgddgd�}|�t ddgddgd��}tjddgdd�}t�||� d S )Nr: r? )r r rZ )rO )r r[ r r r rE )r$ rS r& r' r) r) r* �0test_get_indexer_same_categories_different_order% s z?TestGetIndexer.test_get_indexer_same_categories_different_orderN) r5 r6 r7 r] re rj rm rp rq rr r) r) r) r* rU � s rU c @ s6 e Zd Zej�deeej e j g�dd� �Z dd� Z dS )� TestWhere�klassc C s� t td�td�dd�}dgt|� }|}|�||��}t�||� dgdgt|�d }t tjg|dd � �� |j d�}|�||��}t�||� d S )Nr^ rV F)r r Tr )r ) r r rY �wherer r r r �tolistr )r$ rt rH Zcondr' r& r) r) r* � test_where/ s "zTestWhere.test_wherec C s~ t ddddg�}t�ddddg�}|�|d�}tddddgtd�}t�||� d }tj t |d �� |j �|d� W d Q R X d S ) Nr: r? rA r< TFr )rO z3Cannot setitem on a Categorical with a new category)r ) r r r ru r ro r r r r! r" �_data)r$ rS �maskr& r' r( r) r) r* �test_where_non_categories< s z#TestWhere.test_where_non_categoriesN)r5 r6 r7 r �mark� parametrizer �tupler r r- ZSeriesrw rz r) r) r) r* rs . s " rs c @ s� e Zd Zdd� Zdd� Zej�dddg�dd � �Zejjd e � d d �dfd e � dd�dfde d �dfe � d �dfge d�dd� �Zdd� ZdS )� TestContainsc C sd t td�td�dd�}d|ks"t�d|ks.t�d|ks:t�tj|ksHt�d|ksTt�d |ks`t�d S ) Nr^ �cabdefF)r r r: �zr; r r )r r rC r r )r$ rS r) r) r* � test_containsK s zTestContains.test_containsc C s. t td�tjg td�d�}tj|ks*t�d S )Nr^ r )r )r r r r rC )r$ rS r) r) r* �test_contains_nanX s z TestContains.test_contains_nan�unwrapTFc C s\ t jddd��dt j�}|�d�}||d }t|�}|}|rD|j}tj|ksRt �d |ks^t �t j|kslt �t� d�|ks~t �t� d�|ks�t �t|�}|r�|j}tj|ks�t �d |ks�t �t j|ks�t �t� d�|ks�t �t� d�|ks�t �t|�}|�r|j}tj|k�st �d |k�s t �t j|k�s0t �t� d�|k�sDt �t� d�|k�sXt �d S )Nz 2016-01-01�d )Zperiodsr �Dr r, ) r- Z date_range�insertr, Z to_periodr rx r r rC Z datetime64Z timedelta64) r$ r� Zdti�piZtdirS �obj�obj2Zobj3r) r) r* �test_contains_na_dtype\ s8 z#TestContains.test_contains_na_dtypezitem, expectedr r )g �?Tg �?g �?)r: F)Zidsc C s* t t�td���}||k}||ks&t�d S )Nr )r r Z from_breaks�rangerC )r$ �itemr' rS r& r) r) r* �test_contains_interval� s z#TestContains.test_contains_intervalc C sh t dddg�}d|kst�tjtdd�� dg|k W d Q R X tjtdd�� ddg|k W d Q R X d S )Nr r r r: zunhashable type)r r? )r rC r r! r2 )r$ r% r) r) r* �test_contains_list� s z TestContains.test_contains_listN)r5 r6 r7 r� r� r r{ r| r� r- ZIntervalr Z Timedelta�strr� r� r) r) r) r* r~ J s % r~ )�numpyr r Z pandas.errorsr Zpandasr- r r r r Zpandas._testingZ_testingr r r8 rU rs r~ r) r) r) r* �<module> s xI]
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка