U
|
ý°d ã @ s® d dl Zd dlZd dlmZ d dlmZ d dlZd dl m
|
Z d dl m Z G dd dejZG dd de ZG d d
|
d
|
Zd d Zd d Zej de ¡ e ¡ g¡dd ZdS )é N)Údtypes)Úis_extension_array_dtype)ÚExtensionArrayc @ s e Zd ZdS )Ú
|
DummyDtypeN)Ú__name__Ú
|
__module__Ú __qualname__© r r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_common.pyr s r c @ s8 e Zd ZddddZdd Zedd Zd d
|
d ZdS ) Ú
|
DummyArrayN)Úreturnc C s
|
|| _ d S ©N©Údata)Úselfr r r r
|
Ú__init__ s zDummyArray.__init__c C s | j S r r )r Údtyper r r
|
Ú __array__ s zDummyArray.__array__c C s t S r )r )r r r r
|
r s zDummyArray.dtypeTc C s0 t |tr |r t| | jS | S tj| ||dS )N)r Úcopy)Ú
|
isinstancer Útyper ÚnpÚarray)r r r r r r
|
Úastype s
|
|
zDummyArray.astype)T)r r r r r Úpropertyr r r r r r
|
r s
|
|
r c @ s e Zd Zej de g ¡e g ¡je e g ¡¡e
|
e e ddg¡g¡dd Zej de g ¡e e g ¡¡g¡dd ZdS ) ÚTestExtensionArrayDtypeÚvaluesé é c C s t |s td S r ©r ÚAssertionError©r r r r r
|
Ú test_is_extension_array_dtype& s z5TestExtensionArrayDtype.test_is_extension_array_dtypec C s t |r td S r r r! r r r
|
Ú!test_is_not_extension_array_dtype3 s z9TestExtensionArrayDtype.test_is_not_extension_array_dtypeN)r r r ÚpytestÚmarkÚ parametrizeÚpdZ Categoricalr ZSeriesr r r r r" r# r r r r
|
r % s
|
ûþ
|
|
"r c C sX t t dddg¡} tjdddgtd}| t¡}t ||¡ | d¡}t ||¡ d S )Nr r é ©r Úobject)r r r r* r ÚtmZassert_numpy_array_equal)ÚarrÚexpectedÚresultr r r
|
Ú test_astype8 s
|
|
r/ c C sR t tjdddgtjd} | j| jdd}| |ks6t| | j¡}| |k sNtd S )Nr r r( r) F)r )r r r Úint64r r r )r, r. r r r
|
Útest_astype_no_copyC s
|
r1 r c C s t | tjstt| s td S r )r r ÚExtensionDtyper r r) r r r
|
r" M s r" )Únumpyr r$ Zpandas.core.dtypesr Zpandas.core.dtypes.commonr Zpandasr' Zpandas._testingZ_testingr+ Zpandas.core.arraysr r2 r r r r/ r1 r% r& ZCategoricalDtypeZ IntervalDtyper" r r r r
|
Ú<module> s
|
|
