U
|
ý°dã ã @ sZ d dl Zd dlZd dlmZmZ d dlmZ G dd dZ G dd dZ
|
G dd dZ dS ) é N)ÚIndexÚ
|
RangeIndexc @ sH e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d g¡d d Z
|
dS )ÚTestGetIndexerc C sT t dddd}t d}| |¡}tjddddddddd dg
|
tjd
|
}t ||¡ d S ) Nr é é ©ÚstartÚstopÚstepé
|
éÿÿÿÿé é é ©Údtype©r Ú get_indexerÚnpÚarrayÚintpÚtmÚassert_numpy_array_equal©ÚselfÚindexÚtargetZindexerÚexpected© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/ranges/test_indexing.pyÚtest_get_indexer s
|
|
$z TestGetIndexer.test_get_indexerc C sX t dddd}t d}|j|dd}tjddddddd d d
|
d
|
g
|
tjd }t ||¡ d S ) Nr r r r r Úpad©Úmethodr r r r r r r r r Útest_get_indexer_pad s
|
$z#TestGetIndexer.test_get_indexer_padc C sX t dddd}t d}|j|dd}tjdddddd d d
|
d
|
d g
|
tjd }t ||¡ d S ) Nr r r r r Zbackfillr" r r r é r r r r r r Útest_get_indexer_backfill s
|
$z(TestGetIndexer.test_get_indexer_backfillc C sL t d}t d}|j|ddd}tjdddddd gtjd
|
}t ||¡ d S ) Nr é r! r )r# Úlimitr r r r r r )r Úidxr Úresultr r r r Útest_get_indexer_limit! s
|
z%TestGetIndexer.test_get_indexer_limitr r r éþÿÿÿc
|
C sL t d|d}| td¡}tjdddddddddg tjd}t ||¡ d S ) Né éýÿÿÿé r r r r r )r r Úranger r r r r )r r r r* r r r r Útest_get_indexer_decreasing) s "z*TestGetIndexer.test_get_indexer_decreasingN) Ú__name__Ú
|
__module__Ú __qualname__r r$ r& r+ ÚpytestÚmarkZ parametrizer1 r r r r r s r c @ s e Zd Zdd Zdd ZdS )ÚTestTakec C s2 t dddd}| dddg¡}|j|jks.td S )Nr r% Zfoo©Únamer r )r Útaker9 ÚAssertionError)r r Ztakenr r r Útest_take_preserve_name3 s z TestTake.test_take_preserve_namec C s^ t dddd}| t dddg¡¡}tdddgtjdd }t ||¡ d
|
}tj t
|
|d |jt dddg¡d d W 5 Q R X |jt dddg¡dd d}tdddgtjdd }t ||¡ d
|
}tj t
|
|d |jt dddg¡d d W 5 Q R X tj t
|
|d |jt dddg¡d d W 5 Q R X d}tj t |d | t ddg¡¡ W 5 Q R X d S )Nr r Zxxxr8 r r r r ©r r9 z:Unable to fill values because RangeIndex cannot contain NA)ÚmatchT)Ú
|
fill_valueF)Z
|
allow_fillr? r, éûÿÿÿz3index -5 is out of bounds for (axis 0 with )?size 3) r r: r r r Úint64r Úassert_index_equalr5 ZraisesÚ
|
ValueErrorÚ
|
IndexError)r r) r* r Úmsgr r r Útest_take_fill_value8 s$ $ $$z TestTake.test_take_fill_valueN)r2 r3 r4 r< rF r r r r r7 2 s r7 c @ s e Zd Zdd ZdS )Ú TestWherec C sr t dddd}t dddddg¡}| |d¡}tdddd d
|
gtjdd }t ||¡ | | d¡}t ||¡ d S ) Nr r% Útestr8 TFr r r r r= ) r r r Zputmaskr rA r rB Úwhere)r r) Úmaskr* r r r r Ú test_where_putmask_range_castU s z'TestWhere.test_where_putmask_range_castN)r2 r3 r4 rK r r r r rG T s rG ) Únumpyr r5 Zpandasr r Zpandas._testingZ_testingr r r7 rG r r r r Ú<module> s '"
|
