U
|
ý°dx ã @ sV d dl mZ d dlZd dlZd dlmZmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é )ÚdateN)Ú CategoricalÚCategoricalDtypeÚCategoricalIndexÚIndexÚ IntervalIndexc @ sl e Zd Zdd Zej dddg¡ej dddg¡ej d ddg¡d
|
d Zej d ddg¡d d ZdS )Ú
|
TestAstypec C sÚ t tdtddd}| t¡}t |tt |¡¡ | |¡sDt
|
t |tsRt
|
t |t r`t
|
t j ddgddgd d
|
}t tjd d d g|dd}| d¡}|jd d d gdtjd}t ||¡ t |j}t ||¡ d S )NÚaabbcaÚcabF©Ú
|
categoriesÚorderedgü©ñÒMbP¿g @é é Úright)Úleftr Úclosedr é éÿÿÿÿTÚinterval)Z
|
allow_fillZ
|
fill_value)r ÚlistÚastypeÚobjectÚtmÚassert_index_equalr ÚnpÚarrayÚequalsÚAssertionErrorÚ
|
isinstancer Z from_arraysr Z
|
from_codesZtakeÚnanÚvalues)ÚselfÚciÚresultÚiiÚexpected© r' úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_astype.pyÚ test_astype s
|
ÿ
|
|
zTestAstype.test_astypeÚnameNZfooÚ dtype_orderedTFÚ index_orderedc C s t tdtd|d}|r$| |¡}t|d}| |¡}t | ¡ ||j|d}t ||¡ t| ¡ ¡ d d
|}| |¡}t | ¡ ||d}t ||¡ |dkr¾| d ¡}|}t ||¡ d S )
|
Nr r
|
r )r )r* r r r )r* ÚdtypeFÚcategory)
|
r r Úrenamer r Útolistr r r Úunique)r" r* r+ r, Úindexr- r$ r& r' r' r( Útest_astype_category* s0 ÿ
|
|
|
ü
|
|
z TestAstype.test_astype_categoryÚboxc C sf t ¡ }t||g}|jtks"t|r,|j}| d¡}| t¡}|jtksNtt|d t ksbtd S )Nr. r ) r Útodayr r- r r r r Útype)r" r4 ÚvÚobjÚcatZrtripr' r' r( Ú test_categorical_date_roundtripL s
|
|
z*TestAstype.test_categorical_date_roundtrip) Ú__name__Ú
|
__module__Ú __qualname__r) ÚpytestÚmarkZ parametrizer3 r: r' r' r' r( r s r )Údatetimer Únumpyr r> Zpandasr r r r r Zpandas._testingZ_testingr r r' r' r' r( Ú<module> s
|
|
