U
|
ý°dÅ* ã @ sp d dl Zd dlZd dlmZ d dlm mZ d dl Z
|
d dl m Z d dl m Z d dlmZmZ G dd dZdS )é N)ÚIntIndex)Úisna)Ú SparseArrayÚ SparseDtypec @ s| e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e e de g¡dd Zdd Zdd Ze j
|
ddeddgg¡dd Zdd Ze j
|
d e dd g¡dfe d d g¡ejfd d!gd!fe d"¡gejfg¡d#d$ Ze j
|
d%d&d'd(g¡e j
|
d)dd*g¡ej d+d, Z e j
|
d%d&d'd(g¡ej d-d. Z ej d/d0 Z d1d2 Z d3d4 Z!d5d6 Z"d7d8 Z#d9d: Z$d;d< Z%dS )=ÚTestConstructorsc C s t tjddtjg}|jttjtjks,t|jjtjks>tt |j ¡sNtt tjddtjgdd}|jttjdks|t|j dkstt ddddgtjd}|jttjtjks¸tt |j ¡sÈtt ddddgtj
|
d}|jttj
|
dksôt|j dkstt ddddgdtj
|
d}|jttj
|
dks4t|j dksDtt ddddgd d}|jttj
|
dkspt|j dkstt ddddgdd d}|jttj
|
dks®t|j dks¾td S )Né é r ©Ú
|
fill_valueé ©Údtype©r
|
r ) r ÚnpÚnanr r Úfloat64ÚAssertionErrorÚsubtypeÚisnanr
|
Úint64©ÚselfÚarr© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_constructors.pyÚtest_constructor_dtype s, z'TestConstructors.test_constructor_dtypec C s4 t dddgdd}t dddgtd}t ||¡ d S )Nr r é Úintr )r r ÚtmÚassert_sp_array_equal©r ÚresultÚexpectedr r r Útest_constructor_dtype_str/ s z+TestConstructors.test_constructor_dtype_strc C sX t ddddgtddd}t ddddgdtjd}t ||¡ |jjt d¡ksTtd S )Nr r r éÿÿÿÿr r ) r r r r r r Ú sp_valuesr r r r r r Ú test_constructor_sparse_dtype4 s z.TestConstructors.test_constructor_sparse_dtypec C sP t ddddgdd}t ddddgtjd}t ||¡ |jjt d¡ksLtd S )Nr r z Sparse[int32]r Úint32)r r r' r r r% r r r r r r Ú!test_constructor_sparse_dtype_str: s z2TestConstructors.test_constructor_sparse_dtype_strc C sv t ddtjdgtd}|jttks(tt |j¡s8tt ddtjdgtdd}|jttdksdt|jdksrtd S )NÚAÚBr ©r r
|
) r r r Úobjectr r r r r
|
r r r r Ú test_constructor_object_dtype@ s z.TestConstructors.test_constructor_object_dtypec C sz ddddg}t |tdd}|jttdks.t|jdks<ttj|td}dd t||D }tj |tj
|
d ¡ svtd S ) NFr g Y@g r+ r c s s* | ]"\}}t |t |ko ||kV qd S ©N)Útype)Ú.0ÚxÚyr r r Ú <genexpr>Q s zKTestConstructors.test_constructor_object_dtype_bool_fill.<locals>.<genexpr>) r r, r r r r
|
r ÚarrayÚzipZfromiterÚbool_Úall)r Údatar Z arr_expectedÚitr r r Ú'test_constructor_object_dtype_bool_fillJ s z8TestConstructors.test_constructor_object_dtype_bool_fillr r c C s2 t jtdd tddtjg|d W 5 Q R X d S )NzCannot convert©Úmatchr r r )ÚpytestÚraisesÚ
|
ValueErrorr r r )r r r r r Útest_constructor_na_dtypeT s z*TestConstructors.test_constructor_na_dtypec C s t jdddd}ttj|dd}t t¡ t|}W 5 Q R X t ||¡ t t¡ tt |¡}W 5 Q R X t ||¡ d S )Nz
|
2016-01-01r z
|
US/Pacific)ZperiodsÚtzzdatetime64[ns]r )
|
ÚpdZ
|
date_ranger r Úasarrayr Zassert_produces_warningÚ UserWarningr ZSeries)r Zdtir" r! r r r Ú+test_constructor_warns_when_losing_timezoneY s z<TestConstructors.test_constructor_warns_when_losing_timezonec C s t ddgtdddgd}t ddddgdd}t ||¡ |jttjksNt|j dks\tt dddgtddddgtjdd }t ddddgtjdd
|
}t ||¡ |jttjks¸t|j dksÆtt ddgtdddgdtjd }t ddddgdtjd }t ||¡ |jttjks t|j dks0tt dddgtddddgd dd }t ddddgd d }t ||¡ |jttjkst|j dkstd S )Nr r r )r8 Ú sparse_indexr Úinteger)Úkindr )r8 rF r r
|
r+ ©r8 rF r
|
r r r ©
|
r r r r r r r r r r
|
)r r r" Úexpr r r Ú test_constructor_spindex_dtypei sF ü ÿ ü z/TestConstructors.test_constructor_spindex_dtyperF Nr c C s t d|d d}t dgd d}t ||¡ |jttjks<t|jdksJtt dt ddgd d}t dgd d}t ||¡ |jttjkst|jdkstd S )Nr )r8 rF r r r )
|
r r r r r r r r r
|
r )r rF r rK r r r Ú%test_constructor_spindex_dtype_scalar s z6TestConstructors.test_constructor_spindex_dtype_scalarc C sf t ddgtdddgdd d}t ddddgdd d}t ||¡ |jttjksTt|j dksbtd S )Nr r r r rI r rJ )r r rK r r r Ú0test_constructor_spindex_dtype_scalar_broadcasts s ÿ zATestConstructors.test_constructor_spindex_dtype_scalar_broadcastszdata, fill_valuer ç ð?g @TFz
|
2017-01-01c C s0 t |j}t|r t|s,tn ||ks,td S r. )r r
|
r r )r r8 r
|
r! r r r Ú$test_constructor_inferred_fill_value© s
|
|
z5TestConstructors.test_constructor_inferred_fill_valueÚformatZcooÚcscZcsrÚsizeé
|
c C sL dd l }|jj|dd|d}t |¡}t |¡}| ¡ ¡ }t
|
||¡ d S )Nr r ç à?©ZdensityrQ ) Ú scipy.sparseÚsparseÚrandomr Ú from_spmatrixr rC ÚtoarrayÚravelr Úassert_numpy_array_equal)r rS rQ ÚscipyÚmatr! r" r r r Útest_from_spmatrixº s
|
|
z#TestConstructors.test_from_spmatrixc C sV dd l }|jjddd|d}d|jd< t |¡}t |¡}| ¡ ¡ }t
|
||¡ d S )Nr rT r rU rV ) rW rX rY r8 r rZ r rC r[ r\ r r] )r rQ r^ r_ r! r" r r r Ú*test_from_spmatrix_including_explicit_zeroÇ s
|
|
|
z;TestConstructors.test_from_spmatrix_including_explicit_zeroc C sB dd l }|jjdddd}tjtdd t |¡ W 5 Q R X d S )Nr é r rR )rQ znot '4'r; )rW rX Zeyer= r> r? r rZ )r r^ r_ r r r Útest_from_spmatrix_raisesÔ s z*TestConstructors.test_from_spmatrix_raisesc C s2 t jtdd tt d¡ d¡ W 5 Q R X d S )Nz expected dimension <= 1 datar; rT )r rb )r= r> Ú TypeErrorr r ZarangeZreshape)r r r r Ú%test_constructor_from_too_large_arrayÝ s z6TestConstructors.test_constructor_from_too_large_arrayc C sJ t ddddddddddg
|
dd}t |}|jdks6tt |j|j¡ d S ) Nr r r r r rb é r )r r
|
r r Zassert_almost_equalr% )r ZzarrÚresr r r Ú test_constructor_from_sparseá s z-TestConstructors.test_constructor_from_sparsec C s t t jt jdddt jddt jdg
|
¡}t|}t|dd}d |jd d
< |jd d
d k ¡ rbtt|}d |jd d
< |jd d
d k ¡ std S )
|
Nr r r r rb rf T)Úcopyr )r r4 r r r% Úanyr r7 )r Zarr_datar ÚcpZnot_copyr r r Útest_constructor_copyç s & z&TestConstructors.test_constructor_copyc C s t ddddddg¡}t|dtd}|jttks6tt |j t ddg¡¡ t |j
|
j t ddgt j ¡¡ | ¡ }|jtkstt ||¡ d S )NFTr r r )r r4 r Úboolr r r r r] r% Úsp_indexÚindicesr' Úto_dense©r r8 r Zdenser r r Útest_constructor_booló s z&TestConstructors.test_constructor_boolc C s t dddgd d}|jttjks&t|jr0tt dddgtjd}|jttjksXt|jrbtt dddgtjdd}|jttjdkst|jstd S )NTFr r+ )r r r r r6 r r
|
r r r r Ú test_constructor_bool_fill_value s
|
|
z1TestConstructors.test_constructor_bool_fill_valuec C s t jdt jdgt jd}t|t jd}|jtt jks:tt |j
|
t jddgt jd¡ t |j j t jddgt j d¡ | ¡ }|jt jkstt ||¡ d S )NrO r r r r r )r r4 r Úfloat32r r r r r r] r% rn ro r' rp rq r r r Útest_constructor_float32 s ÿz)TestConstructors.test_constructor_float32)&Ú__name__Ú
|
__module__Ú __qualname__r r# r& r( r- r: r= ÚmarkZ parametrizer r r@ rE rL r rM rN r r4 r rB Ú TimestampZNaTrP ÚtdZskip_if_no_scipyr` ra rc re rh rl rr rs ru r r r r r sJ
|
|
|
(
|
|
üþ
|
|
|
r )Únumpyr r= Zpandas._libs.sparser Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr{ ZpandasrB r Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r r r r r r Ú<module> s
|
