U
|
ý°dÁ9 ã @ s d dl Zd dlZd dlmZ d dlmZ d dlZ d dlm
|
Z
|
m Z d dl m Z d dlmZmZ d dlmZ G dd deZG d d
|
d
|
ZdS ) é N©Úindex)Ú NDArrayBacked)Ú CategoricalÚCategoricalDtype)ÚCategoricalIndexÚIndex)ÚBasec @ sb e Zd ZeZejedddZejdd Zdddd d
|
Z d d Z
|
d d Z dd Z dd Z ej ddddgdddddggededgg¡dd Zd d Zej d dddgdddge dd d g¡e d d dg¡e d d d g¡d!fdddgede dd d g¡e d d dg¡e d d d g¡d!fdd"d#gedejdejd$ejdejd$ejdejd$d!fed%ede ddd g¡e dd dg¡e dd d g¡d!fg¡d&d' Zej d(dddgdddgdgfdddgedejgfddd"gdddgddejgfdd"d#gedejd"d#gfg¡d)d* Zd+d, Zd-d. Zd/d0 Z d1d2 Z d3d4 Z d5d6 Z d7d8 Z dS )9ÚTestCategoricalIndex)Úreturnc C s | j tdtdddS )NÚaabbcaÚcabF©Ú
|
categoriesÚordered)Ú
|
_index_clsÚlist©Úself© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_category.pyÚ simple_index s z!TestCategoricalIndex.simple_indexc C s
|
t d¡S )Néd )ÚtmÚmakeCategoricalIndexr r r r r s zTestCategoricalIndex.indexNFr c C s" |d krt d}tt d||dS )Nr r r )r r )r r r r r r Ú create_index s z!TestCategoricalIndex.create_indexc C s. | j tdd}|d }| |¡dks*td S )NZabcd©r r T)r r Z$_can_hold_identifiers_and_holds_nameÚAssertionError)r ÚidxÚkeyr r r Útest_can_hold_identifiers$ s z.TestCategoricalIndex.test_can_hold_identifiersc C s |}|j }| dd¡}ttd|d}tj||dd | dd¡}ttd|d}tj||dd tg |d dd¡}tdg|d}tj||dd | t¡ dd ¡}| dd ¡}tj||dd tdtj dd
|
d d
|
g}tj t
|
j d fD ]$}ttd d |¡}t ||¡ qæd S )Nr ÚaZaaabbcar T©ÚexactéÿÿÿÿZaabbcaaÚdÚbÚcZaabcbé ) r Úinsertr r r Úassert_index_equalÚastypeÚobjectÚnpÚnanÚpdÚNaT)r r Úcir ÚresultÚexpectedÚnar r r Ú test_insert) s$ z TestCategoricalIndex.test_insertc C sB t dddg}| dtj¡}ttjdddgtd}t ||¡ d S )Nr r( ©Údtype)r r) r/ r0 r r, r r* ©r r1 r2 r3 r r r Ú test_insert_na_mismatched_dtypeG s z4TestCategoricalIndex.test_insert_na_mismatched_dtypec C s |}|j }| d¡}ttd|d}tj||dd | d¡}ttd|d}tj||dd t ttf¡ | d¡ W 5 Q R X d S ) Nr Zabbcar Tr" r$ Zaabbcé
|
) r Údeleter r r r* Zexternal_error_raisedÚ
|
IndexErrorÚ
|
ValueError)r r r1 r r2 r3 r r r Ú test_deleteM s
|
|
z TestCategoricalIndex.test_deletezdata, non_lexsorted_datar( é é é r ÚabcZfabcdc C sX t |}|jdkst|jdks$tt |dd}|jdks>t|jdksLtt |t|d}|jdksjt|jdksxtt |t|dd}|jdkst|jdks¦t|d |d |d g}t |t|d}|jdksÚt|jdksèt|}t |d d
|d}|jdkst|jdks tt |dd
|d}|jdksDt|jdksTtd S )
|
NTF©r r r r r? r( r@ )r Zis_monotonic_increasingr Zis_monotonic_decreasingÚreversed)r ÚdataZnon_lexsorted_datar' Zreordered_datar r r r Útest_is_monotonic] s. z&TestCategoricalIndex.test_is_monotonicc C s t dddgdd}|jdks t|jdks.tt ddgddgdd }|jdksRt|jdks`tt ddddgdddgdd }|jdkst|jdkstd S )
|
Nr Úfoo)ÚnameFTr( r? r@ ©r rH )r Z is_uniquer Zhas_duplicates)r r r r r Útest_has_duplicates s z(TestCategoricalIndex.test_has_duplicateszdata, categories, expectedT)ÚfirstÚlastFr! r& )Úshaper7 Zabbc C sZ t ||dd}| ¡ D ]>\}}t |j|d|¡ || }|j|d}t ||¡ qd S )NrG rI )Úkeep)r Úitemsr Úassert_numpy_array_equalZ
|
duplicatedZdrop_duplicatesr* )r rE r r3 r rN Úer2 r r r Útest_drop_duplicates s *
|
z)TestCategoricalIndex.test_drop_duplicatesz data, categories, expected_datac C s8 t ||d}t||d}t||d}t | ¡ |¡ d S )NrC r6 )r r r r* Úunique)r rE r Z expected_datar r7 r r3 r r r Ú test_uniqueÁ s
|
z TestCategoricalIndex.test_uniquec C s` t ddgddgdd}t| tjtt||dd t| t tjjdddd }t| d S )
|
Nr! r& Tr r" r é r )Úsize) r Ústrr r* ÚevalÚreprr- ÚrandomÚrandint)r r1 r r r Útest_repr_roundtripÑ s z(TestCategoricalIndex.test_repr_roundtripc C s t tdtjg dddgd}t | dg¡t ddddddg¡¡ t | dddg¡t dgd dg ¡¡ t | dddtjg¡t dgd ¡¡ | | td
|
¡¡}t dgd ¡}t ||¡ | | td ¡¡}t dgd dg ¡}t ||¡ d S ) NZaabcar' r! r& r FTrU é Z abcdefghiZdefghi) r r r- r. r rP ÚisinÚarrayÚset_categoriesr8 r r r Ú test_isinÞ s&
|
ÿ ÿ ÿ z TestCategoricalIndex.test_isinc C s` t ddgddgdd}t ddgdddgdd}| |¡s<t| | ¡ ¡sNt| |¡r\td S )Nr! r& Tr r' )r Z identicalr Úcopy)r Zci1Zci2r r r Útest_identicaló s
|
z#TestCategoricalIndex.test_identicalc C sX t |jdd}t ||¡ t |jj|jj¡r2tt |jdd}|jj|jjksTtd S )NT)rb F) r Úvaluesr r* r- Z shares_memoryÚ_dataÚ_codesr )r r r2 r r r Útest_ensure_copied_dataú s
|
z,TestCategoricalIndex.test_ensure_copied_datac C s> t jddddgitdddgd}t|}d }||ks:td S )
|
NÚAr( r? r@ r! r& r' r z A
|
a 1
|
b 2
|
c 3)r/ Z DataFramer rY r )r Zdfr2 r3 r r r Útest_frame_repr s "z$TestCategoricalIndex.test_frame_reprc C s d S ©Nr r r r r Útest_reindex_base s z&TestCategoricalIndex.test_reindex_basec C s d S rj r r r r r Ú test_map_str s z!TestCategoricalIndex.test_map_str)!Ú__name__Ú
|
__module__Ú __qualname__r r ÚpytestZfixturer r r r r5 r9 r> ÚmarkÚ parametrizer rF rJ r- r_ ÚzerosZbool_rR r. rT r\ ra rc rg ri rk rl r r r r r
|
s
|
&þ
|
" ýý
|
ýý
|
ýý
|
ýýäþ)
|
üþ
|
r
|
c
|
@ s¬ e Zd Zdd Zej deje j
|
fej e j fej e jfeje jfg¡dd Zej ddd d fd
|
d d fd d d fd d d fdd dfdd dfg¡dd Zdd ZdS )ÚTestCategoricalIndex2c C s¤ t d¡}d}tjt|d | d¡ W 5 Q R X tjt|d |j d¡ W 5 Q R X |d d
}| d¡}|jj d¡}t ||¡ |j}t | d¡|¡ d S )Nr z;When changing to a larger dtype, its size must be a divisor©ÚmatchÚi8éüÿÿÿ) r r rp Úraisesr= Úviewre ÚcodesrP )r r1 ÚmsgÚresr3 Úcatr r r Ú test_view_i8 s
|
|
z"TestCategoricalIndex2.test_view_i8zdtype, engine_typec C s |t jkr2t jdt jdt jdi| }tt|}n,ttd}|jj d¡}t
|
|j ||j ¡ t |jj |¡srtt|j|std S )Nr( é i Úint64)r- r Úint8Úint16Úint32r Úrangerd Z_ndarrayr+ r Ú__init__re r7 Z
|
issubdtyper{ r Ú
|
isinstanceZ_engine)r r7 Z engine_typeZ num_uniquesr1 Zarrr r r Útest_engine_type, s
|
|
z&TestCategoricalIndex2.test_engine_typez func,op_namec C s | | S rj r ©r r r r Ú<lambda>G ó z TestCategoricalIndex2.<lambda>Ú__sub__c C s | | S rj r r r r r r H r Ú__add__c C s | ddg S ©Nr! r& r r r r r r I r c C s | ddg S r r r r r r r J r c C s ddg| S r r r r r r r K r Ú__rsub__c C s ddg| S r r r r r r r L r Ú__radd__c C sX t tddg}d}d d| ddd| g¡}tjt|d || W 5 Q R X d S )
|
Nr! r& z-'(Categorical|list)' and '(Categorical|list)'ú|zcannot perform z' with this index type: CategoricalIndexzcan only concatenate listz)unsupported operand type\(s\) for [\+-]: ru )r r Újoinrp ry Ú TypeError)r ÚfuncZop_namer Z cat_or_listr| r r r Útest_disallow_addsub_opsD s
|
ýÿz.TestCategoricalIndex2.test_disallow_addsub_opsc C s t tdtdd}| td¡}t |t tdtdd¡ t tdtdd}| td¡}t |t tdtdd¡ | dd ¡}t |t td td
|
d¡ t tdtdd}| d g¡}t |t tdtd d¡ t tdtdd}| d g¡}t |t tdtj g dg tdd¡ t tdtdd}|
|
¡ }t ||¡ t tdtdd}| ¡ }t |t tdtddd¡ d}t j t|d |jtddd W 5 Q R X d S )Nr Zcabdefr r ZefgZffggefc S s | ¡ S rj )Úupper)Úxr r r r l r z>TestCategoricalIndex2.test_method_delegation.<locals>.<lambda>ZAABBCAZCABr% Zcabdr' Zaabbr! ÚabTr z(cannot use inplace with CategoricalIndexru )Zinplace)r r r` r r* Zrename_categoriesZadd_categoriesZremove_categoriesr- r. Z as_unorderedZ
|
as_orderedrp ry r= )r r1 r2 r| r r r Útest_method_delegation^ sR ÿ ÿ ÿ ÿ þ þz,TestCategoricalIndex2.test_method_delegationN)rm rn ro r rp rq rr r- r ÚlibindexZ
|
Int8Enginer Z Int16Enginer Z Int32Enginer Z Int64Enginer r r r r r r rt s,
|
|
|
|
üþ
|
|
|
|
|
|
|
úþ
|
rt )Únumpyr- rp Z pandas._libsr r Zpandas._libs.arraysr Zpandasr/ r r Zpandas._testingZ_testingr Zpandas.core.indexes.apir r Zpandas.tests.indexes.commonr r
|
rt r r r r Ú<module> s
|
