Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_drop.cpython-37.pyc
Назад
B �5�g&K � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m Z m Z m Z d dl mZ ej�dddg�dd� �Zej�d d d g�d d � �Zdd� ZG dd� d�ZdS )� N)�PerformanceWarning)� DataFrame�Index� MultiIndex�Series� Timestampzmsg,labels,level)z labels \[4\] not found in level� �a)z labels \[7\] not found in level� �bc C s� t jdddgdddggddgd �}td d d g|d �}td d d g|d �}tjt| d�� |j||d� W d Q R X tjt| d�� |j||d� W d Q R X d S )N� � � r � � r r )�names� � � )�index)�match)�level)r � from_arraysr r �pytest�raises�KeyError�drop)�msg�labelsr �mi�s�df� r" �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_drop.py�0test_drop_raise_exception_if_labels_not_in_level s "r$ z labels,level)r r )r r c C s� t jdddgdddggddgd �}td d d g|d �}td d d g|d �}|j| |dd�}t�||� |j| |dd�}t�||� d S )Nr r r r r r r r )r r r r )r �ignore)r �errors)r r r r r �tmZassert_series_equal�assert_frame_equal)r r r r r! Z expected_sZ expected_dfr" r" r# �test_drop_errors_ignore& s " r) c C sl t tj�dd�dddgtjdddd�d �} | jd d d dg �� } tj t d d �� | � ddg� W d Q R X d S )Nr r r r �cZ2012�H)�freqZperiods)�columnsr r r znot found in axis)r ) r �np�random�randn�pd� date_range�iloc�copyr r r r )r! r" r" r# �9test_drop_with_non_unique_datetime_index_and_invalid_keys4 s r5 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d d gd d d ggZej � d e ed d gd�e ed dgd�e ed dgd�� dd g�e ed d gd�� dd g�g�dd� �Z ej � dd d d gd d d gg�ej � dg d gd gg�dd� ��Zej � dd d d gd d d gg�ej � dd dgddgg�dd� ��Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zej � d%ee g�d&d'� �Zd(d)� Zej � d*d+d,d-d.g�ej � d/d0d1g�d2d3� ��Zd4d5� Zej � d6d7d7gg�d8d9� �Zd:d;� Zd<d=� Z d>d?� Z d@dA� Z dBdC� Z dDS )E�TestDataFrameDropc C s: t dddgdddgdddggdd d gd d d gd�}d\|j_|j_|�d �}|jd dd�}|�� |�� }}|jd dd�}|d ks�t�|jd ddd�}|d ks�t�x4||||fD ]$}|jjdks�t�|jjdks�t�q�W t|j�d d d gks�t�d}tj t |d�� |�dg� W d Q R X tj t |d�� |jdgdd� W d Q R X |jdgdd�} t dd d gdd�} t � | j| � |jd dgdd�} t dd gdd�} t � | j| � |jdgddd �} t d d d gdd�} t � | j| � |jd dgddd �} t d d gdd�} t � | j| � |jg dd�} t dd d gdd�} t � | j| � d S ) Nr r r r r r r r r r* �d�e�f)r r- )�first�second)�axisT)�inplace)r< r= r: r; z\['g'\] not found in axis)r �gr% )r& )�name)r< r& )r r r? r- r r4 �AssertionError�listr r r r r' �assert_index_equal) �selfr! Z df_dropped_bZ df_dropped_eZ df_inplace_bZ df_inplace_e� return_value�objr Zdropped�expectedr" r" r# �test_drop_namesE sH z!TestDataFrameDrop.test_drop_namesc C s t ddddgddddgd��}t�|jddd�|d g � t�|jdd gd d�|g � t�|jdddgdd�|jdgd d �f � t�|jddgd d�|jddgd d �f � tjtd d �� |�d� W d Q R X tjtdd �� |jddd� W d Q R X tjtd d �� |�ddg� W d Q R X tjtdd �� |jddgdd� W d Q R X t�|jddd�|� t�|jddgdd�|jdddgd d �f � t�|jdddd�|� t�|jddgddd�|d g � t tt t d�t dd�td���dddgd�}t�|jddd�|dg � t�|jdd d�|d � t�|�g �|� |� t dddg��}td�|_ t�|jddd�|jdgd d �f � t�|jddgdd�|jg d d �f � t tj�d d�td�d�}||jdk }|j||jdk jd d �}|d k�s t�t�||� d S ) Nr r r r r )�A�BrH )r< rI r- r z\[5\] not found in axis)r r z\['C'\] not found in axis�Cr% )r& )r< r& ����abcr r )r- �X�Y�rowsr T)r r= )r r' r( r �locr r r rA �zip�range� set_indexr r- r. r/ r0 r r r@ )rC �simpleZnu_dfr! rF rD r" r" r# � test_dropt sB ,, * * &(zTestDataFrameDrop.test_dropc C s� t jdddgddgd�}tddd gg|d �}|j�� s:t�td ddd gdd ddgdddd ggd�}|jd ddgd d�}|�� }|j�� r�t�t� ||� |j d dd�}t� t �� |j d dd�}W d Q R X t� ||� d S )N)r � )�b1�c1)�b2�c2r r* )r r r r )r- r r7 rW rX rY rZ )r- �data)r r- �values)r< ) r � from_tuplesr r- Z _is_lexsortedr@ Z pivot_table� reset_indexr' r( r �assert_produces_warningr )rC Z lexsorted_miZ lexsorted_dfZnot_lexsorted_dfrF �resultr" r" r# �"test_drop_multiindex_not_lexsorted� s & z4TestDataFrameDrop.test_drop_multiindex_not_lexsortedc C s� t dddgdddgdddggdd d gd d d gd�}|�d�}|jdd�}t�||� |jd dd�}|jd d�}t�||� |jd dd�}|jd d�}t�||� |jdgdd�}|jdgd�}t�||� |jdgdd�jd gdd�}|jdgd gd�}t�||� d}tjt|d�� |jdd d� W d Q R X tjt|d�� |jdd d� W d Q R X d}tjt|d�� |jdd� W d Q R X d S )Nr r r r r r r r r r* r7 r8 r9 )r r- )r )r< )r- )r r< r z2Cannot specify both 'labels' and 'index'/'columns')r )r r )r r- z>Need to specify at least one of 'labels', 'index' or 'columns')r r r' r( r r � ValueError)rC r! Zres1Zres2r r" r" r# �test_drop_api_equivalence� s6 z+TestDataFrameDrop.test_drop_api_equivalencer r r �actualr )r[ r r r c C s� t |jt�rdnd }t�d�}tjt|d�� |jd|dd� W d Q R X tjt|d�� |j jd|dd� W d Q R X |jdd|dd�}t � ||� |j jdd|dd�}t � |j |� d S ) Nr z"['c'] not found in axis")r r* )r r< r r% )r< r r& ) � isinstancer r �re�escaper r r r �Tr' r( )rC rd r r Zexpected_no_errr" r" r# �"test_raise_on_drop_duplicate_index� s z4TestDataFrameDrop.test_raise_on_drop_duplicate_indexr � drop_labelsc s8 � fdd�|D �}t |d��� �}t�|t |d�� d S )Nc s g | ]}|� kr|�qS r" r" )�.0�i)rj r" r# � <listcomp> s z:TestDataFrameDrop.test_drop_empty_list.<locals>.<listcomp>)r )r r r' r( )rC r rj Zexpected_index�framer" )rj r# �test_drop_empty_list s z&TestDataFrameDrop.test_drop_empty_listr r c C s. t jtdd�� t|d��|� W d Q R X d S )Nznot found in axis)r )r )r r r r r )rC r rj r" r" r# �test_drop_non_empty_list s z*TestDataFrameDrop.test_drop_non_empty_listc C s: ddddddgddddddgdd d dddgg}t t|� �}t�|�}ttj�d d �|d �}|jddd�}|jdgdd�}t � ||� |jdgdd�}|jdgdd�}|jdgdd�}t � ||� |jddd�}|jdgdd�}t � ||� |jdgdd�}|jddd�}|jdddd�}|jddgdd�}t � ||� d S )Nr �top�routine1�routine2rV �OD�result1Zresult2�wx�wyr r )r- r )r< )r rV rV )rq rt rv )rq rt rw )r r< )rr ru rV )rs ru rV ) �sortedrQ r r] r r. r/ r0 r r' r( )rC ZarraysZtuplesr r! r` rF r" r" r# �test_mixed_depth_drop s, z'TestDataFrameDrop.test_mixed_depth_dropc C s� t ddddgtjdddgddddgd d d d gd ���dddg��� }|jddd�}t d d gdgtjddtjdfgdddgd�d�}t� ||� d S )N�one�twog g �?g @r r r* r r r r )rH rI rJ �DrH rI rJ )r r| )rz g r )r )r- r ) r r. �nanrS Z sort_indexr r r] r' r( )rC r! r` rF r" r" r# �$test_drop_multiindex_other_level_nan0 s z6TestDataFrameDrop.test_drop_multiindex_other_level_nanc C s� t ddddgddddgdddd gdddd gd dd d gd dd d gd dd dgddddgddd dgg ddddgd�}|�d��� }|j|dk }|�dddg�}|j|jdd��� }||j� |j� }|j|_t � ||� d S )Nzx-a�xr g �?g333333�?zz-c�zr* g������@gffffff@zx-br gffffff@g������@zy-a�yzz-bg������ @�var1Zvar2Zvar3Zvar4)r- r r )r ) r �groupby�sizerP rS r r r^ r� �isinr' r( )rC r! Zgrp_sizeZdrop_idxZidfr` rF r" r" r# �test_drop_nonuniqueH s$ z%TestDataFrameDrop.test_drop_nonuniquec C sN t ddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd�}ttj�dd �|tdddgdd�d�}|jddgd d�}|jdd d ddg }t� ||� |jdgd d�}|jdd d dddg }t� ||� |j jddgd d d�}|jdd d ddg j }t� ||� |j jdgd d d�}|jdd d dddg j }t� ||� d S ) NZfoo�barZbazZquxrz r{ �threer r r r r: r; )Zlevels�codesr r rH rI rJ �exp)r? )r r- )r r r r � )r< r ) r r r. r/ r0 r r r3 r' r( rh )rC r rn r` rF r" r" r# �test_drop_levele s( . z!TestDataFrameDrop.test_drop_levelc C s� t dddddgdd�}t�ddd d d g�}tt�d ��dd�td �|d �}||d< |jddd�}t d �}|j j dksxt �|j |dd�}|j|dk }t�||� d S )Nr r r r �id)r? Z 201603231400Z 201603231500Z 201603231600Z 201603231700r �ab)r- r ZtstampT)�appendF)r )r r1 � to_datetimer r. ZarangeZreshaperA rS r r Z is_uniquer@ r rP r' r( )rC �idxZidxdtr! �tsr` rF r" r" r# �"test_drop_level_nonunique_datetime� s z4TestDataFrameDrop.test_drop_level_nonunique_datetime�boxc C s� t ddd�}t ddd�}tj||dd�}|dgt|� |d�}|�|�}t d dd�}tj||dd�}|dgt|� |d�} t�|| � d S ) Nz 2017-10-29z Europe/Berlin)�tzz2017-10-29 04:00:00Z15min)r, r )r[ r z2017-10-29 00:15:00)r r1 r2 �lenr r' Z assert_equal) rC r� �start�endr r[ r` Zexpected_startZ expected_idxrF r" r" r# �'test_drop_tz_aware_timestamp_across_dst� s z9TestDataFrameDrop.test_drop_tz_aware_timestamp_across_dstc C sd t jddddddgddddddggddgd�}ttj�dd�|d �}|�d g�}|jjd ks`t �d S ) Nr r r r rz r{ )r r )r )r r )rz r{ ) r r r r. r/ r0 r r r r@ )rC r r! r` r" r" r# �test_drop_preserve_names� s * z*TestDataFrameDrop.test_drop_preserve_names� operation�__iadd__�__isub__�__imul__�__ipow__r= FTc C s� t dtd�i�}|�� }td�|d< |d }t�d ��F |rN|jdd|d� n|jdd|d�}t||�d� t�||� W d Q R X d S )Nr r r� r )r< r= )r rR r4 r' r_ r �getattrr( )rC r� r= r! rF r� r" r" r# � test_inplace_drop_and_operation� s z1TestDataFrameDrop.test_inplace_drop_and_operationc C sd t �dddgdddgg�}tdddg|d�}|jdd�}tdgt �dgdgg�d�}t�||� d S ) Nr r� rl �jr r r )r )r r r r r' r( )rC r r! r` rF r" r" r# �$test_drop_with_non_unique_multiindex� s z6TestDataFrameDrop.test_drop_with_non_unique_multiindex�indexer)r r c C sj t �ddgddgg�}tdtt|��i|d�}|jdgd�}tdddgit �ddg�d�}t�||� d S ) Nr r r )r )r r r r )r r ) r � from_productr rR r� r r] r' r( )rC r� r� r! r` rF r" r" r# �*test_drop_tuple_with_non_unique_multiindex� s z<TestDataFrameDrop.test_drop_tuple_with_non_unique_multiindexc C s~ t dddgdddgdddggdddgd�}|jdgdd�}t dgdgdggdgd�}t�||� |jddd�}t�||� d S )Nr r g @r� r )r- )r< )r r r' r( )rC r! r` rF r" r" r# � test_drop_with_duplicate_columns� s ( z2TestDataFrameDrop.test_drop_with_duplicate_columnsc C s~ t tj�d�tj�d�tj�d�dddddgd��}|jdd d gd d �}|jd dd d d gd d �}|jd d d �}t�||� d S ) Nr r r r* r7 r8 )rH rI rJ r| r r )r< r rJ )r r. r/ r0 Ztaker r' r( )rC r! rF Zdf2r` r" r" r# �!test_drop_with_duplicate_columns2� s z3TestDataFrameDrop.test_drop_with_duplicate_columns2c C s\ t ddddgi�}d}tjt|d�� |�dd�}W d Q R X t dddgd�}t�||� d S ) Nr r r r ztIn a future version of pandas all arguments of DataFrame\.drop except for the argument 'labels' will be keyword-only)r r )r )r r' r_ � FutureWarningr r( )rC r! r r` rF r" r" r# � test_drop_pos_args_deprecation� s z0TestDataFrameDrop.test_drop_pos_args_deprecationc C sf t ddddgi�}|j}|jdgddd� t�|jtg dd�� ||�� 8 }t�|jtg dd�� d S ) Nr r r r T)r< r= �object)Zdtype)r r r r' rB r- r Zmean)rC r! r r" r" r# �.test_drop_inplace_no_leftover_column_reference� s z@TestDataFrameDrop.test_drop_inplace_no_leftover_column_referencec C sH t t�td�td�g�d�}tjtdd�� |jddd� W d Q R X d S )Nr )r z labels \[5\] not found in level)r r r )r )r r r� rR r r r r )rC r! r" r" r# �(test_drop_level_missing_label_multiindex s z:TestDataFrameDrop.test_drop_level_missing_label_multiindexN) �__name__� __module__� __qualname__rG rU ra rc r[ r �mark� parametrizer rS ri ro rp ry r~ r� r� r� r r� r� r� r� r� r� r� r� r� r� r" r" r" r# r6 D s@ // '& " $! r6 )rf �numpyr. r Z pandas.errorsr Zpandasr1 r r r r r Zpandas._testingZ_testingr' r� r� r$ r) r5 r6 r" r" r" r# �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.09 |
proxy
|
phpinfo
|
Настройка