a Gb0 @sUddlmZmZddlZddlZddlZddlZddlm Z m Z ddl m Z dZdZgdZejddeejedeejejd dd eejed eejegZeeed <eeeeeZejdd eejedeejejd d d eejedegZeeed<eeeeeZeeeddeddee7Ze eedZeeegZ ej!edddZ"ddZ#ej$%dej$&dee'dej$&deej$&de e"gddZ(dS))AnyListN) DataFrameSeries2i)ZjimZjoeZjolieZjolinejolia Z abcdefghijZ20141009)ZperiodsZ ZYXWVUTSRQvals Z abcdefghijkZ ZYXWVUTSRQPkeyscCs |ddS)N)tr r Rlib/python3.9/site-packages/pandas/tests/indexing/multiindex/test_indexing_slow.py&r)columnsr )ZsubsetcCstt|d}t|D]x\}}||jdd|f|kM}|sd|d|d|jvsJq|d|d|jvs~J||}|dt|kr |j t d|dddd}|dusJ|j t |dddd}|dusJt |j|d|d|q|j t dddd}|dus.Jt|dkrxt|dj|jddgd }t |j|d|d|qt |j|d|d|qdS) NboolT)Zaxisinplacer )rrr)nameindex)npZoneslenZastype enumerateZilocanyrcopyZdropcols set_indextmZassert_frame_equalZlocrvaluesZassert_series_equal)midfkeymaskikrightZ return_valuer r rvalidate/s,     r(z(ignore::pandas.errors.PerformanceWarning lexsort_depthr#framecCstjddd|dkr |}n|jtd|d}|tdd}|jj|krVJt|||Wdn1sv0YdS)NT)recordr)Zbyr ) warningscatch_warningsrZ sort_valuesrrrZ_lexsort_depthr()r)r#r+r"r!r r rtest_multiindex_get_locRs r/))typingrrr-ZnumpyrZpytestZpandasZpdrrZpandas._testingZ_testingrmnrZrandomZrandintZchoicelistZ date_rangetolistZrandnr __annotations__maptuplezipr r"concataZdrop_duplicatesbr(ZmarkfilterwarningsZ parametrizeranger/r r r rs@  $ #