U
|
ý°d£ ã @ sr d dl Z d dlZd dlmZ d dlmZ dd Zdd Zdd Z d d
|
Z
|
d d Z d d Z dd Z dd Zdd Zdd Ze j dddg¡e j ddd id d d gfd dd d dd gfd d idd dgfi dd dgfd d idd dgfd d d!dd dgfg¡d"d# Ze j dddg¡e j ddd id d gfd dd d dgfd d idd gfi dd gfg¡d$d% Zd&d' Zd(d) ZdS )*é N)Ú
|
MultiIndexc C s dd | j D t|ks td S )Nc S s g | ]
|
}|j qS © ©Úname©Ú.0Úlevelr r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_names.pyÚ
|
<listcomp> s z%check_level_names.<locals>.<listcomp>)ÚlevelsÚlistÚAssertionError)ÚindexÚnamesr r r Úcheck_level_names s r c C s4 t jdddgddgd} | dd
j| jks0td S )N)ÚaÚb©é é )ÚcÚdÚxÚy©r r )r Ú from_tuplesr r )r r r r Útest_slice_keep_name s r c C s t dddgdddgdddgd¡} | d ¡} d
|
d g| jd < t dddd
|
gdddd gdddd gd¡}| d ¡}t | |¡ d S ) Nr r é é éûÿÿÿr é )r r Úzr! é é
|
)ÚpdZ DataFrameZ set_indexÚlocÚtmZassert_frame_equal)ÚresultZ df_expectedr r r Útest_index_name_retained s $
|
"ÿ
|
r( c C s¦ dd | j D ddgks t| ¡ }| ¡ }| ¡ }dd | jD }|| _t| ddg t|ddg t|ddg t|ddg dd |jD |_t| ddg d S ) Nc S s g | ]
|
}|j qS r r r r r r r
|
s z'test_changing_names.<locals>.<listcomp>ÚfirstÚsecondc S s g | ] }|d qS )r r ©r r r r r r
|
% s ZfirstaZsecondac S s g | ] }|d qS )r r r+ r r r r
|
/ s )r r ÚviewÚcopyZ_viewr r )Úidxr, r- Z shallow_copyZ new_namesr r r Útest_changing_names s r/ c C s$ | dddg¡}|j| jks td S )Né r r )Ztaker r )r. Ztakenr r r Útest_take_preserve_name3 s r1 c C s. t jddgddgd} | ¡ }| |¡s,t| jddgks>t|jddgksPt| jddgd}| |¡snt| jddgkst|jddgkst| jddgd}| |¡s°t| jddgksÂt|jddgksÔttjtd d
|
| jd gd W 5 Q R X tjt d d
|
| jd gd ggd W 5 Q R X d S )Nr )r0 é ZMyName1ZMyName2r ZNewName1ZNewName2r z$Length of new names must be 2, got 1©ÚmatchZmarioz'MultiIndex.name must be a hashable typeZluigi)
|
r r r- Úequalsr r ÚpytestÚraisesÚ
|
ValueErrorÚ TypeError)Z multi_idxZ
|
multi_idx1Z
|
multi_idx2Z
|
multi_idx3r r r Útest_copy_names8 s" r: c C s4 |ddgkst dd | jD }||ks,t | }tjtdd t|dt|jdg W 5 Q R X tjtdd t|dg W 5 Q R X | j\}}| j\}}tjtdd t ||g||gdgd W 5 Q R X tjtdd" t ||g||gdddgd W 5 Q R X d
|
d g|_d d |jD }|d
|
d gks0t d S ) Nr) r* c S s g | ]
|
}|j qS r r r r r r r
|
Y s z test_names.<locals>.<listcomp>z^Length of namesr3 r Úthird)r Úcodesr r r c S s g | ]
|
}|j qS r r r r r r r
|
u s )
|
r r r6 r7 r8 Úsetattrr r r< r )r. Z index_namesZ level_namesr Z
|
major_axisZ
|
minor_axisZ major_codesZ minor_codesr r r Ú
|
test_namesV s2 "
|
|
ýý
|
r> c C s2 ddg| _ tjtdd | d¡ W 5 Q R X d S )NÚfooz name foo occurs multiple timesr3 )r r6 r7 r8 Z_get_level_number©r. r r r Ú(test_duplicate_level_names_access_raisesy s
|
rA c C sH t jdgddggddgd} | jd jdks0t| jd jdksDtd S )Nr r r r r r )r Ú from_productr r r r@ r r r Útest_get_names_from_levels s rC c C s t jdgddggddgd} tjtdd d| jd _W 5 Q R X tjtdd d| jd _W 5 Q R X tjd| jd d
|
}tjtdd d |j _W 5 Q R X tj
|
j d ks®t tj j d ks¾t d S ) Nr r r r r Ú set_namesr3 r? r )r ÚbarF)r rB r6 r7 Ú RuntimeErrorr r r$ ZSeriesr ÚIndexZ_no_setting_namer Z
|
RangeIndex)r. Únewr r r Ú%test_setting_names_from_levels_raises s rI ÚfuncÚrenamerD zrename_dict, exp_namesr r! r )r r r r )r r c C sd t jddgddgddggdddgd }t|| |}t jddgddgddgg|d }t ||¡ d S )
|
Nr r r0 r2 r r r r r ©r Ú from_arraysÚgetattrr& Zassert_index_equal©rJ Z rename_dictZ exp_namesÚmir' Úexpectedr r r Ú+test_name_mi_with_dict_like_duplicate_names s & rR c C sV t jddgddggddgd}t|| |}t jddgddgg|d}t ||¡ d S )Nr r r0 r2 r r r rL rO r r r Útest_name_mi_with_dict_like« s rS c C s> t ddg¡} d}tjt|d | ddi¡ W 5 Q R X d S )Nr r z2Can only pass dict-like as `names` for MultiIndex.r3 r r! )r$ rG r6 r7 r9 rD )ÚixÚmsgr r r Ú&test_index_name_with_dict_like_raising½ s rV c C sR t jddgddggddgd} tjtdd | jdd
|
idd
|
id W 5 Q R X d S ) Nr r r0 r2 r r r z(Can not pass level for dictlike `names`.r3 r! )r r )r rM r6 r7 r9 rD )rP r r r Ú&test_multiindex_name_and_level_raisingÅ s rW )r6 Zpandasr$ r Zpandas._testingZ_testingr& r r r( r/ r1 r: r> rA rC rI ÚmarkZ parametrizerR rS rV rW r r r r Ú<module> sJ # úþ
|
üþ
|
