U
|
ý°dà$ ã @ sn d dl Z d dlZd dlZd dlm mZ d dlZ d dl
|
m Z d dl mZmZ G dd dZG dd dZdS )é N)Ú SparseArrayÚ SparseDtypec @ s¤ e Zd Zdd Zej dddddg¡dd Zej d
|
d Z
|
ej ej d d ddddgddddgdddfdddddgddddgdddfg¡d d Z d d Z d S )!ÚTestSeriesAccessorc C sB t jddddgdd}|j ¡ }t ddddg¡}t ||¡ d S )Nr é é
|
z Sparse[int64]©Údtype)ÚpdÚSeriesÚsparseÚto_denseÚtmÚassert_series_equal)ÚselfÚserÚresultÚexpected© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_accessor.pyÚ test_to_dense s
|
z TestSeriesAccessor.test_to_denseÚattrZnpointsÚdensityÚ
|
fill_valueZ sp_valuesc C s< t ddg}t |¡}t|j|}t||}||ks8td S )Nr r )r r r
|
Úgetattrr ÚAssertionError)r r Zarrr r r r r r Útest_get_attributes s
|
|
|
z&TestSeriesAccessor.test_get_attributesc C s² dd l }ddddg}ddddg}ddddg}|jj|||ffd d
|
}tjj |¡}tj tj ddddgtj
|
d
|
tj ddddgtj
|
d
|
g¡}tjddddg|d d }t ||¡ d S ) Nr é r é é é é é Úintr ú Sparse[int]©Úindexr ) Ú scipy.sparser Ú
|
coo_matrixr r
|
Úfrom_cooÚ
|
MultiIndexÚ from_arraysÚnpÚarrayÚint32r r ) r ÚscipyÚrowÚcolÚdataZsp_arrayr r% r r r r Ú test_from_coo s þÿz TestSeriesAccessor.test_from_cooz>sort_labels, expected_rows, expected_cols, expected_values_posF)Úbr )Úar )r3 r )r4 r )Úzr )r5 r )Úxr )r5 r )r r )r r )r r T©r r ©r r c C sÊ dd l }tdtjddd dgdd}tj ddddd d
|
g¡}tj||d }t d ¡} | ¡ D ]\}
|
\} } |
|
| | | f< qZ|j
|
j d d|d\} }}t | |j
|
j stt | ¡ | ¡ ||ksºt||ksÆtd S )Nr r r ©r )r3 r r5 r )r4 r r5 r )r4 r r5 r )r4 r r6 r )r3 r r5 r )r4 r r5 r ©r% )r r r8 )r r )Ú
|
row_levelsÚ column_levelsÚ sort_labels)r& r r+ Únanr r) Ú from_tuplesr
|
ÚzerosÚitemsr Úto_cooÚ
|
isinstancer' r r Zassert_numpy_array_equalÚtoarray)r r= Z expected_rowsZ expected_colsZexpected_values_posr. Úvaluesr% ÚssZ
|
expected_AÚvaluer/ r0 ÚAÚrowsÚcolsr r r Ú test_to_coo5 s0 úÿ
|
|
ÿ z TestSeriesAccessor.test_to_cooc C s6 t dddg¡}tjtdd |jj W 5 Q R X d S )Nr r r z.sparse©Úmatch)r r
|
ÚpytestÚraisesÚAttributeErrorr r )r r r r r Útest_non_sparse_raisese s z)TestSeriesAccessor.test_non_sparse_raisesN) Ú__name__Ú
|
__module__Ú __qualname__r rN ÚmarkÚ parametrizer ÚtdÚskip_if_no_scipyr2 rK rQ r r r r r s,
|
|
|
|
ü
|
|
üùþ r c @ s| e Zd Zdd Zej ddddg¡ej ddeej dd
g¡ej d
|
d d g¡e
|
j d d Z ej ddddg¡e
|
j dd Z ej dddgej dgddgg¡ddgg¡e
|
j dd Zej ddddejfdd d g¡e
|
j d d Zej d dejg¡e
|
j d!d" Ze
|
j d#d$ Zd%d& Zd'd( Zej d
|
d d g¡ej d)d*d+g¡e
|
j d,d- Ze
|
j d.d/ Zd0d1 Z dS )2ÚTestFrameAccessorc C s6 t dddgi¡}tjtdd |j W 5 Q R X d S )NrH r r r rL )r Ú DataFramerN rO rP r ©r Údfr r r Útest_accessor_raisesl s z&TestFrameAccessor.test_accessor_raisesÚformatZcscZcsrÚcooÚlabelsNr r Úfloat64Úint64c C sv dd l }t|tjd|d ¡ }|jjd||d}tjjj |||d}tjtjd|d||d
|
|¡}t ||¡ d S )Nr r r ©r^ r )r% Úcolumns) r& r r+ r, Úitemr Úeyer rZ Ú from_spmatrixÚastyper Úassert_frame_equal) r r^ r` r r. Zsp_dtypeÚmatr r r r r Útest_from_spmatrixq s ÿþz$TestFrameAccessor.test_from_spmatrixc C sb dd l }|jjddd|d}d|jd< tjj |¡}tdd}t | ¡ ¡ |¡}t
|
||¡ d S )Nr r r ç à?)r r^ ra ç ) r& r Úrandomr1 r rZ rg r Ztodenserh r ri )r r^ r. rj r r r r r r Ú*test_from_spmatrix_including_explicit_zero s
|
|
z<TestFrameAccessor.test_from_spmatrix_including_explicit_zerord r4 r3 rH c C s^ dd l }tdd}|jjdddd}tjjj||d}tj| ¡ |d |¡}t
|
||¡ d S ) Nr ra rm r r rl )r )rd ) r& r r rn r rZ rg rD rh r ri )r rd r. r rj r r r r r Útest_from_spmatrix_columns s
|
z,TestFrameAccessor.test_from_spmatrix_columnsÚcolnames©rH ÚBr7 r )g¹?gÉ?)r6 r6 )r r c C sd dd l }tj|d dddg|d dddgidd}|j ¡ }|j t |¡¡}||kjdks`t d S )Nr r zSparse[int64, 0]r )
|
r& r rZ r rB r' r+ ZasarrayÚnnzr )r rq r. r\ r r r r r rK s ÿ
|
z TestFrameAccessor.test_to_coor c C sX t t|||dg|dt|d||g|dd¡}tjtdd |j ¡ W 5 Q R X d S )Nr r9 rr zfill value must be 0rL )r rZ r rN rO Ú
|
ValueErrorr rB )r r r\ r r r Ú#test_to_coo_nonzero_fill_val_raisesª s
|
ÿ
|
ÿüÿ
|
z5TestFrameAccessor.test_to_coo_nonzero_fill_val_raisesc C s¨ dd l }tj tjtdddtjddgddg¡}tjd|dd}|jjdgdgd d }|jj t
|
ddg¡t
|
ddg¡t
|
ddg¡ffd
|
d }||kj dks¤t d S ) Nr Úabr6 )Únamer Úyr# r$ )r; r< )r r )Úshape)r& r r) r* ZCategoricalIndexÚlistr
|
r rB r' r+ r, rt r )r r. Zmidxr r r r r r Ú test_to_coo_midx_categoricalº s þÿ( ÿz.TestFrameAccessor.test_to_coo_midx_categoricalc C s t jtddgtdddtddgtdddtddgtddddd d
|
gd }|j ¡ }t jddgddgddgdd d
|
gd }t ||¡ d S ) Nr r rb r g ð?rm ra )rH rs ÚCr3 r4 r: )r rZ r r r r r ri )r r\ r r r r r r Í s ýú
|
ÿz TestFrameAccessor.test_to_densec C sL t tddddgddtddddgddd¡}|jj}d}||ksHtd S )Nr r r r9 rr g è?)r rZ r r r r )r r\ Úresr r r r Ú test_densityÜ s þÿz TestFrameAccessor.test_densityÚ dense_indexTFc C s¸ dd l }|jjdd|d}tjjj||d}tj tj ddgtj
|
dtj ddgtj
|
dtj ddgtj
|
dg¡}tjt tj dddg|d|d }|r¨| tj |j¡¡}t ||¡ d S )
|
Nr r r_ rc )r r r r r: )r& r rf r r
|
r( r) r? r+ r, r- r ZreindexÚ from_productÚlevelsr r )r r r r. rH r r% r r r r Útest_series_from_cooç s ýÿ"z&TestFrameAccessor.test_series_from_cooc C sR dd l }|j t ddgddgg¡¡}tjtdd tj j
|
|¡ W 5 Q R X d S )Nr r z,Expected coo_matrix. Got csr_matrix instead.rL ) r& r Z
|
csr_matrixr+ r, rN rO Ú TypeErrorr r
|
r( )r r. Úmr r r Ú,test_series_from_coo_incorrect_format_raisesý s ÿz>TestFrameAccessor.test_series_from_coo_incorrect_format_raisesc C s8 t dt j ddg¡i¡}t|jt jjjjjs4t d S )Nr r r )
|
r rZ Zarraysr rC r ÚcoreÚaccessorZSparseFrameAccessorr r[ r r r Ú test_with_column_named_sparse s z/TestFrameAccessor.test_with_column_named_sparse) rR rS rT r] rN rU rV r{ ÚstringÚ ascii_lettersrW rX rk ro r r) r rp ZNArK r+ r> rv r| r r r r r r r r r rY k sF
|
"þ
|
ÿ
|
|
|
|
rY )r Únumpyr+ rN Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrW Zpandasr Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r r rY r r r r Ú<module> s [
|
