U
|
ý°d» ã @ s& d dl Z d dlmZ G dd dZdS )é Nc @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestDataFrameTakec C sÖ ddddg}|fD ]h}|j |dd}| |j |¡¡}t ||¡ |j |dd}|jd d
dddd gf }tj||d
|
d qddd g}|fD ]}|j |dd}| |j |¡¡}t ||¡ |j |dd}t ||¡ |j |dd}|jd d
dddgf }tj||d
|
d qd }tjt|d |j ddddgdd W 5 Q R X tjt|d |j ddddgdd W 5 Q R X tjt|d |j ddddgdd W 5 Q R X tjt|d |j ddddgdd W 5 Q R X d S )Né é é r ©ZaxisÚDÚBÚCÚAF)Z check_nameséÿÿÿÿzindices are out-of-bounds)Úmatché iáÿÿÿé éûÿÿÿ) ÚtakeÚreindexÚindexÚtmÚassert_frame_equalÚlocÚpytestZraisesÚ
|
IndexError)ÚselfZ float_frameÚorderÚdfÚresultÚexpectedÚmsg© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_take.pyÚ test_take s6
|
|
|
zTestDataFrameTake.test_takec C sô dddddg}|fD ]f}|j |dd}| |j |¡¡}t ||¡ |j |dd}|jd d
ddd d
|
d gf }t ||¡ qddd g}|fD ]b}|j |dd}| |j |¡¡}t ||¡ |j |dd}|jd d
ddd gf }t ||¡ qd S ) Né r r r r r Zfoor r r
|
r éþÿÿÿ©r r r r r r )r Zfloat_string_framer r r r r r r Útest_take_mixed_type. s
|
|
|
z&TestDataFrameTake.test_take_mixed_typec C s~ ddddg}||fD ]d}|j |dd}| |j |¡¡}t ||¡ |j |dd}|jd d
dddd gf }t ||¡ qd S )
|
Nr r r r r r r r
|
r r# )r Zmixed_float_frameZmixed_int_framer r r r r r r Útest_take_mixed_numericG s z)TestDataFrameTake.test_take_mixed_numericN)Ú__name__Ú
|
__module__Ú __qualname__r r$ r% r r r r r s 'r )r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
