U
|
ý°dÉ2 ã @ sT d dl Zd dlZd dlmZmZmZmZ d dlm Z
|
d dl m Z ej ddd dd dd g¡ej d d ed
|
gd gd fd
|
ed d gd
|
d gd fg¡dd Zej dddg¡dd Zdd Zdd Zej ddd edfd d edfd d edfd d e d fd d ed!fd"d ed!fd#d ed$fg¡d%d& Zd'd( Zd)d* Zej d+d,d d-fd.d d/fg¡d0d1 Zd2d3 Zej d4d5d ed fd6d ed d7fd8d ed d7fg¡d9d: Zd;d< Z d=d> Z ej d?d@gdAej gfed d gd7dBgge dCd@ej fg¡dDdEfdFd@ged
|
d d gdGd7dBgge dHdCd@ej fg¡dDdEfd@ged d gd7dBgge dCd@ej fg¡dDdEfd@gdAgfed gd7gge dCg¡dDdEfd@gej gfed gdBgged
|
gdIggdFd@gdAdJggdKdDdEfd@ej fed dBgdDd@ej fdLfg¡dMdN Z ej!dOdP Z"ej dQdRd dSd g¡dTdU Z#dVdW Z$dXdY Z%dZd[ Z&d\d] Z'dS )^é N)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeries)Ú IndexingErrorÚ access_methodc C s | d d
|f S ©N© ©ÚsÚxr r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_getitem.pyÚ<lambda> ó r c C s | j d d
|f S r ©Úlocr
|
r r r r r c C s | j |ddS )Né )Úlevel)Úxsr
|
r r r r r zlevel1_value, expectedr ©Úindexé é c C sL t jdddgddgd}tddd g|d
|
}d|j_| ||}t ||¡ d S ) N)r r )r r )r r ÚAÚB©Únamesr r r r )r Ú from_tuplesr r ÚnameÚtmÚassert_series_equal)r Z level1_valueÚexpectedÚmiZserÚresultr r r Ú test_series_getitem_multiindex s
|
|
r$ Ú level0_valueÚDr c C s t | ddgddddddd d
|
ggdddd d d d d d d g
|
d d dddd d d ddg
|
gddgd}tj t|d ¡}t||dgd}| dkr¦tjtdd |j d W 5 Q R X tjtdd |j d W 5 Q R X |j | }t
|
| ¡ dd
dt dddgdd d }t ||¡ d S ) Nr ÚCr é é é% é9 éC éK éR r r r é é é é ÚtagÚday©ÚlevelsÚcodesr Úval©r Úcolumnsr z^'A'$©Úmatchz^'X'$ÚX©r )r r )r ÚnpÚrandomÚrandnÚlenr ÚpytestÚraisesÚKeyErrorr8 r Zravelr r r )r% r ZarrÚdfr# r! r r r Ú)test_series_getitem_duplicates_multiindex& s& .ý
|
ÿrG c C sL | d }| |jdd
¡}|j d¡ d¡|_||d }t ||¡ d S )Nr é* éA r )éÐ r )Úreindexr Ú droplevelr r ©Ú/multiindex_year_month_day_dataframe_random_dataZ
|
indexer_slr r! r# r r r Útest_series_getitemB s
|
rO c C s. | d }|j d }||d }||ks*td S )Nr é1 )rJ r é
|
)ÚilocÚAssertionErrorrM r r r Ú"test_series_getitem_returns_scalarK s
|
rT z)indexer,expected_error,expected_error_msgc C s
|
| d¡S ©N)rJ r r/ )Ú __getitem__©r r r r r X r z^\(2000, 3, 4\)$c C s | d S rU r rW r r r r Y r c C s
|
| j d S rU r rW r r r r Z r c C s
|
| j d S )N)rJ r r/ r1 r rW r r r r [ r zToo many indexersc C s | t| ¡S r )rV rB rW r r r r \ r Ú c C s | t | S r )rB rW r r r r ] r c C s | j t| S r )rR rB rW r r r r _ r z*single positional indexer is out-of-boundsc C s. | d }t j||d || W 5 Q R X d S )Nr r; )rC rD )rN ÚindexerZexpected_errorÚexpected_error_msgr r r r Ú#test_series_getitem_indexing_errorsU s r[ c C s6 | d }|dd |D }||dk }t ||¡ d S )Nr c s s | ]}|d kV qdS )r Nr )Ú.0r r r r Ú <genexpr>t s z7test_series_getitem_corner_generator.<locals>.<genexpr>r )r r )rN r r# r! r r r Ú$test_series_getitem_corner_generatorp s r^ c C s2 | j }|jd d
df }|d j}t ||¡ d S )Nr )ÚfooZone)ÚTÚvaluesr Zassert_almost_equal)Ú multiindex_dataframe_random_datarF r! r# r r r Útest_getitem_simple~ s
|
rc zindexer,expected_error_msgc C s | d S )N)r_ Zfourr ©rF r r r r r z^\('foo', 'four'\)$c C s | d S )NZfoobarr rd r r r r r z
|
^'foobar'$c C s, | j }tjt|d || W 5 Q R X d S )Nr; )r` rC rD rE )rb rY rZ rF r r r Ú#test_frame_getitem_simple_key_error
s
|
re c C sh t dddgdddgd} ddgdd gd
|
d gg| _| d }t dgdgdgg| jd
|
gd }t ||¡ d S ) NÚ1Ú2Ú3Ú4)ÚaÚbz level1 item1z level1 item2rX z level2 item2z level3 item1z level3 item2r9 )r r: r r Úassert_frame_equal©rF r# r! r r r Ú*test_frame_getitem_multicolumn_empty_level s ý ÿrn zindexer,expected_slicec C s | d S )Nr_ r rd r r r r ¦ r c C s | d S ©NÚbarr rd r r r r § r r1 c C s | j d d
df S ro r rd r r r r ¨ r c C s> | j }|j|j| d}|j d¡|_||}t ||¡ d S )N©r: r )r` rK r: rL r rl )rb rY Zexpected_slicerF r! r# r r r Útest_frame_getitem_toplevel£ s
|
rr c C s¦ ddddddgddddddgdd d
|
dddgg} t t| }t |¡}ttj d d ¡|d }|d }|d d¡}t
|
||¡ |d }|d }| d¡}t
|
||¡ d S )Nrj ÚtopÚroutine1Zroutine2rX ZODÚresult1Zresult2ZwxZwyr/ r0 rq )rj rX rX )rt ru )rt ru rX ) ÚsortedÚzipr r r r? r@ rA Úrenamer r )ÚarraysZtuplesr rF r# r! r r r Útest_frame_mixed_depth_getµ s ý
|
|
rz c C sr | }dddg}t d|dgd|dgd|d gd
|
|d gg|d ddg¡}|d d ¡|d< d|f}|jd |
}t d|dgd|dgg|d ddg¡}|d d ¡|d< t ||¡ |j|d
}t d|dgd|d gd
|
|d gg|d ddg¡}|d d ¡|d< t ||¡ d|fd|f }}|j||
}t d|dgd|d gg|d ddg¡}|d d ¡|d< t ||¡ d S )Nrj rk Úcé é é é é é! é) é+ rq Úint64)r Z set_indexZastyper r rl ) Ú nulls_fixtureÚnÚcolsrF Úidxr# r! Úidx1Úidx2r r r Ú!test_frame_getitem_nan_multiindexÊ s8
|
"þý& ÿþ &r zindexer,expectedrk rp r0 ©rk rp r ©r: Údtyperj r/ ©rj r_ éÿÿÿÿr_ )r7 r6 )r r c C sP t dddgdddggt ddd |fg¡d
|
d }|jd d
| f }t ||¡ d S ) Nr r r r/ r1 r0 r r rk r r )r r r r r Z assert_equal)rY r! r
rF r# r r r Ú&test_frame_getitem_nan_cols_multiindexé s C ÿûr c C s ddddddgddd d
|
d d gddd d
|
d d gg} d ddg}t ddgddddggddddddgdd
|
d
|
ddd ggddgd}t| ||dS )z;Fixture for DataFrame used in tests for gh-4145 and gh-4146rj ÚdÚer{ Úfrk r r/ r1 r r0 r Úh1Úh3Úh5r r ÚA1ÚA2ZB1ÚB2r ÚmainÚsubr5 r9 )r r )Údatar r: r r r Ú dataframe_with_duplicate_index= s .
|
ýr rY c C s | d S ©N©r r r rd r r r r K r c C s | j d d
df S r r rd r r r r K r c C sV | }t dddg}tjdgddgd}tdd d gg||d
|
j}||}t ||¡ d S ) Nr r r r r r r rj r r9 )r r r r r` r rl )r rY rF r r: r! r# r r r Útest_frame_mi_accessJ s r¡ c C s: | }t dddgdddgdd}|d d }t ||¡ d S ) Nrj r r r r r )r r r )r r r ©r rF r! r# r r r Ú#test_frame_mi_access_returns_seriesX s r£ c C sR | }t dddgdddggtddgdddd d
|
gd j}|d d }t ||¡ d S ) Nr r/ r r1 r r r> r r r r9 r )r r r` r rl r¢ r r r Ú"test_frame_mi_access_returns_framec s ý r¤ c C s\ t dtdt dgdgg¡d} | g }t ddgtdgdggg g gdd}t ||¡ d S )Nr r r r9 )r6 r7 )r Úranger Z from_productr rl rm r r r Útest_frame_mi_empty_sliceo s ÿr¦ c C sô ddddgddddgg} t j| dd}tddddg|d gd
|
}|j|jd d
d f d kd d
f j}|j|d d
f }|jd gt|j d d
f }t ||¡ d |j|j|jd d
d f d k jd f< |}tddddg|d gd
|
}t ||¡ d S )Nrj rk )r r r r r r r/ Úvaluer9 r Fr1 )r Z from_arraysr r r rB r rl )ry r rF Zempty_multiindexr# r! r r r Útest_loc_empty_multiindexy s & (r¨ )(Únumpyr? rC Zpandasr r r r Zpandas._testingZ_testingr Zpandas.core.indexingr ÚmarkZ parametrizer$ rG rO rT rE Ú
|
IndexErrorr[ r^ rc re rn Úslicerr rz r Únanr r Zfixturer r¡ r£ r¤ r¦ r¨ r r r r Ú<module> sÚ þ*þ
|
|
ýùþ
|
|
|
þþ
|
ýþ
|
|
ýý ÿûý ýý
|
|
|
ýý
|
|
ÿûý Éþ<
|
|
ÿ
|
|
