x\c @sddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z ddljjZddlmZejdZejjdddd e d d gd d fd d de dgd dfgdZdZdZdZejjdd(dgdgfgdZdZdZejjdddgdZ dZ!dZ"dZ#ejjdd d!gd"Z$d#Z%d$Z&d%Z'd&Z(d'Z)dS()iN(tlrangetproduct(t DataFrametIndext MultiIndextSeriestconcatt date_range(ttestingc Cstjdddddgdddd d gd d d ddgg}tdddgddgdddgdddggddddgdddgdddgdddggdd d!d"d#g}t|d$|d%td&S('NgEJYgr鷯g|?5^ݿg6?gK?g -?gT?g9#?g9]?g46@iiitcodesiiitnamestonettwotthreetfourtindextcolumnstABCDE(tnptarrayRRtlist(tarrR((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pytfour_level_index_dataframe s06skey, level, exp_arr, exp_indexR tlvl0cCs|ddddfS(Nii((R ((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pytstbartfootnametlvl1cCs|ddddfS(Nii((R ((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyRsc Cstjjdd}tdddgddddggd d d d d gd d d d ggdddg}t|d|}|j|d|dd }t||d|}tj||dS(NiR R R RRthellotworldRiiiiRRR!Rtleveltaxis(RtrandomtrandnRRtxsttmtassert_frame_equal( tkeyR$texp_arrt exp_indexRRtdftresulttexpected((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_named_levels_axis_eq_1s$$cCs9|}|jdj}|jd}tj||dS(NRRi(RR(R(tvaluesR)tassert_almost_equal(t multiindex_dataframe_random_dataR.R/R0((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_values's cCs6|}|jd}|jd}tj||dS(NRR(RR(RR(R(tlocR)tassert_series_equal(R4R.R/R0((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_loc_equality.s c Csdddddddg}t|ddddgjddg}tid d d d gd6dtd tjddgdd}|jd dd}tj||dS(NR tabcdeiR tbbcdeitytyzcdeitztxbcdeiitzbcdetybcdeRta1ta2tcntRR R$(R R9i(R R:i(R;R<i(R=R>i(R=Ni(R=R?i(R=R@i( tNoneRt set_indexRRtnanR(R)R*(taccR.R0R/((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_missing_values_in_index5s  !!s key, levelRtsecondcCs||}t|gd}|jjtks1tt|jdddgd}|j|d|}tj||dS(NiRR$RI(RRt is_uniquetFalsetAssertionErrorR(R)R*(R+R$R4tframeR.R0R/((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_with_duplicatesJs "cCsl|}|jddd}||jjddk}tddddgd d |_tj||dS( NRR$RIiRRtbaztquxR tfirst(R(Rtget_level_valuesRR)R*(R4R.R/R0((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyt test_xs_levelYs !c Cstjjdd}tddddgddd gd d d ggd dddgdddgdddgg}t|d|}t|dd!ddgdgg}|jd dd}tj||dS(NiiR R tpR R R R;tctrR=RiiiRR$(RR&R'RRR(R)R*(RRR.R0R/((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyttest_xs_level_eq_2as*-%tindexercCs|jddddgS(NR iR$RR(R i(R((R.((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyRmscCs|jdjdddS(NR iR$R(R((R.((sGlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_xs.pyRnscCs|}dddddgg}tddgdggd d gd ggd d d g}t|d|dtd}||}tj||dS(Ng -?gT?g9#?g9]?g46s>  . $3    !