U
|
¬ý°dh ã @ s: d dl Zd dlZd dlmZ d dlmZ G dd dZdS )é N)Ú Categoricalc @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )ÚTestTakec C sB t ddg}t d ¡ | ddg¡}W 5 Q R X | |¡s>td S )NÚaÚbr éÿÿÿÿ)r ÚtmÚassert_produces_warningÚtakeÚequalsÚAssertionError)ÚselfÚcatÚresult© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_take.pyÚ test_take_default_allow_fill s z%TestTake.test_take_default_allow_fillc C s4 t ddg}t d ¡ | ddg¡ W 5 Q R X d S )Nr r r )r r r r )r r r r r Ú test_take_positive_no_warning s z&TestTake.test_take_positive_no_warningc C sL t dddg}|rd}nd}tjt|d |jddg|d W 5 Q R X d S ) Nr r úindices are out-of-boundsz2index 4 is out of bounds for( axis 0 with)? size 3©Úmatché é ©Ú
|
allow_fill©r ÚpytestÚraisesÚ
|
IndexErrorr ©r r r Úmsgr r r Útest_take_bounds s zTestTake.test_take_boundsc C sL t g ddgd}|rd}nd}tjt|d |jdg|d W 5 Q R X d S ) Nr r ©Ú
|
categoriesr z-cannot do a non-empty take from an empty axesr r r r r r r r Útest_take_empty! s zTestTake.test_take_emptyc C sT t ddddgddg|d}|jdddgdd}t dddg|j|d}t ||¡ d S ) Nr r ©r" Úorderedr é é Fr ©r r r" r Úassert_categorical_equal©r r% r r Úexpectedr r r Útest_positional_take+ s ÿz TestTake.test_positional_takec C sN t ddgdddg|d}|jddgdd}t ddg|j|d}t ||¡ d S ) Nr r Úcr$ r& r Fr r( r* r r r Ú test_positional_take_unobserved3 s z(TestTake.test_positional_take_unobservedc C sL t dddg}|jdddgdd}t dtjtjgddgd}t ||¡ d S )Nr r r r Tr r! )r r ÚnpÚnanr r) ©r r r r+ r r r Útest_take_allow_fill9 s z TestTake.test_take_allow_fillc C sL t dddg}|jdddgddd}t dddgdddgd}t ||¡ d S )Nr r r& T)r Ú
|
fill_valuer! ©r r r r) r1 r r r Ú test_take_fill_with_negative_one@ s z)TestTake.test_take_fill_with_negative_onec C sL t dddg}|jdddgddd}t dddgdddgd }t ||¡ d S )
|
Nr r r- r r& r T©r3 r r! r4 r1 r r r Útest_take_fill_valueG s z TestTake.test_take_fill_valuec C sF t dddg}d}tjt|d |jdddgd d
|
d W 5 Q R X d S ) Nr r r- z9Cannot setitem on a Categorical with a new category \(d\)r r r& r ÚdTr6 )r r r Ú TypeErrorr )r r Zxprr r r Ú test_take_fill_value_new_raisesN s z(TestTake.test_take_fill_value_new_raisesN) Ú__name__Ú
|
__module__Ú __qualname__r r r r# r, r. r2 r5 r7 r: r r r r r s
|
|
r ) Únumpyr/ r Zpandasr Zpandas._testingZ_testingr r r r r r Ú<module> s
|
