U
|
ý°dó
|
ã @ s> d dl Zd dlZd dlmZmZ d dlmZ G dd dZ dS )é N)Ú DataFrameÚ
|
MultiIndexc @ s e Zd Zdd Zdd ZdS )ÚTestReorderLevelsc
|
C sø t dgdddgddggddddddgddddddgddddddggdd d
|
gd }tt d ¡t d ¡d |d}t ||¡}| dddg¡}t ||¡ | dd d
|
g¡}t ||¡ | dddg¡}t dddgddgdggddddddgddddddgddddddggd d
|
dgd }tt d ¡t d ¡d |d}t ||¡}t ||¡ | dddg¡}t dgdgdggddddddgddddddgddddddggdddgd }tt d ¡t d ¡d |d}t ||¡}t ||¡ | dddg¡}t ||¡ d S )NÚbarZoneZtwoZthreer é é ZL0ZL1ZL2)ÚlevelsÚcodesÚnamesé )ÚAÚB)Úindex)r r ÚnpZarangeÚtmÚget_objÚreorder_levelsZ assert_equal)ÚselfZframe_or_seriesr ZdfÚobjÚresultZe_idxÚexpected© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_reorder_levels.pyÚtest_reorder_levels s> ,ý ,ý ,ý z%TestReorderLevels.test_reorder_levelsc C s |}| dddg¡}| dd¡ dd¡}t ||¡ |d dddg¡}|d dd¡ dd¡}t ||¡ |jj dddgdd}|jjddddjdddd}t ||¡ tjtd d
|
|j ddgdd W 5 Q R X tjt d d
|
|j
|
ddd g¡ W 5 Q R X d S ) NÚmonthÚdayÚyearr r r r )Zaxiszhierarchical axis)ÚmatchzToo many levelsé ) r Z swaplevelr Zassert_frame_equalZassert_series_equalÚTÚpytestZraisesÚ TypeErrorÚ
|
IndexErrorr )r Z/multiindex_year_month_day_dataframe_random_dataZymdr r r r r Ú)test_reorder_levels_swaplevel_equivalence5 s z;TestReorderLevels.test_reorder_levels_swaplevel_equivalenceN)Ú__name__Ú
|
__module__Ú __qualname__r r# r r r r r s )r )
|
Únumpyr r Zpandasr r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
