U
    sVc/	                     @   s  d dl Zd dlZd dlZd dlmZ ej	dddgdfddgdfddgdfddgdfddgdfdd	gdfddgd
fddgdfddgdfg	dd Z
ej	dddgdfddgdfddgdfddgdfddgdfddgdfddgd
fddgdfddgdfg	dd ZdS )    Nzto_concat_dtypes, result_dtypeZInt64ZUInt64ZInt8ZInt16ZUInt8ZInt32ZUInt32ZFloat64booleanobjectc                 C   s   t dd | D }t t jddt jgtdgd |}t|| t dd | d d d D }t t jddt jgtdgd |}t|| d S )	Nc                 S   s"   g | ]}t jd dt jg|dqS r      ZdtypepdSeriesNA.0t r   K/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/arrays/integer/test_concat.py
<listcomp>   s     z&test_concat_series.<locals>.<listcomp>r   r   r      c                 S   s"   g | ]}t jd dt jg|dqS r   r   r   r   r   r   r   "   s     )r   concatr	   r
   r   astypetmassert_series_equal)to_concat_dtypesresult_dtyperesultexpectedr   r   r   test_concat_series   s    ""r   Zint64Zuint64Zint8Zint16Zuint32boolc                 C   s   t jddt jg| d d}t tjddg| d d}t j||gdd}t jddt jddgtd|}t	|| t j||gdd}t jddddt jgtd|}t	|| d S )Nr   r   r   T)Zignore_index)
r   r	   r
   nparrayr   r   r   r   r   )r   r   s1s2r   r   r   r   r   test_concat_series_with_numpy*   s      r!   )Znumpyr   ZpytestZpandasr   Zpandas._testingZ_testingr   markZparametrizer   r!   r   r   r   r   <module>   s>   


















