o ۭj@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)rrrf/var/www/html/CropPilot/venv/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 @sveZdZejdejdddgddZddZd d Z d d Z d dZ ddZ ddZ ddZddZejdddgddgfdggfddggfgddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zejd)eeeeejeefejd*eeeeejeefd+d,Zd-d.Zejd/d0d1gd2d3Z ejd4ed5d6fd7ej!fgd8d9Z"d:d;Z#dd?Z%d5S)@TestMultiIndexLocz.ignore:Setting a value on a view:FutureWarninghas_refTFcCsv|}|r |dd}d|jd<|jddksJ|}ttd|_|r+|dd}d|jd<|jddks9JdS)N)r twoBr)r"r )loccopylistrangecolumns)self multiindex_dataframe_random_datar frameviewdfrrr&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)r9r:r9r:)keys)r5r6r T)inplacerdtyperr8)r1r2r3r)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_general7s(    z*TestMultiIndexLoc.test_loc_getitem_generalcCrttjddgdgdggdgdgd}tjtdd |jdWddS1s2wYdS Nr rrr r r4r r4r4rWrWrXr6r*rz^2$match rrrrrpytestraisesKeyErrorr&r+r/rrr0test_loc_getitem_multiindex_missing_label_raisesTs "zBTestMultiIndexLoc.test_loc_getitem_multiindex_missing_label_raisescCs:|d}||jdd}|jddg}t||dS)NA13)rrX)rgr )reindexrr&rBassert_series_equal)r+/multiindex_year_month_day_dataframe_random_dataserrPresultrrr/test_loc_getitem_list_of_tuples_with_multiindex_szATestMultiIndexLoc.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 rrdr$C rrJr>r rrr r rr%rWrJrr>r r)rJr>rrr>rr>) r from_productrr)rfloat64r&rBrjr)r+rxyrPrmy1emptyrrrtest_loc_getitem_seriesgs.       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) Nrorprrrsr rrtrurvryrwr) rrxrr)rryrGr&rBrjrint64)r+rrzr{rPrmr}scalarrrrtest_loc_getitem_arrays,       z(TestMultiIndexLoc.test_loc_getitem_arraycCsttjddgdgdggdgdgd}|jddgd}|jd }t |||jdddgfjddd }|jddd f}t |||jdgdgfdjddd }|jd jddd f}t |||jddg}|jd }t ||dS) Nr rT)irj)rdrdr$)XrYr[rr raxisr)rr rrrrrrE droplevelr&rBrDr+r/rPrmrrrtest_loc_multiindex_labelss"   "  z,TestMultiIndexLoc.test_loc_multiindex_labelscCsbttjddgdgdggdgdgd}|jdd gd}|jd }t ||dS) Nr rTrUrVrYrZr[rr r4rrrrrtest_loc_multiindex_intss z*TestMultiIndexLoc.test_loc_multiindex_intscCrRrSr^rbrrr(test_loc_multiindex_missing_label_raisess "z:TestMultiIndexLoc.test_loc_multiindex_missing_label_raiseszkey, posr r4rr rcCsrttjddgdgdggdgdgd}tjtdd |j|WddS1s2wYdS) Nr rTrUrVrYrZr[z not in indexr\r^)r+rKposr/rrr&test_loc_multiindex_list_missing_labels "z8TestMultiIndexLoc.test_loc_multiindex_list_missing_labelcCsttdtddgddgddggd}tjtd d  |jd Wdn1s+wYtjtd d  |jd Wdn1sGwYtjtdd |jdWddS1sdwYdS)NrWabcdefrz^\('a', 'b'\)$r\rrz^\('a', 'd', 'g'\)$)rrgzToo many indexers)rrrr) rr)rrxr_r`rar&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 )rXr!g?r*r6r r rr4ror) r)rrxrrrrrrBrDarangereshaper&)r+ attributesattribute_valuesrr/rmrPrrr 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) Nrr=r!rrrrrrr4)rr%rWr6rhr ) rrrrrxr)r&rBrjrE)r+rrPrmrrrtest_loc_multiindex_incompletes$     z0TestMultiIndexLoc.test_loc_multiindex_incompletecCs:|}ttjdt||d}|jD]}||qdS)Nr r)rrrrrlenvalues)r+r single_levelrkrrrtest_get_loc_single_level s  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%rWrrr )rrrrWrXrrrXr) rrxrrrrrr&rBrDrEr)r+rr/rmrPrrrtest_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)rndarrayrGslice) indexer_typer;rrrconvert_nested_indexerNs  zQTestMultiIndexLoc.test_loc_getitem_nested_indexer..convert_nested_indexerrrorr=Datarr*rXrr rc3s|] \}}||VqdSrr)rrrrrr as  zDTestMultiIndexLoc.test_loc_getitem_nested_indexer..zas an indexer is not supportedr\)r r r4r!namer)rrxrrrrtuplezipsetr_r` TypeErrorr&rrBrj) r+rrrrrr/r;typesindexerrmrPrrrtest_loc_getitem_nested_indexerCs0   z1TestMultiIndexLoc.test_loc_getitem_nested_indexercCsHtddg}|ddg|d}d|jd<|ddg|d}t||dS)Nrrdrrdr r rrr)r from_tuplesr&rB assert_equal)r+frame_or_seriesmiobjrPrrr)test_multiindex_loc_one_dimensional_tuplers  z;TestMultiIndexLoc.test_multiindex_loc_one_dimensional_tuplerrrcCsPtddg}tddg|d}d|j|ddf<tddg|d}t||dS)Nrrr r rrrrrr&rBrD)r+rrrrPrrr-test_multiindex_one_dimensional_tuple_columnszs 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)Nrrr4r r rr4rrrrrr)r r rrrrrrr)r+r exp_valuerr/rPrrr)test_multiindex_setitem_columns_enlargingsz;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 )rrxpd date_rangerrconcatr'r&rBrDrrir)r+midxser1ser2r/rPrmrrr"test_sorted_multiindex_after_unions   z4TestMultiIndexLoc.test_sorted_multiindex_after_unioncCsttttdtdtdgdgd}|jtjdddddf}tttdtdgdgd}t||dS)NabcdrValrr) rrrxr(r&rs_rBrD)r+r/rMrPrrrtest_loc_no_second_level_indexsz0TestMultiIndexLoc.test_loc_no_second_level_indexcCs`tgdgdgdgdd}tjtdd|jdWddS1s)wYdS) Nr)rrr)rhr)rrl)r)r rr r )r r4z(1, 4)r\)rr r4)rr_r`rar&rbrrrtest_loc_multi_index_key_errors "z0TestMultiIndexLoc.test_loc_multi_index_key_error)&__name__ __module__ __qualname__r_markfilterwarnings parametrizer0rQrcrnr~rrrrrrrrrrr(rrrrrrrrrrnanrrrrrrrrr#sH    ,    )    rz indexer, posrdrDErr bahcCstjgdgdgddgd}ttjddd|d }|j|}|jd krK|gkrKtj t t |d |j |WddS1sDwYdS|t dd d gfkrttj t dd |j |WddS1smwYdS|j |}t||dS)Nrprr r oner#r?rrrr=rrr\rrz'bah')rrxrrrrFrEsizer_r`rarr&rrBrj)rridxrlrPrmrrr7test_loc_getitem_duplicates_multiindex_missing_indexerss  " " rcolumns_indexercCstgdddgf}ttjddtd|d}|jdd d }ttd| gdd}|j dd|f}t ||dS) Nralphabetar r!rr!rrr )levelr) rrxrrrrrr)rFrir&rBrD)r multi_indexr/rPrmrrr4test_loc_getitem_duplicates_multiindex_empty_indexersrcCsLttjtjgddggtddgddgd}|jd}tj}||ks$JdS) Nmeanmedianfunctsr)r r functionrr[)r r )rrrr rrr&r/rmrPrrr=test_loc_getitem_duplicates_multiindex_non_scalar_type_objects  rcCspttdtdtjddtjdddddg}|jd}|jdddf}t ||dS)NrXr rrr)rr) rrrrrrrAr&rBrj)r/rPrmrrr!test_loc_getitem_tuple_plus_slices  rcCs8|}|jd}|dd}|jd|_t||dS)Nr r)r&rrrBrD)rr/rmrPrrrtest_loc_getitem_ints   rcCsB|}tjtdd|jdWddS1swYdS)Nz^3$r\r)r_r`rar&)rr/rrr%test_loc_getitem_int_raises_exceptions "rcCsh|}tjtdd |jdWdn1swYd|jd<d}|jd}||ks2JdS)Nz^\('bar', 'three'\)$r\))r threer$r)r_r`rar&rF)r,r/rPrmrrr 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)anotherrrrlabelsr4))rd1)rd2)r$rr$r )rr(rrxrr&rBassert_numpy_array_equalrasarrayrr)rr'rrErD)r/rPrrr$test_loc_setitem_single_column_slice*s" rc Csddtjfddtjfddtjfddtjfdddd tjfdd tjfg}ttd tgd tj|gd dd}|jdjd}ttdttjg|sLdndddtgd d}t ||dS)Nz Good ThingsrqRz Bad ThingsT) Okay ThingsNr$)r!r"rr!r$r)rWr4)d1d2d3d4)u1u2u3r?r[robjectrr))r>rr) rrronesrrrr&rBrD)using_infer_stringtupsr/rmrPrrrtest_loc_nan_multiindexAs2           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/r3r6rr3r6)r period_rangerrxrr&risnan)rrrr/rmaltrrrtest_loc_period_string_indexing_s  rBcCstddg}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 rr4r!rr%C1C2rJrr*rr) r to_datetimerrxrr&rget_level_valuesrr TimestamprBrj)dt_idxm_idxr/rmrPrrrtest_loc_datetime_mask_slicings""rMcCsPtd}tdtjd|fgddgddd}|jdd|gf}t||dS)N2000r rrr?r)rr)rrJrrrr&rBrj)daterlrmrrr&test_loc_datetime_series_tuple_slicings rPcCstddgddgddgddggtjgddd gd d d gd }tjddgdd gd }|j|ddf}tddgddgddggtjgddd gd d d gd}t||dS)Nrr rrrr )r)r rr r rQrrOr?authorpricerGrrQ)rrQrQr)rrrr&rBrD)r/rrmrPrrrtest_loc_with_mi_indexers rTcCstjdddd}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_framerArrget_locr&rErrBrDrj)dtirlr/df2rmrPrrrrtest_loc_mi_with_level1_named_0s r[cCstgdgdgdgdgdgdgddd }|d d g}|jd td fddfd}|jd jtd ddf}t||dS)N)z20160525 13:30:00.023MSFT51.95r])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.135r\z51.92r])rbr_r`raztime,ticker,bid,ask,rtickertimer_z2016-05-25 13:30:00r) rsplitrArFr&rrrBrD)r/rZrMrPrrrtest_getitem_str_slices rgcCsltjgdddd}gd}gd}t|||g}ttd|tjd}|j|d d d f}|d ks4JdS) N)z 20181101 1100z 20181101 1200z 20181102 1300z 20181102 1400datetimer)rr2)rdrdZW)r$rqr1Fr4rwrrdr$g) r PeriodIndexrrHrr)rryr&)pilev2lev3rrlrmrrr!test_3levels_leading_period_indexsrpc@s$eZdZddZddZddZdS)TestKeyErrorsWithMultiIndexcCsjttdddgdd}|ddg}tjtdd |jd WddS1s.wYdS) Nr6r4rrprrdr$rr\r r) rrrrrAr_r`rar&)r+r/rZrrr!test_missing_keys_raises_keyerrors  "z=TestKeyErrorsWithMultiIndex.test_missing_keys_raises_keyerrorcCsZtdtddggdd}tjtdd|jdWddS1s&wYdS) Nrrr r rz\(0, 3\)r\)rr)rrrxr_r`rar&)r+rlrrr!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 r4rrz'\('b', '1', slice\(None, None, None\)\)r\rrz \('b', '1'\))rr)rrHrrGrrrr_r`rar&rrget_locs)r+rr/msgrrrtest_missing_key_combinations$   "z8TestKeyErrorsWithMultiIndex.test_missing_key_combinationN)rrrrsrtrzrrrrrqs rqcCs2|}|d}|d}|jdd}t||dS)Nrd)rgr!)r&rBrj)rkr/rlrmrPrrrtest_getitem_loc_commutabilitys r{cCstgdddtjggddddg}|jdg}tdd gitjd gddgd d }t|||jd}tdd git d gdd d }t||dS)N)r r r!rr)r r4r!)r9ind1ind2r|r}r9r )rr r?rr) rrrrAr&rrrBrDrr rrrtest_loc_with_nans    rcCs^tgdggddgd}tjtdd|jdWddS1s(wYdS)Nrrrrz\(2\.0, 2\.0, 3\.0\)r\)@r@)rrAr_r`rar&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 ) rrrrHrXrrepeatrVrBrnonzero)rrr&rPrrrtest_get_loc_datetime_index2s   rcCstddgddgg}tddgddgdd gd d gg|d }dddgf}td dgddgg|j|ddf<tddgd dgdd gd d gg|d }t||dS)Nrrrr r rr4r!rr%rWrrrrXr7r6)rrxrrrGr&rBrD)rr/rrPrrr,test_loc_setitem_indexer_differently_orderedHs $ $$rcCstddgddgddgddgggd gd gd d gd }|jtdddgfddf}tddgddgddgddgggdgdgd d gd }t||dS)Nr r rr4r!rr%rWrur r r r rrr)rrrr)r r r r )rr&rrBrDr rrr5test_loc_getitem_index_differently_ordered_slice_noneSsrr)r r r%rr rrWr%)r r r%rrrWcCstdgdtgddgd}|jtd|fddf}tdgdgdgdgdgd}t|||j|jj|ddddf}t||dS) Nr rW)rQr)r r%rrr)r r)r rW)r r%rr)r r r r r r r r )r r r r%r%rrrW)r) rrrr&rrBrDrisin)rr/rmrPrrr@test_loc_getitem_index_differently_ordered_slice_none_duplicatescs  rcCstjdgdgdgggdd}tddgi|d}|dd g}|jddddf}t||tdg|d}|jddddf}tdgtdgd d d}t ||dS) Nrzr{zrr?rrrr rr~) rrHrrr&rBrDrrrj)rr/rPrmrlrrr3test_loc_getitem_drops_levels_for_one_row_dataframeys rcCsjtgdddgg}ttd|d}|jddddgf}ttdtgd d}t||dS) Nrpr r rrr$rd)r!r4))r$r )r$r )rdr )rdr ) rrxrrzerosr&rrBrD)colsr/rmrPrrr$test_mi_columns_loc_list_label_orders rcCsttddgdgdggdgdgd}dd g|j_d d g|j_tjt d d |j ddgdfWddS1sAwYdS)Nr6rwrur)OhiorColorado)GreenRedrrkey1key2statecolorz\[2\] not in indexr\rr r) rrrrrrr*r_r`rar&)r-rrr$test_mi_partial_indexing_list_raisess  "rcCsdttdtddgddggd}tjtdd|jd d gWddS1s+wYdS) Nr4r r rrrz\['not' 'found'\] not in indexr\notfound)rr)rrxr_r`rar&)rrrr(test_mi_indexing_list_nonexistent_raisess""rcCstgdgdgddgtddgddggd}d |jd <d |jd <d |jd <tgdgdgddtjtjtjgggdtddgddggd}t||dS)N)r r r!r)rr4r%rWrr r rdr$rrrr)rr rd)rr)rrg@g@)rr5g@g @)rgrrr)rrrr)rrrxr&rrrBrD)rmrPrrr'test_mi_add_cell_missing_row_non_uniques$    rcCstdddtjdgdggddgdd }|jd }|dksJt|tjs'J|jd gd fjd }|dks8Jt|tjs@JdS)Nrr rrr4rrr?r)rrrrr)rrrHr& isinstancerrrE)r/rmrrr$test_loc_get_scalar_casting_to_floats    rcCsftjddgddggddgd}t|tjd}|jd}ttjtjgtddgdd d }t ||dS) Nrrrdr$r rr?rwr~r) rrxrrryr&rrrBrj)rs2rmrPrrr)test_loc_empty_single_selector_with_namess   rcCshtgdgdtdd}|ddg}tjtdd|jd WddS1s-wYdS) N)rrr,r)rXrXrr!rrprdr$z^1$r\)rr )rr)rAr_r`rar&rrrr'test_loc_keyerror_rightmost_key_missings "rcCs8tddg}tddg|d}|jd}|dksJdS)Nr)r)r4r!r r r)rrrr&)rrlrmrrr*test_multindex_series_loc_with_tuple_labels r)8numpyrr_ pandas.errorsrrpandasrrrrrpandas._testing_testingrBfixturerrrrrr IndexSlicerrrrrrrrr.rBrMrPrTr[rgrprqr{rrrrrrrrrrrrrrrrrrrsx             $