U
|
¬ý°d}- ã @ s d dl m Z d dlZd dlZd dlZd dlmZ d dlmZ d dl
|
m m Z ejeje d¡ejfeje d¡ejfejdejfejdejfejdejfejd ejfejd
|
ejfejdejfejdejfejdejfejd ejfejd
|
ejfejdejfejdejfejdejfejd ejfejd
|
ejfejdejfejdejfejdejfejd ejfejd
|
ejfgd d d ZG dd dZG dd dZdS )é )ÚdatetimeN)ÚiNaTé é é g @y @ @TÚ )Úparamsc C s | j S )N)Úparam)Úrequest© r úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_take.pyÚdtype_fill_out_dtype s r c @ sv e Zd Ze d¡Zdd Zdd Zdd Zdd Z d
|
d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd ZdS )ÚTestTakez Incompatible type for fill_valuec C sÌ |\}}}t j ddd¡ |¡}ddddg}tj|||d}|dddg |dddg k ¡ s`t|d |kspt|j|ks~tddddg}tj|||d}|ddddg || k ¡ sºt|j|ksÈtd S )Nr é é r éÿÿÿÿ)Ú
|
fill_valueé © ÚnpÚrandomÚrandintÚastypeÚalgosÚtake_ndÚallÚAssertionErrorÚdtype©Úselfr r r Z out_dtypeÚdataÚindexerÚresultr r r Útest_1d_fill_nonna/ s
|
$ zTestTake.test_1d_fill_nonnac C sÆ |\}}}t j ddd¡ |¡}ddddg}tj||d|d}|dddgd d
f |dddgd d
f k ¡ srt|dd d
f |k ¡ st|j|ksttj||d|d}|d d
dddgf |d d
dddgf k ¡ sât|d d
df |k ¡ sþt|j|kstddddg}tj||d|d}|ddddgd d
f ||d d
f k ¡ s^t|j|ksnttj||d|d}|d d
ddddgf |d d
|f k ¡ s²t|j|ksÂtd S )Nr r ©é r r r ©Úaxisr r r r r r r Útest_2d_fill_nonna? s$
|
4 4 22zTestTake.test_2d_fill_nonnac C sî |\}}}t j ddd¡ |¡}ddddg}tj||d|d}|dddgd d
d d
f |dddgd d
d d
f k ¡ s~t|dd d
d d
f |k ¡ s t|j|ks®ttj||d|d}|d d
dddgd d
f |d d
dddgd d
f k ¡ st|d d
dd d
f |k ¡ s&t|j|ks6ttj||d|d}|d d
d d
dddgf |d d
d d
dddgf k ¡ st|d d
d d
df |k ¡ s®t|j|ks¾tddddg}tj||d|d}|ddddgd d
d d
f ||d d
d d
f k ¡ st|j|ks*ttj||d|d}|d d
ddddgd d
f |d d
|d d
f k ¡ szt|j|ksttj||d|d}|d d
d d
ddddgf |d d
d d
|f k ¡ sÚt|j|ksêtd S )Nr r )r% r r r r r& r r r r r r Útest_3d_fill_nonnaW s2
|
@"B$B$ >>>zTestTake.test_3d_fill_nonnac C sP t j d¡ t j¡}ddddg}t ||¡}| |¡}t j|d< t
|
||¡ d S )Né
|
r r r r © r r Zrandnr Úfloat32r r ÚtakeÚnanÚtmÚassert_almost_equal©r Úarrr! r" Úexpectedr r r Útest_1d_other_dtypesy s
|
|
z TestTake.test_1d_other_dtypesc C s t j dd¡ t j¡}ddddg}tj||dd}|j|dd}t j|d< t
|
||¡ tj||dd}|j|dd}t j|d d
df< t
|
||¡ d S ) Nr* r% r r r r r ©r' r+ r1 r r r Útest_2d_other_dtypes s
|
z TestTake.test_2d_other_dtypesc C sl t jdddgtd}t |ddddg¡}| ddddg¡}t ||¡ t |dddg¡}|jt j ksht
|
d S )Nr r ©r r r © r ÚarrayÚboolr r r- r/ Úassert_numpy_array_equalr Úobject_r ©r r2 r" r3 r r r Ú test_1d_bool s zTestTake.test_1d_boolc C s¼ t jdddgdddgdddggtd}t |ddddg¡}|jddddgdd}t ||¡ tj|ddddgdd}|jddddgdd}t ||¡ t |dddg¡}|jt j ks¸t
|
d S )Nr r r7 r r5 r r8 r= r r r Ú test_2d_bool s & zTestTake.test_2d_boolc C s¨ t j dd¡ t j¡}dddddg}tj||dd}|j|dd}t j|ddgd d
f< t
|
||¡ tj||dd}|j|dd}t j|d d
ddgf< t
|
||¡ d S )Nr r r r r r r5 r+ r1 r r r Útest_2d_float32« s zTestTake.test_2d_float32c C s^ t j ddd¡d }|jdd}ddd d
|
d g}tj||dd }|j|dd }t| t j¡dd gd d
f< t
|
||¡ tj||dt d d
|
d
|
d}|j|dd }t d d
|
d
|
|dd gd d
f< t
|
||¡ tj||d
|
d }|j|d
|
d }t| t j¡d d
dd gf< t
|
||¡ tj||d
|
t d d
|
d
|
d}|j|d
|
d }t d d
|
d
|
|d d
dd gf< t
|
||¡ d S )Ni ¨ iàY r$ l hí] zdatetime64[ns]r7 r r r r r5 r i× r& ) r r r Úviewr r r- r Úint64r/ r0 r r1 r r r Útest_2d_datetime64¼ s& zTestTake.test_2d_datetime64c C s t d¡ dd¡}t |ddg¡}t dddgdd d
|
gg¡}t ||¡ tj|ddgd dd }t dddgdddgg¡}t ||¡ d S ) Né r r r r r r é r* é T©Ú
|
allow_fillr )r ÚarangeÚreshaper r- r9 r/ r; r= r r r Útest_take_axis_0Ø s zTestTake.test_take_axis_0c C sÐ t d¡ dd¡}tj|ddgdd}t ddgdd gd
|
d gd d gg¡}t ||¡ tj|ddgdddd}t ddgddgd
|
dgd dgg¡}t ||¡ tj t
|
dd tj|ddgdddd W 5 Q R X d S )NrD r r r r r r5 r r% é é rE rF T)r' rH r úindices are out-of-bounds©Úmatch) r rI rJ r r- r9 r/ r; ÚpytestÚraisesÚ
|
IndexErrorr= r r r Útest_take_axis_1ã s " " zTestTake.test_take_axis_1c C s t dddg¡}t ddg¡}tjtdd tj||ddgd W 5 Q R X t jdddgtd }tj||ddgd}t jddggtd }t ||¡ d S )
|
Nr r r r zfill_value must be a scalarrO TrG r7 )
|
r r9 rQ rR Ú
|
ValueErrorr r- Úobjectr/ r; r1 r r r Ú!test_take_non_hashable_fill_valueò s z*TestTake.test_take_non_hashable_fill_valueN)Ú__name__Ú
|
__module__Ú __qualname__ÚreÚcompileZ
|
fill_errorr# r( r) r4 r6 r> r? r@ rC rK rT rW r r r r r + s
|
"
|
r c @ sF e Zd Zdd Zdd Zej dddg¡dd Zd
|
d Z d d Z
|
dS )ÚTestExtensionTakec C sv t ddg¡}d}tjt|d tj|ddgdd W 5 Q R X d}tjt|d tj|ddgd d W 5 Q R X d S )
|
Nr r rN rO r T©rH z2index 2 is out of bounds for( axis 0 with)? size 2F)r r9 rQ rR rS r r- )r r2 Úmsgr r r Útest_bounds_check_large s z)TestExtensionTake.test_bounds_check_largec C s t jdddgt jd}dddg}d}tjt|d tj||d
|
d W 5 Q R X t ||¡}t jdddgt jd}t ||¡ d S ) Nr r r r7 r r éþÿÿÿz7'indices' contains values less than allowed \(-2 < -1\)rO Tr^ )
|
r r9 rB rQ rR rU r r- r/ r; )r r2 r! r_ r" r3 r r r Útest_bounds_check_small s
|
z)TestExtensionTake.test_bounds_check_smallrH TFc C sj t jg t jd}tj|g |d}t ||¡ d ddg¡}tj t
|
|d tj|dg|d W 5 Q R X d S )Nr7 r^ ú|z.cannot do a non-empty take from an empty axes.rN rO r ) r r9 rB r r- r/ r; ÚjoinrQ rR rS )r rH r2 r" r_ r r r Útest_take_empty s þÿz!TestExtensionTake.test_take_emptyc C s: t jt g ¡ddgddd}t ddg¡}t ||¡ d S )Nr Tg rG ©r r- r r9 r/ r; )r r" r3 r r r Útest_take_na_empty) s z$TestExtensionTake.test_take_na_emptyc C s8 dddg}t |ddg¡}t ddg¡}t ||¡ d S )Nr r r r rf r= r r r Útest_take_coerces_list. s
|
z(TestExtensionTake.test_take_coerces_listN) rX rY rZ r` rb rQ ÚmarkZ parametrizere rg rh r r r r r] ÿ s
|
r] )r r[ Únumpyr rQ Z pandas._libsr Zpandas._testingZ_testingr/ Zpandas.core.algorithmsÚcoreZ
|
algorithmsr ZfixtureZint8Úint16Úint32Úfloat64Z
|
complex128r< Zbool_r r r] r r r r Ú<module> sF êÿ
|
U
|
