U
|
ý°d
|
ã @ s d dl Zd dlZd dlZd dlmZ d dlm Z G dd dZ
|
ej dej dddgd d
|
e d d ¡fej d d d gdd
|
e dd ¡fej dddged
|
e ed ¡fej dddgdd
|
e d ¡fej d d d gdd
|
e d¡fej dddgdd
|
e e¡fg¡dd ZdS )é N)Ú SparseArrayc @ s@ e Zd Zej dddg¡dd Zej dddg¡dd ZdS ) ÚTestSparseArrayConcatÚkindÚintegerÚblockc C sn t ddddg|d}t ddddg|d}t ||g¡}tjdddddgdd}t |j|¡ |j|ksjtd S )Né r é ©r Úint64©Údtype© r Z_concat_same_typeÚnpÚarrayÚtmZassert_numpy_array_equalZ sp_valuesr ÚAssertionError)Úselfr ÚaÚbÚresultÚexpected© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/sparse/test_combine_concat.pyÚ
|
test_basic
|
s z TestSparseArrayConcat.test_basicc C s~ |dkr dnd}t ddddg|d}t ddddg|d}t ||g¡}tjdddddgdd}t |j|¡ |j|ksztd S ) Nr r r r r r r
|
r r )r r Úotherr r r r r r r Útest_uses_first_kind s z*TestSparseArrayConcat.test_uses_first_kindN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer r r r r r r s
|
r zother, expected_dtypeé é é r
|
r g ø?g @g @Úfloat64r r ÚcÚcategoryc C s t jdddgt dd¡d}t j|| gdd}t t|t| ¡ |¡}t ||¡ t j| |gdd}t t| t| ¡ |¡}t ||¡ d S )Nr r r r
|
r T)Z ignore_index)ÚpdÚSeriesÚ SparseDtypeÚconcatÚlistZastyper Zassert_series_equal)r Zexpected_dtypeZs_sparser r r r r Útest_concat_with_non_sparse# s r- )Únumpyr r Zpandasr( Zpandas._testingZ_testingr Zpandas.core.arrays.sparser r r r! r) r* Úobjectr r- r r r r Ú<module> s ôþ
|
