U
|
ý°dÀ ã @ s` d dl Zd dlZd dlZd dlmZ d dlm Z ej
|
dddgej ddgej dfej ddgddej ddgej dfej ddgddej ddgej dfe ddg¡ej ddgej dfd d
|
d gej d d
|
d gejdfe d d
|
d g¡ej d d
|
d gejdfej d d
|
d gd dej d d
|
d gejdfg ej g ej dfg¡d d Zej
|
dd d
|
dgej d d
|
dgd dg¡dd Zej
|
dd d
|
gej d d
|
gd dej d d
|
gejdg¡dd Zej
|
dd ddgej d dejgddg¡dd Zej
|
dddgej ddgddej d d
|
gedejd d
|
gedg¡dd Zd d Zej
|
ddeed d d g¡d d! ZdS )"é N)Úcheck_array_indexerzindexer, expectedé é ©ZdtypeÚint64ZInt32TFÚbooleanc C s* t dddg¡}t|| }t ||¡ d S ©Nr r é )ÚnpÚarrayr ÚtmÚassert_numpy_array_equal)ÚindexerÚexpectedÚarrÚresult© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_check_indexer.pyÚtest_valid_input s
|
r r c C s> t dddg¡}t|| }t jdddgtd}t ||¡ d S )Nr r r TFr )r
|
r r Úboolr r )r r r r r r r Ú test_boolean_na_returns_indexer" s
|
r c C s< t dddg¡}d}tjt|d t|| W 5 Q R X d S )Nr r r z Boolean index has wrong length©Úmatch©r
|
r ÚpytestÚraisesÚ
|
IndexErrorr ©r r Úmsgr r r Útest_bool_raise_length/ s r ZInt64c C s< t dddg¡}d}tjt|d t|| W 5 Q R X d S )Nr r r z9Cannot index with an integer indexer containing NA valuesr )r
|
r r r Ú
|
ValueErrorr r r r r Ú test_int_raise_missing_values? s r! g g ð?g @Úfloat64c C s< t dddg¡}d}tjt|d t|| W 5 Q R X d S )Nr r r ú9arrays used as indices must be of integer or boolean typer r r r r r Ú test_raise_invalid_array_dtypesJ s
|
r$ c C sN t jddg| d}t dddg¡}d}tjt|d t|| W 5 Q R X d S ) NÚaÚbr r r r r# r )Úpdr r
|
r r r r )Znullable_string_dtyper r r r r r Ú test_raise_nullable_string_dtype[ s
|
r( r )Nc C s* t dddg¡}t|| }|| ks&td S r )r
|
r r ÚAssertionError)r r r r r r Ú!test_pass_through_non_array_likesd s
|
r* )Únumpyr
|
r Zpandasr' Zpandas._testingZ_testingr Zpandas.api.indexersr ÚmarkZ parametrizer ZintpZIndexZbool_r r r ZNAr! Úobjectr$ r( ÚEllipsisÚslicer* r r r r Ú<module> sb $$ $þòþ
|
ÿ
|
|
ýþ
|
ÿ
|
üþ
|
|
