U
|
¬ý°dR ã @ sN d dl Zd dlZd dlmZmZmZmZmZm Z d dl
|
m Z G dd dZ dS )é N)Ú CategoricalÚCategoricalDtypeÚNaTÚ TimestampÚarrayÚ to_datetimec @ sr e Zd Zdd Zdd Zej dddg¡dd Zej d
|
ddg¡ej d ddg¡d d Z dd Z
|
dd Z dS )Ú
|
TestAstypec C sZ t dd tdD }tjjddd}tj||d}| d¡}t|dd}t
|
||¡ d S )Nc S s g | ] }t |qS © ©Ústr©Ú.0Úir r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_astype.pyÚ
|
<listcomp> s zMTestAstype.test_astype_str_int_categories_to_nullable_int.<locals>.<listcomp>é é ©Úsize©ÚdtypeZInt64© r ÚrangeÚnpÚrandomÚrandintr Z
|
from_codesÚastyper ÚtmZ assert_extension_array_equal©Úselfr ÚcodesZarrÚresÚexpectedr r r Ú.test_astype_str_int_categories_to_nullable_int s
|
z9TestAstype.test_astype_str_int_categories_to_nullable_intc C s^ t dd tdD }tjjddd}tj||d}| d¡}t|ddd }t
|
||¡ d S ) Nc S s g | ]}t |d qS )é r
|
r r r r r s zOTestAstype.test_astype_str_int_categories_to_nullable_float.<locals>.<listcomp>r r r r ZFloat64r$ r r r r r Ú0test_astype_str_int_categories_to_nullable_float s
|
z;TestAstype.test_astype_str_int_categories_to_nullable_floatÚorderedTFc
|
C sæ t td|d}| t¡}t |¡}t ||¡ d}tj t
|
|d | t ¡ W 5 Q R X t dddddddddg |d}| t¡}tj|td}t ||¡ | t ¡}tj|d d}t ||¡ | t ¡}tj|t d}t ||¡ d S )
|
NZabbaaccc©r& z#Cannot cast object dtype to float64)Úmatchr é r$ r Úint) r Úlistr Úobjectr r r Úassert_numpy_array_equalÚpytestZraisesÚ
|
ValueErrorÚfloatr* )r r& ÚcatÚresultr" Úmsgr r r Ú test_astype$ s"
|
|
|
|
|
zTestAstype.test_astypeÚ dtype_orderedÚ cat_orderedc C s t d}t|t d|d}t|d}| |¡}t||j|d}t ||¡ tt d|}| |¡}t||d}t ||¡ |dkr| d¡}|}t ||¡ d S ) NZ abcaacbabZbac)Ú
|
categoriesr& r' Zadcr FÚcategory)r+ r r r r7 r Zassert_categorical_equal)r r5 r6 Údatar1 r r2 r" r r r Útest_astype_category> s
|
|
|
|
z TestAstype.test_astype_categoryc C s@ t tdtg}| t¡}tjtdtgdd}t ||¡ d S )Nz
|
2021-03-27z2021-03-27 00:00:00r, r )
|
r r r r r, r r r r r- ©r r1 r2 r" r r r Ú&test_astype_object_datetime_categoriesW s
|
z1TestAstype.test_astype_object_datetime_categoriesc C s< t tdg}| t¡}tjtdgdd}t ||¡ d S )Nz
|
2014-01-01z2014-01-01 00:00:00r, r )r r r r, r r r r- r; r r r Ú'test_astype_object_timestamp_categories^ s
|
z2TestAstype.test_astype_object_timestamp_categoriesN) Ú__name__Ú
|
__module__Ú __qualname__r# r% r. ÚmarkZ parametrizer4 r: r< r= r r r r r s
|
|
|
r )Únumpyr r. Zpandasr r r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
