Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/__pycache__/test_getitem.cpython-37.pyc
Назад
B �5�g. � @ s� d dl Z d dlZd dlZd dlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZ d dlmZ d dlmZ G dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZdS )� N) � Categorical�CategoricalDtype�CategoricalIndex� DataFrame� DatetimeIndex�Index� MultiIndex�Series� Timestamp�concat� get_dummies� period_range)� SparseArrayc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestGetitemc C s^ t dddgddggddgddggd�}td td �|d �}tjtdd �� |d W d Q R X d S ) NZa_lotZonlyoneZ notevenonei� � � r )�levels�codes���� )�index�columns)�match)r r �range�pytest�raises�KeyError)�self�mi�df� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/indexing/test_getitem.py� test_getitem_unused_level_raises s z,TestGetitem.test_getitem_unused_level_raisesc C sv t ddd�}ttj�dd�|d�}||d }t�||jd d �df � t|� |d }t�||jd d �df � d S )Nz1/1/2000� )Zperiods� )r r ) r r �np�random�randn�tm�assert_series_equal�iloc�repr)r �rngr �tsr r r! �test_getitem_periodindex$ s z$TestGetitem.test_getitem_periodindexc C sZ t td�td�g�}tddgddggdddg|d�}t|�}|t|j� }t�||� d S )Nz 12-31-1999z 12-31-2000r r Zuint8)�dtyper r )r r r r �listr r( �assert_frame_equal)r Zcats�expectedZdummies�resultr r r! �1test_getitem_list_of_labels_categoricalindex_cols1 s z=TestGetitem.test_getitem_list_of_labels_categoricalindex_colsc C sx t ddg�}td|i�}t|dd�}|d }t�||� |jd d �df }t�||� |jd d �df }t�||� d S )Nr r �A)�name)r r r r( r) r* �loc)r �datar r2 r3 r r r! �0test_getitem_sparse_column_return_type_and_dtype< s z<TestGetitem.test_getitem_sparse_column_return_type_and_dtypeN)�__name__� __module__� __qualname__r" r. r4 r9 r r r r! r s r c @ s� e Zd Zdd� Zdd� Zdd� Zejjde e e e dd � d d � gd d d dddgd�ej�dddg�dd� ��Z dd� Zdd� ZdS ) �TestGetitemListLikec C sN t dgdgdgd��}dddg|_tjtdd�� |dd dg W d Q R X d S ) Ng �?g @g @)�x�y�zr> r@ z\['y'\] not in index)r r? )r r r r r )r r r r r! � test_getitem_list_missing_keyM s z1TestGetitemListLike.test_getitem_list_missing_keyc C sd t tj�dd�td�d�}d|j_|ddg }|jjdks>t�|jd d �dd �f }t � ||� d S )N� ZAABC)r �foo�B�C� ) r r% r& r' r0 r r6 �AssertionErrorr* r( r1 )r r r3 r2 r r r! � test_getitem_list_duplicatesW s z0TestGetitemListLike.test_getitem_list_duplicatesc C sT t dddgdddggdddgd �}d }tjtt�|�d �� |d g W d Q R X d S ) Nr rF r rB r# � �a�b)r z?"None of [Index(['baf'], dtype='object')] are in the [columns]")r Zbaf)r r r r �re�escape)r r �msgr r r! �test_getitem_dupe_colsb s "z*TestGetitemListLike.test_getitem_dupe_cols�idx_typec C s t t| tt| ����S )N)�dict�zipr �len)�lr r r! �<lambda>o � z TestGetitemListLike.<lambda>c C s t t| tt| ������ S )N)rQ rR r rS �keys)rT r r r! rU p rV r0 �iterr �setrQ � dict_keys)Zidsr r rF c C s� |dkr|d }}n(t tj�dd�tdddgdd �d �}d }|jd |jd g}||�}t||��}|| } |jd d �|f } |jj| j_t � | | � |||g �}t j t d d�� || W d Q R X d S )Nr �food� r )rC �bar)ZbazZqux)�peekZaboo)ZsthZsth2)r6 )r )Zgoodr[ r z not in index)r )r r% r& r' r r r0 r7 �namesr( r1 r r r ) r rP r � float_frame�frame�missingrW �idxZ idx_checkr3 r2 r r r! �test_getitem_listlikeh s$ z)TestGetitemListLike.test_getitem_listlikec C s` t dddgdddgd��}dd � d D �}|j| }t ddgddgd�ddgd �}t�||� d S ) Nr rF r rB r# rI )rJ rK c s s | ] }|V qd S )Nr )�.0r> r r r! � <genexpr>� s zBTestGetitemListLike.test_getitem_iloc_generator.<locals>.<genexpr>)r rF )r )r r* r( r1 )r r �indexerr3 r2 r r r! �test_getitem_iloc_generator� s z/TestGetitemListLike.test_getitem_iloc_generatorc C s\ t dddgdddgd��}dd � d D �}|j|df }tddgd ddgd �}t�||� d S ) Nr rF r rB r# rI )rJ rK c s s | ] }|V qd S )Nr )re r> r r r! rf � s zRTestGetitemListLike.test_getitem_iloc_two_dimensional_generator.<locals>.<genexpr>)r rF rK )r6 r )r r* r r( r) )r r rg r3 r2 r r r! �+test_getitem_iloc_two_dimensional_generator� s z?TestGetitemListLike.test_getitem_iloc_two_dimensional_generatorN)r: r; r< rA rH rO r �mark� parametrizer0 rX r rY rd rh ri r r r r! r= L s r= c @ s e Zd Zdd� Zdd� ZdS )�TestGetitemCallablec C s� |dd� }|j d d �df }t�||� |dd� }|j d d �ddgf }t�||j d d �ddgf � |d d� }|dd� }|jdd gd d �f }t�||� d S ) Nc S s dS )Nr5 r )r> r r r! rU � rV z;TestGetitemCallable.test_getitem_callable.<locals>.<lambda>r5 c S s ddgS )Nr5 rD r )r> r r r! rU � rV rD r c S s dddgS )NTFr )r> r r r! rU � rV r rF )r7 r( r) r1 r* )r r` r3 r2 r r r r! �test_getitem_callable� s z)TestGetitemCallable.test_getitem_callablec C s` t ddggddggd�}t dgdggd�}|d }t�||� |jd d �df }t�||� d S )Nr rF rJ rK )r )r r( r1 r7 )r r r2 r3 r r r! �%test_loc_multiindex_columns_one_level� s z9TestGetitemCallable.test_loc_multiindex_columns_one_levelN)r: r; r< rm rn r r r r! rl � s rl c @ s� e Zd Zdd� Zej�dddgddggdd gd d ggejd gejd gejdgddggfddgddggdd gd d ggejd gejd gdd gd d ggff�dd� �Z ej dd� �Z dd� Z dd� Z dd� Zd d � Zd S ) �TestGetitemBooleanMaskc C s� t dtjddd�itddddddgtdddgdd �d d �d �}t dtjddd�itddddddgtdddgd d �d d �d �}||jdk }|jg }t�||� ||jdk }|jg }t�||� ||jdk }|jdddg }t�||� ||jdk }|jdddg }t�||� ||jdk }|jdg }t�||� ||jdk }|jg }t�||� d}t j t |d�� ||jdk W d Q R X t j t |d�� ||jdk W d Q R X d S )Nr5 rI Zint64)r/ r rF r T)ZorderedrD )r/ r6 )r FrJ r rB z7Unordered Categoricals can only compare equality or not)r ) r r% �aranger r r r* r( r1 r r � TypeError)r Zdf3Zdf4r3 r2 rN r r r! �(test_getitem_bool_mask_categorical_index� sF z?TestGetitemBooleanMask.test_getitem_bool_mask_categorical_indexzdata1,data2,expected_datar rF r rB g �?rI � r\ g @g @g @g @g @r# c C sp t t�|��}t t�|��}t||gdd�}||dk }dd� t|�D �}t |�jddd�d�} t�|| � d S ) Nr )ZaxisrF c S s i | ]\}}t �|�|�qS r )r% �array)re �i�colr r r! � <dictcomp> s z`TestGetitemBooleanMask.test_getitem_bool_mask_duplicate_columns_mixed_dtypes.<locals>.<dictcomp>r )rF r )r )r r% rt r � enumerate�renamer( r1 ) r Zdata1Zdata2Z expected_dataZdf1�df2r r3 Zexdictr2 r r r! �5test_getitem_bool_mask_duplicate_columns_mixed_dtypes s zLTestGetitemBooleanMask.test_getitem_bool_mask_duplicate_columns_mixed_dtypesc C s, ddddg}t t�d��dd�|dd�}|S ) Nr5 rE �D� r rB �float64)r r/ )r r% rp �reshape)r Zdupsr r r r! � df_dup_cols$ s z"TestGetitemBooleanMask.df_dup_colsc C s4 |}d}t jt|d�� ||jdk W d Q R X d S )Nz$cannot reindex from a duplicate axis)r rI )r r � ValueErrorr5 )r r� r rN r r r! �;test_getitem_boolean_frame_unaligned_with_duplicate_columns* s zRTestGetitemBooleanMask.test_getitem_boolean_frame_unaligned_with_duplicate_columnsc C sj t t�d��dd�ddddgdd �}||jd k }|j|_|}||jd k }t�||� |j t |� d S ) Nr} r rB r5 rD rE r| r~ )r r/ rI ) r r% rp r rE r r( r1 �dtypes�str)r r� r r2 r3 r r r! �2test_getitem_boolean_series_with_duplicate_columns3 s " zITestGetitemBooleanMask.test_getitem_boolean_series_with_duplicate_columnsc C sf t t�d��dd�ddddgdd �}||d k }|j|_|}||d k }t�||� |j t|� d S ) Nr} r rB r5 rD rE r| r~ )r r/ rI ) r r% rp r r r( r1 r� r� )r r� r r2 r3 r r r! �1test_getitem_boolean_frame_with_duplicate_columnsC s " zHTestGetitemBooleanMask.test_getitem_boolean_frame_with_duplicate_columnsc C s" t � }||dk }t�||� d S )Nr )r r( r1 )r r rz r r r! �%test_getitem_empty_frame_with_booleanT s z<TestGetitemBooleanMask.test_getitem_empty_frame_with_booleanN)r: r; r< rr r rj rk r% �nanr{ Zfixturer� r� r� r� r� r r r r! ro � s B". ro c @ s e Zd Zdd� Zdd� ZdS )�TestGetitemSlicec C s� t �ddd�}t|�}|ddg \}}t j�dd�}|tk rL|d d �df }|||d �}|||� }|jdd � } t�|| � |j ||� }t�|| � d S ) Ng $@g I@rF r# � � r r )r � ) r% rp r r& r' r r* r( Z assert_equalr7 ) r Zframe_or_series�valuesr �start�endr8 �objr3 r2 r r r! �test_getitem_slice_float64] s z+TestGetitemSlice.test_getitem_slice_float64c C sh t dditddddg�d�}t�t�� |dd � }W d Q R X t dditdddg�d�}t�||� d S ) NrJ r z11.01.2011 22:00z11.01.2011 23:00z12.01.2011 00:00z2011-01-13 00:00)r z 2011-01-01z 2011-11-01)r r r( Zassert_produces_warning� FutureWarningr1 )r r r3 r2 r r r! �test_getitem_datetime_slicep s z,TestGetitemSlice.test_getitem_datetime_sliceN)r: r; r< r� r� r r r r! r� \ s r� )rL �numpyr% r Zpandasr r r r r r r r r r r r Zpandas._testingZ_testingr( Zpandas.core.arraysr r r= rl ro r� r r r r! �<module> s 8 4X
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка