U
|
ý°du" ã @ s` d dl Zd dlZd dlm mZ d dlmZm Z m
|
Z
|
m Z d dl m Z G dd dZdd ZdS )é N)Ú DataFrameÚ
|
MultiIndexÚ
|
date_rangeÚ to_datetimec
|
@ sº e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d d Z e j deeg¡dd Zdd Ze j deddedgddgfeddedddd d d ddgfg¡d d ZdS )!ÚTestMultiIndexPartialc C sÐ ddg}ddg}t tdt ||g¡d}t td|d}|d }t ||¡ t tdt |dd
|g¡d}|dg }t ||¡ tjtdd |d W 5 Q R X tjtd
|
d |dg W 5 Q R X d S ) Né
|
é ÚaÚbé ©ÚindexÚcolumnsé Ú1©Úmatchz'\[1\] not in index') r Úranger Ú from_productÚtmÚassert_frame_equalÚpytestÚraisesÚKeyError)ÚselfÚl1Úl2ÚdfÚexpectedÚresult© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_partial.pyÚtest_getitem_partial_int s ÿ
|
z.TestMultiIndexPartial.test_getitem_partial_intc C s d S )Nr )r r r r! Útest_series_slice_partial' s z/TestMultiIndexPartial.test_series_slice_partialc
|
C sô |}|}| d¡}|jd }|jd j}t ||¡ t ||¡ | d¡}|jd }t ||¡ tddgddgddggddddddddgddddddddgddddddddggd }ttj d
|
d ¡|t
|
d d } | d¡}| jd }t ||¡ d S )NÚfoo©éÐ é ÚbarÚoneZtwoéÿÿÿÿr r )ÚlevelsÚcodesé r' Zabcdr )r$ r) ) ÚxsÚlocÚTr r r r ÚnpÚrandomÚrandnÚlist)
|
r Ú multiindex_dataframe_random_dataÚ/multiindex_year_month_day_dataframe_random_dataÚframeÚymdr Zresult2r r r r r r! Útest_xs_partial* s*
|
|
|
|
ýþ
|
|
z%TestMultiIndexPartial.test_xs_partialc C sT |}|j }|d }|j|j|jjd dk d}|j d¡ d¡|_t ||¡ d S )N©r& r r )r r )r0 Zreindexr r, Ú droplevelr r )r r6 r8 r r r r r! Útest_getitem_partialJ s z*TestMultiIndexPartial.test_getitem_partialc C sl |}|j dd
}|dd
}t ||¡ |}|j dd
}|jjd }||dk|dk@ }t ||¡ d S )Nr( Zbazé é r: r% r )r/ r r r r, )r r5 r6 r7 r r r8 Zlevr r r! Útest_fancy_slice_partialS s z.TestMultiIndexPartial.test_fancy_slice_partialc C sÔ t dddgdddgdddggddgddgddggd }ttj d
|
d ¡|d }|jd d d
f }|jd }t ||¡ |jd ddgf }|jd ddg }t ||¡ tj t
|
dd |jdd d
f W 5 Q R X d S )Nr r r r
|
ÚxÚyÚpÚq)r, r+ r= r ©r )r rA z\('a', 'foo'\)r )r r$ ) r r r1 r2 Zrandr/ r r r r r )r Úidxr r r r r r! Ú"test_getitem_partial_column_selectc s þ
|
z8TestMultiIndexPartial.test_getitem_partial_column_selectc C s( |}| ¡ }| ¡ }d|jd< d|jdd
< t ||¡ |rjt ¡ d|d jd< W 5 Q R X d|jd< nd|d jd< d|jdd
df< t ||¡ d|jd < d|jd d
|
< t ||¡ |rþt ¡ d|d jd < W 5 Q R X |d jd |d jd k n&d|d jd < |d jd dks$td S ) Nr r% éA éU r ÚA)r% rI é r& éd é )Úcopyr/ Úilocr r Z raises_chained_assignment_errorÚAssertionError)r r6 Zusing_copy_on_writer8 r Úexpr r r! Útest_partial_setw s,
|
|
|
|
z&TestMultiIndexPartial.test_partial_setÚdtypec C sÖ |}|j j}|j |d |¡g|dd
¡|_ |d }|j }t|tsNt|tkrn|jd jt j
|
kstn|jd jt j kstd|jd kst|jd j r¦t|j r°tt jtdd |d W 5 Q R X d S )Nr r rI rL Z14r )r r+ Z
|
set_levelsZastypeÚ
|
isinstancer rO ÚintrR r1 Úint_Úfloat64Z _should_fallback_to_positionalr r r )r r6 rR r8 r+ ZserÚmir r r! Ú!test_getitem_intkey_leading_level s &
|
z7TestMultiIndexPartial.test_getitem_intkey_leading_levelc C s |}| ¡ }| ¡ }d|jddg< d|jd< d|jd< t ||¡ | ¡ }| ¡ }d|jdd
< d|jd< d|jd< t ||¡ |d ¡ }|d ¡ }d|jddg< d|jd< d|jd< t ||¡ |d ¡ }|d ¡ }d|jdd
< d|jd< d|jd< t ||¡ d S )Nr r$ r( rI )rM r/ r r Zassert_series_equal)r r5 r7 r r r r r! Ú test_setitem_multiple_partial² s2
|
|
|
|
|
|
|
|
z3TestMultiIndexPartial.test_setitem_multiple_partialz indexer, exp_idx, exp_valuesz2019-2Nz
|
2019-02-01r r= Ú2019ÚMS©ZperiodsÚfreqr r c C s t dddd}tttdtj|ddggdd gd
|
d }t|tj|ddggdd gd
|
d }|| }t ||¡ |j| }t ||¡ |jdd | }t ||¡ |j|d d
f }t ||¡ | dd¡
|
¡ }| dd¡
|
¡ }|jd d
|d d
f }t ||¡ d S ) NrZ r r[ r\ r' r r r@ rA ©ÚnamesrD ©Zaxis) r r r4 r r r r r r/ Z swaplevelZ
|
sort_index) r ZindexerZexp_idxZ
|
exp_valuesZdate_idxr r r Zdf2r r r! Ú!test_partial_getitem_loc_datetimeÐ s*
|
þþ
|
z7TestMultiIndexPartial.test_partial_getitem_loc_datetime)Ú__name__Ú
|
__module__Ú __qualname__r" r# r9 r< r? rF ÚtdZ skip_array_manager_invalid_testrQ r ÚmarkZ parametrizerT ÚfloatrX rY Úslicer r ra r r r r! r s*
|
!
|
|
ýþþ r c C s ddgddgg} t j| ddgd}t j| dd gd}ttj d
|
d
|
¡||d }|jd d
dd
f d¡jddd }|jd }t
|
||¡ d S )Nr r
|
r r Zcol1Zcol2r^ Zrow1Zrow2r' r r` )r r
|
) r r r r1 r2 r3 rN r; r/ r r )Ú iterablesr Úrowsr r r r r r! Ú"test_loc_getitem_partial_both_axisø s &
|
rk )Únumpyr1 r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsre Zpandasr r r r Zpandas._testingZ_testingr r rk r r r r! Ú<module> s j
|
