U
|
ý°dÅ ã @ sf d dl Zd dlZd dlmZ d dlmZmZmZ d dl m
|
Z d dl m Z mZ G dd dZdd ZdS ) é N)ÚIntIndex)Ú DataFrameÚSeriesÚ Timestamp)Ú SparseArrayÚ SparseDtypec @ sJ e Zd Zdd Zdd Zdd Zej de dd gd
|
e d d ge
|
e d d fe dd ge e ddgfe dd gd de e ddge
|
e dd fe e dd gde ejdd gdd e
|
dedd ¡e dd dgee dddge
|
edd fe ddge e ddgfe dd dgee dd dge
|
edd fg¡dd Zd d Zd d Zd d! Zd"S )#Ú
|
TestAstypec C s t d d ddg}| d¡}t d d ddgt d¡d}t ||¡ tddd}| |¡}t tjdd g|j dt
|
d
|
dd g|¡}t ||¡ td d}| |¡}t tjddgtj dt
|
d
|
dd g|¡}t ||¡ t dtj dd gdd}t jtdd | d¡ W 5 Q R X d S )Nr é zSparse[float32]Úfloat32©ÚdtypeÚfloat64©Z
|
fill_valueç ç @é é Úint64é ZNA©Úmatchz
|
Sparse[i8])r ÚastypeÚnpr ÚtmÚassert_sp_array_equalr Z _simple_newÚarrayÚsubtyper r ÚnanÚpytestÚraisesÚ
|
ValueError)ÚselfÚarrÚresultÚexpectedr © r% ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_astype.pyÚ test_astype s.
|
|
ÿ
|
|
ÿ zTestAstype.test_astypec C s t ddddgttdd}| t¡}tjddddgtd}t ||¡ | ttd¡}t ddddgttdd}t ||¡ d S )Nr r r FT)
|
r r Úintr Úboolr r r Úassert_numpy_array_equalr )r! Úar# r$ r% r% r& Útest_astype_bool, s
|
|
ÿzTestAstype.test_astype_boolc C sF t dddg¡}t|dd}t |¡}| |¡}t || |¡¡ d S )Nr r r r )r r r r r r r* )r! Zany_real_numpy_dtypeÚvalsr" ÚtypÚresr% r% r& Útest_astype_all9 s
|
|
|
zTestAstype.test_astype_allzarr, dtype, expectedr r Úfloatr ç ð?r FTr údatetime64[ns]Z1970é
|
Ú0Ú1Ú10Z20g $@g 4@c C s" | |j |¡¡}t ||¡ d S )N)r r Z update_dtyper r )r! r" r r$ r# r% r% r& Útest_astype_more@ s $zTestAstype.test_astype_morec C s6 t dtjg}tjtdd | t¡ W 5 Q R X d S )Nr2 zCannot convert non-finiter )r r r r r r r r( )r! r" r% r% r& Útest_astype_nan_raisesg s z!TestAstype.test_astype_nan_raisesc C sH t dddg}ttd}|j|dd}t ddd gd
|
d }t ||¡ d S ) Nr r r r F)Úcopyr2 r g @r r )r r r1 r r r )r! r" r r# r$ r% r% r& Útest_astype_copy_falsel s
|
|
z!TestAstype.test_astype_copy_falsec
|
C s t jdddgdd}t|}| d¡}| d¡}t ||¡ tdt t j¡j }| |¡}t |
|
¡ |¡ td|d }t||d}| d¡} t | |¡ d S ) NZNaTz
|
2016-01-02z
|
2016-01-03zM8[ns]r r r3 r ) r r r r r r* r Ziinfor ÚminZto_numpy)
|
r! Úvaluesr" r# r$ Z dtype_int64Zresult2r Zarr3Zresult3r% r% r& Útest_astype_dt64_to_int64v s
|
|
|
|
z$TestAstype.test_astype_dt64_to_int64N)Ú__name__Ú
|
__module__Ú __qualname__r' r, r0 r ÚmarkZ parametrizer r r1 r) Úparamr r r ÚstrÚobjectr8 r9 r; r> r% r% r% r& r sJ
|
ýý
|
þý ý ýæþ#
|
|
r c C sd t ddgddgddggddgd} | d td d
|
d ¡}tdddgdtd d
|
d d }t ||¡ d S )Nr+ r Úbr r ÚAÚB)ÚcolumnsÚcategoryÚcr rE )Únamer )r r r r r Zassert_series_equal)Zdfr# r$ r% r% r& Ú5test_dtype_sparse_with_fill_value_not_present_in_data s "
|
ÿrM )Únumpyr r Zpandas._libs.sparser Zpandasr r r Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r r rM r% r% r% r& Ú<module> s z
|
