U
|
ý°dôC ã @ s@ d dl Z d dlZd dlZd dlmZ d dlZd dlmZ d dl m
|
Z d dl m Z mZ ejdd Zejdd Zejd d
|
ZG d d d ZG d d dZdd Zdd Zej ddddgdd fd ddgddfd ddgddfd ddddgddfddddgddfg ddfddd d ddgdd fddd d ddgddfddd d ddgddfddd d ddgddfdd d dddgd dfdd d dddgddfg ¡dd Zej dddejejgdejdejgddejgejdd d ejdgejd d dddgg¡ej dejd dg¡d d Zd d Z d d! Z d"d# Z ej dejdg¡d$d% Z d&d' Z dS )(é N)ÚIntIndex)Úisna)Ú SparseArrayÚ SparseDtypec C s& t t jt jdddt jddt jdg
|
¡S )z<Fixture returning numpy array with valid and missing entriesé é é é é é )ÚnpÚarrayÚnan© r r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_array.pyÚarr_data s r c C s t | S )z-Fixture returning SparseArray from 'arr_data'©r )r r r r Úarr s r c C s t ddddddddddg
|
ddS ) zEFixture returning SparseArray with integer entries and 'fill_value=0'r r r r r r
|
r ©Ú
|
fill_valuer r r r r Úzarr s r c
|
@ sZ e Zd Zej dddejg¡dd Zdd Z ej dd d
|
d ge
|
d d
|
g¡d g¡d d Z dd Z dd Z ej ddddddgddfg ddfdgddfddejdgdefg¡dd Zej d ejejejejejgd ejejd ejgd ejdd dgg¡ej dddg¡d d Zej d d d!g¡d"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- ZdS ).ÚTestSparseArrayr r Nc C sf t t dddddg¡dd}|jd|d}t|r:|jj}t t |ddddg¡dd}t ||¡ d S )Nr r r g @r ) r r r Úshiftr ÚdtypeZna_valueÚtmÚassert_sp_array_equal)Úselfr ÚsparseÚresÚexpr r r Útest_shift_fill_value$ s z%TestSparseArray.test_shift_fill_valuec C s t dtjdgtjd}d|_|jdks*tt dddgdtjd}d|_|jdksTtd|_|jdkshttj|_t |j¡stt d d
|
d gd
|
tjd}d |_|js¦ttj|_t |j¡s¾td S ) Nç ð?g @r r r r ©r r gÍÌÌÌÌÌ@TF)r r r r ÚAssertionErrorÚint64ÚisnanÚbool_©r r r r r Útest_set_fill_value. s
|
z#TestSparseArray.test_set_fill_valueÚvalr r r )r r r c C s> t dddgdtjd}d}tjt|d ||_W 5 Q R X d S )NTFr" zfill_value must be a scalar©Úmatch)r r r& ÚpytestÚraisesÚ
|
ValueErrorr )r r) r Úmsgr r r Ú test_set_fill_invalid_non_scalarU s z0TestSparseArray.test_set_fill_invalid_non_scalarc C s, | ¡ }|j|jk st|j|jks(td S ©N)ÚcopyZ sp_valuesr# Úsp_index)r r Zarr2r r r Ú test_copy] s zTestSparseArray.test_copyc C s t | ¡ |¡ d S r1 )r Zassert_almost_equalÚto_dense)r r r r r r Útest_values_asarrayb s z#TestSparseArray.test_values_asarrayzdata,shape,dtype)r
|
)r )r ÚAÚB)r c C s t ||d}|j|kstd S )N©r )r Úshaper# )r Údatar: r Úoutr r r Ú
|
test_shapee s zTestSparseArray.test_shapeÚvalsc C s. t |¡}t||d}| ¡ }t ||¡ d S ©Nr )r r r r5 r Úassert_numpy_array_equal)r r> r r r r r r Útest_dense_reprs s
|
|
z TestSparseArray.test_dense_reprÚfixr r c C s$ | |¡}t |¡}t ||¡ d S r1 )Zgetfixturevaluer Zround_trip_pickler )r rB ÚrequestÚobjZ unpickledr r r Ú test_pickle s
|
|
zTestSparseArray.test_picklec C s2 t dddg}t d ¡ |D ]}q W 5 Q R X d S )Nr r r )r r Zassert_produces_warning)r Zsp_arrÚ_r r r Útest_generator_warnings s z'TestSparseArray.test_generator_warningsc C sx t tjdgdd}t ddg¡}| | d¡}t ddgdd}t ||¡ t |¡}| | d¡}t
|
|t |¡¡ d S )Nr! r r TFr ) r r r r Z_wherer r ÚpdÚSeriesÚwhereZassert_series_equal)r r Úmaskr r Zserr r r Ú test_where_retain_fill_value s
|
z,TestSparseArray.test_where_retain_fill_valuec C s t dtjtjdtjg}| d¡}t dddddgdtjd}t ||¡ t dtjtjdtjgdd}| d¡}t dddddgdtjd}t ||¡ t dtjdddg}| d¡}t dddddgdtjd}t ||¡ t dtjdddgdd}| d¡}t dddddgdtjd}t ||¡ t tjtjtjtjg}| d¡}t ddddgdtjd}t ||¡ t tjtjtjtjgdd}| d¡}t ddddgdtjd}t ||¡ t ddddg}| d¡}t ddddgdd}t ||¡ t ddddg}|jttj kst
|
|j dks t
|
| d¡}t ||¡ t ddddgdd}|jttj ks`t
|
|j dkspt
|
| d¡}t ddddgdd}t ||¡ t ddddgtjd}|jttj tjdksÌt
|
t |j ¡sÞt
|
| d¡}t ddddgdd}t ||¡ d S )Nr r éÿÿÿÿr" r r ç ) r r r ÚfillnaÚfloat64r r r r r$ r# r r% ©r Úsr r r r r Ú test_fillna sZ
|
|
|
|
|
|
|
|
|
|
zTestSparseArray.test_fillnac C s t dtjtjdtjg}| d¡}tjdddddgtjd}t | ¡ |¡ t dtjtjdtjgdd}| d¡}t dddddgdtjd}t ||¡ d S )Nr r r9 r r r" )
|
r r r rO r rP r r@ r5 r rQ r r r Útest_fillna_overlapÙ s
|
|
z#TestSparseArray.test_fillna_overlapc C s t tdtdddddddddddg }tjdddgtjd}| ¡ \}t ||¡ t ddddddddddddg }| ¡ \}t ||¡ d S ) Nr r r r r r
|
é r9 )r Úfloatr r Úint32Znonzeror r@ )r ÚsaÚexpectedÚresultr r r Ú test_nonzeroæ s (
|
|
z TestSparseArray.test_nonzero)Ú__name__Ú
|
__module__Ú __qualname__r, ÚmarkÚ parametrizer r r r( r r0 r4 r6 Úobjectr= rA rE rG rL rS rT r[ r r r r r # s@
|
'"
|
|
üþ
|
ýþ
|
< r c @ sè e Zd Zej dejdddddgede e dd d
|
d d g¡fejddej
|
ddgede e dd ej
|
d dg¡fg¡ej dddg¡dd Z dd Z dd Z ej ddej
|
g¡dd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&S )'ÚTestSparseArrayAnalyticsz data,expectedr r r r r
|
r9 r! g @g @g $@g .@g @g (@ÚnumpyTFc C s |r
|
t jndd }|t|}t ||¡ |t|t jd}t ||¡ |t|dd}t ||¡ |rÌd}tjt|d t jt|t j d W 5 Q R X d}tjt|d t jt||d W 5 Q R X n@d
|
}t
|
d | d ¡}tjt|d t|j|d W 5 Q R X d S )Nc S s | ¡ S r1 )Úcumsum)rR r r r Ú<lambda> ó z6TestSparseArrayAnalytics.test_cumsum.<locals>.<lambda>r r z&the 'dtype' parameter is not supportedr* r9 z$the 'out' parameter is not supported)r< r zaxis(=z) out of bounds)Úaxis) r rd r r r r r, r- r. r$ ÚreÚescape)r r; rY rc rd r< r/ rg r r r Ú test_cumsumó s$ z$TestSparseArrayAnalytics.test_cumsumc C s¸ t dtjdtjdg}t dtjdtjdg}t t||¡ t t |¡|¡ t ddddgdd}t dddg|jdd}t t||¡ t t |¡|¡ t ddddgdd}t ddddgdd}t t||¡ t t |¡|¡ t dtjdtjdg}t t dtjdtjdg¡}t t |¡|¡ t ddddgdd}t t ddddg¡t d¡d}t t |¡|¡ t ddddgdd}t t ddddg¡t d¡d}t t |¡|¡ d S )Nr r éþÿÿÿrM r )Ú sparse_indexr r )r r r r r Úabsr3 Úsin)r r rZ r r r r Ú
|
test_ufunc s* z#TestSparseArrayAnalytics.test_ufuncc C s¼ t dtjdtjdg}t dtjdtjdg}t t |d¡|¡ t ddddgdd}t ddddgdd}t t |d¡|¡ t ddddgdd}t ddddgdd}t t |d¡|¡ d S )Nr r rk r rM r r )r r r r r Úadd)r r rZ r r r Útest_ufunc_args8 s z(TestSparseArrayAnalytics.test_ufunc_argsr rN c C sh t |gd ddg |d}t |¡\}}t t |¡¡\}}t |t ||d¡ t |t ||d¡ d S )Né
|
gñ?g@r )r r ÚmodfÚasarrayr r )r r r Zr1Zr2Úe1Úe2r r r Ú test_modfF s
|
z"TestSparseArrayAnalytics.test_modfc C s, t dddddgdd}|j}|dks(td S )Nr r r Úinteger©Úkindé ©r Únbytesr# ©r r rZ r r r Útest_nbytes_integerO s z,TestSparseArrayAnalytics.test_nbytes_integerc C s, t dddddgdd}|j}|dks(td S )Nr r r Úblockry r{ r| r~ r r r Útest_nbytes_blockU s z*TestSparseArrayAnalytics.test_nbytes_blockc C s$ t t dd d dg¡}t |¡ d S )NZ2012Z2013)r rH Ú to_datetimer rt )r rR r r r Útest_asarray_datetime64\ s z0TestSparseArrayAnalytics.test_asarray_datetime64c C s t ddg}|jdkstd S )Nr r g à?)r Zdensityr# r' r r r Ú test_density` s z%TestSparseArrayAnalytics.test_densityc C s t ddg}|jdkstd S )Nr r )r Znpointsr# r' r r r Ú test_npointsd s z%TestSparseArrayAnalytics.test_npointsN)r\ r] r^ r, r_ r` r r rV r r rj ro rq rw r r r r r
r r r r rb ò s* þþûþ
|
rb c C sN t dtjdgdd} tj| _| ¡ }t |¡}t dddg¡}t ||¡ d S )Nr! rN r FT) r r r r r rt r r r@ ©r rZ rY r r r Ú*test_setting_fill_value_fillna_still_worksi s
|
r c C sT t dtjgdd} tj| _t jt tjg¡tddgtttjd}t
|
| |¡ d S )NrN r r r r )Z sparse_arrayrl r ) r r r r Z _simple_newr r r rV r r )r rY r r r Ú test_setting_fill_value_updatesw s
|
|
ýr zarr,fill_value,locr r r rM r
|
c C s t | |d ¡ }||ks td S r? )r Z_first_fill_value_locr# )r r ÚlocrZ r r r Útest_first_fill_value_loc s r r c C sF t | |d ¡ }t | ¡ ¡ }t|t s,tt |¡}t ||¡ d S r? )
|
r ÚuniquerH rI Ú
|
isinstancer# r rt r r@ )r r ÚaÚbr r r Útest_unique_na_fill s
|
|
r c C s. t ddg} | ¡ }t dg}t ||¡ d S )Nr )r r r r r r r r Útest_unique_all_sparse s
|
r c C s t dddg} t dddgdd}| dddd¡}t ||¡ | t dddd¡¡}t ||¡ | t dddd¡¡}t dddgdd}t ||¡ d S ) Nr r r rr é é r )r r r )r Úmapr r rH rI ©r rY rZ r r r Útest_mapµ s r c C s@ t dddg} t ddd gdd}| ddd¡}t ||¡ d S )Nr r r rr r r )r r )r r r r r r r r Útest_map_missingÇ s r c C sv t tjdg| d}t dg| d}t | ¡ |¡ t ddg|d¡}tjdg|dt dg¡d}t | ¡ |¡ d S )Nr r r! r )r r )Úindex)
|
r r r r r ZdropnarH Ú DataFrameZIndexZ assert_equal)r r r ÚdfZ expected_dfr r r Ú test_dropnaÏ s r c C sJ t t d¡¡ dd ¡} | ¡ }t dd tdD ¡}t ||¡ d S )N)r
|
r
|
c S s t | ddS )Nr r r )Úxr r r re Ý rf z1test_drop_duplicates_fill_value.<locals>.<lambda>c S s i | ]}|t d gddqS )rN r r r )Ú.0Úir r r Ú
|
<dictcomp>ß s z3test_drop_duplicates_fill_value.<locals>.<dictcomp>r
|
) rH r r ÚzerosÚapplyZdrop_duplicatesÚranger Zassert_frame_equal)r rZ rY r r r Ú test_drop_duplicates_fill_valueÛ s r¢ )!rh rc r r, Zpandas._libs.sparser ZpandasrH r Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r Zfixturer r r r rb r r r_ r` r r r r r r r r¢ r r r r Ú<module> sh
|
|
|
Pw ôþ
|
|
ûþ
|
|
|
