U
|
ý°dvL ã @ sÀ d dl Z d dlZd dlZd dlm mZ d dlm m
|
Z d dl m Z d dlmZ d dlmZmZmZ dZd ddgdddgd d
|
d gd ddgd d
|
dgd dd ggZd dgd d gd gd gd gdggZd gdgd dgddgd dgddggZdgdgd dgddgdgdggZd dgd dgddgd d gg g gZd gdgg g g g gZg g g g g g gZ eeeeeegZ ddddddgZ G dd dZ G dd dZ G d d d Z!G d d d Z"G d d! d!Z#G d"d# d#Z$dS )$é N)ÚSeries)Ú
|
BlockIndexÚIntIndexÚmake_sparse_indexé é é é é é é é é é é
|
é é é Ú
|
plain_caseÚ delete_blocksÚ split_blocksÚ
|
skip_blockÚ no_intersectÚ one_emptyc @ s6 e Zd Zej ddgdgdgdgdgdggddgddgddgddgdddgd ddggd
|
gdgd gdgd
|
gd ggddgddgdgd gdgd ggddgd dgdgd gdgdggddgddgddgd dgdgdggdgdgdddgd ddgdgdggddgd d gddgddgddddgd dd dggg¡dd Zdd ZdS )ÚTestSparseIndexUnionú"xloc, xlen, yloc, ylen, eloc, elenr r
|
r r r r r r r r é r é r r c C s¤ t t||}t t||}| |¡} t| t s0tt | jtj |tj
|
d¡ t | j tj |tj
|
d¡ | ¡ }
|
| ¡ } |
|
| ¡} t| t stt | j| ¡ j¡ d S )N©Zdtype)r Ú TEST_LENGTHÚ
|
make_unionÚ
|
isinstanceÚAssertionErrorÚtmÚassert_numpy_array_equalÚblocsÚnpÚarrayÚint32ÚblengthsÚ to_int_indexr Úindices) ÚselfÚxlocÚxlenÚylocÚylenÚelocÚelenÚxindexÚyindexZbresultZixindexZiyindexZiresult© r5 ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_libsparse.pyÚtest_index_make_union[ s ,
|
|
z*TestSparseIndexUnion.test_index_make_unionc C sü t dtjdddgtjd}t dtjddgtjd}| |¡}t dt ddddgtj¡}| |¡sjtt dtjg tjd}t dtjddgtjd}| |¡}t dt ddgtj¡}| |¡sÊtt dtjg tjd}t dtjg tjd}| |¡}t dt g tj¡}| |¡s$tt dtjdddddgtjd}t dtjdddddgtjd}| |¡}t dt dddddgtj¡}| |¡stt dtjddgtjd}t dtjddgtjd}d}tjt |d | |¡ W 5 Q R X d S )
|
Nr
|
r r r r r r ú-Indices must reference same underlying length©Úmatch)
|
r r& r' r( r Úequalsr" ÚpytestÚraisesÚ
|
ValueError)r, ÚaÚbÚresÚexpÚmsgr5 r5 r6 Útest_int_index_make_union s2
|
|
|
|
z.TestSparseIndexUnion.test_int_index_make_unionN)Ú__name__Ú
|
__module__Ú __qualname__r< ÚmarkÚ parametrizer7 rD r5 r5 r5 r6 r Z s * "".øþ
|
,r c @ s¤ e Zd Zejejjdee ddd Z
|
dd Z ej de de jd d
|
ge jd e de jd d
|
d ge jd e d e jg e jd e de jg e jd g¡dd ZdS )ÚTestSparseIndexIntersectr ©Úidsc C sÈ t t||}t t||}t t||} t td ||}
|
| |¡} | | ¡sLt| ¡ | ¡ ¡} | | ¡ ¡sptd} tjt| d | |
|
¡ W 5 Q R X tjt| d | ¡ |
|
¡ ¡ W 5 Q R X d S )Nr r8 r9 ) r r Ú intersectr; r" r* r< r= Ú Exception) r, r- r. r/ r0 r1 r2 r3 r4 ÚexpectedZ longer_indexÚresultrC r5 r5 r6 Útest_intersect¶ s
|
z'TestSparseIndexIntersect.test_intersectc C s t dtjg tjd}t dtjddgtjd}| |¡ |¡sDt| |¡ |¡sXt| ¡ }| ¡ }| |¡ |¡s|t| |¡ |¡std S )Nr r r r )r r& r' r( rM r; r" Úto_block_index)r, r3 r4 r5 r5 r6 Útest_intersect_emptyÉ s z-TestSparseIndexIntersect.test_intersect_emptyÚcaser
|
r r r r r c C s4 | |¡ |¡st| ¡ }| |¡ |¡s0td S ©N)rM r; r" rR )r, rT r5 r5 r6 Útest_intersect_identicalÔ s
|
z1TestSparseIndexIntersect.test_intersect_identicalN)rE rF rG ÚtdZskip_if_windowsr< rH rI ÚCASESÚIDSrQ rS r r& r' r( rV r5 r5 r5 r6 rJ µ s üþ rJ c @ s¢ e Zd Zdd Zdd Zej dddg¡dd Zej dddg¡d
|
d Z ej d d dgdd gddgddgddgddgddgddgddgddgg
|
¡dd Z
|
d S ) ÚTestSparseIndexCommonc C s t dtjddgtjddd}t|ts,t|jdks:tt |j
|
tjddgtjd¡ t dtjg tjddd}t|tst|jdkstt |j
|
tjg tjd¡ t dtjddddgtjddd}t|tsØt|jdksætt |j
|
tjddddgtjd¡ d S © Nr r r r Úinteger©Úkindr r © r r& r' r( r! r r" Únpointsr# r$ r+ ©r, Úidxr5 r5 r6 Útest_int_internalä s ÿz'TestSparseIndexCommon.test_int_internalc C sÖ t dtjddgtjddd}t|ts,t|jdks:tt |j
|
tjdgtjd¡ t |j tjdgtjd¡ t dtjg tjddd}t|tst|jdks¨tt |j
|
tjg tjd¡ t |j tjg tjd¡ t dtjddddgtjddd}t|tst|jdks tt |j
|
tjdgtjd¡ t |j tjdgtjd¡ t dtjdddgtjddd}t|tst|jdkstt |j
|
tjddgtjd¡ t |j tjddgtjd¡ d S © Nr r r r Úblockr] r r © r r& r' r( r! r r" r` r# r$ r% r) ra r5 r5 r6 Útest_block_internalö s( " z)TestSparseIndexCommon.test_block_internalr^ r\ re c C sú t dtjddgtjd|d}| d¡dks0t| d¡dksBt| d¡dksTt| d¡dksft| d¡dksxt| d¡dkstt dtjg tjd|d}tdd D ]}| |¡dks®tq®t dtjddddgtjd|d}| d¡dksút| d¡dkst| d¡dks"t| d¡dks6t| d¡dksJt| d¡dks^tt dtjdddgtjd|d}| d¡dkst| d¡dks¦t| d¡dksºt| d¡dksÎt| d¡dksât| d¡dksötd S ©
|
Nr r r r r] éÿÿÿÿr r r
|
)r r& r' r( Úlookupr" Úrange)r, r^ rb Úir5 r5 r6 Ú test_lookup s0 " z!TestSparseIndexCommon.test_lookupc C sD t dtjddgtjd|d}| tjdddgtjd¡}tjdddgtjd}t ||¡ | tjddddgtjd¡}tjddddgtjd}t ||¡ t dtjg tjd|d}| tjddddgtjd¡}tjddddgtjd}t ||¡ t dtjddddgtjd|d}| tjdddgtjd¡}tjdddgtjd}t ||¡ | tjddddgtjd¡}tjddddgtjd}t ||¡ t dtjdddgtjd|d}| tjddddgtjd¡}tjddddgtjd}t ||¡ | tjdddd gtjd¡}tjddddgtjd}t ||¡ d S rh )r r& r' r( Z lookup_arrayr# r$ )r, r^ rb rA rB r5 r5 r6 Útest_lookup_array. s2 " z'TestSparseIndexCommon.test_lookup_arrayz idx, expectedr ri r
|
r r r r r é r r r é c C sD t dddgddg}| |¡|ks&t| ¡ }| |¡|ks@td S )Nr r
|
r r r )r rj r" r* )r, rb rO ZbindexZiindexr5 r5 r6 Útest_lookup_basicsQ s z(TestSparseIndexCommon.test_lookup_basicsN) rE rF rG rc rg r< rH rI rm rn rq r5 r5 r5 r6 rZ ã s*
|
|
"öþrZ c @ sR e Zd Zdd Zej dddddg¡dd Zd
|
d Zd d Z dd Z
|
dd Z dS )ÚTestBlockIndexc C sÖ t dtjddgtjddd}t|ts,t|jdks:tt |j
|
tjdgtjd¡ t |j tjdgtjd¡ t dtjg tjddd}t|tst|jdks¨tt |j
|
tjg tjd¡ t |j tjg tjd¡ t dtjddddgtjddd}t|tst|jdks tt |j
|
tjdgtjd¡ t |j tjdgtjd¡ t dtjdddgtjddd}t|tst|jdkstt |j
|
tjddgtjd¡ t |j tjddgtjd¡ d S rd rf ra r5 r5 r6 rg i s( " z"TestBlockIndex.test_block_internalrl r
|
r éd ée c C sb t |tjd|dtjddd}tjd|dtjd}t |j|¡ t |jtjt |tjd¡ d S )Nr r r re r] )
|
r r& Úaranger( r# r$ r% r) ZonesÚlen)r, rl rb rB r5 r5 r6 Útest_make_block_boundary s z'TestBlockIndex.test_make_block_boundaryc C sD t dddgddg}| |¡s"t| t dddgddg¡r@td S )Nr r r r r
|
r )r r; r" ©r, Úindexr5 r5 r6 Ú test_equals s zTestBlockIndex.test_equalsc C s g }g }t d|| t d|| d}tjt|d t ddgdg W 5 Q R X d}tjt|d t dddgdd g W 5 Q R X d S )
|
Nr r zBlock 0 extends beyond endr9 r r
|
zBlock 0 overlapsr r )r r< r= r> )r, ÚlocsÚlengthsrC r5 r5 r6 Útest_check_integrity s z#TestBlockIndex.test_check_integrityc
|
C sZ ddg}ddg}ddddddd d
|
d d g
|
}t d ||}| ¡ }t |jtj|tjd¡ d S )Nr r r r r r r ro r r r r r r )r r* r# r$ r+ r& r' r( )r, r{ r| Zexp_indsre Zdenser5 r5 r6 Útest_to_int_index¢ s z TestBlockIndex.test_to_int_indexc C s( t dddgddg}| ¡ |ks$td S )Nr r r
|
r )r rR r" rx r5 r5 r6 Útest_to_block_index¬ s z"TestBlockIndex.test_to_block_indexN) rE rF rG rg r< rH rI rw rz r} r~ r r5 r5 r5 r6 rr h s
|
|
rr c @ sF e Zd Zdd Zdd Zdd Zejjde e
|
dd d
|
Z d d Z d S )Ú TestIntIndexc C sL d}t jt|d tddddgd W 5 Q R X d}t jt|d tddd dgd W 5 Q R X d}t jt|d tddd dgd W 5 Q R X d
|
}t jt|d tddddgd W 5 Q R X t jt|d tdddd gd W 5 Q R X d }t jt|d tddddgd W 5 Q R X t jt|d tddddgd W 5 Q R X d S ) NzToo many indicesr9 r r r )Úlengthr+ z No index can be less than zeror
|
éþÿÿÿz(All indices must be less than the lengthr z#Indices must be strictly increasing)r< r= r> r )r, rC r5 r5 r6 r} ² s& z!TestIntIndex.test_check_integrityc C s t dtjddgtjddd}t|ts,t|jdks:tt |j
|
tjddgtjd¡ t dtjg tjddd}t|tst|jdkstt |j
|
tjg tjd¡ t dtjddddgtjddd}t|tsØt|jdksætt |j
|
tjddddgtjd¡ d S r[ r_ ra r5 r5 r6 rc × s ÿz TestIntIndex.test_int_internalc C sB t ddddddg}| |¡s"t| t dddddg¡r>td S )Nr r r r r r )r r; r" rx r5 r5 r6 rz é s zTestIntIndex.test_equalsr rK c C s^ t t||}t t||}| ¡ ¡ } | ¡ ¡ }
|
t| t s>t| |¡sLt|
|
|¡sZtd S rU )r r r* rR r! r" r; ) r, r- r. r/ r0 r1 r2 r3 r4 ZxbindexZybindexr5 r5 r6 r î s z TestIntIndex.test_to_block_indexc C s( t ddddddg}| ¡ |ks$td S )Nr r r r r
|
r )r r* r" rx r5 r5 r6 r~ ú s z TestIntIndex.test_to_int_indexN) rE rF rG r} rc rz r< rH rI rX rY r r~ r5 r5 r5 r6 r ± s %
|
r c @ s> e Zd Zej ddddddg¡ejjdeedd d
|
Zd S ) ÚTestSparseOperatorsÚopnameÚaddÚsubÚmulÚtruedivÚfloordivr rK c C sH t td| d}t t|} tt||}
|
tt||} |
|
¡ } | ¡ } t |
|
j¡d d }t | j¡d d }d}d}|||
|
||| |\}}}||| ||| |\}}}| ¡ |¡s¶t
|
t ||¡ ||ksÎt
|
t || j}| t t¡¡ |¡}t || j}| t t¡¡ |¡}| ||}| |j¡}t ||j¡ t ||j¡ d S )NZsparse_Z_float64g $@r g Y@r r )ÚgetattrÚsplibÚoperatorr r r* r& ru r` r; r" r# r$ r r+ ZreindexZfillnaÚvalues)r, r r- r. r/ r0 r1 r2 Z sparse_opZ python_opr3 r4 ZxdindexZydindexÚxÚyZxfillZyfillZresult_block_valsZrb_indexZbfillZresult_int_valsZri_indexZifillZxseriesZyseriesZ series_resultr5 r5 r6 Útest_op sJ
|
ÿ
|
ÿ
|
|
zTestSparseOperators.test_opN) rE rF rG r< rH rI rX rY r r5 r5 r5 r6 r ÿ s r )%r Únumpyr& r< Zpandas._libs.sparseZ_libsÚsparser Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrW Zpandasr Zpandas._testingZ_testingr# Zpandas.core.arrays.sparser r r r r r r r r r Z
|
both_emptyrX rY r rJ rZ rr r r r5 r5 r5 r6 Ú<module> s ú ú ú ú
|
ú
|
ú
|
ú ú
|
[. IN
|
