U
|
¬ý°dO ã @ sh d dl Zd dlZd dlZd dlmZ dd Zdd Z dd Z
|
dd Z d
|
d Z d d Z dd Zdd ZdS )é Nc C sô t jddd gdd} tjtdd | d¡ W 5 Q R X tjtdd | d ¡ W 5 Q R X | d
|
¡}tjddtjgd
|
d}t ||¡ t jd d d gdd} | d¡}tjdddgdd}t ||¡ | d ¡}tjdddgd d}t ||¡ d S )Nç¹?çÉ?ÚFloat64©Údtypez cannot convert NA to integer)ÚmatchÚint64z cannot convert float NaN to boolÚboolÚfloat64ç ç ð?g à?r é FT)
|
ÚpdÚarrayÚpytestZraisesÚ
|
ValueErrorÚastypeÚnpÚnanÚtmÚassert_numpy_array_equal©ÚarrÚresultÚexpected© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/floating/test_astype.pyÚ test_astype s
|
|
|
r c C sr t jddd gdd} | d¡}t || ¡ | t ¡ ¡}t || ¡ | d¡}t jddd gdd}t ||¡ d S )Nr r r r ÚFloat32)r r r r Ú assert_extension_array_equalÚ Float64Dtyper r r r Ú test_astype_to_floating_array! s
|
|
r! c C s\ t jddd gdd} | d¡}t jddd gdd}t ||¡ | t ¡ ¡}t ||¡ d S )Nr r r r ÚbooleanFT)r r r r r Z BooleanDtyper r r r Ú test_astype_to_boolean_array. s
|
r# c C sB t jddd gdd} | d¡}t jddd gdd}t ||¡ d S )Nr g ø?r r ZInt64r r )r r r r r r r r r Ú test_astype_to_integer_array9 s
|
r$ c C sP t jddd gdd} tjdddgdd}t | t¡|¡ t | d ¡|¡ d S )
|
Nr r r r z0.1z0.2z<NA>ZU32Ústr)r r r r r r r% )Úar r r r Útest_astype_strB s r' c C sh t jddd gdd} t jddd gdd}| jddd}|| k sBtt || ¡rRtd|d< t | |¡ t j|d< t | |¡ | jdd d}|| kstt |j | j ¡sªtt |j
|
| j
|
¡s¾td|d< | d dksÖtt j|d< | d t jksòtt jddd gdd} t jddd gdd}| jd
|
d d}t || ¡r:td|d< t | |¡ t j|d< t | |¡ d S ) Nr r r r T)Úcopyé
|
r Fr ) r r r ÚAssertionErrorr Z shares_memoryr ÚNAr Ú_dataZ_mask)r Úorigr r r r Útest_astype_copyJ s2
|
|
|
r. c C sf t jdt jg| d}| t¡}tjdt jgtd}t ||¡ t|d t sPt
|
|d t jksbt
|
d S )Nr r r r ) r r r+ r Úobjectr r r Ú
|
isinstanceÚfloatr* )r r r r r r r Útest_astype_objectn s
|
r2 c C sN t jddddgdd} | t ¡ ¡}t jddd d
|
gt ¡ d}t ||¡ d S ) NÚ1Ú2Ú3Ú4r/ r r g @g @g @)r ZSeriesr r r Zassert_series_equal)Z
|
testseriesr r r r r Útest_Float64_conversiony s r7 )Únumpyr r Zpandasr Zpandas._testingZ_testingr r r! r# r$ r' r. r2 r7 r r r r Ú<module> s $
|
