U
|
ý°d- ã @ s d Z ddlZddlZddlmZ ddlmZmZ ddl m
|
Z
|
m Z m Z m Z mZmZmZmZ ddlmZ G dd dZG dd d ZG d
|
d d ZG d d d ZG dd dZG dd dZej de dddge dddge dddgg¡d d Z ej d ej d ej gej gej ddgej d fej d ej gej d gej dddgej d fej ej d ej ge!d ej gej ddgej d fe d!eegegej ddgej d fdddej gej gej dgej d fej dej e"d"dge!d e ej ge!d ej ddgej d fg¡d#d$ Z#d%d& Z$dS )'aS
|
test_indexing tests the following Index methods:
|
__getitem__
|
get_loc
|
get_value
|
__contains__
|
take
|
where
|
get_indexer
|
get_indexer_for
|
slice_locs
|
asof_locs
|
|
The corresponding tests.indexes.[index_type].test_indexing files
|
contain tests for the corresponding methods specific to those Index subclasses.
|
é N)ÚInvalidIndexError)Úis_float_dtypeÚ is_scalar)ÚNAÚ DatetimeIndexÚIndexÚ IntervalIndexÚ
|
MultiIndexÚNaTÚ PeriodIndexÚTimedeltaIndexc @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestTakec C s ddg}d}t jt|d |j|dd W 5 Q R X d}t jt|d |j||d W 5 Q R X d}t jt|d |j|d d
|
W 5 Q R X d S ) Né é z1take\(\) got an unexpected keyword argument 'foo'©Úmatch)Úfooz$the 'out' parameter is not supported)Úoutz%the 'mode' parameter is not supportedZclip)Úmode)ÚpytestÚraisesÚ TypeErrorÚtakeÚ
|
ValueError)ÚselfÚindexÚindicesÚmsg© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_indexing.pyÚtest_take_invalid_kwargs) s z!TestTake.test_take_invalid_kwargsc C st ddddg}t |dk r d S | |¡}|| }| |¡s<tt|tttfspd}tj t
|
|d |j W 5 Q R X d S )Né é r r é z*'(.*Index)' object has no attribute 'freq'r ) Úlenr ÚequalsÚAssertionErrorÚ
|
isinstancer r r r r ÚAttributeErrorÚfreq)r r ÚindexerÚresultÚexpectedr r r r Ú test_take8 s
|
zTestTake.test_takec C s@ t ddddg}d}d}tjt|d | |¡ W 5 Q R X d S )Nr r r r" z!Expected indices to be array-liker )r r r r r )r Z integer_indexZ scalar_indexr r r r Útest_take_indexer_typeH s
|
z TestTake.test_take_indexer_typec C sH t |dkrd S | dddg¡}| ddt |d g¡}t ||¡ d S )Nr éÿÿÿÿr )r$ r ÚtmZassert_index_equal)r r r+ r, r r r Ú test_take_minus1_without_fillP s
|
z&TestTake.test_take_minus1_without_fillN)Ú__name__Ú
|
__module__Ú __qualname__r r- r. r1 r r r r r ( s r c @ s e Zd Zej dedddgdfedddgdfedddejdgdfedddej dgdfedddejgejfedddej gej fg¡dd Z
|
ej dedddgdfedddgdfedddejgdfedddej gdfedddejgej fedddej gejfedddgej d ejfedddgej d ej fedddgej d ejfedddgej d ej fg
|
¡d
|
d Z ej dedddgdfedddgdfg¡d d Zej dedddgdfedddgdfg¡dd Zdd Zdd ZdS )Ú TestContainsz index,valr r r Ú2r! c C s ||ks t d S ©N©r& ©r r Úvalr r r Útest_index_contains] s z TestContains.test_index_contains©Údtypec C s ||ks t d S r7 r8 r9 r r r Útest_index_not_containsk s z$TestContains.test_index_not_containsc C s ||ks t d S r7 r8 r9 r r r Útest_mixed_index_contains s z&TestContains.test_mixed_index_containsÚ1c C s ||ks t d S r7 r8 r9 r r r Ú test_mixed_index_not_contains s z*TestContains.test_mixed_index_not_containsc C s |}t |sddddgn
|
ddddg}t||d }t |js`d|ksFtd
|
|ksRtd|kstn$d|ksltd
|
|ksxtd|kstd S ) Nr r r r" ç¹?gñ?g@gffffff
|
@r< ç ð?)r r r= r& )r Zany_real_numpy_dtyper= Údatar r r r Ú test_contains_with_float_index s
|
z+TestContains.test_contains_with_float_indexc C sp t |trd S d}tjt|d g |k W 5 Q R X d dddddg¡}tjt|d i |jk W 5 Q R X d S ) Nzunhashable type: 'list'r ú|zunhashable type: 'dict'z must be real number, not dictzan integer is requiredz\{\}z6pandas\._libs\.interval\.IntervalTree' is not iterable)r' r r r r ÚjoinZ_engine)r r r r r r Ú&test_contains_requires_hashable_raises s
|
ûÿ z3TestContains.test_contains_requires_hashable_raisesN)r2 r3 r4 r ÚmarkÚ parametrizer ÚnpÚinfÚnanr; Úint64Zuint64r> r? rA rE rH r r r r r5 \ sL úþ
|
ôþ
|
"ÿ
|
"ÿ
|
r5 c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú
|
TestGetLocc C s2 t jttfdd | tdd¡ W 5 Q R X d S )NÚslicer r r )r r r r Úget_locrP )r r r r r Útest_get_loc_non_hashable² s z$TestGetLoc.test_get_loc_non_hashablec C sv ddl m} G dd d|}t|jr*tt}d}t|ttt t
|
frLt }d}t j ||d | |j¡ W 5 Q R X d S )Nr )ÚEnumc @ s e Zd ZdZdS )z6TestGetLoc.test_get_loc_non_scalar_hashable.<locals>.EÚx1N)r2 r3 r4 ÚX1r r r r ÚE¼ s rV z <E.X1: 'x1'>zE.X1r )ÚenumrS r rU r& ÚKeyErrorr' r r r r r r r rQ )r r rS rV Úexcr r r r Ú test_get_loc_non_scalar_hashable¸ s" üþ
|
z+TestGetLoc.test_get_loc_non_scalar_hashablec C sR t }t|tttttfr t}tj |dd |
|
dd t dD ¡ W 5 Q R X d S )Nzgenerator objectr c s s | ]
|
}|V qd S r7 r )Ú.0Úxr r r Ú <genexpr>â s z4TestGetLoc.test_get_loc_generator.<locals>.<genexpr>r# ) rX r' r r r r r r r r rQ Úrange)r r rY r r r Útest_get_loc_generatorÒ s ûþ z!TestGetLoc.test_get_loc_generatorc C s@ t ddttgdd}| t¡}t ddddg¡}t ||¡ d S )Nr r ÚInt64r< FT)r r rQ rK Úarrayr0 Úassert_numpy_array_equal©r Úidxr+ r, r r r Ú!test_get_loc_masked_duplicated_naä s
|
z,TestGetLoc.test_get_loc_masked_duplicated_naN)r2 r3 r4 rR rZ r_ re r r r r rO ± s rO c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestGetIndexerc C s |j r0tj|jtjd}| |¡}t ||¡ n(d}tj t
|
|d | |¡ W 5 Q R X tj t dd |j|dd W 5 Q R X d S )Nr< ú8Reindexing only valid with uniquely valued Index objectsr zInvalid fill methodÚinvalid)Úmethod) Ú_index_as_uniquerK ZarangeÚsizeÚintpÚ get_indexerr0 rb r r r r )r r r, Úactualr r r r Útest_get_indexer_baseí s
|
z$TestGetIndexer.test_get_indexer_basec C s¤ |j r:| |dd
¡}t|tjs(t|jtjksjtn0d}tj t
|
|d | |dd
¡ W 5 Q R X | |dd
¡\}}t|tjst|jtjks td S )Nr r rg r ) rj rm r' rK Zndarrayr& r= rl r r r Úget_indexer_non_unique)r r r* r Ú_r r r Ú test_get_indexer_consistencyú s z+TestGetIndexer.test_get_indexer_consistencyc C sP t ddttgdd}| t dtgdd¡}tjdddg|jd}t ||¡ d S )Nr r r` r< r r" )r r Úget_indexer_forrK ra r= r0 rb rc r r r Ú%test_get_indexer_masked_duplicated_na
|
s z4TestGetIndexer.test_get_indexer_masked_duplicated_naN)r2 r3 r4 ro rr rt r r r r rf ì s rf c @ s e Zd Zdd ZdS )ÚTestConvertSliceIndexerc C sp t d d d}t|trBd}tjt|d | |d¡ W 5 Q R X n*d}tjt|d | |d¡ W 5 Q R X d S )Nr zClabel-based slicing with step!=1 is not supported for IntervalIndexr Úlocz7'>=' not supported between instances of 'str' and 'int')rP r' r r r r Z_convert_slice_indexerr )r r Úkeyr r r r Ú test_convert_almost_null_slice s
|
z6TestConvertSliceIndexer.test_convert_almost_null_sliceN)r2 r3 r4 rx r r r r ru s ru c @ s e Zd Zdd ZdS )Ú TestPutmaskc C s² t |s d S |d }d}tjt|d$ | t t |d tj¡|¡ W 5 Q R X tjt|d$ | t t |d tj¡|¡ W 5 Q R X tjt|d | d|¡ W 5 Q R X d S )Nr z,putmask: mask and data must be the same sizer r r )r$ r r r ZputmaskrK ZonesZbool_)r r Úfillr r r r Ú test_putmask_with_wrong_mask# s ((z(TestPutmask.test_putmask_with_wrong_maskN)r2 r3 r4 r{ r r r r ry " s ry rd r r r" rB gÉ?g333333Ó?ÚaÚbÚcc C s* d}t jt|d | d W 5 Q R X d S )Nz,Indexing with a float is no longer supportedr rC )r r Ú
|
IndexError)rd r r r r Ú test_getitem_deprecated_float5 s r zidx,target,expectedZvar1r< z
|
2020-08-05ÚNaNc C s" t | }| |¡}t ||¡ d S r7 )r rs r0 rb )rd Útargetr, Zaxisrn r r r Ú)test_get_indexer_non_unique_multiple_nans@ s
|
r c C sd t ddg}t d| gdd}| |¡\}}t |tjddgtjd¡ t |tjdgtjd¡ d S )NrC g @r Úobjectr< r r/ )r rp r0 rb rK ra rl )Z nulls_fixturerd r Z
|
result_idxZresult_missingr r r Ú7test_get_indexer_non_unique_nans_in_object_dtype_targetb s
|
r
)%Ú__doc__ÚnumpyrK r Z pandas.errorsr Zpandas.core.dtypes.commonr r Zpandasr r r r r r
|
r r Zpandas._testingZ_testingr0 r r5 rO rf ru ry rI rJ r rM ra rl r Úfloatr r
r r r r Ú<module> sR (
|
4U;& &ÿ
|
& ýý ý$ýîþ
|
|
