Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/__pycache__/test_indexing.cpython-37.pyc
Назад
B �5�g�1 � @ s� d dl Zd dlZd dlmZmZmZmZmZm Z m Z m Z m Z d dl mZ d dlm mZ d dlmZ G dd� de�ZG dd� d�ZG dd � d �Zej�d d d g�d d� �Zej�d d d g�dd� �Z ej dd� �Z dd� Z dS )� N) � Categorical�CategoricalIndex�Index�Interval� IntervalIndex� PeriodIndex�Series� Timedelta� Timestamp)�TestCategoricalc @ s e Zd Zdd� Zdd� Zej�deddg�eddgddgd�g�d d � �Z ej�deddgddd gd�eddgddd gd�eddgdgd�eddgdgd�g�d d � �Z ej�deddg�eddgddgdd�eddgddd gdd�g�dd� �Z dd� Z dd� Z dS )�!TestCategoricalIndexingWithFactorc C s� | j d dkst�| j d dks$t�| j dddg }t�|jtjdddgtjd�� | j t�| j �dk }t�|jtjdddgtjd�� d S )Nr �a����c� � )�dtype) �factor�AssertionError�tm�assert_numpy_array_equal�_codes�np�array�int8Zasarray)�selfZsubf� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/categorical/test_indexing.py� test_getitem s z.TestCategoricalIndexingWithFactor.test_getitemc C s� | j �� }d|d< |d dks"t�d|d< |d dks:t�| j �� }tjt|�dd�}d|d< d|d< d||< tddddddddgdd �}t�||� d S ) N�br r r �bool)r Tr )�ordered) r �copyr r Zzeros�lenr r �assert_categorical_equal)r r �indexer�expectedr r r � test_setitem s z.TestCategoricalIndexingWithFactor.test_setitem�otherr r )� categoriesc C sR t ddgddgd�}t�ddg�}|| ||< t ddgddgd�}t�||� d S )Nr r )r) TF)r r r r r$ )r r( �target�maskr& r r r � test_setitem_same_but_unordered2 s zATestCategoricalIndexingWithFactor.test_setitem_same_but_unorderedr c C sP t ddgddgd�}t�ddg�}d}tjt|d�� || ||< W d Q R X d S )Nr r )r) TFzCCannot set a Categorical with another, without identical categories)�match)r r r �pytest�raises� ValueError)r r( r* r+ �msgr r r �'test_setitem_different_unordered_raises> s zITestCategoricalIndexingWithFactor.test_setitem_different_unordered_raisesT)r) r! c C sR t ddgddgdd�}t�ddg�}d}tjt|d�� || ||< W d Q R X d S )Nr r T)r) r! FzCCannot set a Categorical with another, without identical categories)r- )r r r r. r/ r0 )r r( r* r+ r1 r r r � test_setitem_same_ordered_raisesO s zBTestCategoricalIndexingWithFactor.test_setitem_same_ordered_raisesc C s. t dddg�}|d |d< |d dks*t�d S )N)r r )r r r r )r r )r �catr r r �test_setitem_tuple_ s z4TestCategoricalIndexingWithFactor.test_setitem_tuplec C s� t j�d� tt jjdddd��t j���dg�}t �dg��t j �}d||< |j t �dg��t j � }t � |t jdgdd �� d S ) Nr r � i�I )�sizei���i�� r )r ) r �random�seedr �randint�astyper Zadd_categoriesr �int64�codesr r )r r r% �resultr r r �test_setitem_listlikeg s z7TestCategoricalIndexingWithFactor.test_setitem_listlikeN)�__name__� __module__� __qualname__r r' r. �mark� parametrizer r, r2 r3 r5 r? r r r r r s$ * $ r c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d d gd d gg�dd� �Z ej� dd d d d gd d d d gd d d d gd d d d gg�ej� dd d gd dgd d gddgg�ej� de e g�ej� ddddg�dd� ����Z dd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )$�TestCategoricalIndexingc C s^ t dddddddg�}|d }|dks*t�|dd� }t ddgddddgd�}t�||� d S )Nr r r �d� r6 )r) )r r r r$ )r r4 Zslicedr& r r r �test_getitem_slicey s z*TestCategoricalIndexing.test_getitem_slicec C sn t j�d� tt jjdddd��t j��}|jt �dg��t j � }|t �dg��t j � j}t � ||� d S )Nr r r6 i�I )r7 i�� ) r r8 r9 r r: r; r r= r r< r r )r r r> r& r r r �test_getitem_listlike� s z-TestCategoricalIndexing.test_getitem_listlikec C sf t ddddddgdd�}t|�}t|� tjddddddgtjd �}t dddgdd�}t�|j|� t� |j |� t ddddddgdd�}t|d d �}t|� tjddddddgtjd �}t dddgdd�}t�|j|� t� |j |� t d d dddddgdd�}t|d d �} tjdddddddgtjd �}t dddddd d gdd�}t�| j|� t� | j |� d S )Nz2014-01z2014-02z2014-03�M)�freqr r r )r T)r! z2013-12z2013-11z2013-10z2013-09z2013-08z2013-07z2013-05� r6 � rG ) r r �strr r r r r r �assert_index_equalr) ) r Zidx1Zcat1Zexp_arrZexp_idxZidx2�cat2Zexp_idx2Zidx3Zcat3r r r �test_periodindex� sL z(TestCategoricalIndexing.test_periodindexc C s` t ddddg�}tjddddgtjd�}dddg|_t�|�� |� t�|jt dddg�� d S )Nr r r r r rG )r ) r r r r< r) r r � __array__rO r )r �s�expr r r �test_categories_assignments� s z3TestCategoricalIndexing.test_categories_assignments�new_categoriesr r rG rM c C s8 t ddddg�}d}tjt|d�� ||_W d Q R X d S )Nr r r zKnew categories need to have the same number of items as the old categories!)r- )r r. r/ r0 r) )r rV r4 r1 r r r �/test_categories_assignments_wrong_length_raises� s zGTestCategoricalIndexing.test_categories_assignments_wrong_length_raises� idx_values� key_valuesr6 � key_classr N�category�keyc C s� ||t dd�d�}|dkr |j}t||d�}|�|�\}}|�|�\} } t�|| � t�|| � |�� �|�} |�� �|�} t�| | � d S )Nr r6 )r) r\ )r )�ranger r Zget_indexer_non_uniquer r �uniqueZ get_indexer) r rX rY rZ r r\ �idxr& Zexp_missr> Zres_missZ exp_uniqueZ res_uniquer r r �test_get_indexer_non_unique� s z3TestCategoricalIndexing.test_get_indexer_non_uniquec C s� t tddg��}|�ddg�}t tdd gddgd��}t�||� t tddg��}|�ddg�}t td d gddgd��}t�||� d S )Nr r TF)r) )r r �wherer �assert_series_equal)r �serr> r& r r r �test_where_unobserved_nan� s z1TestCategoricalIndexing.test_where_unobserved_nanc C s\ t tdddgddddgd��}|jdddgdd�}t tdddg|jjd��}t�||� d S ) Nr r r rF )r) TF)r( )r r ra r4 r) r rb )r rc r> r& r r r � test_where_unobserved_categories� s z8TestCategoricalIndexing.test_where_unobserved_categoriesc C sr t tdddgddddgd��}tdddgddddgd�}|�dddg|�}t tdddg|jd��}t�||� d S ) Nr r r rF )r) TF)r )r r ra r r rb )r rc r( r> r& r r r � test_where_other_categorical� s z4TestCategoricalIndexing.test_where_other_categoricalc C sF t tdddg��}d}tjt|d�� |�dddgd� W d Q R X d S ) Nr r r z3Cannot setitem on a Categorical with a new category)r- TFrF )r r r. r/ r0 ra )r rc r1 r r r � test_where_new_category_raises s z6TestCategoricalIndexing.test_where_new_category_raisesc C sl t tdddgddddgdd��}tdddgddddgdd�}tjtdd�� |�dd dg|� W d Q R X d S ) Nr r r rF T)r) r! z without identical categories)r- F)r r r. r/ r0 ra )r rc r( r r r �!test_where_ordered_differs_rasies s z9TestCategoricalIndexing.test_where_ordered_differs_rasies)r@ rA rB rH rI rQ rU r. rC rD rW r r r` rd re rf rg rh r r r r rE x s 3& .$" rE c @ sf e Zd Zdd� Zejjdedd�dfdedd �d fd ed�d fe d�d fge d �d d� �Z dd� Z dS )� TestContainsc C s� t td�td�d�}d|ks t�d|ks,t�tj|ks:t�tjtdd�� dg|ksXt�W d Q R X d |ksnt�d|kszt�t td�tjg td�d�}tj|ks�t�d S ) NZaabbcaZcab)r) r �zzunhashable type: 'list')r- r r )r �listr r �nanr. r/ � TypeError)r r r r r � test_contains s zTestContains.test_containszitem, expectedr r T)g �?Tg �?g �?F)r F)Zidsc C s* t t�td���}||k}||ks&t�d S )NrG )r r Z from_breaksr] r )r �itemr& r4 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 rG r zunhashable type)r- r )r r r. r/ rm )r r4 r r r �test_contains_list9 s z TestContains.test_contains_listN) r@ rA rB rn r. rC rD r r r rN rp rq r r r r ri s ri �indexTFc C sZ t td��}tdddg�}| r&t|�}t�|�s4t�|| }||�d� }t� ||� d S )NrG TF�object) r r] r r �comZis_bool_indexerr r; r rb )rr rS r_ r> r& r r r �test_mask_with_booleanF s ru c C sL t td��}tddd g�}| r&t|�}|| }||�d� }t�||� d S )NrG TF)r r] r r Zfillnar rb )rr rS r_ r> r& r r r �*test_mask_with_boolean_na_treated_as_falseS s rv c c s6 ddd�}| � � �}|�td|� dV W dQ R X dS )z� Monkeypatch Categorical.__array__ to ensure no implicit conversion. Raises ------ ValueError When Categorical.__array__ is called. Nc S s t d��d S )NzI cannot be converted.)r0 )r r r r r r m s z(non_coercible_categorical.<locals>.arrayrR )N)�context�setattrr )Z monkeypatchr �mr r r �non_coercible_categoricala s rz c C s0 t dddg�}t|�}|jd }|dks,t�d S )Nr r r r )r r �atr )rz Zarrrc r> r r r �test_series_atu s r| ) �numpyr r. Zpandasr r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.common�core�commonrt Z&pandas.tests.arrays.categorical.commonr r rE ri rC rD ru rv Zfixturerz r| r r r r �<module> s , d 1
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка