U
|
ý°dT) ã @ s d dl Z d dlmZ d dlZe jdd Ze jg dgddgdddggdd d
|
Ze jd d Ze jd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z dS )9é N)Úarray_equivalentc C s$ t dddgdddgddd gd
|
¡S ) z+DataFrame with columns 'L1', 'L2', and 'L3'é é é é é é ÚAÚBÚC)ÚL1ÚL2ÚL3)ÚpdZ DataFrame© r r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_label_or_level_utils.pyÚdf
|
s r r r r )Úparamsc C s | j }|r| |¡}|S )z;DataFrame with columns or index levels 'L1', 'L2', and 'L3')ÚparamÚ set_index)Úrequestr Úlevelsr r r Ú df_levels s
|
r c C s | ddg¡} | d | d< | S )z<DataFrame with levels 'L1' and 'L2' and labels 'L1' and 'L3'r r r )r ©r r r r Údf_ambig s r c C s& | dg¡} tj| | d gdd} | S )z9DataFrame with level 'L1' and labels 'L2', 'L3', and 'L2'r r r ©Úaxis)r r Úconcatr r r r Ú df_duplabels% s r c C s$ t | j}dd | jjD }||fS )Nc S s g | ]}|d k r|qS )Nr )Ú.0Únamer r r Ú
|
<listcomp>2 s z%get_labels_levels.<locals>.<listcomp>)ÚlistÚcolumnsÚindexÚnames)r Úexpected_labelsÚexpected_levelsr r r Úget_labels_levels0 s
|
r( c C sD |D ]:}| j ||dst| j||dr,t| j||dstqd S ©Nr )Ú_is_label_referenceÚAssertionErrorÚ_is_level_referenceÚ _is_label_or_level_reference)ÚframeÚlabelsr Úlabelr r r Úassert_label_reference6 s r1 c C sD |D ]:}| j ||dst| j||dr,t| j||dstqd S r) )r, r+ r* r- )r. r r Úlevelr r r Úassert_level_reference= s r3 c C sD | |¡}t| \}}|dkr$| j} t| ||d t| ||d d S ©Nr r )Ú_get_axis_numberr( ÚTr3 r1 ©r r r& r' r r r Ú*test_is_level_or_label_reference_df_simpleF s
|
r8 c C sL | |¡}|dkr| j} t| dg|d t| dg|d t| dg|d d S )Nr r r r r )r5 r6 r1 r3 ©r r r r r Ú test_is_level_reference_df_ambigT s
|
r: c C s^ | d¡j}t|dgdd | d¡r*t| ddg¡j}t|ddgdd | d¡rZtd S ©Nr r r r r )r r r3 r, r+ r ©r Úsr r r Ú+test_is_level_reference_series_simple_axis0i s r> c C s8 | d¡j}tjtdd |jddd W 5 Q R X d S ©Nr zNo axis named 1©Úmatchr r )r r ÚpytestÚraisesÚ
|
ValueErrorr, r< r r r Ú*test_is_level_reference_series_axis1_erroru s rE c C sn | |¡}|dkr | j} d}nd}tjt|d | jd|d W 5 Q R X | jd|d | jd|drjtd S ) Nr z.'L1' is both a column level and an index labelz.'L1' is both an index level and a column labelr@ r r r r )r5 r6 rB rC rD Ú _check_label_or_level_ambiguityr+ )r r Úmsgr r r Ú&test_check_label_or_level_ambiguity_df s
|
rH c C sf | d¡j}|jddd |jddd | ddg¡j}|jddd |jddd |jddd d S r; )r r rF r r< r r r Ú*test_check_label_or_level_ambiguity_series s rI c C s8 | d¡j}tjtdd |jddd W 5 Q R X d S r? )r r rB rC rD rF r< r r r Ú6test_check_label_or_level_ambiguity_series_axis1_error© s rJ c C sT | |¡}|D ]@}|dkr&| | j}n | j| j}| j||d}t||stqd S )Nr r )r5 Ú_valuesÚlocÚ_get_label_or_level_valuesr r+ )r. r/ r r0 ÚexpectedÚresultr r r Úassert_label_values³ s
|
rP c C s^ | |¡}|D ]J}|dkr,| jj|dj}n| jj|dj}| j||d}t||stqd S )Nr )r2 r )r5 r$ Zget_level_valuesrK r# rM r r+ )r. r r r2 rN rO r r r Úassert_level_values¿ s
|
rQ c C sD t | \}}| |¡}|dkr$| j} t| ||d t| ||d d S r4 )r( r5 r6 rP rQ r7 r r r Ú(test_get_label_or_level_values_df_simpleÍ s
|
rR c C s< | |¡}|dkr| j} t| dg|d t| dg|d d S )Nr r r r )r5 r6 rQ rP r9 r r r Ú'test_get_label_or_level_values_df_ambigÛ s
|
|
rS c C sx | |¡}|dkr| j} t| dg|d t| dg|d |dkrFd}nd}tjt|d t| d g|d W 5 Q R X d S )
|
Nr r r r r z#The column label 'L2' is not uniquez"The index label 'L2' is not uniquer@ r )r5 r6 rQ rP rB rC rD )r r Z expected_msgr r r Ú+test_get_label_or_level_values_df_duplabelsè s
|
rT c C sB | d¡j}t|dgdd | ddg¡j}t|ddgdd d S )Nr r r r )r r rQ r r< r r r Ú+test_get_label_or_level_values_series_axis0 s rU c C s8 | d¡j}tjtdd |jddd W 5 Q R X d S r? )r r rB rC rD rM r< r r r Ú1test_get_label_or_level_values_series_axis1_error
|
s rV c C sh | |¡}|D ]T}| j||d}|dkrF|| jks6t||jksbtq|| jksTt||jkstqd S ©Nr r )r5 Ú_drop_labels_or_levelsr# r+ r$ )r. r/ r r0 Ú
|
df_droppedr r r Úassert_labels_dropped s
|
rZ c C sp | |¡}|D ]\}| j||d}|dkrJ|| jjks8t||jjksjtq|| jjksZt||jjkstqd S rW )r5 rX r$ r% r+ r# )r. r r r2 rY r r r Úassert_levels_dropped! s
|
r[ c C sl t | \}}| |¡}|dkr$| j} t| ||d t| ||d tjtdd | jd|d W 5 Q R X d S )Nr r únot valid labels or levelsr@ ÚL4) r( r5 r6 rZ r[ rB rC rD rX r7 r r r Ú test_drop_labels_or_levels_df0 s
|
r^ c C s | d¡j}t|dgdd tjtdd |jddd W 5 Q R X | ddg¡j}t|ddgdd tjtdd |jddd W 5 Q R X d S )Nr r r r\ r@ r] r )r r r[ rB rC rD rX r r< r r r Ú!test_drop_labels_or_levels_seriesC s r_ ) rB Zpandas.core.dtypes.missingr Zpandasr Zfixturer r r r r( r1 r3 r8 r: r> rE rH rI rJ rP rQ rR rS rT rU rV rZ r[ r^ r_ r r r r Ú<module> s>
|
|
|
|
|
|
|
|
|
|
