o
    ŀg[
                     @   s   d dl Zd dlZd dlZd dlmZ d dlm	Z	 G dd dZ
ejdejg dddedd fejg d	d
ded
d fejg dedeed fejg dddedfejg d	dded
fejg dddeefgdd ZdS )    N)SparseArrayc                   @   s@   e Zd Zejdddgdd Zejdddgdd ZdS )	TestSparseArrayConcatkindintegerblockc                 C   s`   t g d|d}t g d|d}t ||g}tjg ddd}t|j| |j|ks.J d S )N   r   r      r   r   r   r	   r	   r   r	   r   r	   r	   int64dtyper   _concat_same_typenparraytmassert_numpy_array_equal	sp_valuesr   )selfr   abresultexpected r   b/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/arrays/sparse/test_combine_concat.py
test_basic
   s   z TestSparseArrayConcat.test_basicc                 C   sp   |dkrdnd}t g d|d}t g d|d}t ||g}tjg ddd}t|j| |j|ks6J d 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   zother, expected_dtype)         r   r   )g      ?g      @g      @float64)r   r   ccategoryc                 C   s   t jg dt ddd}t j|| gdd}t t|t|  |}t|| t j| |gdd}t t| t| |}t|| d S )N)r   r   r	   r   r   r   T)ignore_index)pdSeriesSparseDtypeconcatlistastyper   assert_series_equal)r   expected_dtypes_sparser   r   r   r   r   test_concat_with_non_sparse#   s   r7   )numpyr   r$   pandasr.   pandas._testing_testingr   pandas.core.arrays.sparser   r   r%   r&   r/   r0   objectr   r7   r   r   r   r   <module>   s"    