U
|
ý°d ã @ sj d dl Z d dlZd dlZd dlZd dlmZ ej dddej
|
fddej
|
fde d d
|
¡fd e d d
|
¡fg¡d d Z dd Zdd Zej ddddddddej
|
fdg¡dd Zd d Zej d ed ed fed ed d fed edej
|
feeejeeejfed e d ¡fg¡d!d" Zd#d$ Zej d%eed&feed&feed&feed'feed'fg¡d(d) Zd*d+ Z ej d,d-ee d ¡fd.ee d ¡fd/ee d¡fd0ee d1¡fd2ee d¡fd3ee d¡ej
|
fg¡d4d5 Z ej d6eed7ee d¡d7d&feed eed d&feeed8eeej
|
d&feed eeej
|
d'feed7eed7d'fg¡d9d: Z ej d,d;d<d=d>d?g¡d@dA Z ej dBdCdDdEg¡dFdG Z ej dHeed eeed7feedIeeedJfeedIeeedKfeedLeeedIfg¡dMdN Z!ej dOeeej
|
ee "dP¡feedQee "dR¡fg¡dSdT Z#dUdV Z$dS )Wé N)Ú SparseDtypezdtype, fill_value)Úintr Úfloat)ÚboolFÚobjectúdatetime64[ns]ÚNaTÚnsútimedelta64[ns]c C sH t | }|j}t |¡r8t |¡r2t|t|ksDtn ||ksDtd S ©N)r Ú
|
fill_valueÚpdZisnaÚtypeÚAssertionError)Údtyper Z sparse_dtypeÚresult© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_dtype.pyÚtest_inferred_dtype
|
s
|
|
r c C s$ t dd} t | }|jdks td S )Nr r )r r r )r r r r r Útest_from_sparse_dtype s
|
r c C s0 t dd} t | dd}t dd}||ks,td S )Nr é é ©r ©r r )r r Úexpectedr r r Ú!test_from_sparse_dtype_fill_value$ s
|
|
r )r N)r N)r N)r N)r N)r
|
Nr )r r c C s0 t | |}t | |}||ks t||ks,td S r r )r r ÚaÚbr r r Ú
|
test_equal+ s
|
|
r c C s6 t ttd} t ttj}| |ks&t|| ks2td S )NÚnan)r r Únpr r ©r r r r r Útest_nans_equal? s r" za, bÚfloat64Úfloat32c C s | |ks t d S r )r r! r r r Útest_not_equalF s r% c C s( t jtdd t d¡ W 5 Q R X d S )Nz3Cannot construct a 'SparseDtype' from 'not a dtype'©Úmatchz not a dtype©ÚpytestÚraisesÚ TypeErrorr Úconstruct_from_stringr r r r Ú!test_construct_from_string_raisesT s
|
ÿr- zdtype, expectedTFc C s | j |kstd S r )Z _is_numericr )r r r r r Útest_is_numeric[ s r. c C s t tj} | t d¡ks td S )Nr )r ÚstrÚsubtyper r r )r r r r Útest_str_uses_objecti s
|
r1 zstring, expectedzSparse[float64]zSparse[float32]ú Sparse[int]z Sparse[str]r/ zSparse[datetime64[ns]]ZSparsec C s t | ¡}||kstd S r )r r, r )Ústringr r r r r Útest_construct_from_stringn s
|
r4 za, b, expectedg r c C s4 | |k}||kst t| t|k}||ks0t d S r )r Úhash)r r r r r r r Útest_hash_equal~ s r6 )r2 r )zSparse[int, 0]r )z Sparse[int64]Úint64)úSparse[int64, 0]r7 )zSparse[datetime64[ns], 0]r c C s t | ¡\}}||kstd S r )r Z_parse_subtyper )r3 r r0 Ú_r r r Útest_parse_subtype s r: r3 zSparse[int, 1]zSparse[float, 0.0]zSparse[bool, True]c C s( t jtdd t | ¡ W 5 Q R X d S )Nz fill_value in the string is notr& r( )r3 r r r Ú,test_construct_from_string_fill_value_raises s r; zoriginal, dtype, expectedr g ð?Ú1g ø?c C s | |¡}||kstd S r )Ú update_dtyper )Úoriginalr r r r r r Útest_update_dtype§ s
|
|
r? z#original, dtype, expected_error_msgz7Cannot convert non-finite values (NA or inf) to integerÚabcz-invalid literal for int() with base 10: 'abc'c C s( t jt|d | |¡ W 5 Q R X d S )Nr& )r) r* Ú
|
ValueErrorr= )r> r Zexpected_error_msgr r r Útest_update_dtype_raisesµ s rB c C sD t tddd} d}| |ks tt ttdd} d}| |ks@td S )Nr7 r r r8 Ú0zSparse[object, '0'])r/ r r r )r r r r r Ú test_reprÉ s rD )%ÚreÚnumpyr r) Zpandasr Zpandas.core.arrays.sparser ÚmarkZ parametrizer Z
|
datetime64Z timedelta64r r r r r" r r r r r% r- r r r/ r. r1 r4 r6 r: r; r? ÚescaperB rD r r r r Ú<module> sà úþ
|
øþ
|
ûþ
|
|
|
|
|
|
|
ûþ
|
|
úþ
|
ûþ
|
|
ûþ
|
|
ÿ
|
üþ
|
|
ýýúþ
|
|
