o g @sBddlZddlmZmZmZddlmZddZ GdddZ dS)N) MultiIndexSeries date_rangecCs|tdddjj}t|td}tt|D] }|||j|<q|jtks%Jt|dt j s/J| d}t|t j s|}|d}|dddf}|jdddd}t||dS)NAtwolevel)rtmassert_series_equal)self multiindex_dataframe_random_datadfrexpectedrrrrtest_xs_level_seriess z)TestXSWithMultiIndex.test_xs_level_seriescCstgd}tgd|d}|jjddgdd}|dusJtdd gd d gd}|jjdgdd}|dus8J|jd dd }t||dS)N))aone)r&r)br')r(r)rrindexL1L2T)inplacerrr&r(r'r)r from_tuplesrr, set_namesrrr )r!idxr return_valuer$rrrr*test_series_getitem_multiindex_xs_by_label!s  z?TestXSWithMultiIndex.test_series_getitem_multiindex_xs_by_labelcCsdttddd}ttd|g}tgd|d}tddgtdd}|jddd}t||dS) N20130903rrAB)rrr*rrr*r+rr)listrr from_productrrrr )rdtr2rr$rrrr!test_series_getitem_multiindex_xs0s z6TestXSWithMultiIndex.test_series_getitem_multiindex_xscCshtjgdddgd}tgd|d}|jddd d }td d gtjd d gddgdd}t||dS)N)r&xr&y)r(r<level1level2names)rrrr+r&rFaxis drop_levelrr;r=)rr0rrrr )r!mirrr$rrrtest_series_xs_droplevel_false:s  z3TestXSWithMultiIndex.test_series_xs_droplevel_falsecCstjdgddgd}tdg|d}tt|jddgd d d Wdn1s+wYtt|jdgd d d WddS1sJwYdS) Nr;r?r@rArr+r&r<rFrC)rr0rrassert_produces_warning FutureWarningr)r!rFrrrrtest_xs_key_as_listIs  "z(TestXSWithMultiIndex.test_xs_key_as_listN)__name__ __module__ __qualname__r%r4r:rGrJrrrrrs  r) numpyrpandasrrrpandas._testing_testingrrrrrrrs