U b0 @sUddlmZmZddlZddlZddlZddlZddlm Z m Z ddl m Z dZdZdddd d gZejdd eejed eejejd d deejedeejegZeeed<eeeeeZejddeejedeejejd ddeejedegZeeed<eeeeeZeeeddeddee7Ze eedZeeegZ ej!edddZ"ddZ#ej$%dej$&dee'dej$&deej$&d e e"gd!d"Z(dS)#)AnyListN) DataFrameSeries2iZjimZjoeZjolieZjolinejolia Z abcdefghijZ20141009)ZperiodsZ ZYXWVUTSRQvals Z abcdefghijkZ ZYXWVUTSRQPkeyscCs |ddS)N)tr r W/tmp/pip-unpacked-wheel-xg7vt6ln/pandas/tests/indexing/multiindex/test_indexing_slow.py&r)columnsr )ZsubsetcCstt|d}t|D]x\}}||jdd|f|kM}|sd|d|d|jkstq|d|d|jks~t|| }|dt|kr |j t d|dddd}|dkst|j t |dddd}|dkstt |j|d|d|q|j t dddd}|dks.tt|dkrxt|dj|jddgd }t |j|d|d|qt |j|d|d|qdS) NboolT)Zaxisinplacer )rrr)nameindex)npZoneslenZastype enumerateZilocanyrAssertionErrorcopyZdropcols set_indextmZassert_frame_equallocrvaluesZassert_series_equal)midfkeymaskikrightZ return_valuer r rvalidate/s0     r*z(ignore::pandas.errors.PerformanceWarning lexsort_depthr%framec CsptjddZ|dkr |}n|jtd|d}|tdd}|jj|krVtt |||W5QRXdS)NT)recordr)Zbyr ) warningscatch_warningsrZ sort_valuesrrrZ_lexsort_depthrr*)r+r%r-r$r#r r rtest_multiindex_get_locRs r1))typingrrr/ZnumpyrZpytestZpandaspdrrZpandas._testingZ_testingr mnrrandomrandintchoicelistZ date_rangetolistZrandnr __annotations__maptuplezipr r$concataZdrop_duplicatesbr*markfilterwarningsZ parametrizeranger1r r r rsB  $ #