U
|
ý°dÑ ã @ sL d dl Zd dlZd dlZd dlmZmZmZ d dl m
|
Z
|
G dd de
|
Z dS )é N)Ú infer_dtypeÚis_object_dtypeÚis_string_dtype)ÚBaseExtensionTestsc @ sÊ e Zd ZdZdd Zdd Zdd Zdd 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% Zd&d' Zd(d) Zej d*d+d,g¡d-d. Zd/S )0ÚBaseDtypeTestsz%Base class for ExtensionDtype classesc C s t |jtstd S ©N)Ú
|
isinstanceÚnameÚstrÚAssertionError©ÚselfÚdtype© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/dtype.pyÚ test_name s zBaseDtypeTests.test_namec C s t d}|j|kstd S )NZ biufcmMOSUV)ÚsetÚkindr )r r Zvalidr r r Ú test_kind s zBaseDtypeTests.test_kindc C s t | |j¡}|dks td S ©NT)ÚtypeÚis_dtyper r ©r r Úresultr r r Útest_is_dtype_from_name s z&BaseDtypeTests.test_is_dtype_from_namec C s | |¡dkstd S r ©r r ©r Údatar r r r Útest_is_dtype_unboxes_dtype s z*BaseDtypeTests.test_is_dtype_unboxes_dtypec C s t | |¡}|dkstd S r )r r r r r r r Útest_is_dtype_from_self s z&BaseDtypeTests.test_is_dtype_from_selfc C s | dddg¡dkstd S )Né é é Fr r r r r Útest_is_dtype_other_input" s z(BaseDtypeTests.test_is_dtype_other_inputc C s t |r td S r )r r r r r r Útest_is_not_string_type% s z&BaseDtypeTests.test_is_not_string_typec C s t |r td S r )r r r r r r Útest_is_not_object_type( s z&BaseDtypeTests.test_is_not_object_typec C s$ ||j kst||j d ks td S )Nz-suffix©r r r r r r Útest_eq_with_str+ s z BaseDtypeTests.test_eq_with_strc C s |t d¡kstd S )NÚobject)Únpr r r r r r Útest_eq_with_numpy_object/ s z(BaseDtypeTests.test_eq_with_numpy_objectc C s ||ks t |t kst d S r )r r( r r r r Útest_eq_with_self2 s z BaseDtypeTests.test_eq_with_selfc C s | ¡ t|kstd S r )Zconstruct_array_typer r r r r r Útest_array_type6 s z BaseDtypeTests.test_array_typec C sÊ |j }t tj||d|ddd¡}|jt|k}zt d¡dk}W n tk
|
r\ d}Y nX |sfttjddddgt d d
|
}|
|
||¡ tjddddgt d d
|
}|j t¡t|k}|
|
||¡ d S ) N)r Zfoor )ÚAÚBÚCÚDÚint64ZInt64TFÚABCD)Úindex) r ÚpdZ DataFrameZSeriesZdtypesr
|
r) Ú TypeErrorr ÚlistZassert_series_equalÚapply)r r r Zdfr Znew_numpy_behaviorÚexpectedr r r Útest_check_dtype9 s ÿ
|
z BaseDtypeTests.test_check_dtypec C s t | d S r )Úhashr r r r Ú test_hashableR s z BaseDtypeTests.test_hashablec C s t ||jkstd S r )r
|
r r r r r r Útest_strU s zBaseDtypeTests.test_strc C s ||j kst|dkstd S )NZ anonther_typer& r r r r Útest_eqX s zBaseDtypeTests.test_eqc C sH | |j¡}t|t|ks tt| |j¡}t|t|ksDtd S r )Úconstruct_from_stringr r r r r r r Ú#test_construct_from_string_own_name\ s z2BaseDtypeTests.test_construct_from_string_own_namec C s> dt |j d}tjt|d t | d¡ W 5 Q R X d S )NzCannot construct a 'z' from 'another_type'©ÚmatchZ another_type)r Ú__name__ÚpytestÚraisesr5 r> )r r Úmsgr r r Ú.test_construct_from_string_another_type_raisesd s z=BaseDtypeTests.test_construct_from_string_another_type_raisesc C s, t jtdd t| d¡ W 5 Q R X d S )Nz;'construct_from_string' expects a string, got <class 'int'>r@ r )rC rD r5 r r> r r r r Ú,test_construct_from_string_wrong_type_raisesi s
|
þz;BaseDtypeTests.test_construct_from_string_wrong_type_raisesc C s | |g¡|kstd S r )Z_get_common_dtyper r r r r Útest_get_common_dtypep s z$BaseDtypeTests.test_get_common_dtypeÚskipnaTFc C s8 t ||d}t|tstt ||d}t|ts4td S )N)rI )r r r
|
r )r r Z data_missingrI Úresr r r Útest_infer_dtypew s z BaseDtypeTests.test_infer_dtypeN) rB Ú
|
__module__Ú __qualname__Ú__doc__r r r r r r# r$ r% r' r* r+ r, r9 r; r< r= r? rF rG rH rC ÚmarkZ parametrizerK r r r r r s. r ) Únumpyr) rC Zpandasr4 Zpandas.api.typesr r r Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
|
