U
|
ý°drO ã @ s¦ d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
m Z m Z m Z d dlmZ ej dddg¡dd Zej d d
|
d g¡d d Zdd ZG dd dZdS )é N)ÚPerformanceWarning)Ú DataFrameÚ DatetimeIndexÚ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 5 Q R X tjt| d |j||d W 5 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' ú[d:\z\workplace\vscode\pyvenv\venv\Lib\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¡dddgtjddddd } | jd
|
d d dg ¡ } tj t
|
d d | ddg¡ W 5 Q R X d S )Nr r r
|
r ÚcZ2012ÚH)ÚfreqZperiods©Úcolumnsr r r únot found in axisr ) 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_keys5 s ýr< c
|
@ s, e Zd Zdd Zdd Zdd Zdd Zd d
|
d gd d
|
d ggZej
|
d e ed d gde ed dgde 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 Zej
|
dg i e g ¡eg d d eg eg g¡d d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Z ej
|
d.d/d0d1d2g¡ej
|
d3d4d5g¡d6d7 Z d8d9 Z ej
|
d:d;d;gg¡d<d= Z d>d? Z d@dA Z!dBdC Z"dDdE Z#ej
|
dFd dgdfd gdGfg¡dHdI Z$dGS )JÚTestDataFrameDropc C s6 t dddgdddgdddggdd d
|
gd d d gd}d\|j_|j_| d ¡}|jd dd}| ¡ | ¡ }}|jd dd}|d kst|jd ddd}|d ks¢t||||fD ]$}|jjdksÂt|jjdks®tq®t|jd d d gksìtd}tj t
|
|d | dg¡ W 5 Q R X tj t
|
|d |jdgdd W 5 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 r3 )ÚfirstÚsecond©ÚaxisT)Úinplace©rE rF rB rC z\['g'\] not found in axisr Úgr* ©r+ ©Úname©rE r+ )r r rK r3 r! r; Ú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_namesF sJ ý
|
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 5 Q R X tjtdd |jddd W 5 Q R X tjtd d | ddg¡ W 5 Q R X tjtdd |jddgdd W 5 Q R X tjtdd |jdddgdd W 5 Q R X t |jddd|¡ t |jddgdd|jdddgd d
f ¡ t |jdddd|¡ t |jddgddd|d g ¡ t tt t
|
dt
|
ddtddddgd}t |jddd|dg ¡ t |jdd
|
d|d ¡ t | g ¡|¡ | t d d d g¡}td|_ t |jd d d|jd gd d
f ¡ t |jd d gdd|jg d d
f ¡ t tj d d¡tdd}||jdk }|j||jdk jd d!}|d ks.tt ||¡ d S )"Nr r r r r )ÚAÚBrU rD rV r3 r z\[5\] not found in axisr r z\['C'\] not found in axisÚCz#\['C', 'D', 'F'\] not found in axisÚDÚFr* rI rL éýÿÿÿÚabcr
|
r ©r3 ÚXÚYÚrowsr T)r# rF )r r, r- r! Úlocr r r rN ÚzipÚrangeÚ set_indexr r3 r5 r6 r7 r r rM )rP ÚsimpleZnu_dfr& rS rQ r' r' r( Ú test_dropu sR ,, ÿ ÿ ÿ
|
&(zTestDataFrameDrop.test_dropc C sÚ t jdddgddgd}tddd gg|d
|
}|j ¡ s:ttd ddd gdd ddgdddd ggd}|jd ddgd d}| ¡ }|j ¡ rtt ||¡ |j
|
d dd}t t ¡ |j
|
d dd}W 5 Q R X t ||¡ d S )N)r
|
Ú )Úb1Úc1)Úb2Úc2r r/ r r r r r\ r
|
r> rg rh ri rj )r3 Údata)r r3 ÚvaluesrD ) r Ú from_tuplesr r3 Z _is_lexsortedrM Z pivot_tableÚ reset_indexr, r- r! Úassert_produces_warningr )rP Z lexsorted_miZ lexsorted_dfZnot_lexsorted_dfrS Ú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 ||¡ |jdgddjd gdd}|jdgd gd}t ||¡ d}tjt|d |jdd d W 5 Q R X tjt|d |jdd d W 5 Q R X d}tjt|d |jdd W 5 Q R X d S )Nr r r r r r r r
|
r r/ r> r? r@ rA r rD r\ )r# rE r z2Cannot specify both 'labels' and 'index'/'columns'r )r# r )r# r3 z>Need to specify at least one of 'labels', 'index' or 'columns')r r! r, r- r r Ú
|
ValueError)rP r& Zres1Zres2r" r' r' r( Útest_drop_api_equivalenceÅ s8 ý
|
z+TestDataFrameDrop.test_drop_api_equivalencer r r Úactualr
|
©rk r r r c C s¸ t |jtrdnd }t d¡}tjt|d |jd|dd W 5 Q R X tjt|d |j jd|dd W 5 Q R X |jdd|dd}t
|
||¡ |j jdd|dd}t
|
|j |¡ d S ) Nr z"['c'] not found in axis"r r/ ©r rE r r* )rE r r+ ) Ú
|
isinstancer r ÚreÚescaper r r r! ÚTr, r- )rP rt 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©r| r' r( Ú
|
<listcomp> s z:TestDataFrameDrop.test_drop_empty_list.<locals>.<listcomp>r ©r r! r, r- )rP r r| Zexpected_indexÚframer' r r( Útest_drop_empty_list s z&TestDataFrameDrop.test_drop_empty_listr r c C s. t jtdd t|d |¡ W 5 Q R X d S )Nr4 r r )r r r r r! )rP r r| r' r' r( Útest_drop_non_empty_list s z*TestDataFrameDrop.test_drop_non_empty_listÚempty_listlikezdatetime64[ns]©Zdtypec C sP ddgddgd}t dt dg}t||d}| ¡ }| |¡}t ||¡ d S )Nr r ÚoneÚtwo)Zcolumn_aZcolumn_bz
|
2021-01-01r )r r r; r! r, r- )rP r
rk r r& rS rp r' r' r( Ú2test_drop_empty_listlike_non_unique_datetime_index s
|
zDTestDataFrameDrop.test_drop_empty_listlike_non_unique_datetime_indexc 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Úroutine2rf ÚODÚresult1Zresult2ÚwxÚwyr r r\ r rD )r
|
rf rf )r r r )r r r rv )r r rf )r r rf ) Úsortedra r rm r r5 r6 r7 r! r, r- )rP ZarraysZtuplesr r& rp rS r' r' r( Útest_mixed_depth_drop) s2 ý
|
ÿz'TestDataFrameDrop.test_mixed_depth_dropc C s t ddddgtjdddgddddgd d
|
d d gd dddg¡ ¡ }|jddd}t d
|
d gdgtjddtjdfgdddgdd}t ||¡ d S )Nr r ç g ð?g @r
|
r r/ r r r r ©rU rV rW rX rU rV rW r rX )r r r r r2 )
|
r r5 Únanrc Z
|
sort_indexr! r rm r, r- ©rP r& rp rS r' r' r( Ú$test_drop_multiindex_other_level_nanJ 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ÍÌÌÌÌÌ @Úvar1Zvar2Zvar3Zvar4r\ r r r ) r ÚgroupbyÚsizer` rc r! r rn r Úisinr, r- )rP r& Zgrp_sizeZdrop_idxZidfrp rS r' r' r( Útest_drop_nonuniqueb s(
|
|
|
|
|
|
|
|
|
÷
|
ôz%TestDataFrameDrop.test_drop_nonuniquec C sÜ |}|j ddgdd}|jddddd g }t ||¡ |j d
|
gd d}|jddd d d dg }t ||¡ |jj ddgddd}|jddddd g j}t ||¡ |jj d
|
gdd d}|jddd d d dg j}t ||¡ d S )NÚbarZquxrB r r r r r r r rC r r é )rE r )r! r: r, r- rz )rP Z multiindex_dataframe_random_datar rp rS r' r' r( Útest_drop_level 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 ÚidrJ z2016-03-23 14:00z2016-03-23 15:00z2016-03-23 16:00z2016-03-23 17:00r Úabr2 ZtstampT)ÚappendZ 201603231600Fr )r r8 Ú to_datetimer r5 ZarangeZreshaperN rc r r Z is_uniquerM r! r` r, r- )rP ÚidxZidxdtr& Útsrp rS r' r' r( Ú"test_drop_level_nonunique_datetime s" ûÿ z4TestDataFrameDrop.test_drop_level_nonunique_datetimec
|
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)r1 r ru z2017-10-29 00:15:00)r r8 r9 Úlenr! r, Z assert_equal)
|
rP Zframe_or_seriesÚstartÚendr rk rp Zexpected_startZ expected_idxrS 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 r r r r r )r r )r r )
|
r r r r5 r6 r7 r! r r rM )rP r r& rp r' r' r( Útest_drop_preserve_names´ s ÿ z*TestDataFrameDrop.test_drop_preserve_namesÚ operationÚ__iadd__Ú__isub__Ú__imul__Ú__ipow__rF FTc C s t dtdi}| ¡ }td|d< |d }t d ¡F |rN|jdd|d n|jdd|d}t||d t ||¡ W 5 Q R X d S )Nr r r r rG )r rb r; r, ro r! Úgetattrr- )rP r° rF r& rS 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 r~ Újr r r r )r r r r! r, r- )rP r$ r& rp rS 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 Ú from_productr rb r« r! rm r, r- )rP r¹ r§ r& rp rS 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\ rD r r 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/ r> r? r r r rD r rW )r r5 r6 r7 Ztaker! r, r- )rP r& rS Zdf2rp r' r' r( Ú!test_drop_with_duplicate_columns2ð s
|
|
|
üÿ z3TestDataFrameDrop.test_drop_with_duplicate_columns2c C sf t ddddgi}|j}|jdgddd t |jtg dd¡ || ¡ 8 }t |jtg dd¡ d S ) Nr
|
r r r TrG Úobjectr )r r
|
r! r, rO r3 r Zmean)rP 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 tdtdg¡d}tjtdd |jddd W 5 Q R X d S )Nr r z labels \[5\] not found in levelr r r r )r r r» rb r r r r! )rP r& r' r' r( Ú(test_drop_level_missing_label_multiindex s z:TestDataFrameDrop.test_drop_level_missing_label_multiindexz
|
idx, levelNc C sd t dddtjgdd|d |¡}|jtdtjg|d}t dgdd|d |¡}t ||¡ d S )Nr r éd )r
|
r r r )r r8 ZNArc r! r r, r- )rP Zany_numeric_ea_dtyper§ r r& rp rS r' r' r( Útest_drop_index_ea_dtype s ÿþ
|
ÿþz*TestDataFrameDrop.test_drop_index_ea_dtype)%Ú__name__Ú
|
__module__Ú __qualname__rT re rq rs rk r ÚmarkÚ parametrizer rc r{ r r r5 Úarrayr r r r r r r r¢ r© r® r¯ r¶ r¸ r¼ r½ r¾ rÀ rÁ rà r' r' r' r( r= E sj /3 'üþ
|
|
úþ
|
!
|
|
ÿ
|
|
|
r= )rx Únumpyr5 r Z pandas.errorsr Zpandasr8 r r r r r r Zpandas._testingZ_testingr, rÇ rÈ r) r. r< r= r' r' r' r( Ú<module> s" þþ
|
|
|
