U
|
ý°d ã @ s> d dl Z d dlZd dlZd dlZd dlmZ G dd dZ dS )é Nc @ sh e Zd Zej dde d¡fdg¡dd Zdd Z d d
|
Z
|
d d Z d d Z dd Z dd Zdd ZdS )ÚTestConvertDtypeszconvert_integer, expectedFÚint32)TÚInt32c C s¬ t t jdddgt d¡dt jdddgt d ¡dd
|
¡}t d |¡ | d d |d ¡}W 5 Q R X t t jdddg|dt jdddgd| ddd
|
¡}t ||¡ d S )Né é é r ©ÚdtypeÚxÚyÚzÚO©ÚaÚbÚstring_storageTFzstring[ú]) ÚpdÚ DataFrameÚSeriesÚnpr Zoption_contextÚconvert_dtypesÚtmÚassert_frame_equal)ÚselfZconvert_integerÚexpectedr ÚdfÚresult© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_convert_dtypes.pyÚtest_convert_dtypes s þÿþÿz%TestConvertDtypes.test_convert_dtypesc C s t ¡ }t || ¡ ¡ d S )N)r r r r r )r Zempty_dfr r r Útest_convert_empty! s z$TestConvertDtypes.test_convert_emptyc C sL t ddgddgd¡}d|j_| ¡ }t |j|j¡ |jjdksHtd S )Nr r r é r Úcols)r r ÚcolumnsÚnamer r Zassert_index_equalÚAssertionError)r r r r r r Ú'test_convert_dtypes_retain_column_names& s
|
z9TestConvertDtypes.test_convert_dtypes_retain_column_namesc C s t d¡}t tjdddgt d¡dtjddd gt d ¡dtjd
|
d d gt d ¡dtjtjd d gt d¡dt tjddd¡t tj ddd¡d¡}|j
|
dd}t tj |j dddg| ¡ d¡tj | ddd g¡¡tj | d
|
d d g¡¡tj | d d dg¡¡tj |j t ddd¡t ddd¡t ddd¡g|jddd¡tj |j t d¡t d¡t d¡g| d¡d¡d¡}t ||¡ d S )NÚpyarrowr r r r r r
|
r r TFç Y@éÈ ÚfloatZ2022)ZperiodsZ1D)r r ÚcÚdÚeÚf©Z dtype_backend©Útypeç i@iæ Úns)Úunit)ÚpytestÚ importorskipr r r r r ÚnanZ
|
date_rangeZtimedelta_ranger ÚarraysÚArrowExtensionArrayÚarrayr ÚdatetimeÚ timestampÚ timedeltaÚdurationr r ©r Úpar r r r r r Útest_pyarrow_dtype_backend/ sN
|
úÿ
|
ÿ ý
|
úÿ
|
ýúÿïÿ z,TestConvertDtypes.test_pyarrow_dtype_backendc C s: t d¡ tjdddgdd}|jdd}t ||¡ d S )Nr( r r r zint64[pyarrow]r r0 ©r6 r7 r r r r r ©r r r r r r Ú*test_pyarrow_dtype_backend_already_pyarrow\ s
|
z<TestConvertDtypes.test_pyarrow_dtype_backend_already_pyarrowc C sæ t d¡}t tjddd gddtjddd gddtjd d
|
d gd dtjd d d gddd¡}|jdd}t tj |jddd g| ¡ d¡tj | ddd g¡¡tj | d d
|
d g¡¡tj | d d dg¡¡d¡}t
|
||¡ d S )Nr( r r r r r
|
r zstring[python]TFÚbooleanr) r* ZFloat64)r r r, r- r0 r1 r3 ) r6 r7 r r r r r9 r: r; r r r r@ r r r Ú/test_pyarrow_dtype_backend_from_pandas_nullableb s&
|
üÿ ÿúÿ
|
zATestConvertDtypes.test_pyarrow_dtype_backend_from_pandas_nullablec C s4 t d¡ tjdgd}|jdd}t ||¡ d S )Nr( r )r$ r0 rC rD r r r Ú test_pyarrow_dtype_empty_objecty s
|
z1TestConvertDtypes.test_pyarrow_dtype_empty_objectc C sB t ddddgi¡}d}tjt|d |jdd W 5 Q R X d S ) Nr r r r zPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.)ÚmatchÚnumpyr0 )r r r6 ZraisesÚ
|
ValueErrorr )r r Úmsgr r r Ú test_pyarrow_engine_lines_false s
|
ÿz1TestConvertDtypes.test_pyarrow_engine_lines_falseN)Ú__name__Ú
|
__module__Ú __qualname__r6 ÚmarkZ parametrizer r r r! r' rB rE rG rH rM r r r r r
|
s ÿ
|
-r )
|
r< rJ r r6 Zpandasr Zpandas._testingZ_testingr r r r r r Ú<module> s
|
|
