U
|
ý°dÍL ã @ s d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ d dlmZ ejdd Zejdd ZG d d
|
d
|
ZdS ) é N)Úis_categorical_dtype)Ú CategoricalÚCategoricalIndexÚ DataFrameÚIndexÚIntervalÚSeriesÚ TimedeltaÚ Timestamp)ÚCategoricalDtypec C s0 t dtjdddittdttdddd S )
|
NÚAé Úint64©ÚdtypeÚaabbcaÚcabÚB©r Úname©Úindex©r ÚnpÚaranger ÚlistÚCDT© r r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_categorical.pyÚdf s ÿür c C s0 t dtjdddittdttdddd S )
|
Nr r r r r Úcaber r r r r r r r Údf2! s ÿür! c @ s¨ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zej d'd(d)d*gd+d,d-gd.d/d0gd1d2d3gfd4d5 ejD d6d5 ejD e j d(d7d0ge d8e d(d9e d9d:e d:d;ge d<d(d(e d<d)d(e d<d*d(ge!d(d=e!d)d=e!d*d>gfd?d5 ej"D e#j$ %d(d9d:d;g¡j e#j&d@d*dAj e#j'dBd*dCj f¡dDdE Z(dFdG Z)dHS )IÚTestCategoricalIndexc C s( t td}|jd }ttddd |¡}|j|ks:ttdddd git|d
|
}t
|
||¡ | ¡ }d |jd< ttd dd |¡}|j|ksttdd d d ddd git|d
|
}t
|
||¡ t j tdd |jd W 5 Q R X | ¡ }| ¡ }|j t¡|_d|jd< d|jd< t
|
||¡ d S )Nr ÚaZaaar ©r r r é é r é r é é é z^'d'$©ÚmatchÚdé
|
)r r Úlocr Úastyper ÚAssertionErrorr r ÚtmÚassert_frame_equalÚcopyÚpytestÚraisesÚKeyErrorr Úobject)Úselfr r ÚresultZbidxÚexpectedZbidx2r! r r r Útest_loc_scalar, s2
|
|
ÿü
|
|
z$TestCategoricalIndex.test_loc_scalarc C s¼ d|j d< | ¡ }d|j d< ttddd}tdddd d
|
d dd git|d }t ||¡ | ¡ }d|j d< tddd d
|
d dtjgtjtjtjtjtjtjdgdt|d }t ||¡ d S )Nr' r# r. )r- r Zaabbcadr r$ r r( r) r* g $@r )r- ÚC)r r= ) r/ r4 r r r r2 r3 r Únan)r9 r Zdf3Zbidx3Z expected3Zdf4r r r Ú,test_loc_setitem_with_expansion_non_categoryL s(
|
|
ÿü
|
þûzATestCategoricalIndex.test_loc_setitem_with_expansion_non_categoryc C s( t jtdd |jd W 5 Q R X d S )Nz^1$r+ r% ©r5 r6 r7 r/ )r9 r r r r Ú$test_loc_getitem_scalar_non_categoryh s z9TestCategoricalIndex.test_loc_getitem_scalar_non_categoryc C sP t tddddg}|d d d
}tjddddgtjd}t | ¡ |¡ tdt d¡d
|
d ¡i}t j |j d
|
d d d dgd|d< t dtd
|
d gddgdd}|jd }t ||¡ tdt dd¡
|
d ¡it dd¡
|
d ¡d}t j |j d
|
d d d dgd|d< |jdd
}t ||¡ t dtd
|
d gddgdd}|jd }t ||¡ d S )Nr% r( r) r* éÿÿÿÿr Úvalueéd r r é é2 éK )ZbinsÚDé r. ©r r é r' r é é )r r r Úarrayr r2 Zassert_numpy_array_equalZ __array__r r r0 ÚpdZcutrC r ÚilocÚassert_series_equalr3 r/ )r9 ÚcatÚreverseÚexpr r; r: r r r Ú test_slicingl s&
|
þ
|
z!TestCategoricalIndex.test_slicingc C sð t dddddddgdddgd}tddddd d
|
d g}d d dddddg}t||d|d}t ddgdddgd}tddg}ddg}t||d|d}t||dd} tddgddgddd}
|
d} |jd d
d d
f } t | |¡ t| d jsôt |jd d d
f } t
|
| |
|
¡ t | d t s&t |jd d
df }t
|
|| ¡ t|jsTt |jd }|| kslt |j dd
d d
f } t | |¡ t| d js¢t |j dd d
f } t
|
| |
|
¡ t | d t sÔt |j d d
df }t
|
|| ¡ t|jst |j d }|| kst |j d|jd f }|| ks<t |jd }|| ksTt |jd }|| kslt |jd g }||d dk }t ||¡ ||d dk }t ||¡ |jd }|| ksÈt |jd } t
|
| |
|
¡ t | d t sòt |jtd d } t | |¡ t| d js"t |jd dg } t | |¡ t| d jsPt |jd d
df }t
|
|| ¡ t|js~t |jd d
tdd f } t | |¡ t| d js¶t |jd d
dd gf } t | |¡ t| d jsìt d S ) Nr# ÚcÚb©Ú
|
categoriesÚhÚiÚjÚkÚlÚmÚnr% r( r) r* r& r é ©ÚcatsÚvaluesr rc rJ rd r8 )r r r r )r( r )r\ rc )r r r r rP r2 r3 r r r1 rQ Ú
|
isinstanceÚstrr/ ÚcolumnsZiatÚatÚslice)r9 rc Úidxrd r Zcats2Zidx2Zvalues2Zexp_dfZexp_colZexp_rowZexp_valZres_dfZres_rowZres_colZres_valZ exp_fancyZ res_fancyr r r Ú test_slicing_and_getting_ops
s ÿ
|
|
|
|
|
|
z1TestCategoricalIndex.test_slicing_and_getting_opsc C sz t dddddddgdddgd}tddddd d
|
d g}d d d d dddg}t||d|d}|jd d
d d
f }tt ddgdddgdd d gdddgd}t ||¡ |jd d
d d
f j}tddgddg}t ||¡ |j dd
df }tt dddgdddgddddgdd}t ||¡ |j dd
|j
|
dd
f }tdt dddgdddgdidddgd}t ||¡ d S )Nr# rW rV rX rZ r[ r\ r] r^ r_ r` r% r( r) r* r& rb r Úcategoryr rc rd rJ r ) r r r rP r2 r3 Zdtypesr rQ r/ rg )r9 rc rj rd r r: r; r r r Útest_slicing_doc_examplesþ s> ÿþû ý þz.TestCategoricalIndex.test_slicing_doc_examplesc C s4 |j ddg }|jddddg }tj||dd d S ) NrV r# r* r r% r& T©Zcheck_index_type)r/ rP r2 r3 )r9 r r: r; r r r Ú test_loc_getitem_listlike_labels$ s z5TestCategoricalIndex.test_loc_getitem_listlike_labelsc C s4 t jtt d¡d |jdddg W 5 Q R X d S )Nú['e'] not in indexr+ r# rW Úe©r5 r6 r7 ÚreÚescaper/ ©r9 r! r r r Ú)test_loc_getitem_listlike_unused_category* s z>TestCategoricalIndex.test_loc_getitem_listlike_unused_categoryc C s( t jtdd |jd W 5 Q R X d S )Nz^'e'$r+ rq r@ ru r r r Ú&test_loc_getitem_label_unused_category0 s z;TestCategoricalIndex.test_loc_getitem_label_unused_categoryc C s2 t jtt d¡d |jddg W 5 Q R X d S )Nz['d'] not in indexr+ r# r- rr ru r r r Ú test_loc_getitem_non_category5 s z2TestCategoricalIndex.test_loc_getitem_non_categoryc C sd | ¡ }d|jd< |jdddg }ttdtddd}td d
|
d d d ddgi|d}t ||¡ d S )Nr' rq r# rW Zaaabber r )rY r r r r% r& r( r) r )r4 r/ r r r r2 r3 )r9 r! r r: Ú exp_indexr; r r r Ú0test_loc_setitem_expansion_label_unused_category: s
|
zETestCategoricalIndex.test_loc_setitem_expansion_label_unused_categoryc C sú t dddg}tdddgddd gd
|
|d }|jddg }t ddg|jd }tddgddgd
|
|d }tj||d d |jdddg }t dddg|jd }tdddgdddgd
|
|d }tj||d d tjtt
|
d¡d |jddg W 5 Q R X d S )Nr# rW rV r% r( r) r* r& r ©r r r rX Trn ú['x'] not in indexr+ Úx) r r r/ rY r2 r3 r5 r6 r7 rs rt )r9 r r Úresry rT r r r Útest_loc_listlike_dtypesC s z-TestCategoricalIndex.test_loc_listlike_dtypesc C sø t dddg}tdddgdddgd |d
|
}|jddg }tdddgdddgd t dddgd
|
}tj||d d |jdddg }tdddddgdddddgd t dddddgd
|
}tj||d d tjtt d ¡d |jddg W 5 Q R X d S )Nr# rW r% r( r) r* r& r r{ r Trn r| r+ r} )
|
r r r/ r2 r3 r5 r6 r7 rs rt ©r9 r r r~ rT r r r Ú8test_loc_listlike_dtypes_duplicated_categories_and_codesZ s ÿ þzMTestCategoricalIndex.test_loc_listlike_dtypes_duplicated_categories_and_codesc C s t ddddgtdd}tdddd gd
|
d d d gd|d}|jddg }tdddgd
|
d d gdt dddgtddd}tj||dd |jdddg }tdddddgd
|
d d
|
d d gdt dddddgtddd}tj||dd tjtt
|
d¡d |jddg W 5 Q R X d S )Nr# rW rV ÚabcderX r% r( r) r* r& r ra rM r{ r Trn r| r+ r} ) r r r r/ r2 r3 r5 r6 r7 rs rt r r r r Ú(test_loc_listlike_dtypes_unused_categoryq s "þ þz=TestCategoricalIndex.test_loc_listlike_dtypes_unused_categoryc C s t ddddgtdd}tdddd gd
|
d d d gd|d}tjtdd |jd W 5 Q R X tjtt d¡d |jddg W 5 Q R X d S )Nr# rW rV r rX r% r( r) r* r& r ra rM r{ r rq r+ rp ) r r r r5 r6 r7 r/ rs rt )r9 r r r r r Ú9test_loc_getitem_listlike_unused_category_raises_keyerror s "zNTestCategoricalIndex.test_loc_getitem_listlike_unused_category_raises_keyerrorc C s` t tj dd¡tdtdd}| ¡ }t|j|_t|j|_t |j
|
dd d
f |jdd}t |j
|
dd d
f |¡ t |j
|
d d
df |jdd}t |j
|
d d
df |¡ ttddd d
|
gd }t |j
|
dd gd d
f |j|d }t |j
|
dd gd d
f |¡ ttd dddgd }t |j
|
d d
ddgf |j|d}t |j
|
d d
ddgf |¡ d S )Nr) ÚABCZXYZ©r rg r rJ ÚXZABr r= rX ©rg r ZXYÚYÚZ)r r ÚrandomÚrandnr r4 r r rg r r/ r2 rQ r3 )r9 r ÚcdfÚexpectry Ú exp_columnsr r r Útest_ix_categorical_index s " "z.TestCategoricalIndex.test_ix_categorical_indexc C sl t tj dd¡tdtdd}| ¡ }t|j|_t|j|_ttdddgd}t |j dd d
f |j|d }t
|
|j dd d
f |¡ ttd
|
d d gd}t |j d d
d f |j|d}t
|
|j d d
d f |¡ t |j ddgd d
f |jttd d }t
|
|j ddgd d
f |¡ t |j d d
d d gf |jttdd}t
|
|j d d
d d gf |¡ d S )Nr) ZABAZXYXr ZAAr r rX r ZXXr r ZAABZXXY) r r r r r r4 r r rg r/ r2 r3 )r9 r r ry r r r r r Ú$test_ix_categorical_index_non_unique© s,
|
ý
|
ýz9TestCategoricalIndex.test_ix_categorical_index_non_uniquec C sZ d}t jt|d |jdd
W 5 Q R X |jdd
}|jddd g }t ||¡ d S )
|
NzRcannot do slice indexing on CategoricalIndex with these indexers \[1\] of type intr+ r% r& rW rV r( r) r* )r5 r6 Ú TypeErrorr/ rP r2 r3 )r9 r Úmsgr: r; r r r Útest_loc_sliceÆ s ÿz#TestCategoricalIndex.test_loc_slicec C s| t ddgddgddggtddd gd
|
}|d }|jd dksBt|jd dksTt|jd dksft|jd dksxtd S ) Nr% r( r) r* r& r r r r= r r )r r% )r r r/ r1 rh )r9 r Úsr r r Ú&test_loc_and_at_with_categorical_indexÓ s ÿz;TestCategoricalIndex.test_loc_and_at_with_categorical_indexÚ
|
idx_valuesr% r( r) rB éþÿÿÿéýÿÿÿç ø?ç @ç @g ø¿g Àg Àc c s" | ]}t jd ddg|dV qdS ©r% r( r) r N©r rN ©Ú.0r r r r Ú <genexpr>é s z TestCategoricalIndex.<genexpr>c c s" | ]}t jd ddg|dV qdS )r r r r Nr )r Zdtypr r r r¡ ë s rW r r* r rL iã r- rH c c s" | ]}t jd ddg|dV qdS r )rO rN r r r r r¡ ó s z
|
2019-01-01)ÚperiodsZ1d)Ústartr¢ c C s t ||d}tddddgi|d}t|d |d }|j|d }tdgdg|d d }t ||¡ |j|d d
|
}tddg|d d
|
dgd }t ||¡ |j| }tddg|d d
|
dgd }t ||¡ | ¡ }d |j|d < tdd ddgi|d}t ||¡ | ¡ }d d g|j|d d
|
df< tdd d dgi|d}t ||¡ | ¡ }d d g|j|df< tdd d dgi|d}t ||¡ d S )N)Úorderedr ZfooÚbarZbazr r r% rJ r( r ZquxZqux2) r r ri r/ r r2 rQ r3 r4 )r9 r r¤ Zcat_idxr Úslr: r; r r r Ú+test_loc_getitem_with_non_string_categoriesà s0
|
z@TestCategoricalIndex.test_loc_getitem_with_non_string_categoriesc C s t ddtjg}ttd|d}|tj dks2t|jtj dksFtt|}|jtjdf dksft|jtj d dks~td S )Nr r r) r r( r )r r r> r Úranger1 r/ r )r9 ÚciZserr r r r Ú!test_getitem_categorical_with_nan! s z6TestCategoricalIndex.test_getitem_categorical_with_nanN)*Ú__name__Ú
|
__module__Ú __qualname__r< r? rA rU rk rm ro rv rw rx rz r r r r r r r r r5 ÚmarkZ parametrizer2 ZALL_INT_NUMPY_DTYPESZFLOAT_NUMPY_DTYPESr rN r8 r r
|
r ZALL_INT_EA_DTYPESrO Z IntervalIndexZ from_breaksZ
|
date_rangeZtimedelta_ranger§ rª r r r r r" + sR y& û ñëþ
|
'r" )rs Únumpyr r5 Zpandas.core.dtypes.commonr ZpandasrO r r r r r r r r
|
Zpandas._testingZ_testingr2 Zpandas.api.typesr r Zfixturer r! r" r r r r Ú<module> s (
|
|
|
|
