ó šxŠ\c @sddlZddlmZddlZddlZddlZddlmZm Z m Z m Z ddl m Zejd„ƒZejd„ƒZejjdƒdefd „ƒYƒZejjd gedfd gedfd d gedfd ged fejdd…dgfedfejdd…ddgfedfgƒd„ƒZejjdƒejjdd„d„gƒd„ƒƒZejjdgedƒfdggfgƒd„ƒZd„Zd„Z d„Z!d„Z"d„Z#dS(iÿÿÿÿN(tcatch_warnings(t DataFrametIndext MultiIndextSeries(ttestingcCs:tdddddggdddd d ggd d gƒS( ssingle level MultiIndextlevelstfootbartbaztquxtcodesiiiitnamestfirst(R(((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pytsingle_level_multiindex scCsddgdddgg}ddddddgddddddgg}td|d|ƒ}ttjjddƒd|ƒS(NiiiRR itindex(RRtnptrandomtrandn(RR R((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt%frame_random_data_integer_multi_indexs0signore:\n.ix:DeprecationWarningtTestMultiIndexLocc BsªeZd„Zd„Zd„Zd„Zd„Zd„Zd„Ze j j de e eeejeefƒe j j de e eeejeefƒd „ƒƒZRS( c Cs’tjdddgdddggƒ}td|dtd ƒd tjƒ}tddgƒ}tdd ddd d dgdtjddgdddggƒd tjƒ}|j|}tj||ƒ|jddg}tj||ƒtddgdddgƒ}|j|}tj||ƒtdgd tjƒ}tgdtd|j dgggd tjƒƒ}|j|}tj||ƒdS(NiiitAtBtCRtdatai tdtypeiiiiRR ( Rt from_productRtrangeRtfloat64tlocttmtassert_series_equalR(tselfRtxtytexpectedtresultty1tempty((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_getitem_seriess&'$$   $ c CsŠtjdddgdddggƒ}td|dtd ƒd tjƒ}tjddgƒ}tdd ddd d dgdtjddgdddggƒd tjƒ}|j|}tj ||ƒtjgƒ}tgdtd|j dgggd tjƒƒ}|j|}tj ||ƒtj dƒ}tdd ddgddddgd tjƒ}|j|}tj ||ƒdS(NiiiRRRRRi RiiiiRR ( RRRRRRtarrayRRRRtint64(R RR!R"R#R$R&tscalar((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_getitem_array9s*'$$  $   c sxttjjddƒddddgdddggddddgddd ggƒ}ttjjddƒdd d d gd d dggdd d d gd ddggƒ‰|jd}tdtƒ|jd}WdQXtj ||ƒ|jdd…df}tdtƒ|jdd…df}WdQXtj ||ƒ|jdjdd…df}tdtƒ%|jdjdd…df}WdQXtj ||ƒ|jd}tdtƒ|jd}WdQXtj ||ƒˆjd }tdtƒˆjd }WdQXtj ||ƒt j t ‡fd†ƒtdtƒt j t ‡fd†ƒWdQXdS(NitcolumnstitjRRRtXtYiiiii i trecordcs ˆjdS(Ni(R((tmi_int(sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt~scs ˆjdS(Ni(tix((R2(sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyR3s(R-R/(R-R/( RRRRRRtTrueR4Rtassert_frame_equaltpytesttraisestKeyError(R t mi_labelstrstxp((R2sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_multiindexVs:!!  &  cCsDgtdƒD]}dt|ƒ^q }gtdƒD]}dt|ƒ^q6}tj||gƒ}dtjjddƒd}t|d|ƒ}||}tj ||ƒttj d ƒj d dƒd tjdd d dgdd d ggƒƒ}|j dd gfdd…f}|j dd g}tj ||ƒdS(Nit AttributeitValuegš™™™™™¹?i gà?R,i iÿÿÿÿRiiii( RtstrRRRRRRRR6tarangetreshapeR(R R-t attributestattribute_valuesRtdfR$R#((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt test_loc_multiindex_indexer_noneƒs)) "cCs©ttjdddƒtjtdƒdddggƒƒ}|jdd…dd…f}|jdd …dd…f}tj||ƒtj||ƒ|jdd …dd…f}tj||ƒtj||ƒ|jdd…dd…f}tj||ƒtj||ƒttjdddƒtjtdƒdddggƒƒ}|j d d d d ddg}|jdd d…dd…f}tj||ƒdS(NiRR)itatbtciiiiii i ii( RRRARRRRRRtiloc(R tsR#R$((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_multiindex_incompletešs"''"cCsJ|}ttjjt|ƒƒd|ƒ}x|jD]}||q4WdS(NR(RRRRtlentvalues(R Rt single_levelRKtk((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_get_loc_single_level¶s  cCs¬tjgtjdddgddgƒD] }|^q(ƒ}ttjjddƒ||ƒ}|jdd…dd…f}|}t j ||ƒtjgtjdddgddgƒD] }|^q»ƒ}ttjjddƒ||ƒ}|jdd…dd…f}|j d }t j ||ƒ|jddd…f}|j d d !}ddg|_ t j ||ƒ|jdd…df}|d}t j ||ƒdS( NiiiRGRHi iiii( Rt from_tuplest itertoolstproductRRRRRRR6RJR(R ttRRER$R#((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_getitem_int_slice½s(.! .!  tindexer_type_1tindexer_type_2c s d„‰dddg}dddg}tj||gƒ}ttjt|ƒdd ƒd |d d gƒ}ddgddgf}||f}t‡fd †t||ƒDƒƒ} |j| d f} t ddddgdd d tj|ƒƒ} t j | | ƒdS(NcSs<|tjkrtj|ƒS|tkr2t|ŒS||ƒS(N(RtndarrayR(tslice(t indexer_typetkeys((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pytconvert_nested_indexeræs    i iiiiiRR)RR,tDatac3s$|]\}}ˆ||ƒVqdS(N((t.0R[RP(R](sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pys úsiitname( RRRRRARMttupletzipRRRR( R RWRXRGRHRRER\ttypestindexerR$R#((R]sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_getitem_nested_indexerÛs"   (t__name__t __module__R'R+R=RFRLRQRVR7tmarkt parametrizetlistRatsetRZRRYRRRe(((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyRs   -      s"indexer, is_level1, expected_errorRtDs\['D'\] not in indexRtbahcCsrtjdddgdddggddd gƒ}ttjd d d ƒd |ƒjƒ}|gkrv|jg}n¥|rÇtdddgd tjdddgdggddd gƒƒjƒ}nTtjdgdddggddd gƒ}ttjdd d ƒd |ƒjƒ}|dk rQtj t d|ƒ|j |WdQXn|j |}t j ||ƒdS(NRRRRRR R tonettwoi RR)Riiitmatch(RRRRRAt sort_indexRJtNoneR7R8R9RRR(Rdt is_level1texpected_errortidxRKR#texp_idxR$((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt7test_loc_getitem_duplicates_multiindex_missing_indexerss$ ' 3  RdcCs|jddgS(NiÐii i (iÐii (iÐii (R(RK((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyR3+scCs|jddgS(NiÐii i (iÐii (iÐii (R4(RK((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyR3,scCsC|d}|j|jdd!ƒ}||ƒ}tj||ƒdS(NRi1i3(treindexRRR(t/multiindex_year_month_day_dataframe_random_dataRdRKR#R$((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_series_loc_getitem_fancy)s  tcolumns_indexercCs¾tjdddgddgfƒ}ttjjddƒdtdƒd |ƒ}|jd d d d ƒ}tdtdƒd |jgƒd ƒ}|j dd…|f}t j ||ƒdS(NRRR talphatbetaiiRR,tlevelitaxisi( RRRRRRRRqRxRRR6(R{t multi_indexRER#R$((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt4test_loc_getitem_duplicates_multiindex_empty_indexer7s-(cCswttjtjgddggdtjdd gƒdddgƒ}|jdd f}tj}||ksst‚dS( NtmeantmedianR,tfunctsRtfunctionR`(R„R‚(R„Rƒ(R„R‚(RRR‚RƒRRRRtAssertionError(RER$R#((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt=test_loc_getitem_duplicates_multiindex_non_scalar_type_objectHs!   cCs›titjdƒd6tjdƒd6tjjdƒd6tjjdƒd6ƒjddgƒ}|jd}|jddd…f}tj||ƒdS( Ni RGRHRItdi(ii(ii( RRRARRt set_indexRRR(RER#R$((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt!test_loc_getitem_tuple_plus_sliceTs cCsF|}|jd}|d}|jjdƒ|_tj||ƒdS(Niiýÿÿÿi(RRt droplevelRR6(RRER$R#((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyttest_loc_getitem_int`s   cCs1|}tjtddƒ|jdWdQXdS(NRps^3L?$i(R7R8R9R(RRE((sHlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_loc.pyt%test_loc_getitem_int_raises_exceptionhscCsi|}tjtddƒ|jdWdQXd|jd s:    "é  "7