U
|
ý°dh= ã @ s d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ ejdd ZG dd dZG d d
|
d
|
ZdS ) é N)ÚSettingWithCopyError)Ú DataFrameÚIndexÚ
|
IndexSliceÚ
|
MultiIndexÚSeriesÚconcat)ÚBDayc C s t dddddgdddd d
|
gd d d ddgg¡} tddgddgdddgdddggdddgdddgddd gd ddggd d d d gd!}t| |td"d#S )$NçEØðôJYà¿çrùé·¯Àçî|?5^ºÝ¿çÌÑã÷6ý©?ç²ï§ÆK×?ç à- Ü?çÉTÁ¨¤ö?çúíëÀ9#Ò?ç9]ð?ç46<½RÆ?gá ©Qå¿g^)ËÇà¿g¶óýÔxéÖ¿gI&Âì?g=U«@ÚaÚxÚbÚqçı.n£$@ç 4@g >@é é é r é é ÚoneÚtwoÚthreeÚfour©ÚlevelsÚcodesÚnamesÚABCDE©ÚindexÚcolumns)ÚnpÚarrayr r Úlist)Úarrr) © r/ úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_xs.pyÚfour_level_index_dataframe s ýÿ "
|
ýr1 c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestXSc C st | ¡ }|jd }| |¡}| ¡ D ]>\}}t |¡rNt || | ¡sbtq$||| | ks$tq$dddddddd } t| }
|
|
|
d¡}|jtj kst|d
|
dks®t|d dks¾tt
|
j t t d ¡d | |jd t ¡ W 5 Q R X |jd
|
dd} |d
|
} t | | ¡ |jd
|
dd} d| d d
< |r^t |d
|
|d
|
¡ | dk ¡ rptn| dk ¡ sptd S )Nr r r )Ú1Ú2r3 r4 Ú3)r3 r4 r5 )ÚAÚBr6 r7 z Timestamp('1999-12-31 00:00:00')©Úmatchr )Úaxis)Úcopyr) ÚxsÚitemsr+ ÚisnanÚAssertionErrorr ÚdtypeZobject_ÚpytestÚraisesÚKeyErrorÚreÚescaper ÚtmÚassert_series_equalÚall) ÚselfZ float_frameZdatetime_frameÚusing_copy_on_writeZfloat_frame_origÚidxr< ÚitemÚvalueZ test_dataÚframeZseriesÚexpectedr/ r/ r0 Útest_xs' s6
|
|
|
|
ÿ zTestXS.test_xsc C s t dgd}d|d< d|d< d|d< d |d
|
< d |d < | d¡}tdddd d gtd dd}t ||¡ t dddgd}| d¡}tg dtjd}t ||¡ d S )Nr ©r) g ð?r6 Úfoor7 g @ÚCÚbarÚDg @ÚEr' )r) Únamer r Úc)rW r@ )r r< r r- rF rG r+ Úfloat64)rI Údfr< ÚexpÚresultrO r/ r/ r0 Útest_xs_cornerM s
|
|
zTestXS.test_xs_cornerc C sD t tj dd¡dddddgd}| d¡}|jd }t ||¡ d S )Nr r r rX r rQ )r r+ ÚrandomÚrandnr< ÚilocrF rG )rI rZ Zcrossr[ r/ r/ r0 Útest_xs_duplicates` s
|
|
zTestXS.test_xs_duplicatesc C s t ddddddddddddd d d
|
d g¡}|jdd d d }|d d
}t ||¡ |jdd d gd d }t ||¡ d S )NÚsatÚsun)r r Z
|
strawberryé
|
é éØ )ÚdayÚflavourZsalesÚyearri rh rg F)ÚlevelÚ
|
drop_levelr )rf rb )r Ú set_indexr< rF Úassert_frame_equal©rI rZ r\ rO r/ r/ r0 Útest_xs_keep_levelg s üÿù zTestXS.test_xs_keep_levelc C sÜ t t d¡ dd¡tdtdd}| ¡ }|rdt ¡ d| d¡d d
< W 5 Q R X t ||¡ nt|r°d}t
|
j t |d d| d¡d d
< W 5 Q R X | d¡dk ¡ rØtn(d| d¡d d
< | d¡dk ¡ sØtd S ) Nr r r r( é r zC\nA value is trying to be set on a copy of a slice from a DataFramer8 )r r+ ZarangeZreshapeÚranger; rF Z raises_chained_assignment_errorr< rm rA rB r Úanyr? rH )rI Úusing_array_managerrJ ÚdmÚdf_origÚmsgr/ r/ r0 Ú test_xs_vieww s $
|
zTestXS.test_xs_viewN)Ú__name__Ú
|
__module__Ú __qualname__rP r] ra ro rw r/ r/ r/ r0 r2 & s
|
&r2 c @ s2 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddgdgfg¡dd Z dd Z e j
|
ddddd eddgd d fdd d d edgdd fg¡d d Ze j
|
d!d"d d#d g¡d$d% Ze j
|
d!d&d d'd g¡d(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:S );ÚTestXSWithMultiIndexc C s ddddddddgddddddddgg}t t| }tj|ddgd }ttj d
|
d ¡d d dg|d}|jdddd}|j d d
dgf }t
|
||¡ d S )NrT ÚbazrR Úquxr r ÚfirstÚsecond©r& r é r6 r7 rS r( )r rT )r r~ r ©rj r: r ) r- Úzipr Ú from_tuplesr r+ r^ r_ r< r` rF rm )rI ZarraysZtuplesr) rZ r\ rO r/ r/ r0 Útest_xs_doc_example s þ z(TestXSWithMultiIndex.test_xs_doc_examplec C sx t dd}td}tj||gddgd}ttj t|d¡|dd d
|
g}|j ddd }|j
|
dd d
f }t ||¡ d S ) Niaß2ieß2ÚabcdeÚdateZsecidr r ÚXÚYÚZ©rj ) rq r- r Z from_productr r+ r^ r_ Úlenr< ÚlocrF rm )rI ÚdatesÚidsr) rZ r\ rO r/ r/ r0 Útest_xs_integer_key s
|
z(TestXSWithMultiIndex.test_xs_integer_keyc C sL |}|j ddd}||j d¡dk }tddddgd d
|
|_t ||¡ d S ) Nr r r r rR rT r| r} r~ ©rW )r< r) Zget_level_valuesr rF rm ©rI Ú multiindex_dataframe_random_datarZ r\ rO r/ r/ r0 Ú test_xs_level« s
|
z"TestXSWithMultiIndex.test_xs_levelc C s t j dd¡}tdddgdddgd d
|
d ggd d dgd d dgd d dggd}t||d}t|dd
dgdggd}|jd d d}t ||¡ d S )Nr r r Úpr r r ÚyrX ÚrÚzr r r ©r$ r% rQ r ©r+ r^ r_ r r r< rF rm )rI r. r) rZ rO r\ r/ r/ r0 Útest_xs_level_eq_2² s þ z'TestXSWithMultiIndex.test_xs_level_eq_2c C sf |}| ¡ }|jddd}|r,d|d d
< n*d}tjt|d d|d d
< W 5 Q R X t ||¡ d S )Nr r r rd úAA value is trying to be set on a copy of a slice from a DataFramer8 ©r; r< rA rB r rF rm )rI r rJ rZ ru r\ rv r/ r/ r0 Ú test_xs_setting_with_copy_error½ s z4TestXSWithMultiIndex.test_xs_setting_with_copy_errorc C sj |}| ¡ }|jdddgd}|r0d|d d
< n*d}tjt|d d|d d
< W 5 Q R X t ||¡ d S )N©r r r r" r rd r r8 r )rI r1 rJ rZ ru r\ rv r/ r/ r0 Ú(test_xs_setting_with_copy_error_multipleÏ s z=TestXSWithMultiIndex.test_xs_setting_with_copy_error_multiplez
|
key, level)r r r r c C sv |}t |gd }|jjdks"tt |jdddgd }t|trX|jt||d}n|j||d}t ||¡ d S )Nr Fr r r )
|
r r) Z is_uniquer? r< Ú
|
isinstancer- ÚtuplerF rm )rI Úkeyrj r rN rZ rO r\ r/ r/ r0 Útest_xs_with_duplicatesá s
|
z,TestXSWithMultiIndex.test_xs_with_duplicatesc C sv dddddddg}t |dd d
|
gd dd g¡}t d
|
d d dd gitdtjddgd dd}|jddd}t ||¡ d S )N)r r r )r Zbbcder )r Zyzcdeé )r Úxbcdeé )r Né )r Úzbcder¥ )r Úybcder¨ Za1Za2Zcnt©r* r§ r¨ r¥ r¦ r© rª r rQ r r )r rl r r+ Únanr< rF rm )rI ÚaccrZ rO r\ r/ r/ r0 Ú test_xs_missing_values_in_indexï s ù þz4TestXSWithMultiIndex.test_xs_missing_values_in_indexz key, level, exp_arr, exp_indexr Úlvl0c C s | d d
dd
f S )Nr r r/ ©r r/ r/ r0 Ú<lambda> ó z TestXSWithMultiIndex.<lambda>rT rR Úlvl1r c C s | d d
dd
f S )Nr r r/ r° r/ r/ r0 r± r² c
|
C s t j dd¡}tddgddddggddd d gdd d
|
d ggd d gd}t||d}|j||d d}t|||d} t || ¡ d S )Nr r r rT rR ZhelloZworldr r r r r¯ r³ r# r« r r )
|
rI r£ rj Zexp_arrZ exp_indexr. r) rZ r\ rO r/ r/ r0 Ú test_xs_named_levels_axis_eq_1 s ý z3TestXSWithMultiIndex.test_xs_named_levels_axis_eq_1Úindexerc C s | j dddgdS )Nr r r" r ©r< ©rZ r/ r/ r0 r± r² c C s | d¡j dddS )Nr r r" r r¶ r· r/ r/ r0 r± r² c C s` |}dddddgg}t dgdggdgdggd d
|
gd }t||td d }||}t ||¡ d S )Nr r r r r r r r r r! r# r' r( ©r r r- rF rm ©rI rµ r1 rZ Zexpected_valuesZexpected_indexrO r\ r/ r/ r0 Útest_xs_level_multiple s
|
|
ÿ ÿz+TestXSWithMultiIndex.test_xs_level_multiplec C s | j dddS )Nr r r r¶ r· r/ r/ r0 r± , r² c C s
|
| d¡S )Nr r¶ r· r/ r/ r0 r± , r² c C s |}dddddgdddd d
|
gg}t d d gd dgddggddgddgddggdddgd}t||tdd}||}t ||¡ d S )Nr
|
r r r r r r r r r r r r r r r r r r r! r" r# r' r( r¸ r¹ r/ r/ r0 Útest_xs_level0+ s þý ÿz#TestXSWithMultiIndex.test_xs_level0c C s* |}| d¡j}|jd }t ||¡ d S )N©rT r r )r< ÚvaluesrF Zassert_almost_equalr r/ r/ r0 Útest_xs_values@ s
|
z#TestXSWithMultiIndex.test_xs_valuesc C s( |}| d¡}|jd }t ||¡ d S )Nr¼ )r< r rF rG r r/ r/ r0 Útest_xs_loc_equalityF s
|
|
z)TestXSWithMultiIndex.test_xs_loc_equalityc C s¸ t dddgddggddddddgddddddggd}ttj dd ¡|d
|
}|tkr\|d }|jd d
d¡}| t dd d
f ¡}t
|
||¡ |j t dd d
f }t
|
||¡ d S ) N)rR rT r )rR r| r )rR r} r r r r r é r rQ éþÿÿÿ) r r r+ r^ r_ r r` Z droplevelr< r rF Z assert_equalr )rI Úframe_or_seriesr) ÚobjrO r\ r/ r/ r0 Ú+test_xs_IndexSlice_argument_not_implementedL s þ z@TestXSWithMultiIndex.test_xs_IndexSlice_argument_not_implementedc C sR t ddddgi}|tkr"|d }d}tjt|d |jddd W 5 Q R X d S )
|
Nr6 r r r zIndex must be a MultiIndexr8 r Úasr )r r rA rB Ú TypeErrorr< )rI r rà rv r/ r/ r0 Útest_xs_levels_raises` s z*TestXSWithMultiIndex.test_xs_levels_raisesc C sp t jdddgddgd}tddd gg|d
|
}|jd dd d }tddggt jddgddgdd
|
}t ||¡ d S )N)r r )r r )r r Zlevel1Zlevel2r r r r r« r F©r: rk )r r r r< rF rm )rI ÚmirZ r\ rO r/ r/ r0 Ú"test_xs_multiindex_droplevel_falsei s ÿ ÿþz7TestXSWithMultiIndex.test_xs_multiindex_droplevel_falsec C sL t dddggtdddgd}|jdddd }t ddgi}t ||¡ d S )
|
Nr r r r r rX r« FrÈ )r r r< rF rm rn r/ r/ r0 Útest_xs_droplevel_falsex s z,TestXSWithMultiIndex.test_xs_droplevel_falsec C s t dddggtdddgd}|jdddd }t |jd d
d
|
f j|jd d
d
|
f j¡s^td|jd < |r|t ddgi}nt ddgi}t ||¡ t dd dggtdddgd}|jdddd }d|jd < |rât ddgi}n"|röt ddgi}nt ddgi}t ||¡ d S ) Nr r r r r rX r« FrÈ r )r r g @)
|
r r r< r+ Z shares_memoryr` Z_valuesr? rF rm )rI rs rJ rZ r\ rO r/ r/ r0 Ú test_xs_droplevel_false_view s" 0
|
|
z1TestXSWithMultiIndex.test_xs_droplevel_false_viewc C sZ t dddg¡}tdddgddd gg|d
|
}tjtd d |jd ddd W 5 Q R X d S )N)r Úmr )r Únr )r ÚorX r r r r r rÀ r« r r8 )r r F)rk r: )r r r rA rB rC r< )rI rÉ rZ r/ r/ r0 Ú$test_xs_list_indexer_droplevel_false s z9TestXSWithMultiIndex.test_xs_list_indexer_droplevel_falseN)rx ry rz r
r r r r r rA ÚmarkZ parametrizer¤ r® r r´ rº r» r¾ r¿ rÄ rÇ rÊ rË rÌ rÐ r/ r/ r/ r0 r{ sH
|
þþ
|
þþ
|
ÿ
|
!r{ )rD Únumpyr+ rA Z pandas.errorsr Zpandasr r r r r r Zpandas._testingZ_testingrF Zpandas.tseries.offsetsr Zfixturer1 r2 r{ r/ r/ r/ r0 Ú<module> s
|
h
|
