ó šxŠ\c @sddlZddlZddlmZmZddlmZmZm Z m Z ddl m Z ddl mZejjdd„d„d „gƒejjd d e d gd d gƒfd e ddgd d dgƒfgƒd„ƒƒZejjdddgƒd„ƒZejjdd„d„gƒd„ƒZejjdd„d„gƒd„ƒZejjdd„edfd„edfd „e d!fd"„ed#fd$„ed#fd%„ed&fgƒd'„ƒZd(„Zd)„Zejjd*d+„d,fd-„d.fgƒd/„ƒZd0„Zejjd1d2„edƒfd3„edd4ƒfd5„edd4ƒfgƒd6„ƒZejjd7ee gƒd8„ƒZ!ej"d9„ƒZ#ejjdd:„d;„gƒd<„ƒZ$d=„Z%d>„Z&dS(?iÿÿÿÿN(tutzip(t DataFrametIndext MultiIndextSeries(t IndexingError(ttestingt access_methodcCs|dd…|fS(N((tstx((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pytscCs|jdd…|fS(N(tloc(R R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR scCs|j|ddƒS(Ntleveli(txs(R R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR sslevel1_value, expectediitindexiicCsStdddgƒ}tjdddgƒ|_|||ƒ}tj||ƒdS(Niiii(ii(ii(ii(RRt from_tuplesRttmtassert_series_equal(Rt level1_valuetexpectedR tresult((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_series_getitem_multiindexs t level0_valuetDtAcCsxtd|ddgdddddd d d ggd dddd ddddddg d dddddddddg gdddgƒ}tjjt|ƒd ƒ}t|d|ddgƒ}|dkrötjtddƒ|j dWdQXntjtddƒ|j dWdQX|j |}t |j ƒdd!dddt dddgddƒƒ}t j||ƒdS( NtlevelstBtCiiii%i9iCiKiRtcodesiiiiiiitnamesttagtdayRtcolumnstvalRtmatchs^'A'$s^'X'$tXtname(RtnptrandomtrandntlenRtpytesttraisestKeyErrorR"RtravelRRR(RRtarrtdfRR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt)test_series_getitem_duplicates_multiindex!s !!'  tindexercCs|dS(NiÐi(iÐi((R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR =scCs |jdS(NiÐi(iÐi(R (R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR >scCsa|d}|j|jdd!ƒ}|jjdƒjdƒ|_||ƒ}tj||ƒdS(NRi*iAi(treindexRt droplevelRR(t/multiindex_year_month_day_dataframe_random_dataR1R RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_series_getitem<s   cCs|dS(NiÐii (iÐii ((R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR KscCs |jdS(NiÐii (iÐii (R (R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR LscCs9|d}|jd}||ƒ}||ks5t‚dS(NRi1(tiloctAssertionError(R4R1R RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt"test_series_getitem_returns_scalarJs   s)indexer,expected_error,expected_error_msgcCs |jdƒS(NiÐii(iÐii(t __getitem__(R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR Xss^356L?$cCs|dS(NiÐii(iÐii((R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR YscCs |jdS(NiÐii(iÐii(R (R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ZssToo many indexerscCs|jt|ƒƒS(N(R9R)(R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR [ssindex out of boundscCs|t|ƒS(N(R)(R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR \scCs|jt|ƒS(N(R6R)(R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ]ss*single positional indexer is out-of-boundscCs4|d}tj|d|ƒ||ƒWdQXdS(NRR#(R*R+(R4R1texpected_errortexpected_error_msgR ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt#test_series_getitem_indexing_errorsWs cCsB|d}|d„|Dƒ}||dk}tj||ƒdS(NRcss|]}|dkVqdS(iN((t.0R ((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pys ksi(RR(R4R RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt$test_series_getitem_corner_generatorhs cCsC|j}|jdd…df}|dj}tj||ƒdS(Nitfootone(R?R@(tTtvaluesRtassert_almost_equal(t multiindex_dataframe_random_dataR/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_getitem_simplets  sindexer,expected_error_msgcCs|dS(NR?tfour(R?RF((R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR |ss^\('foo', 'four'\)$cCs|dS(Ntfoobar((R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR }ss ^'foobar'$cCs3|j}tjtd|ƒ||ƒWdQXdS(NR#(RAR*R+R,(RDR1R;R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt#test_frame_getitem_simple_key_error{s cCsžtidddgd6dddgd6ƒ}ddgd d gd d gg|_|d}tdgdgdggd |jdd gƒ}tj||ƒdS(Nt1t2t3tat4tbs level1 item1s level1 item2ts level2 item2s level3 item1s level3 item2RR!(RR!RRtassert_frame_equal(R/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt*test_frame_getitem_multicolumn_empty_level†s, $ sindexer,expected_slicecCs|dS(NR?((R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ’scCs|dS(Ntbar((R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR “sicCs|jdd…dfS(NRR(R (R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ”scCsW|j}|jd|j|ƒ}|jjdƒ|_||ƒ}tj||ƒdS(NR!i(RAR2R!R3RRP(RDR1texpected_sliceR/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_frame_getitem_toplevel‘s   tunicode_stringsc Cs&ddddddgddddddgdd d dddgg}|r€g|D]%}g|D]}t|ƒ^q_^qR}ntt|Œƒ}tj|ƒ}ttjjd d ƒd |ƒ}|d}|dj dƒ}t j ||ƒ|d}|d}|j dƒ}t j ||ƒdS(NRLttoptroutine1troutine2ROtODtresult1tresult2twxtwyiiR!(RLRORO(RWRZ(RWRZRO(RWRZ( RtsortedRRRRR&R'R(trenameRR( RUtarraysR.R ttuplesRR/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_frame_mixed_depth_getŸs5!   c CsÍddddddgddd d d d gddd d d d gg}d ddg}tdddgddddggdddddddgdd d ddd ggdddgƒ}t|d|d|ƒS(s;Fixture for DataFrame used in tests for gh-4145 and gh-4146RLtdtetctfRNiiiiiith1th3th5RRRtA1tA2tB1tB2RiRtmaintsubRR!(RR(tdataRR!((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pytdataframe_with_duplicate_index¾s0cCs|dS(NRRj(RRj((R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ÍscCs|jdd…dfS(NRRj(RRj(R (R/((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyR ÎscCs€|}tdddgƒ}tjd gdddgƒ}td d d ggd |d |ƒj}||ƒ}tj||ƒdS(NRgRhRiRRjRRnRoRLiRR!(RRj(RRRRRARRP(RqR1R/RR!RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyttest_frame_mi_accessÌs ' cCsR|}tdddgddddgddƒ}|d d}tj||ƒdS( NRLiRRgRhRiR%RjR(RRR(RqR/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt#test_frame_mi_access_returns_seriesÛs*cCsv|}tdddgdddggdtddgddƒd d d d gƒj}|d d}tj||ƒdS(NRciRdiRRmR%RoR!RgRhRiR(RRRARRP(RqR/RR((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pyt"test_frame_mi_access_returns_frameæs !('tnumpyR&R*t pandas.compatRRtpandasRRRRtpandas.core.indexingRt pandas.utilRRtmarkt parametrizeRR0R5R8R,t IndexErrorR<R>RERHRQtsliceRTtTruetFalseRbtfixtureRqRrRsRt(((sLlib/python2.7/site-packages/pandas/tests/indexing/multiindex/test_getitem.pytsV  "  6 $        ' $