a
    Pf[
                     @   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   Z_concat_same_typenparraytmZassert_numpy_array_equalZ	sp_valuesr   )selfr   abresultexpected r   g/var/www/ai-form-bot/venv/lib/python3.9/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slJ 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      @Z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)Zignore_index)pdSeriesSparseDtypeconcatlistZastyper   Zassert_series_equal)r   Zexpected_dtypeZs_sparser   r   r   r   r   test_concat_with_non_sparse#   s    r.   )numpyr   r!   Zpandasr)   Zpandas._testingZ_testingr   Zpandas.core.arrays.sparser   r   r"   r#   r*   r+   objectr   r.   r   r   r   r   <module>   s    