o h@s<ddlZddlZddlmZmZddlZddlmZm Z m Z m Z ddl m ZejddZejddZGdd d Zejd ggfd ged fd d ggfd dggfd ggfejdddgfeddd fejddddgfeddd fgddZejdgedfdggfgddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Z d'd(Z!d)d*Z"d+d,Z#d-d.Z$d/d0Z%d1d2Z&Gd3d4d4Z'd5d6Z(d7d8Z)d9d:Z*d;d<Z+d=d>Z,d?d@Z-ejdAgdBgdCgdDdEZ.dFdGZ/dHdIZ0dJdKZ1dLdMZ2dNdOZ3dPdQZ4dRdSZ5dTdUZ6dVdWZ7dS)XN) IndexingErrorPerformanceWarning) DataFrameIndex MultiIndexSeriescCstgdggdgdgdS)zsingle level MultiIndex)foobarbazqux)rfirst)levelscodesnames)rrr[/var/www/html/env/lib/python3.10/site-packages/pandas/tests/indexing/multiindex/test_loc.pysingle_level_multiindexsrcCsFddggdg}gdgdg}t||d}ttjdd|d S) Nrr rr r )rrrr r r )rr r rr r )rrr )r index)rrnprandom default_rngstandard_normal)rrrrrr%frame_random_data_integer_multi_indexs rc @sXeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ e j dddgddgfdggfddggfgddZddZddZdd Zd!d"Zd#d$Ze j d%eeeeejeefe j d&eeeeejeefd'd(Zd)d*Ze j d+d,d-gd.d/Ze j d0ed1d2fd3ej fgd4d5Z!d6d7Z"d8d9Z#d:d;Z$d1S)<TestMultiIndexLoccCsV|}d|jd<|jddksJ|}ttd|_d|jd<|jddks)JdS)N)r twoBr)r!r )loccopylistrangecolumns)self multiindex_dataframe_random_dataframedfrrr&test_loc_setitem_frame_with_multiindex$s  z8TestMultiIndexLoc.test_loc_setitem_frame_with_multiindexc Cs*|}ddddddddddddd d d d d dd }t|}|||d }|jd dgd}d}ttt|j||jddWdn1sNwY|j dd}|dus_J|j|}t j dgd|d}t j d gd|d} t j || gd dgd} tdgdi| d} t|| dS)NiiXM)rr r rg @@ )amountcolnum)r7r8r7r8)keys)r3r4r T)inplacerdtyperr6)r/r0r1r)rastype set_indextmassert_produces_warningrassert_frame_equalr%iloc sort_indexrarrayr from_arrays) r*any_real_numpy_dtyper<datar-key return_valuerescol_arryear_arrrexpectedrrrtest_loc_getitem_general/s(    z*TestMultiIndexLoc.test_loc_getitem_generalcCrttjddgdgdggdgdgd}tjtdd |jdWddS1s2wYdS Nr rrr r r2r r2r2rUrUrVr4r)rz^2$match rrrrrpytestraisesKeyErrorr%r*r-rrr0test_loc_getitem_multiindex_missing_label_raisesLs "zBTestMultiIndexLoc.test_loc_getitem_multiindex_missing_label_raisescCs:|d}||jdd}|jddg}t||dS)NA13)rrV)rer )reindexrr%r@assert_series_equal)r*/multiindex_year_month_day_dataframe_random_dataserrNresultrrr/test_loc_getitem_list_of_tuples_with_multiindexWszATestMultiIndexLoc.test_loc_getitem_list_of_tuples_with_multiindexcCstgdgdg}t|tdtjd}tddg}tgdtddggdgtjd}|j|}t|||jddg}t||tddgdd gd }|j|}t||tgtjd }tgt|j gggtjd tjd }|j|}t||dS)Nr r rrbr#C rrHr<r rrr r rr$rUrHrr<r r)rHr<rrr<rr<) r from_productrr(rfloat64r%r@rhr)r*rxyrNrky1emptyrrrtest_loc_getitem_series_s.       z)TestMultiIndexLoc.test_loc_getitem_seriescCstgdgdg}t|tdtjd}tddg}tgdtddggdgtjd}|j|}t ||tg}tgt|j gggtjd d d }|j|}t ||t d}tgd gdtjd}|j|}t ||dS) Nrmrnrprqr rrrrsrtrwrur) rrvrr(rrwrEr%r@rhrint64)r*rrxryrNrkr{scalarrrrtest_loc_getitem_array~s,       z(TestMultiIndexLoc.test_loc_getitem_arraycCsttjddgdgdggdgdgd}|jddgd}|jd }t |||jdddgfjddd }|jddd f}t |||jdgdgfdjddd }|jd jddd f}t |||jddg}|jd }t ||dS) Nr rR)irj)rbrbr#)XrYrYrr raxisr)rr rrrrrrC droplevelr%r@rBr*r-rNrkrrrtest_loc_multiindex_labelss"   "  z,TestMultiIndexLoc.test_loc_multiindex_labelscCsbttjddgdgdggdgdgd}|jdd gd}|jd }t ||dS) Nr rRrSrTrWrXrYrr r2rrrrrtest_loc_multiindex_intss z*TestMultiIndexLoc.test_loc_multiindex_intscCrPrQr\r`rrr(test_loc_multiindex_missing_label_raisess "z:TestMultiIndexLoc.test_loc_multiindex_missing_label_raiseszkey, posr r2rr rcCsrttjddgdgdggdgdgd}tjtdd |j|WddS1s2wYdS) Nr rRrSrTrWrXrYz not in indexrZr\)r*rIposr-rrr&test_loc_multiindex_list_missing_labels "z8TestMultiIndexLoc.test_loc_multiindex_list_missing_labelcCsttdtddgddgddggd}tjtd d  |jd Wdn1s+wYtjtd d  |jd Wdn1sGwYtjtdd |jdWddS1sdwYdS)NrUabcdefrz^\('a', 'b'\)$rZrrz^\('a', 'd', 'g'\)$)rrgzToo many indexers)rrrr) rr(rrvr]r^r_r%r)r*srrr(test_loc_multiindex_too_many_dims_raisess   "z:TestMultiIndexLoc.test_loc_multiindex_too_many_dims_raisescCsddtdD}ddtdD}t||g}dtjddd }t||d }||}t ||tt d  d dtgd gdgd}|j ddgfddf}|j ddg}t ||dS)NcSg|]}dt|qS) Attributestr.0rrrr zFTestMultiIndexLoc.test_loc_multiindex_indexer_none..r cSr)Valuerrrrrrrr g?r )rVr g?r)r4r r rr2rmr) r(rrvrrrrrr@rBarangereshaper%)r* attributesattribute_valuesrr-rkrNrrr test_loc_multiindex_indexer_nones  z2TestMultiIndexLoc.test_loc_multiindex_indexer_nonecCsttjdddttdgdg}|jddddf}|jdd ddf}t|||jdd ddf}t|||jddddf}t||ttjdddttdgdg}|j gd }|jd d d ddf}t||dS) Nr}r;r rrrrrrr2)rr$rUr4rfr ) rrrrrvr(r%r@rhrC)r*rrNrkrrrtest_loc_multiindex_incompletes$     z0TestMultiIndexLoc.test_loc_multiindex_incompletecCs:|}ttjdt||d}|jD]}||qdS)Nr r)rrrrrlenvalues)r*r single_levelrkrrrtest_get_loc_single_levels  z+TestMultiIndexLoc.test_get_loc_single_levelcCstgdddgg}ttjdd||}|jddddf}|}t ||tgdddgg}ttjdd||}|jd d ddf}|j dd}t |||jd ddf}|j d d}ddg|_ t |||jddd f}|d }t ||dS) N)rr$rUrrr )rrrrUrVrrrVr) rrvrrrrrr%r@rBrCr)r*rr-rkrNrrrtest_loc_getitem_int_slice!s"    z,TestMultiIndexLoc.test_loc_getitem_int_sliceindexer_type_1indexer_type_2c sddgd}gd}t||g}ttjt|dd|dgd}d d gd d gf}||f}tfd dt||D} |tusD|turet j t dd|j | dfWddS1s^wYdS|j | df} t gddt|d} t| | dS)NcSs,|tjkr t|S|tkrt|S||SN)rndarrayrEslice) indexer_typer9rrrconvert_nested_indexerFs  zQTestMultiIndexLoc.test_loc_getitem_nested_indexer..convert_nested_indexerrrmr}r;Datarr)rVrr rc3s|] \}}||VqdSrr)rrrrrr Ys  zDTestMultiIndexLoc.test_loc_getitem_nested_indexer..zas an indexer is not supportedrZ)r r r2r namer)rrvrrrrtuplezipsetr]r^ TypeErrorr%rr@rh) r*rrrrrr-r9typesindexerrkrNrrrtest_loc_getitem_nested_indexer;s0   z1TestMultiIndexLoc.test_loc_getitem_nested_indexercCsHtddg}|ddg|d}d|jd<|ddg|d}t||dS)Nrrbrrbr r rrr)r from_tuplesr%r@ assert_equal)r*frame_or_seriesmiobjrNrrr)test_multiindex_loc_one_dimensional_tuplejs  z;TestMultiIndexLoc.test_multiindex_loc_one_dimensional_tuplerrrcCsPtddg}tddg|d}d|j|ddf<tddg|d}t||dS)Nrrr r rrrrrr%r@rB)r*rrrrNrrr-test_multiindex_one_dimensional_tuple_columnsrs z?TestMultiIndexLoc.test_multiindex_one_dimensional_tuple_columnszindexer, exp_valueN?r r cCsptddg}tddgddgg|ddgd }d |j|d d gf<tgd dd||gg|gdd }t||dS)Nrrr2r r rr2rrrrrr)r r rrrrrrr)r*r exp_valuerr-rNrrr)test_multiindex_setitem_columns_enlarging{sz;TestMultiIndexLoc.test_multiindex_setitem_columns_enlargingcCsttjdddtddgg}td|d}td|ddd}tj||gdd}|}|jd d }t ||t ||d }|jd d }t ||tj|| |j gdd}|jd d }t ||dS) N20110101r periodsrrr rrz 2011-01-01z 2011-01-02rr )rrvpd date_rangerrconcatr&r%r@rBrrgr)r*midxser1ser2r-rNrkrrr"test_sorted_multiindex_after_unions   z4TestMultiIndexLoc.test_sorted_multiindex_after_unioncCsttttdtdtdgdgd}|jtjdddddf}tttdtdgdgd}t||dS)NabcdrValrr) rrrvr'r%rs_r@rB)r*r-rKrNrrrtest_loc_no_second_level_indexsz0TestMultiIndexLoc.test_loc_no_second_level_indexcCs`tgdgdgdgdd}tjtdd|jdWddS1s)wYdS) Nr)rrr)rhr)rrl)r)r rr r )r r2z(1, 4)rZ)rr r2)rr]r^r_r%r`rrrtest_loc_multi_index_key_errors "z0TestMultiIndexLoc.test_loc_multi_index_key_error)%__name__ __module__ __qualname__r.rOrarlr|rrrrr]mark parametrizerrrrrrr'rrrrrrrrrrnanrrrrrrrrr#sD   ,    )    rz indexer, posrbrDErr bahcCstjgdgdgddgd}ttjddd|d }|j|}|jd krK|gkrKtj t t |d |j |WddS1sDwYdS|t dd d gfkrttj t dd |j |WddS1smwYdS|j |}t||dS)Nrnrr r oner"r=rpr}r;rrrZrrz'bah')rrvrrrrDrCsizer]r^r_rr%rr@rh)rridxrjrNrkrrr7test_loc_getitem_duplicates_multiindex_missing_indexerss  " " rcolumns_indexercCstgdddgf}ttjddtd|d}|jdd d }ttd| gdd}|j dd|f}t ||dS) Nralphabetar r rr rrr )levelr) rrvrrrrrr(rDrgr%r@rB)r multi_indexr-rNrkrrr4test_loc_getitem_duplicates_multiindex_empty_indexersrcCsLttjtjgddggtddgddgd}|jd}tj}||ks$JdS) Nmeanmedianfunctsr)rrfunctionrrY)r r)rrrrrrr%r-rkrNrrr=test_loc_getitem_duplicates_multiindex_non_scalar_type_objects  r cCspttdtdtjddtjdddddg}|jd}|jdddf}t ||dS)NrVr rrr)rr) rrrrrrr?r%r@rh)r-rNrkrrr!test_loc_getitem_tuple_plus_slices  r cCs8|}|jd}|dd}|jd|_t||dS)Nr r)r%rrr@rB)rr-rkrNrrrtest_loc_getitem_ints   rcCsB|}tjtdd|jdWddS1swYdS)Nz^3$rZr)r]r^r_r%)rr-rrr%test_loc_getitem_int_raises_exceptions "rcCsh|}tjtdd |jdWdn1swYd|jd<d}|jd}||ks2JdS)Nz^\('bar', 'three'\)$rZ))r threer#r)r]r^r_r%rD)r+r-rNrkrrr test_loc_getitem_lowerdim_corners  rcCstdtdtdgdgd}d|d<|j|jdddf<tt |dt |jttj t dt gd d}| }td|jddd f<td|jddd f<t||dS) NstringabcdMain)anotherrrrlabelsr2))rb1)rb2)r#rr#r )rr'rrvrr%r@assert_numpy_array_equalrasarrayrr(rr&rrCrB)r-rNrrr$test_loc_setitem_single_column_slice"s" rc Csddtjfddtjfddtjfddtjfdddd tjfdd tjfg}ttd tgd tj|gd dd}|jdjd}ttdttjg|sLdndddtgd d}t ||dS)Nz Good ThingsroRz Bad ThingsT) Okay ThingsNr#)rrrrr#r)rUr2)d1d2d3d4)u1u2u3r=rYrobjectzstring[pyarrow_numpy]r&)r<rr) rrronesrrrr%r@rB)using_infer_stringtupsr-rkrNrrrtest_loc_nan_multiindex9s2           r+cCstjdddd}d}tj||fdd}t|dd }|jd }|j|d d fd f}t|s0Jt|s7J|jd }t|sCJdS)N2013Q12013Q4Q)freq)Wii )PeriodCVRr=) OMSOMKRES DRIFT_IND OEVRIG_INDFIN_INDVARE_UDLOEN_UDFIN_UDr))r,r0r3rr0r3)r period_rangerrvrr%risnan)rrrr-rkaltrrrtest_loc_period_string_indexingWs  r?cCstddg}tj||gddgd}tddgdd gd d gd d gg|d dgd}|j|d|jddkfd f}tdgd tj t dt dfgddgdd}t ||dS)Nz 2017-05-04z 2017-05-05Idx1Idx2r=r r rr2r rr$C1C2rHrr)rr) r to_datetimerrvrr%rget_level_valuesrr Timestampr@rh)dt_idxm_idxr-rkrNrrrtest_loc_datetime_mask_slicingws""rJcCsPtd}tdtjd|fgddgddd}|jdd|gf}t||dS)N2000r rrr=r)rr)rrGrrrr%r@rh)daterjrkrrr&test_loc_datetime_series_tuple_slicings rMcCstddgddgddgddggtjgddd gd d d gd }tjddgdd gd }|j|ddf}tddgddgddggtjgddd gd d d gd}t||dS)Nrr rrrr )r)r rr r rNrrLr=authorpricerDrrN)rrNrNr)rrrr%r@rB)r-rrkrNrrrtest_loc_with_mi_indexers rQcCstjdddd}ttd|d}|}||d<|jddd }|jjd ks'J|j|d|j |d}|j dg d}t |||d}|jjd ksQJ|j |d}|j dg d}t ||dS) Nz 2016-01-01rz US/Pacific)rtzrr rT)append)Nr)rrrr(to_framer?rrget_locr%rCrr@rBrh)dtirjr-df2rkrNrrrrtest_loc_mi_with_level1_named_0s rXcCstgdgdgdgdgdgdgddd }|d d g}|jd td fddfd}|jd jtd ddf}t||dS)N)z20160525 13:30:00.023MSFT51.95rZ)z20160525 13:30:00.048GOOGz720.50z720.93)z20160525 13:30:00.076AAPLz98.55z98.56)z20160525 13:30:00.131r\98.6198.62)20160525 13:30:00.135rYz51.92rZ)r_r\r]r^ztime,ticker,bid,ask,rtickertimer\z2016-05-25 13:30:00r) rsplitr?rDr%rrr@rB)r-rWrKrNrrrtest_getitem_str_slices rdcCsltjgdddd}gd}gd}t|||g}ttd|tjd}|j|d d d f}|d ks4JdS) N)z 20181101 1100z 20181101 1200z 20181102 1300z 20181102 1400datetimer)rr/)rbrbZW)r#ror.Fr2rurrbr#g) r PeriodIndexrrFrr(rrwr%)pilev2lev3rrjrkrrr!test_3levels_leading_period_indexsrmc@s$eZdZddZddZddZdS)TestKeyErrorsWithMultiIndexcCsjttdddgdd}|ddg}tjtdd |jd WddS1s.wYdS) Nr4r2rrnrrbr#rrZr r) rrrrr?r]r^r_r%)r*r-rWrrr!test_missing_keys_raises_keyerrors  "z=TestKeyErrorsWithMultiIndex.test_missing_keys_raises_keyerrorcCsZtdtddggdd}tjtdd|jdWddS1s&wYdS) Nrrr r rz\(0, 3\)rZ)rr)rrrvr]r^r_r%)r*rjrrr!test_missing_key_raises_keyerror2s "z=TestKeyErrorsWithMultiIndex.test_missing_key_raises_keyerror2cCs(tjtgdtgdtgdggdd}ttjdd|d}d }tjt |d |j d d t dfddfWdn1sJwYtjt |d |j d d t dfWdn1slwYtjt d d |j dddfWddS1swYdS)Nrrrr)rrr3)rrrr)rr"rr=r r2rrz'\('b', '1', slice\(None, None, None\)\)rZrrz \('b', '1'\))rr)rrFrrErrrr]r^r_r%rrget_locs)r*rr-msgrrrtest_missing_key_combinations$   "z8TestKeyErrorsWithMultiIndex.test_missing_key_combinationN)rrrrprqrwrrrrrns rncCs2|}|d}|d}|jdd}t||dS)Nrb)rer )r%r@rh)rir-rjrkrNrrrtest_getitem_loc_commutability s rxcCstgdddtjggddddg}|jdg}tdd gitjd gddgd d }t|||jd}tdd git d gdd d }t||dS)N)r r r rr)r r2r )r7ind1ind2ryrzr7r )rr r=rr) rrrr?r%rrr@rBrr rrrtest_loc_with_nans    r|cCs^tgdggddgd}tjtdd|jdWddS1s(wYdS)Nrrrrz\(2\.0, 2\.0, 3\.0\)rZ)@r}@)rr?r]r^r_r%r-rrrtest_getitem_non_found_tuple!s  "rcCstjddd}t|g}|dtdddksJ|dtdddks'J|dddd}|dddd}||ksAJ|dd}|dd}||ksWJ||d}||d}t | d|dS)Nz 2001-01-01drz2001-01rr ) rrrrFrUrrepeatrSr@rnonzero)rrr%rNrrrtest_get_loc_datetime_index*s   rcCstddgddgg}tddgddgdd gd d gg|d }dddgf}td dgddgg|j|ddf<tddgd dgdd gd d gg|d }t||dS)Nrrrr r rr2r rr$rUrrprVr5r4)rrvrrrEr%r@rB)rr-rrNrrr,test_loc_setitem_indexer_differently_ordered@s $ $$rcCstddgddgddgddgggd gd gd d gd }|jtdddgfddf}tddgddgddgddgggdgdgd d gd }t||dS)Nr r rr2r rr$rUrrr r r r rrr)rrrr)r r r r )rr%rr@rBr rrr5test_loc_getitem_index_differently_ordered_slice_noneKsrr)r r r$rr rrUr$)r r r$rrrUcCstdgdtgddgd}|jtd|fddf}tdgdgdgdgdgd}t|||j|jj|ddddf}t||dS) Nr rU)rNr)r r$ror)r r)r rU)r r$rr)r r r r r r r r )r r r r$r$rrrU)r) rrrr%rr@rBrisin)rr-rkrNrrr@test_loc_getitem_index_differently_ordered_slice_none_duplicates[s  rcCstjdgdgdgggdd}tddgi|d}|dd g}|jddddf}t||tdg|d}|jddddf}tdgtdgd d d}t ||dS) Nrxryzrr=rrrr rr{) rrFrrr%r@rBrrrh)rr-rNrkrjrrr3test_loc_getitem_drops_levels_for_one_row_dataframeqs rcCsjtgdddgg}ttd|d}|jddddgf}ttdtgd d}t||dS) Nrnr r rrr#rb)r r2))r#r )r#r )rbr )rbr ) rrvrrzerosr%rr@rB)colsr-rkrNrrr$test_mi_columns_loc_list_label_orders rcCsttddgdgdggdgdgd}dd g|j_d d g|j_tjt d d |j ddgdfWddS1sAwYdS)Nr4rtrrr)OhiorColorado)GreenRedrrkey1key2statecolorz\[2\] not in indexrZrr r) rrrrrrr)r]r^r_r%)r,rrr$test_mi_partial_indexing_list_raisess  "rcCsdttdtddgddggd}tjtdd|jd d gWddS1s+wYdS) Nr2r r rrrz\['not' 'found'\] not in indexrZnotfound)rr(rrvr]r^r_r%)rrrr(test_mi_indexing_list_nonexistent_raisess""rcCstgdgdgddgtddgddggd}d |jd <d |jd <d |jd <tgdgdgddtjtjtjgggdtddgddggd}t||dS)N)r r r r)rr2r$rUrr r rbr#rrrr)rr rb)rr)rr}g@g@)r~r3g@g @)r~grrr~)rrrr)rrrvr%rrr@rB)rkrNrrr'test_mi_add_cell_missing_row_non_uniques$    rcCstdddtjdgdggddgdd }|jd }|dksJt|tjs'J|jd gd fjd }|dks8Jt|tjs@JdS)Nrr rrr2rrr=r)rrrrr)rrrFr% isinstancerr}rC)r-rkrrr$test_loc_get_scalar_casting_to_floats    rcCsftjddgddggddgd}t|tjd}|jd}ttjtjgtddgdd d }t ||dS) Nrrrbr#r rr=rur{r) rrvrrrwr%rrr@rh)rs2rkrNrrr)test_loc_empty_single_selector_with_namess   rcCshtgdgdtdd}|ddg}tjtdd|jd WddS1s-wYdS) N)rrr,r)rVrVrr!rrnrbr#z^1$rZ)rr )rr(r?r]r^r_r%rrrr'test_loc_keyerror_rightmost_key_missings "rcCs8tddg}tddg|d}|jd}|dksJdS)Nr)r)r2r r r r)rrrr%)rrjrkrrr*test_multindex_series_loc_with_tuple_labels r)8numpyrr] pandas.errorsrrpandasrrrrrpandas._testing_testingr@fixturerrrrrr IndexSlicerrr r rrrrr+r?rJrMrQrXrdrmrnrxr|rrrrrrrrrrrrrrrrrrsx            $