U
|
ý°d; ã @ sT d dl Zd dlmZ d dlmZ d dlZd dlm Z
|
d dl m Z G dd de Z dS )é N)Úis_extension_array_dtype)ÚExtensionDtype)Ú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 S ) ÚBaseInterfaceTestsz,Tests that the basic interface is satisfied.c C s t |dkstd S ©Néd )ÚlenÚAssertionError©ÚselfÚdata© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/interface.pyÚtest_len s zBaseInterfaceTests.test_lenc C s |j dkstd S r )Úsizer r
|
r r r Ú test_size s z BaseInterfaceTests.test_sizec C s |j dkstd S )Né )Úndimr r
|
r r r Ú test_ndim s z BaseInterfaceTests.test_ndimc C s |j dkstd S )NT)Z _can_hold_nar r
|
r r r Útest_can_hold_na_valid s z)BaseInterfaceTests.test_can_hold_na_validc C s |j j}|| ¡ }|d |ks&t|d |ks6t||ksBt||ksNttjD ]6}||ksTt|t|krrqT||ks~t||ksTtqTd S )Nr )ÚdtypeÚna_valueÚisnar ÚtmZ NULL_OBJECTSÚtype)r r Ú data_missingr Z na_value_objr r r Ú test_contains s
|
z BaseInterfaceTests.test_containsc C s( t |¡}|jdd}||jks$td S )NF)Úindex)ÚpdÚSeriesZ memory_usageÚnbytesr )r r ÚsÚresultr r r Útest_memory_usage9 s
|
z$BaseInterfaceTests.test_memory_usagec C sN t |¡}|d |d ks tt j|td}t jt|td}t ||¡ d S )Nr ©r )ÚnpÚarrayr ÚobjectÚlistr Zassert_numpy_array_equal©r r r" Úexpectedr r r Útest_array_interface> s
|
|
z'BaseInterfaceTests.test_array_interfacec C s@ t |s tt |jstt t |¡s,tt|jts<td S ©N)r r r r r Ú
|
isinstancer r
|
r r r Ú test_is_extension_array_dtypeF s z0BaseInterfaceTests.test_is_extension_array_dtypec C s t |drtt |dr td S )NÚvaluesZ_values)Úhasattrr r
|
r r r Útest_no_values_attributeL s z+BaseInterfaceTests.test_no_values_attributec C s4 t |¡}t|jdr0|jjd j|jjks0td S )NÚblocksr ) r r r0 Z_mgrr2 Z
|
is_numericr Z _is_numericr ©r r r" r r r Útest_is_numeric_honoredR s
|
z*BaseInterfaceTests.test_is_numeric_honoredc C sT | ¡ }t|rP| d¡s t| ¡ s*t| d¡r8t| ¡ rDt|jjsPtd S )NÚanyÚall)r r Z_reducer r5 r6 r Z _is_boolean)r r Únar r r Útest_isna_extension_arrayW s z,BaseInterfaceTests.test_isna_extension_arrayc C s@ |d |d kst | ¡ }|d |d< |d |d ks<t d S )Nr r )r Úcopyr3 r r r Ú test_copyd s z BaseInterfaceTests.test_copyc C sl |d |d kst | ¡ }||k s(t t|t|ks<t |d |d< |d |d ks\t |jd d d S )Nr r r$ )r Úviewr r3 r r r Ú test_viewl s z BaseInterfaceTests.test_viewc C s. | ¡ }t|}t|ts t||ks*td S r, )Útolistr( r- r r) r r r Ú test_tolist{ s z BaseInterfaceTests.test_tolistN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r r r r# r+ r. r1 r4 r8 r: r< r> r r r r r s r )Únumpyr% Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr Zpandasr Zpandas._testingZ_testingr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
