U
|
ý°d ã @ sP d dl Zd dlZd dlmZmZ d dlZd dlmZm Z m
|
Z
|
m Z d dl m Z ejdd Zejdd ZG dd d Zej d
|
g g fd ged fd d gg fd dgg fd gg fejdd
dgf eddd fejdd
ddgf eddd fg¡dd Zej dg edfdgg fg¡dd Zdd Zdd Zdd Zd d Z d d Z d!d" Z d#d$ Z d%d& Z d'd( Z!d)d* Z"d+d, Z#d-d. Z$d/d0 Z%d1d2 Z&G d3d4 d4Z'd5d6 Z(d7d8 Z)d9d: Z*d;d< Z+d=d> Z,d?d@ Z-ej dAdBddCdDdd dEdCgdBddCdDd dEgg¡dFdG Z.dHdI Z/dJdK Z0dLdM Z1dNdO Z2dPdQ Z3dRdS Z4dTdU Z5dVdW Z6dXdY Z7dS )Zé N)Ú IndexingErrorÚPerformanceWarning)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesc C s$ t ddddggddddggd gd
|
S ) zsingle level MultiIndexÚfooÚbarÚbazZquxr é é é Úfirst)ÚlevelsÚcodesÚnames)r © r r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_loc.pyÚsingle_level_multiindex s
|
ÿr c C sT ddgdddgg} ddddddgddddddgg}t | |d}ttj dd¡|dS )Nr r r )r r é ©Úindex)r r ÚnpÚrandomÚrandn)r r r r r r Ú%frame_random_data_integer_multi_index s r c @ sP e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j dddgddgfdgg fddgg fg¡dd Zdd Zd d Zd d Zd!d" Zd#d$ Ze j d%eeeeejee f¡e j d&eeeeejee f¡d'd( Z d)d* Z e j d+d,d-g¡d.d/ Z e j d0ed1d2fd3ej fg¡d4d5 Z!d6d7 Z"d8d9 Z#d1S ):ÚTestMultiIndexLocc C sV |}d|j d< |j d dks t| ¡ }ttd|_d|j d< |j d dksRtd S )Né )©r ÚtwoÚBr é )r r )ÚlocÚAssertionErrorÚcopyÚlistÚrangeÚcolumns)ÚselfÚ multiindex_dataframe_random_dataÚframeÚdfr r r Ú&test_loc_setitem_frame_with_multiindex$ s
|
|
z8TestMultiIndexLoc.test_loc_setitem_frame_with_multiindexc C s |}ddddddddddddd d
|
d d d dd }t |}| ||d ¡}|jd dgd}d}t t¡" t |j| |jdd
¡ W 5 Q R X |j dd}|d ksªt
|
|j| }t j dgd |d}t j d gd |d} t j|| gd dgd}
|
t ddddgi|
|
d} t || ¡ d S )Ni¼ iX éÞ iM i¼ )r r r r é g @ç @é é )ÚamountÚcolÚnum)r3 r4 r3 r4 )Úkeys)r/ r0 r T)Zinplacer ©Údtype©r r2 r )r ZastypeÚ set_indexÚtmZassert_produces_warningr Úassert_frame_equalr" ÚilocÚ
|
sort_indexr# r Úarrayr Ú from_arrays) r( Zany_real_numpy_dtyper7 Údatar+ ÚkeyZ return_valueÚresZcol_arrZyear_arrr Úexpectedr r r Útest_loc_getitem_general/ s& ý &
|
z*TestMultiIndexLoc.test_loc_getitem_generalc C s` t tj dd¡dddgdddggdddgdddggd}tjtd d
|
|jd W 5 Q R X d S © Nr r r. r é é
|
r0 ©r' r z^2$©Úmatch©r r r r ÚpytestÚraisesÚKeyErrorr" ©r( r+ r r r Ú0test_loc_getitem_multiindex_missing_label_raisesL s ýzBTestMultiIndexLoc.test_loc_getitem_multiindex_missing_label_raisesc C s: |d }| |jdd
¡}|jddg }t ||¡ d S )NÚAé1 é3 )éÐ r rG )rT r é )Úreindexr r" r: Úassert_series_equal)r( Ú/multiindex_year_month_day_dataframe_random_dataÚserrC Úresultr r r Ú/test_loc_getitem_list_of_tuples_with_multiindexW s zATestMultiIndexLoc.test_loc_getitem_list_of_tuples_with_multiindexc C s t dddgdddgg¡}t|tdtjd}tddg}td ddd
|
d d gt ddgdddgg¡tjd }|j| }t ||¡ |jddg }t ||¡ tddgddgd}|j| }t ||¡ tg tjd}tg t |j g g gtjdtjd}|j| }t ||¡ d S )Nr r r rQ r ÚCé ©r r@ r7 r r r! rF ©r@ r r7 r )r@ r7 ©r r r7 ©r r7 )
|
r Ú from_productr r& r Úfloat64r" r: rW r )r( r ÚxÚyrC rZ Úy1Úemptyr r r Útest_loc_getitem_series_ s. ý
|
|
ý
|
z)TestMultiIndexLoc.test_loc_getitem_seriesc C s t dddgdddgg¡}t|tdtjd}t ddg¡}td ddd
|
d d gt ddgdddgg¡tjd }|j| }t ||¡ t g ¡}tg t |j
|
g g gtjddd}|j| }t ||¡ t d¡}td ddgdddgtjd }|j| }t ||¡ d S )Nr r r rQ r r\ r] r^ r r r! rF r_ r` rc ra ) r rb r r& r rc r> r" r: rW r Úint64)r( r rd re rC rZ rg Zscalarr r r Útest_loc_getitem_array~ s, ý
|
|
ý
|
|
|
z(TestMultiIndexLoc.test_loc_getitem_arrayc C s t tj dd¡dddgdddggdddgdddggd}|jd d
|
g d ¡}|jd }t ||¡ |jd d
d gf jd d
|
d }|jd d
df }t ||¡ |jd gd gf d ¡jd d
|
d }|jd jd d
df }t ||¡ |jd d
|
g }|jd }t ||¡ d S )Nr ÚiÚjrQ r ÚXÚYrH r r r ©Úaxis)rk rm © r r r r r< Ú droplevelr" r: r; ©r( r+ rC rZ r r r Útest_loc_multiindex_labels s" ý
|
"
|
z,TestMultiIndexLoc.test_loc_multiindex_labelsc C sf t tj dd¡dddgdddggdddgdddggd}|jd d
|
g d ¡}|jd }t ||¡ d S ) Nr r r. r rF rG r0 rH r r rq rs r r r Útest_loc_multiindex_ints· s ý
|
z*TestMultiIndexLoc.test_loc_multiindex_intsc C s` t tj dd¡dddgdddggdddgdddggd}tjtd d
|
|jd W 5 Q R X d S rE rK rO r r r Ú(test_loc_multiindex_missing_label_raisesÁ s ýz:TestMultiIndexLoc.test_loc_multiindex_missing_label_raiseszkey, posr r. r r r c C s` t tj dd¡dddgdddggdddgdddggd}tjtd d
|
|j| W 5 Q R X d S ) Nr r r. r rF rG r0 rH z not in indexrI rK )r( rA Úposr+ r r r Ú&test_loc_multiindex_list_missing_labelË s ýz8TestMultiIndexLoc.test_loc_multiindex_list_missing_labelc C s t tdt ddgddgddgg¡d}tjtd d
|
|jd W 5 Q R X tjtd d
|
|jd W 5 Q R X tjtdd
|
|jd W 5 Q R X d S )NrF ÚaÚbÚcÚdÚeÚfr z^\('a', 'b'\)$rI ©ry rz z^\('a', 'd', 'g'\)$)ry r| ÚgzToo many indexers)ry r| r rl ) r r& r rb rL rM rN r" r )r( Úsr r r Ú(test_loc_multiindex_too_many_dims_raises× s þz:TestMultiIndexLoc.test_loc_multiindex_too_many_dims_raisesc C sÐ dd t dD }dd t dD }t ||g¡}dtj dd¡ d }t||d }|| }t ||¡ tt d
|
¡
|
d d¡t dd d dgdd d gg¡d}|j dd gfd d
f }|j dd g }t ||¡ d S )Nc S s g | ]}d t | qS )Ú Attribute©Ústr©Ú.0rk r r r Ú
|
<listcomp>è s zFTestMultiIndexLoc.test_loc_multiindex_indexer_none.<locals>.<listcomp>r c S s g | ]}d t | qS )ÚValuer r r r r r é s r g¹?rG g à?©r' r0 éÿÿÿÿr r r. r ) r& r rb r r r r r: r; ÚarangeÚreshaper" )r( Ú
|
attributesZattribute_valuesr r+ rZ rC r r r Ú test_loc_multiindex_indexer_noneå s þz2TestMultiIndexLoc.test_loc_multiindex_indexer_nonec C s
|
t tjdddt tddddgg¡}|jd d
dd
f }|jdd
dd
f }t ||¡ |jd d
dd
f }t ||¡ |jdd
dd
f }t ||¡ t tjdddt tddddgg¡}|j d
|
d d d ddg }|jdd d
dd
f }t ||¡ d S )Né ri r6 r ry rz r{ r r. r r! rF r0 rU é r )
|
r r r r rb r& r" r: rW r< )r( r rC rZ r r r Ú test_loc_multiindex_incompleteü s$ þ þz0TestMultiIndexLoc.test_loc_multiindex_incompletec C s4 |}t tj t|¡|d}|jD ] }|| q"d S )Nr )r r r r ÚlenÚvalues)r( r Z single_levelr Úkr r r Útest_get_loc_single_level s
|
z+TestMultiIndexLoc.test_get_loc_single_levelc C s t dddgddgg¡}ttj dd¡||}|jdd
d d
f }|}t ||¡ t dddgddgg¡}ttj dd¡||}|jdd
d d
f }|j d d
}t ||¡ |jdd d
f }|j d
|
d
}ddg|_
|
t ||¡ |jd d
df }|d }t ||¡ d S ) Nr r! rF ry rz rG é é r r ) r rb r r r r r" r: r; r< r )r( r r+ rZ rC r r r Útest_loc_getitem_int_slice s"
|
z,TestMultiIndexLoc.test_loc_getitem_int_sliceÚindexer_type_1Úindexer_type_2c sò dd dddg}dddg}t ||g¡}ttjt|d d
|
|d gd }ddgddgf}||f}t fd dt||D } |tks|tkr¸t j
|
t dd |j | d f W 5 Q R X d S |j | d f }
|
t ddddgd t |¡d} t |
|
| ¡ d S )Nc S s, | t jkrt |¡S | tkr$t| S | |S ©N)r Úndarrayr> Úslice)Ú indexer_typer5 r r r Úconvert_nested_indexerC s
|
|
|
zQTestMultiIndexLoc.test_loc_getitem_nested_indexer.<locals>.convert_nested_indexerrG r r r r r ri r6 ÚData©r r' c 3 s | ]\}} ||V qd S r r )r r r ©r r r Ú <genexpr>V s ÿzDTestMultiIndexLoc.test_loc_getitem_nested_indexer.<locals>.<genexpr>z as an indexer is not supportedrI r. r ©Únamer )r rb r r r r ÚtupleÚzipÚsetrL rM Ú TypeErrorr" r r: rW ) r( r r ry rz r r+ r5 ÚtypesÚindexerrZ rC r r£ r Ú test_loc_getitem_nested_indexer8 s2
|
|
ÿ þ
|
ÿz1TestMultiIndexLoc.test_loc_getitem_nested_indexerc C sH t ddg¡}|ddg|d}d|jd< |ddg|d}t ||¡ d S )N©ry rQ ©rz rQ r r r r ©ry )r Ú from_tuplesr" r: Z assert_equal)r( Zframe_or_seriesÚmiÚobjrC r r r Ú)test_multiindex_loc_one_dimensional_tupleg s
|
|
z;TestMultiIndexLoc.test_multiindex_loc_one_dimensional_tupler¬ r° ry c C sP t ddg¡}tddg|d}d|j|d d
f< tddg|d}t ||¡ d S )Nr® r¯ r r r r ©r r± r r" r: r; )r( r¬ r² r³ rC r r r Ú-test_multiindex_one_dimensional_tuple_columnso s
|
z?TestMultiIndexLoc.test_multiindex_one_dimensional_tuple_columnszindexer, exp_valueNç ð?©r r c C sx t ddg¡}tddgddgg|ddgd }d
|
|j|d d gf< tddd
|
d
|
gdd||gg|ddd d gd }t ||¡ d S ) Nr¸ ©r r. r r r r. ry rz r¢ r· r{ r| rµ )r( r¬ Z exp_valuer² r+ rC r r r Ú)test_multiindex_setitem_columns_enlargingx s
|
ýz;TestMultiIndexLoc.test_multiindex_setitem_columns_enlargingc C sÔ t tjdddtddgg¡}td|d}td|d d
d}tj||gdd}| ¡ }|jd d
|
}t
|
||¡ t ||d }|jd d
|
}t
|
||¡ tj|| |j ¡gdd}|jd d
|
}t
|
||¡ d S ) NZ20110101r ©Úperiodsry rz r r ro z
|
2011-01-01z
|
2011-01-02©r r )r rb ÚpdÚ
|
date_ranger r Úconcatr$ r" r: r; r rV r )r( ZmidxZser1Úser2r+ rC rZ r r r Ú"test_sorted_multiindex_after_union s ÿ z4TestMultiIndexLoc.test_sorted_multiindex_after_unionc C st t t tdtdtdg¡dgd}|jtjd d
dd d
f }t t tdtdg¡dgd}t ||¡ d S )NÚabZcdr} ZValr¢ r{ ) r r rb r% r" r Zs_r: r; )r( r+ rB rC r r r Ú test_loc_no_second_level_index s þ ÿz0TestMultiIndexLoc.test_loc_no_second_level_index)$Ú__name__Ú
|
__module__Ú __qualname__r, rD rP r[ rh rj rt ru rv rL ÚmarkÚ parametrizerx r r r r r r% r§ r© r r r r r r r´ r¶ Únanrº r rÄ r r r r r # sF
|
|
,
|
ÿ ÿ )
|
ÿ
|
r z indexer, posrQ r ÚDÚEr r Úbahc C sÖ t jdddgdddggddgd }ttjd
|
d d |d ¡ }|j| }|jdkr| g krtj t
|
t | d |j | W 5 Q R X nP| t d ddgfkr¼tj t
|
dd |j | W 5 Q R X n|j | }t ||¡ d S )NrQ r r\ r r r
|
Úoner r8 r] ri r6 r r rI rÍ z'bah')r rb r r r r= r< ÚsizerL rM rN r
r" r r: rW )r¬ rw ÚidxrY rC rZ r r r Ú7test_loc_getitem_duplicates_multiindex_missing_indexers¨ s ÿ
|
|
rÑ Úcolumns_indexerc C s~ t dddgddgf¡}ttj dd¡td|d}|jd d
|
d }ttd| g ¡d d}|j d d
| f }t
|
||¡ d S ) Nr r r
|
ÚalphaÚbetar r r¢ r r )Úlevelrp ) r rb r r r r r& r= rV r" r: r; )rÒ Z multi_indexr+ rC rZ r r r Ú4test_loc_getitem_duplicates_multiindex_empty_indexerÉ s rÖ c C sL t tjtjgddggt ddg¡ddgd} | jd }tj}||ksHtd S ) NÚmeanÚmedian©Úfunctsr× )rÚ rØ Úfunctionr¦ rH )rÛ rÙ )r r r× rØ r r± r" r# ©r+ rZ rC r r r Ú=test_loc_getitem_duplicates_multiindex_non_scalar_type_objectÖ s ý
|
rÝ c C sd t t d¡t d¡tj d¡tj d¡d ddg¡} | jd }| jdd d
f }t ||¡ d S )NrG )ry rz r{ r| ry rz )r r ) r r r r r r9 r" r: rW )r+ rC rZ r r r Ú!test_loc_getitem_tuple_plus_sliceã s
|
|
üÿù
|
rÞ c C s8 | }|j d }|dd
}|j d¡|_t ||¡ d S )Nr éýÿÿÿr )r" r rr r: r; )r r+ rZ rC r r r Útest_loc_getitem_intò s
|
|
rà c C s, | }t jtdd |jd W 5 Q R X d S )Nz^3$rI r )rL rM rN r" )r r+ r r r Ú%test_loc_getitem_int_raises_exceptionú s rá c C sT | }t jtdd |jd W 5 Q R X d|jd< d}| ¡ jd }||ksPtd S )Nz^\('bar', 'three'\)$rI ))r Úthreer r )rL rM rN r" r= r# )r) r+ rC rZ r r r Ú test_loc_getitem_lowerdim_corner s
|
rã c C sÀ t dtdt dgdg¡d} d| d< | j| jd d
df< t t | d ¡t | j¡¡ t tj
|
t dt d d
|
d g¡d} | ¡ }t d¡| jd d
d f< t d¡|jd d
d f< t | |¡ d S )NÚstringZabcdZMain)ZanotherrÎ r¢ ry Úlabelsr. )rQ Ú1)rQ Ú2)r ræ r r )r r% r rb r r" r: Úassert_numpy_array_equalr ZasarrayrÊ r& r± r$ r r< r; )r+ rC r r r Ú$test_loc_setitem_single_column_slice s" ý ýré c
|
C sÆ ddt jfddt jfddt jfddt jfdddd t jfdd
|
t jfg} tt d ¡td d ddgtj| dddgdd}|jd jd }tt d¡tt jgdddtd d ddgddd}t ||¡ d S )Nz Good Thingsr\ ÚRz
|
Bad ThingsÚT)ú Okay ThingsÚNr )rì rí rË rì r rË )rF r. Úd1Zd2Zd3Zd4Úu1Úu2Úu3r8 rH )r r. Úobject)r7 r¦ r6 r¢ )
|
r rÊ r Zonesr r r± r" r: r; )Ztupsr+ rZ rC r r r Útest_loc_nan_multiindex% s*
|
|
|
|
|
|
ø
|
ýýró c C sr t jdddd} d}tj| |fdd}t|dd }|jd
|
}|j| d d fd f }t |¡s`tt |¡sntd S )NÚ2013Q1Z2013Q4ÚQ)Úfreq)éW i® i )ZPeriodZCVRr8 ) ÚOMSZOMKZRESZ DRIFT_INDZ
|
OEVRIG_INDZFIN_INDZVARE_UDZLOEN_UDZFIN_UDr¢ ))rô r÷ rø r r÷ rø ) r¾ Z period_ranger rb r r" r Úisnanr# )ry rk rÐ r+ rZ Zaltr r r Ú test_loc_period_string_indexing? s þ
|
rú c C s° t ddg¡} tj| | gddgd}tddgdd gd
|
d gd d gg|d dgd}|j| d |j d¡dkfd f }tdgd tj t
|
d¡t
|
d¡fgddgdd}t ||¡ d S )Nz
|
2017-05-04z
|
2017-05-05ZIdx1ZIdx2r8 r r r r. r r r! ZC1ZC2©r@ r r' r r¥ ) r¾ Ú to_datetimer rb r r" r Zget_level_valuesr r± Ú Timestampr: rW )Zdt_idxZm_idxr+ rZ rC r r r Ú test_loc_datetime_mask_slicing` s" ÿ"þýrþ c C sP t d¡} tdtjd| fgddgddd}|jd d
| gf }t ||¡ d S )NÚ2000r ry rz r8 r{ )r r¦ )r¾ rý r r r± r" r: rW )ÚdaterY rZ r r r Ú&test_loc_datetime_series_tuple_slicings s
|
ýr c C s° t ddgddgddgddggtjddd d gd
|
d gd d dgd} tjdd gd
|
d gd }| j|d d
f }t ddgddgddggtjdd d gd
|
d gd d dgd}t ||¡ d S )Nry r r rz r{ r r½ )r r ©r r r r r8 ÚauthorZpricerû r¢ )r r r± r" r: r; )r+ rÐ rZ rC r r r Útest_loc_with_mi_indexer s
|
ÿûýr c C sÒ t jdddd} ttd| d}| ¡ }| |d< |jddd }|jjd
|
ksNt|j | d ¡ |j
|
| d }|j dg d ¡}t ||¡ |d }|jjd
|
ks¢t|j
|
| d }|j dg d ¡}t ||¡ d S ) Nz
|
2016-01-01r z
|
US/Pacific)r¼ Útzr r r T)Úappend)Nr )r¾ r¿ r r& Zto_framer9 r r r# Úget_locr" r< rr r: r; rW )ZdtirY r+ Údf2rZ rC rÁ r r r Ú test_loc_mi_with_level1_named_0 s r c C s| | dddd}t j|dgd}| ddg¡ ¡ }|jdtd fd d
f d
|
¡}|jd jtd d d
f }t ||¡ d S ) Nr Úmerger@ z quotes2.csvÚtime)Z parse_datesZtickerZAAPLz2016-05-25 13:30:00r ) r¾ Zread_csvr9 r= r" r rr r: r; )ÚdatapathÚpathr+ r rB rC r r r Útest_getitem_str_sliceª s r c C sx t jddddgddd} ddd d
|
g}dd d d g}t | ||g¡}ttd|tjd}|j| d ddf }|dkstt d S )Nz 20181101 1100z 20181101 1200z 20181102 1300z 20181102 1400Údatetimer )r¦ rö rQ ÚZÚWr\ rõ ÚFr. ra r g )
|
r¾ Z PeriodIndexr r? r r& r rc r" r# )ÚpiZlev2Zlev3r² rY rZ r r r Ú!test_3levels_leading_period_indexµ s
|
ý r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestKeyErrorsWithMultiIndexc C sV t t d¡ dd¡dddgd}| ddg¡}tjtdd |jd
|
W 5 Q R X d S ) Nr0 r. r rQ r r\ r ræ rI ©r r ) r r r r r9 rL rM rN r" )r( r+ r r r r Ú!test_missing_keys_raises_keyerrorÆ s z=TestKeyErrorsWithMultiIndex.test_missing_keys_raises_keyerrorc C sD t dt ddggd ¡d}tjtdd |jd W 5 Q R X d S ) Nr r r r r z\(0, 3\)rI )r r )r r rb rL rM rN r" )r( rY r r r Ú!test_missing_key_raises_keyerror2Î s z=TestKeyErrorsWithMultiIndex.test_missing_key_raises_keyerror2c C sô t jt ddddg¡t ddddg¡t ddddg¡gdd d
|
gd }ttj d d ¡|d}d}tjt |d" |j
|
ddt d fd d
f W 5 Q R X tjt |d |j ddt d f¡ W 5 Q R X tjt dd |j
|
dd d
f W 5 Q R X d S )Nry rz ræ rç Ú3r{ r| rÎ r râ r8 r. r r z'\('b', '1', slice\(None, None, None\)\)rI z \('b', '1'\))rz ræ )r r? r r> r r ZrandrL rM rN r" r r Zget_locs)r( r² r+ Úmsgr r r Ú test_missing_key_combinationÕ s ýú& z8TestKeyErrorsWithMultiIndex.test_missing_key_combinationN)rÅ rÆ rÇ r r r r r r r r Å s r c C s2 | }|d }|d }|j d d }t ||¡ d S )NrQ )rT r )r" r: rW )rX r+ rY rZ rC r r r Ú test_getitem_loc_commutabilityé s
|
r c C s t dddgddtjgdddgd dd g¡} | jdg }t d
|
dgitjd gdd gd d }t ||¡ | jd }t d
|
dgit dgd dd }t ||¡ d S )Nr r r ry r| r. )r3 Úind1Úind2r r r3 )ry r r8 r ©r¦ )
|
r r rÊ r9 r" r r± r: r; r rÜ r r r Útest_loc_with_nanñ s ÿþ ÿ
|
r c C sR t ddddggddddgd dddg¡} tjtd
|
d | jd W 5 Q R X d S ) Nr r r r. ry rz r{ r| r z\(2\.0, 2\.0, 3\.0\)rI )ç @r! ç @)r r9 rL rM rN r" ©r+ r r r Ú test_getitem_non_found_tuple s
|
ÿr$ c C sæ t jddd} t | g¡}| d¡tddd ks4t| d¡tddd ksNt|d d d
d¡}| d d d
d¡}||kst| d¡ d¡}| d¡ d¡}||ks®t| |¡ d¡}| | ¡ d¡}t
|
| ¡ d |¡ d S )Nz
|
2001-01-01éd r» z2001-01r é r ) r¾ r¿ r r? r r r# Úrepeatr r: rè Znonzero)r r² r" rC r r r Útest_get_loc_datetime_index
|
s r( c C s t ddgddgg¡} tddgddgdd gd
|
d gg| d }dddgf}t d dgddgg¡|j|d d
f< tddgd dgdd gd
|
d gg| d }t ||¡ d S )Nry rz r r r r r. r r r! rF r r] rG r1 r0 )r rb r r r> r" r: r; )r² r+ r¬ rC r r r Ú,test_loc_setitem_indexer_differently_ordered s $ $$r) c C sª t ddgddgddgddggd d d
|
d
|
gddddggd d
|
gd } | jtd ddgfd d
f }t ddgddgddgddggd d
|
d d
|
gddddggd d
|
gd }t ||¡ d S ) Nr r r r. r r r! rF ry rz r¢ )r r" r r: r; rÜ r r r Ú5test_loc_getitem_index_differently_ordered_slice_none+ s ý ýr* r¬ r r! r rF c C sº t dgd t ddddddd d
|
g¡d gd }|jtd | fd d
f }t dgd ddd dd dd d gdd d dddddggd gd }t ||¡ |j|jj| ddd d
f }t ||¡ d S )Nr rF r r¸ )r r! r )r r )r r )r rF )r r! ry r¢ r r! r r )rÕ ) r r r± r" r r: r; r Úisin)r¬ r+ rZ rC r r r Ú@test_loc_getitem_index_differently_ordered_slice_none_duplicates; s ÿû&ý r, c C sª t jdgdgdggdddgd} tdd gi| d
|
}| d d g¡}|jdd d
df }t ||¡ td g| d
|
}|jdd d
df }td gtdgdd d
|
}t ||¡ d S ) Nrd re Úzry rz r{ r8 r| r r r r )
|
r r? r rr r" r: r; r r rW )r² r+ rC rZ rY r r r Ú3test_loc_getitem_drops_levels_for_one_row_dataframeQ s r. c C sp t dddgddgg¡} tt d¡| d}|jd d
ddgf }tt d¡t d d
|
d d g¡d}t ||¡ d S ) NrQ r r\ r r )r r r )r r. )r r )r r )rQ r )rQ r ) r rb r r Úzerosr" r± r: r; )Úcolsr+ rZ rC r r r Ú$test_mi_columns_loc_list_label_order_ s þr1 c C s t t d¡ d¡ddddgddddggdddgd d
|
d ggd } d d g| j_ddg| j_tjt dd | j
|
ddgdf W 5 Q R X d S )Nr0 )r. r ry rz r r ZOhioZColoradoZGreenZRedr¢ Úkey1Úkey2ÚstateÚcolorz\[2\] not in indexrI ) r r r r r r r' rL rM rN r" )r* r r r Ú$test_mi_partial_indexing_list_raisesk s ý r6 c C sN t tdt ddgddgg¡d} tjtdd | jd d
|
g W 5 Q R X d S ) Nr. r r ry rz r z \['not' 'found'\] not in indexrI ÚnotÚfound)r r& r rb rL rM rN r" )r r r r Ú(test_mi_indexing_list_nonexistent_raisesx s "r9 c C s¼ t ddddgddddggd d gt ddgd
|
d gg¡d } d | jd< d| jd< d| jd< t ddddgddddgddd d gdtjtjtjggd d ddgt ddgd
|
d gg¡d }t | |¡ d S )Nr r r r r r. r! rF ry rQ r r¢ r r{ )r{ ©r rQ )r| r: r· r! g @g @r" r/ g @g @g ð¿r| )r r rb r" r rÊ r: r; )rZ rC r r r Ú'test_mi_add_cell_missing_row_non_unique s$ ý
|
|
|
|
|
|
ü
|
ø
|
r; c C s t dddtjdgdggddgdd } | jd
|
}|dks>tt|tjsNt| jd gd f jd }|dksptt|tjstd S )Nr· r r r r. r{ r| r8 r )r¹ rz r¹ rz r ) r r r? r" r# Ú
|
isinstancer ri r< )r+ rZ r r r Ú$test_loc_get_scalar_casting_to_float s ÿ
|
r= c C sf t jddgddggddgd} t| tjd}|jd }ttjtjgtddgdd d
|
}t ||¡ d S ) Nry rz rQ r r r r8 ra r r )
|
r rb r r rc r" rÊ r r: rW )rÐ Ús2rZ rC r r r Ú)test_loc_empty_single_selector_with_names£ s
|
|
r? c C sb t ddddddgddddddgtd d
|
} | d d g¡} tjtd d | jd W 5 Q R X d S )Nr% éÈ i, rG r é r& é! r )rQ r r\ rQ r z^1$rI )r% r )r r& r9 rL rM rN r" r# r r r Ú'test_loc_keyerror_rightmost_key_missing¬ s ýÿrC c C s8 t ddg¡} tddg| d}|jd }|dks4td S )Nr¸ )r )r. r r r r )r r± r r" r# )r² rY rZ r r r Ú*test_multindex_series_loc_with_tuple_label» s
|
rD )8Únumpyr rL Z pandas.errorsr r Zpandasr¾ r r r r Zpandas._testingZ_testingr: Zfixturer r r rÈ rÉ r Z
|
IndexSlicerÑ rÖ rÝ rÞ rà rá rã ré ró rú rþ r r r r r r r r r$ r( r) r* r, r. r1 r6 r9 r; r= r? rC rD r r r r Ú<module> sv
|
|
|
|
ùþ
|
|
! $ ,
|
|
