U
|
¬ý°dN ã @ s d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ G dd dZG dd dZG d d
|
d
|
ZdS ) é N)ÚPY311)Ú CategoricalÚCategoricalIndexÚ DataFrameÚIndexÚSeriesÚ StringDtype)Úrecode_for_categoriesc @ s6 e Zd Zdd Zdd Zdd Zej ddd d
|
d gdd gg¡d d Z dd Z
|
dd Z dd Z ej ddgdddgddddgg¡dd Z dd Zd d Zd d Zd d! Zej d"dddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgddgfdddgddgdgfdddgddgdgfdddgddgdgfdddgddgdgfdddgddgdd#gfg ¡ej d$d%d&g¡d'd( Zd)d* Zd+d, Zd-d. Zej d/dgdejgdddgg¡d0d1 Zd2d3 Zd4S )5ÚTestCategoricalAPIc C sØ t tddd}t |jtdddg¡ |jr2tt tdtddd}t |jtdddg¡ |jrjtt tdd d}t |jtdddg¡ |jstt tdtdd d}t |jtdddg¡ |jsÔtd S )
|
NZacbF©ÚorderedÚaÚbÚcZbca©Ú
|
categoriesr T)r ÚlistÚtmÚassert_index_equalr r r ÚAssertionError)ÚselfZcat1Úcat2Zcat3Zcat4© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_api.pyÚtest_ordered_api s
|
|
|
z#TestCategoricalAPI.test_ordered_apic C s¨ t ddddgdd}| ¡ }|jr&t| ¡ }|js8t| d¡jsHt| d¡jrXttr`dnd}tjt |d d|_W 5 Q R X tjt |d d|_W 5 Q R X d S )
|
Nr r r Tr Fz8property 'ordered' of 'Categorical' object has no setterúcan't set attribute©Úmatch)
|
r Z as_unorderedr r Z
|
as_orderedÚ set_orderedr ÚpytestÚraisesÚAttributeError)r Úcatr Úmsgr r r Útest_set_ordered' s
|
|
ÿýz#TestCategoricalAPI.test_set_orderedc C sÎ t ddddg}| dddg¡}t | ¡ tjddddgtjd¡ t |j t
|
dddg¡ tjddddgtj d}t | ¡ |¡ t
|
dddg}t |j |¡ | dd ¡}t d
|
d d d
|
g}t ||¡ d S ) Nr r r é é é ©Zdtypec S s | ¡ S )N)Úupper)Úxr r r Ú<lambda>M ó z;TestCategoricalAPI.test_rename_categories.<locals>.<lambda>ÚAÚBÚC) r Úrename_categoriesr Úassert_numpy_array_equalÚ __array__ÚnpÚarrayÚint64r r r Úobject_Úassert_categorical_equal)r r" ÚresZexp_catÚresultÚexpectedr r r Útest_rename_categories< s ÿz)TestCategoricalAPI.test_rename_categoriesÚnew_categoriesr% r& r' é c C s< t ddddg}d}tjt|d | |¡ W 5 Q R X d S )Nr r r zKnew categories need to have the same number of items as the old categories!r )r r r Ú
|
ValueErrorr0 ©r r< r" r# r r r Ú*test_rename_categories_wrong_length_raisesQ s
|
ÿz=TestCategoricalAPI.test_rename_categories_wrong_length_raisesc C sB t ddg}| tddgddgd¡}t ddg}t ||¡ d S )Nr r r r% ©Úindex)r r0 r r r7 )r r r9 r: r r r Ú test_rename_categories_series[ s z0TestCategoricalAPI.test_rename_categories_seriesc C s t ddddg}| ddddd ¡}tddddg}t |j|¡ t ddddg}| ddd
|
¡}tddddg}t |j|¡ t ddddg}| ddddd d d ¡}tddddg}t |j|¡ t ddddg}| ddd¡}tddddg}t |j|¡ d S )Nr r r Údr= r' r& r% )r r r rD )r r é é )r r r rD ÚeÚf)rH Úg)r r0 r r r r )r r" r8 r: r r r Útest_rename_categories_dictb s z.TestCategoricalAPI.test_rename_categories_dictc C sd t ddddgdd}| ¡ }t ddddgdddgdd}| dddg¡}t ||¡ t ||¡ d S ©Nr r r Tr r )r ÚcopyÚreorder_categoriesr r7 ©r r" ÚoldÚnewr8 r r r Útest_reorder_categories} s
|
ÿ z*TestCategoricalAPI.test_reorder_categoriesr r rD r c C s@ t ddddgdd}d}tjt|d | |¡ W 5 Q R X d S )Nr r r Tr z=items in new_categories are not the same as in old categoriesr )r r r r> rM r? r r r Ú test_reorder_categories_raises s z1TestCategoricalAPI.test_reorder_categories_raisesc C s t ddddgdd}| ¡ }t ddddgddddgdd}| d¡}t ||¡ t ||¡ | dg¡}t ||¡ t ||¡ t tddd}t tdtd dd}| tdd
|
g¡}t ||¡ | t dd
|
g¡¡}t ||¡ | t dd
|
g¡}t ||¡ | dd
|
g¡}t ||¡ d S ) Nr r r Tr rD r ÚabcZabcderG )
|
r rL Úadd_categoriesr r7 r r r3 r4 r )r r" rO rP r8 r: r r r Útest_add_categories s.
|
|
ÿ
|
z&TestCategoricalAPI.test_add_categoriesc C sH t ddddgdd}t d¡}tjt|d | dg¡ W 5 Q R X d S ) Nr r r rD Tr z5new categories must not include old categories: {'d'}r )r ÚreÚescaper r r> rT )r r" r# r r r Ú#test_add_categories_existing_raises´ s
|
z6TestCategoricalAPI.test_add_categories_existing_raisesc C sÌ t tddgdd}tdgdd}| |¡}t tddgddtdddgddd}t ||¡ t tdddgt d}td gt d}| |¡}t tdddgt dtddd gt dd}t ||¡ d S )
|
Nr% r& ZInt64r( r= ©r r r rD )r r rT r r7 r )r r" Zserr9 r: r r r Ú,test_add_categories_losing_dtype_information» s
|
ÿ
|
þz?TestCategoricalAPI.test_add_categories_losing_dtype_informationc C s t ddddgdd}tdddg}tjddddgtjd}| dddg¡}| dddg¡}t |j|¡ t |
|
¡ |¡ tdddg}t |j|¡ t |
|
¡ |¡ t ddddgdd}| dg¡}t |j tjddddgtj d¡ | ddd g¡}t |j tjdd
|
ddgtj d¡ t |jtddd g¡ | dddd g¡}tdddd g}t |j|¡ t d
|
d d d d
|
gd
|
d d d gdd}t |j tjdd
|
d d dgtj d¡ t |jtd
|
d d d g¡ tjd
|
d d d d
|
gtjd}t t |¡|¡ | d d d d
|
g¡}t |j tjd d d
|
dd gtj d¡ t |jtd d d d
|
g¡ tjd
|
d d d d
|
gtjd}t t |¡|¡ | ¡ d ksxt| ¡ d
|
kst|jd d d d
|
gdd}|jr¬tt t |¡t |¡¡ | d¡ d d d d
|
g¡}|jrètt t |¡t |¡¡ d S )Nr r r Tr r( r éÿÿÿÿrD r% r& r' r= r F)r r r3 r4 r6 Úset_categoriesr r r r1 r2 ÚcodesÚint8Ú_codesr5 ZasarrayÚminr Úmaxr r ) r r" Zexp_categoriesZ
|
exp_valuesr8 Zexp_categories_backr ÚexpÚc2r r r Útest_set_categoriesÎ sJ "" $$ z&TestCategoricalAPI.test_set_categoriesz"values, categories, new_categoriesrG r TFc C s4 t ||}t |||}|j||d}t ||¡ d S )Nr )r r\ r r7 )r Úvaluesr r< r r r: r9 r r r Útest_set_categories_many s
|
z+TestCategoricalAPI.test_set_categories_manyc C s: t ddg}|jdgdd}t dtjg}t ||¡ d S )Nr- r. T)Úrename)r r\ r3 Únanr r7 )r r" r9 r: r r r Ú test_set_categories_rename_less0 s z2TestCategoricalAPI.test_set_categories_rename_lessc C s¤ t dddgddddgd}| ddddg¡ t dddgtdd}t ||¡ t dddgddddgd}|jddddgdd t dddgtdd}t ||¡ d S )
|
Nr r r rD rY rG ZacdeT)Zfastpath)r Z_set_categoriesr r r7 )r r" r: r r r Útest_set_categories_private7 s z.TestCategoricalAPI.test_set_categories_privatec C s t ddddgdd}| ¡ }t ddtjdgddgdd}| d¡}t ||¡ t ||¡ | dg¡}t ||¡ t ||¡ d S rK )r rL r3 rh Úremove_categoriesr r7 rN r r r Útest_remove_categoriesC s
|
z)TestCategoricalAPI.test_remove_categoriesÚremovalsc C s@ t dddg}t d¡}tjt|d | |¡ W 5 Q R X d S )Nr r z-removals must all be in old categories: {'c'}r )r rV rW r r r> rk )r rm r" Úmessager r r Ú test_remove_categories_raisesP s
|
z0TestCategoricalAPI.test_remove_categories_raisesc
|
C sÞ t dddddgdddddgd}tdddddg}tddddg}t |j|¡ | ¡ }t |j|¡ t |j|¡ t dddtjgdddddgd}| ¡ }t |jtt dddg¡¡ tjddd d
|
gtj d }t
|
|j |¡ t |j|¡ d tjd dd d tjg}t |t dd}| ¡ }t |jtdd d g¡ tjd d
|
dddd d
|
gtj d }t
|
|j |¡ | ¡ |ksrtt d} tj | d d d
d¡ d¡}tj|tj t|d¡< t || d}| ¡ }| ¡ | ¡ ksÚtd S )Nr r r rD rG rY r r% r& r[ r( ÚFÚDr. ZABCDEFG)re r Úabcdefghijklmnopqrstuvwxyzi' Úobjectéd )r r r r r Zremove_unused_categoriesr3 rh r4 r^ r1 r] r Útolistr ÚrandomÚchoiceZastypeÚlen)
|
r r Zexp_categories_allZexp_categories_droppedr8 Z exp_codesÚvalr" ÚoutÚalphar r r Ú test_remove_unused_categoriesX s4 z0TestCategoricalAPI.test_remove_unused_categoriesN)Ú__name__Ú
|
__module__Ú __qualname__r r$ r; r ÚmarkÚ parametrizer@ rC rJ rQ rR rU rX rZ rd rf ri rj rl r3 rh ro r| r r r r r
|
sZ
|
|
ýþ
|
Bìþ
|
r
|
c @ s e Zd Zdd ZdS )Ú TestCategoricalAPIWithFactorc C sz | ¡ }|jsttdddgd|jd}tdddgdd dgd
|
|d }t ||¡ | ¡ }| dddd g¡}| ¡ }tt d |jdd}tddddgdd ddgd
|
|d }t ||¡ t
|
dddddddddddg }| ¡ }tdddg|jdd}tdddgdddgd
|
|d }t ||¡ t
|
t j dddg}| ¡ }tdddgd dd gd
|
tddt j gddgddd }t ||¡ d S )Nr r r r )Únamer r' r& g Ø?g Ð?)ÚcountsZfreqsrA rD Zabcd)r r r r% rE g]tÑEÝ?gtÑE]tÑ?g à?)r r ) Zdescriber r r r r Zassert_frame_equalrL r\ r r r3 rh )r ZfactorÚdescZ exp_indexr: r" r r r Ú test_describe} sX
|
ÿ ÿ ÿþ þ
|
ÿþz*TestCategoricalAPIWithFactor.test_describeN)r} r~ r r r r r r r | s r c @ s e Zd Zdd Zej dddgddgddgddgfddgddgddgddgfddgddgddgddgfddgddgddgddgfddddgddgdddgddddgfddd d gdddgddgddd
|
d
|
gfddd
|
gdddgdddgddd
|
gfddd
|
gdddgdgd
|
dd
|
gfddd
|
gdddgd gd
|
d
|
d
|
gfddd
|
gdddgg d
|
d
|
d
|
gfd
|
d
|
gg ddgd
|
d
|
gfddgddgddgddgfg ¡d d Zdd ZdS )ÚTestPrivateCategoricalAPIc C s t ddddtjg}tjdddddgdd }t |j|¡ trBd
|
nd }tj t
|
|d tjdddddgdd |_W 5 Q R X |j}tj t d d d|d< W 5 Q R X d|d< tjdddddgdd }t |j|¡ d|j d< tjdddddgdd }t |j|¡ d S )Nr r r r r% r& r[ r^ r( z6property 'codes' of 'Categorical' object has no setterr r z#assignment destination is read-onlyr= ) r r3 rh r4 r r1 r] r r r r! r> r_ )r r rb r# r] r r r Útest_codes_immutable¯ s$ ÿý$
|
z.TestPrivateCategoricalAPI.test_codes_immutablezcodes, old, new, expectedr r% r r r r& r[ rD c C sL t j|t jd}t j|t jd}t|}t|}t|||}t ||¡ d S )Nr( )r3 Z
|
asanyarrayr^ r r r r1 )r r] rO rP r: r9 r r r Útest_recode_to_categoriesÍ s z3TestPrivateCategoricalAPI.test_recode_to_categoriesc C sR d}t |¡}t|}t j|d ddt jd}t|}t|||}t ||¡ d S )Niè r% r[ r( )r3 Zaranger Úint16r r r1 )r ÚNr] rO r: rP r9 r r r Ú test_recode_to_categories_largeæ s
|
z9TestPrivateCategoricalAPI.test_recode_to_categories_largeN) r} r~ r r r r r r r r r r r r ® s& $$" ôþ
|
r )rV Únumpyr3 r Z pandas.compatr Zpandasr r r r r r Zpandas._testingZ_testingr Z pandas.core.arrays.categoricalr r
|
r r r r r r Ú<module> s j2
|
