U
|
ý°d ã @ sj d dl Zd dlZd dlZd dlmZmZ d dlmZ G dd dZ
|
G dd dZ G dd dZ d d
|
Z dS ) é N)ÚIndexÚNaTc @ sn e Zd Zej dddg¡dd Zej dddg¡ej d ed
|
d d fed ddfg¡dd Zdd Z dS )ÚTestGetSliceBoundszside, expected)Úlefté )Úrighté c C s* t td}|jd|d}||ks&td S )NÚabcdefÚe©Úside)r ÚlistÚget_slice_boundÚAssertionError)Úselfr ÚexpectedÚindexÚresult© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/base_class/test_indexing.pyÚ test_get_slice_bounds_within s z/TestGetSliceBounds.test_get_slice_bounds_withinr r r zdata, bound, expectedr Úxé ZbcdefgÚar c C s& t |}|j||d}||ks"td S )Nr )r r r )r r r ÚdataÚboundr r r r r Ú test_get_slice_bounds_outside s z0TestGetSliceBounds.test_get_slice_bounds_outsidec C s0 t jtdd tg jddd W 5 Q R X d S )Nz Invalid value for side kwarg©Úmatchr Zmiddler )ÚpytestÚraisesÚ
|
ValueErrorr r )r r r r Ú"test_get_slice_bounds_invalid_side s z5TestGetSliceBounds.test_get_slice_bounds_invalid_sideN)
|
Ú__name__Ú
|
__module__Ú __qualname__r ÚmarkZ parametrizer r r r" r r r r r s
|
ÿ r c @ s e Zd Zdd ZdS )ÚTestGetIndexerNonUniquec C sT t ddg t dg¡\}}t tjdgtjd|¡ t tjdgtjd|¡ d S )NÚAÚBr éÿÿÿÿ©Zdtype)r Zget_indexer_non_uniqueÚtmÚassert_numpy_array_equalÚnpÚarrayZintp)r ZindexesÚmissingr r r Ú*test_get_indexer_non_unique_dtype_mismatch" s zBTestGetIndexerNonUnique.test_get_indexer_non_unique_dtype_mismatchN)r# r$ r% r1 r r r r r' ! s r' c @ s$ e Zd Zejjdd Zdd ZdS )Ú
|
TestGetLocc C sb t d}tjddd}tj |td|g¡}| ¡ }t|d }|| }| |¡}||ks^t d S )NÚABCDEFGHIJKLMNOPQRSTUVWXYZz
|
2016-01-01éd )Zperiodsiè é )
|
r ÚpdZ
|
date_rangeZ
|
MultiIndexZ from_productÚrangeZ to_flat_indexÚlenÚget_locr )r ZlevZdtiÚmiZoidxÚlocÚtupÚresr r r Ú.test_get_loc_tuple_monotonic_above_size_cutoff* s
|
z9TestGetLoc.test_get_loc_tuple_monotonic_above_size_cutoffc C s t dtjd ddd gtd}| tj¡}|dks2t| d ¡}t ddddddg¡}t ||¡ t j
|
t dd | t ¡ W 5 Q R X d S ) NZfoog ð?r+ é FTr r ) r r. ÚnanÚobjectr9 r r/ r, r- r r ÚKeyErrorr )r Úidxr= r r r r Ú4test_get_loc_nan_object_dtype_nonmonotonic_nonunique; s
|
z?TestGetLoc.test_get_loc_nan_object_dtype_nonmonotonic_nonuniqueN)r# r$ r% r r& Zslowr> rD r r r r r2 ) s
|
r2 c C s: t jddt jgdd} | j| }tdg}t ||¡ d S )NTFÚbooleanr+ r )r6 ZSeriesZNAr r r, Zassert_index_equal)Zserr r r r r Ú test_getitem_boolean_ea_indexerM s
|
|
rF )Únumpyr. r Zpandasr6 r r Zpandas._testingZ_testingr, r r' r2 rF r r r r Ú<module> s $
|
