U
|
ý°d;< ã @ s d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m
|
Z
|
d dl m Z G dd dZG dd dZG dd d ZG d
|
d d ZG d d d ZdS )é N)ÚInvalidIndexError)ÚCategoricalIndexÚIndexÚ IntervalIndexÚ Timestampc @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestTakec C s t dddgdd}| t dddg¡¡}t dddgdd}t ||¡ t |j|j¡ |jt dddg¡dd }t ddtjgdddgdd
|
}t ||¡ t |j|j¡ |jt dddg¡d dd }t dddgdd}t ||¡ t |j|j¡ t t d t dddd}| t dddg¡¡}t t dt dddd}t ||¡ t |j|j¡ |jt dddg¡dd }t ddtjgt dddd}t ||¡ t |j|j¡ |jt dddg¡d dd }t t dt dddd}t ||¡ t |j|j¡ d}t
|
j t |d |jt dddg¡dd W 5 Q R X t
|
j t |d |jt dddg¡dd W 5 Q R X d}t
|
j t |d | t ddg¡¡ W 5 Q R X d S )Né é é Úxxx©Únamer éÿÿÿÿT©Ú
|
fill_value)Ú
|
categoriesr F©Z
|
allow_fillr ZCBAÚABC)r Úorderedr ZBCAÚBÚCúJWhen allow_fill=True and fill_value is not None, all indices must be >= -1©Úmatchéþÿÿÿéûÿÿÿú3index -5 is out of bounds for (axis 0 with )?size 3)r ÚtakeÚnpÚarrayÚtmÚassert_index_equalZassert_categorical_equalÚvaluesÚnanÚlistÚpytestÚraisesÚ
|
ValueErrorÚ
|
IndexError)ÚselfÚidxÚresultÚexpectedÚmsg© r. úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_indexing.pyÚtest_take_fill_value sn ÿ ÿ
|
ÿ ÿ ÿ$$z TestTake.test_take_fill_valuec C s t jdddgdd}t|}| t dddg¡¡}t jdddgdd}t|}t ||¡ |jt dddg¡d d
|
}t jddd gdd}t dddg¡}t||d }t ||¡ |jt dddg¡d d d}t jdddgdd}t|}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 )Nz
|
2011-01-01z
|
2011-02-01z
|
2011-03-01r r r r r Tr ÚNaT©r Fr r r r r r ) ÚpdZ DatetimeIndexr r r r r r! r% r& r' r( )r) r* r+ r, Zexp_catsr- r. r. r/ Ú test_take_fill_value_datetimeO s< ÿ ÿ ÿ$$z&TestTake.test_take_fill_value_datetimec C s¤ t dddgdd}dddg}d}tjt|d |j|dd
|
W 5 Q R X d }tjt|d |j||d W 5 Q R X d }tjt|d |j|dd W 5 Q R X d S )Nr r r
|
Úfoor r r z1take\(\) got an unexpected keyword argument 'foo'r )r5 z$the 'out' parameter is not supported)Úoutz%the 'mode' parameter is not supportedZclip)Úmode)r r% r& Ú TypeErrorr r' )r) r* Úindicesr- r. r. r/ Útest_take_invalid_kwargsv s
|
z!TestTake.test_take_invalid_kwargsN)Ú__name__Ú
|
__module__Ú __qualname__r0 r4 r: r. r. r. r/ r s >'r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestGetLocc C sú t tdtdd}ttd}| d¡| d¡ks8t| d¡| d¡ksPt||fD ](}tjtdd | d¡ W 5 Q R X qXt td tdd}ttd }| d
|
¡}t || d
|
¡¡ t |t
|
d d d d d d g¡¡ | d¡}|| d¡ksøt|d kst||fD ]*}tjtdd | d¡ W 5 Q R X qt tdtdd}ttd}| d¡}|| d¡ksxt|t ddd kst| d¡}|| d¡ks¬t|t ddd ksÂt||fD ]*}tjtdd | d¡ W 5 Q R X qÊd S )NZabcdeÚedabcr2 ÚaÚez 'NOT-EXIST'r z NOT-EXISTZaacdedÚdFTé ZaabbbÚabcr r Úbé z'c'Úc) r r$ r Úget_locÚAssertionErrorr% r& ÚKeyErrorr Úassert_numpy_array_equalr r Úslice) r) Zcidx1Úidx1ÚiZcidx2Úidx2ÚresZcidx3Zidx3r. r. r/ Ú test_get_loc s@
|
ÿ
|
|
|
zTestGetLoc.test_get_locc C s& t td}| d¡}|dks"td S )NrD rE r )r r$ rH rI )r) Úcidxr+ r. r. r/ Útest_get_loc_unique· s
|
z TestGetLoc.test_get_loc_uniquec C s2 t td}| d¡}tddd }||ks.td S )NZabbcrE r r
|
)r r$ rH rL rI ©r) rR r+ r, r. r. r/ Ú test_get_loc_monotonic_nonunique¼ s
|
z+TestGetLoc.test_get_loc_monotonic_nonuniquec C s< t td}| d¡}tjddddgtd}t ||¡ d S )NZabcbrE FT©Údtype)r r$ rH r r Úboolr rK rT r. r. r/ Ú#test_get_loc_nonmonotonic_nonunique s
|
z.TestGetLoc.test_get_loc_nonmonotonic_nonuniquec C s, t ddtjg}| tj¡}|dks(td S )NÚAr r )r r r# rH rI )r) ÚcirP r. r. r/ Útest_get_loc_nanÈ s zTestGetLoc.test_get_loc_nanN)r; r< r= rQ rS rU rY r\ r. r. r. r/ r> s
|
/r> c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestGetIndexerc C sj t tdtdd}tjt|tjd}| |¡}t ||¡ t j
|
t dd |j|dd W 5 Q R X d S )NÚcabr2 rV zInvalid fill methodr Úinvalid©Úmethod) r r$ r ZarangeÚlenÚintpÚ get_indexerr rK r% r& r' )r) r* r, Úactualr. r. r/ Útest_get_indexer_baseÑ s
|
z$TestGetIndexer.test_get_indexer_basec
|
C sÆ t j d¡ ttdtddd}tt |¡}d}ddd t|fD ]B}|t jjd
|
t||d }t j
|
t |d | |¡ W 5 Q R X qDtdtd fD ](}t j
|
t |d | |¡ W 5 Q R X qd S )NiÍ[Úaabbcar^ F©r r ú8Reindexing only valid with uniquely valued Index objectsr r rF r )Úsizer Zaababca) r ÚrandomÚseedr r$ r r rb Úrandintr% r& r rd )r) r[ Zoidxr- ÚnÚfinderr. r. r/ Ú test_get_indexer_requires_uniqueÜ s z/TestGetIndexer.test_get_indexer_requires_uniquec
|
C s t tdtdd}t td}|tdttdfD ]^}d}tjt|d | |¡ W 5 Q R X | |¡\}}tj ddd d
|
gtj
|
d }t ||¡ q6d S ) NÚaabcder? r2 Úabfri r r r r r rV ) r r$ r r% r& r rd Zget_indexer_non_uniquer r rc r Zassert_almost_equal)r) rM rO Zindexerr- Zr1Ú_r, r. r. r/ Útest_get_indexer_non_uniqueô s z*TestGetIndexer.test_get_indexer_non_uniquec C s¨ t tdtdd}t td}d}tjt|d |j|dd W 5 Q R X d }tjt|d |j|d
|
d W 5 Q R X d }tjt|d |j|d d W 5 Q R X d S ) Nrq r? r2 rr z3method pad not yet implemented for CategoricalIndexr Úpadr` z8method backfill not yet implemented for CategoricalIndexZbackfillz7method nearest not yet implemented for CategoricalIndexZnearest)r r$ r% r& ÚNotImplementedErrorrd )r) rM rO r- r. r. r/ Útest_get_indexer_method s z&TestGetIndexer.test_get_indexer_methodc C sf t jtdtdgtd}tdtdg}t||ddd}| |¡}t jddgd d}t ||¡ d S )
|
Nz1999-12-31 00:00:00z2000-12-31 00:00:00rV FÚcategory)r r rW r r rc )r r r Úobjectr rd r rK )r) ZarrZcatsr[ r+ r, r. r. r/ Útest_get_indexer_array s þ
|
z%TestGetIndexer.test_get_indexer_arrayc C sP t ddgddgd}| t ddgddgd¡}tjddgdd}t ||¡ d S ©Nr@ rE r2 r rc rV ©r rd r r r rK ©r) r[ r+ r, r. r. r/ Ú+test_get_indexer_same_categories_same_order s z:TestGetIndexer.test_get_indexer_same_categories_same_orderc C sP t ddgddgd}| t ddgddgd¡}tjddgdd}t ||¡ d S r{ r| r} r. r. r/ Ú0test_get_indexer_same_categories_different_order" s z?TestGetIndexer.test_get_indexer_same_categories_different_orderc C s t ddtjdg}dddtjg}| |¡}tjddddgtjd}t ||¡ ddddg}| |¡}tjddddgtjd}t ||¡ d S )Nr r r
|
rC r rV r )r r r# rd r rc r rK )r) r[ Zother1Zres1Z expected1Zother2Zres2Z expected2r. r. r/ Ú)test_get_indexer_nans_in_index_and_target* s
|
|
z8TestGetIndexer.test_get_indexer_nans_in_index_and_targetN) r; r< r= rf rp rt rw rz r~ r r r. r. r. r/ r] Ð s r] c @ s e Zd Zdd Zdd ZdS )Ú TestWherec C s |}t tdtddd}dgt| }|}| ||¡}t ||¡ dgdgt|d }t tjg|dd
¡ |j d}| ||¡}t ||¡ d S )Nrg r^ Frh Tr r2 )
|
r r$ rb Úwherer r! r r# Útolistr )r) Z listlike_boxÚklassrN Zcondr, r+ r. r. r/ Ú
|
test_where8 s "zTestWhere.test_wherec C s~ t ddddg}t ddddg¡}| |d¡}tddddgtd}t ||¡ d }tj t
|
|d
|
|j |d¡ W 5 Q R X d S ) Nr@ rE rG rB TFr rV z3Cannot setitem on a Categorical with a new categoryr ) r r r r r ry r r! r% r& r8 Ú_dataZ_where)r) r[ Úmaskr+ r, r- r. r. r/ Útest_where_non_categoriesF s z#TestWhere.test_where_non_categoriesN)r; r< r= r
r r. r. r. r/ r 7 s r c
|
@ s e Zd Zdd Zdd Zej dddg¡dd Zejjd
|
e
|
d d ¡dfd e
|
dd¡dfde d dfe d ¡dfge ddd Zdd ZdS )Ú TestContainsc C sd t tdtddd}d|ks"td|ks.td|ks:ttj|ksHtd|ksTtd |ks`td S )
|
Nrg ÚcabdefFrh r@ ÚzrA r r )r r$ rI r r# ©r) r[ r. r. r/ Ú test_containsU s zTestContains.test_containsc C s. t tdtjg tdd}tj|ks*td S )Nrg r r2 )r r$ r r# rI r r. r. r/ Útest_contains_nana s z TestContains.test_contains_nanÚunwrapTFc C s\ t jddd dt j¡}| d¡}||d }t|}|}|rD|j}tj|ksRt d |ks^t t j|kslt t
|
d¡|ks~t t d¡|kst t|}|r¢|j}tj|ks°t d |ks¼t t j|ksÊt t
|
d¡|ksÜt t d¡|ksît t|}|r|j}tj|kst d |ks t t j|ks0t t
|
d¡|ksDt t d¡|ksXt d S )Nz
|
2016-01-01éd )Zperiodsr ÚDr r1 ) r3 Z
|
date_rangeÚinsertr1 Z to_periodr r r r# rI Z
|
datetime64Z timedelta64) r) r ZdtiÚpiZtdir[ ÚobjÚobj2Zobj3r. r. r/ Útest_contains_na_dtypee s8
|
z#TestContains.test_contains_na_dtypezitem, expectedr r )ç ø?Tg à?r )r@ F)Úidsc C s* t t td¡}||k}||ks&td S )Nr
|
)r r Z from_breaksÚrangerI )r) Úitemr, r[ r+ r. r. r/ Útest_contains_interval s z#TestContains.test_contains_intervalc C sh t dddg}d|ksttjtdd dg|k W 5 Q R X tjtdd ddg|k W 5 Q R X d S )Nr r r
|
r@ zunhashable typer rE )r rI r% r& r8 )r) r* r. r. r/ Útest_contains_list s z TestContains.test_contains_listN)r; r< r= r r r% ÚmarkZ parametrizer r3 ZIntervalr Z TimedeltaÚstrr r r. r. r. r/ r T s"
|
$
|
úö
|
r )Únumpyr r% Z pandas.errorsr Zpandasr3 r r r r Zpandas._testingZ_testingr r r> r] r r r. r. r. r/ Ú<module> s wIg
|
