o
    5³cª
  ã                   @   s>   d dl Zd dlZd dlmZmZ d dlmZ G dd„ dƒZ	dS )é    N)Ú	DataFrameÚ
MultiIndexc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestReorderLevelsc                 C   sœ  t dgg d¢ddggg d¢g d¢g d¢gg d¢d	}tt d
¡t d
¡dœ|d}t ||¡}| g d¢¡}t ||¡ | g d¢¡}t ||¡ | g d¢¡}t g d¢ddgdggg d¢g d¢g d¢gg d¢d	}tt d
¡t d
¡dœ|d}t ||¡}t ||¡ | g d¢¡}t dgdgdggg d¢g d¢g d¢gg d¢d	}tt d
¡t d
¡dœ|d}t ||¡}t ||¡ | g d¢¡}t ||¡ d S )NÚbar)ÚoneÚtwoÚthreer   é   )r   r   r   r   r   r   )r   r	   é   r   r	   r
   )r   r	   r   r	   r   r	   )ÚL0ÚL1ÚL2)ÚlevelsÚcodesÚnamesé   )ÚAÚB)Úindex)r   r	   r
   )r	   r
   r   )r   r   r   )r   r   r   )r   r   r   )r   r   ÚnpÚarangeÚtmÚget_objÚreorder_levelsÚassert_equal)ÚselfÚframe_or_seriesr   ÚdfÚobjÚresultÚe_idxÚexpected© r"   úd/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/frame/methods/test_reorder_levels.pyÚtest_reorder_levels   s>   ýýýz%TestReorderLevels.test_reorder_levelsc                 C   s*  |}|  g d¢¡}| dd¡ dd¡}t ||¡ |d   g d¢¡}|d  dd¡ dd¡}t ||¡ |jj g d¢dd}|jjddddjdddd}t ||¡ tjtdd |j ddgdd W d   ƒ n1 snw   Y  tjt	d	d |j
  g d
¢¡ W d   ƒ d S 1 sŽw   Y  d S )N)ÚmonthÚdayÚyearr   r	   r
   r   )Úaxiszhierarchical axis)ÚmatchzToo many levels)r	   r
   é   )r   Ú	swaplevelr   Úassert_frame_equalÚassert_series_equalÚTÚpytestÚraisesÚ	TypeErrorÚ
IndexErrorr   )r   Ú/multiindex_year_month_day_dataframe_random_dataÚymdr   r!   r"   r"   r#   Ú)test_reorder_levels_swaplevel_equivalence5   s    ÿ"ÿz;TestReorderLevels.test_reorder_levels_swaplevel_equivalenceN)Ú__name__Ú
__module__Ú__qualname__r$   r5   r"   r"   r"   r#   r      s    )r   )
Únumpyr   r/   Úpandasr   r   Úpandas._testingÚ_testingr   r   r"   r"   r"   r#   Ú<module>   s
    