U
|
¬ý°d# ã @ sl d dl Zd dlZd dlmZ d dlZd dlmZ d dl
|
m Z m Z dd Z ej dddd d
|
g¡d d Zd d Zej dddg¡dd Zej dddg¡dd Zdd Zdd Zdd Zej d e d e d g¡d d Zd!d" Zd#d$ Zd%d& Zej d'ddg¡d(d) Zej d'ddg¡ej d d*d+d,g¡d-d. Z ej d d/d+d,g¡d0d1 Z d2d3 Z d4d5 Z dS )6é N)ÚABCIndex)Ú Int8DtypeÚ UInt32Dtypec C sF | j r t | j¡jdks4tnt | j¡jdks4t| jd k sBtd S )NÚiÚu)Zis_signed_integerÚnpÚdtypeÚtypeÚkindÚAssertionErrorÚname©r © r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/integer/test_dtypes.pyÚ test_dtypes s r ÚopÚsumÚminÚmaxÚprodc C sÀ t dddgdd dgt jdd dgddd¡}t|j| }| dkrVt|tjsdtnt|t sdtt|
|
d ¡| }t jt d
|
d g¡t jddgddd t j ddgd d d}t ||¡ d S )NÚaÚbé é ÚInt64r )ÚAÚBÚC> r r r r r ç ð?g @)r r )r )Úindex)ÚpdZ DataFrameÚarrayÚgetattrr Ú
|
isinstancer Úint64r ÚintÚgroupbyÚIndexÚtmZassert_frame_equal)r ZdfÚresultÚexpectedr r r Útest_preserve_dtypes s ýÿ þr+ c C sB t jtjddgdd} d}tjt|d | d¡ W 5 Q R X d S )Nr é ÚInt8r ú cannot convert NA to integer©ÚmatchZuint32)r r! r ÚnanÚpytestÚraisesÚ
|
ValueErrorÚastype)ÚarrÚmsgr r r Útest_astype_nansafe6 s r8 ÚdropnaTFc C sp | d d
} |r&t | | ¡ ¡}n| }t tj|| jd¡}tj|| jd}| j|jks`tt ||¡ d S )Né
|
r ) r r! Úisnar r' r r r( Úassert_index_equal)Úall_datar9 Úotherr) r* r r r Útest_construct_index? s r? c C sr | d d
} |r | | ¡ }n| }| j}t t |¡¡}t|tsHt| |¡}| t
|
¡ |¡}t ||¡ d S )Nr: ) r; r r r' r r! r# r r r5 Úobjectr( r< )r= r9 r> r Úidxr) r* r r r Útest_astype_indexP s
|
rB c C s| | d d
} | | ¡ }| }t }t |¡}| | j¡}t |¡}t ||¡ t |¡}| |¡}tj||d}t ||¡ t |¡}| | jj¡}t |j | jj¡¡}t ||¡ t |¡}| | j¡}t |¡}t ||¡ t |¡}| |¡}tj||d}t ||¡ t |¡}d}t
|
j t |d | | jj¡ W 5 Q R X t |¡}| d¡}t t |¡¡}t ||¡ d S )Nr: r r. r/ r@ )r; r r ÚSeriesr5 r r( Úassert_series_equalZ numpy_dtypeÚ_datar2 r3 r4 r Zasarray)r= ÚintsÚmixedr Úsr) r* r7 r r r Ú test_astypec s@
|
|
|
|
|
|
|
|
|
|
|
|
rI c C sp t jdddd gdd} t jdddd gdd}| jddd}|| k sFtt || ¡rVtd|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 jdddd gdd} t jdddd gdd}| jd d
|
d}t || ¡rBtd|d < t | |¡ t j|d < t | |¡ d S ) Nr r, r r r T)Úcopyr: r FÚInt32) r r! r5 r r( Z shares_memoryÚ assert_extension_array_equalÚNAr rE Z_mask)r6 Úorigr) r r r Útest_astype_copy s2
|
|
|
rO c C sx t jddgdd} | d¡}tjddgdd}t ||¡ t jddgdd} | d¡}tjddgdd}t ||¡ d S )Nr r, rK r r$ ÚUInt32Zuint64)r r! r5 r r( Úassert_numpy_array_equal©r r) r* r r r Útest_astype_to_larger_numpy¹ s
|
|
rS r r- rP c C s t jdddgdd}| | ¡}t jdddg| d}t ||¡ t jdddd gdd}| | ¡}t jdddd g| d}t ||¡ d S )Nr r, r r r )r rC r5 r( rD )r rH r) r* r r r Ú test_astype_specific_castingÅ s
|
|
rT c C sB t jddd gdd} | d¡}t jddd gdd}t ||¡ d S )Nr r, r r ZFloat64r g @©r r! r5 r( rL ©r6 r) r* r r r Útest_astype_floatingÒ s
|
rW c C sN t dddt jg¡d } | d¡}tjddddgdd d ¡}t ||¡ d S )
|
Nr r, r i Ê;zdatetime64[ns]ZNaTzM8[s]r zM8[ns])r r! rM r5 r r( rQ rV r r r Útest_astype_dt64Ù s
|
rX c C sÄ d}dddg}t jt|d tj|| d W 5 Q R X t jt|d t |¡ | ¡ W 5 Q R X dddtjg}t jt|d tj|| d W 5 Q R X t jt|d t |¡ | ¡ W 5 Q R X d S )Nz cannot safelyg333333ó?gffffff@g @r/ r ) r2 r3 Ú TypeErrorr r! rC r5 r r1 )r r7 r6 r r r Útest_construct_cast_invalidã s
|
rZ Ú in_seriesc C s´ t jddd gdd}| r"t |¡}|jdtjd}tjddtjgdd}t ||¡ |jd d
|
d}tjddd
|
gd d}t ||¡ |jd d d}tjd d d gd d}t ||¡ d S )Nr r r r Úfloat64)r Zna_valueg r r$ éÿÿÿÿÚboolFT)r r! rC Úto_numpyr r1 r( rQ )r[ r r) r* r r r Útest_to_numpy_na_nanô s
|
r` Úint32r$ r^ c C sN t jddgdd}|r t |¡}|j| d}tjddg| d}t ||¡ d S )Nr r r r )r r! rC r_ r r( rQ )r r[ r r) r* r r r Útest_to_numpy_dtype s
|
rb r\ c C s> t jddd gdd}tjt| d |j| d W 5 Q R X d S )Nr r r r r/ )r r! r2 r3 r4 r_ )r r r r r Útest_to_numpy_na_raises s rc c C sX t jddd gdd} tjdddgtj dd}t | t¡|¡ t | d ¡|¡ d S )
|
Nr r, r r Ú1Ú2z<NA>ZU21Ústr)r r! r r( ZENDIANrQ r5 rf )r r* r r r Útest_astype_str s rg c C sJ t jddddd gdd} | d¡}t jdd ddd gdd}t ||¡ d S )
|
Nr r r] r, r r ÚbooleanTFrU rR r r r Útest_astype_boolean" s
|
ri ) Únumpyr r2 Zpandas.core.dtypes.genericr Zpandasr Zpandas._testingZ_testingr( Zpandas.core.arrays.integerr r r ÚmarkZ parametrizer+ r8 r? rB rI rO rS rT rW rX rZ r` rb rc rg ri r r r r Ú<module> s<
|
|
|
|
2$
|
|
|
|
|
|
