o
    vh(                     @   s^   d dl Zd dlZd dlmZ d dlZd dlmZmZm	Z	m
Z
mZ d dlmZ G dd dZdS )    N)using_string_dtype)	DataFrame
RangeIndexSeriesconcat
date_rangec                   @   sZ  e Zd Zdd Zdd Zejdddgejdg g d	gd
d Zejde	j
e	je	jfe	j
e	je	jfde	j
e	jfde	je	jfde	j
e	jfde	je	jfddgdd Zejdg ddd Zejdg dejdg ddd Zdd Zdd Zdd  Zd!d" Zd#d$ Zejje d%d&d'd( Zd)d* Zd+d, Zd-d. Zd/d0 ZdS )1TestEmptyConcatc           
      C   s<  t tjddtdd}|d d  }d|d< |dd }||||dd  g}t|d|d	}|jg d
d}|d 	|sBt
nd|d< d|jdddf< t|| t dtditddddd}t  }t||gdd}	t|	| t||gdd}	t|	| t||g}	t|	| t||g}	t|	| d S )N   )
      abcdcolumns   barfoor   )axissort)abcdr   strr   Ai'  20130101s)periodsfreqindex   r   )r   nprandomdefault_rngstandard_normallistcopyr   reindexastypeobjectloctmassert_frame_equalranger   )
selfr   using_infer_stringdfdfcopyemptyframes	concattedexpectedresult r8   d/var/www/html/hyperkenya/venv/lib/python3.10/site-packages/pandas/tests/reshape/concat/test_empty.pytest_handle_empty_objects   s4   
z)TestEmptyConcat.test_handle_empty_objectsc                 C   s<  t g ddd}t ddd}t||gdd}tg dtjtjtjgd	td
d}t|| t g ddd}t ddd}d}tjt	|d t||gdd}W d    n1 sYw   Y  t g d}t
|| t g ddd}t d dd}t||gdd}tg dtjtjtjgdddgtd
d}t|| d S )Nr    r	      x)nameyfloat64)r>   dtyper    r!   )r=   r?   r<   r   zDThe behavior of array concatenation with empty entries is deprecated)matchr   )r=   r   )r   r   )r   r   r   r"   nanr   r,   r-   assert_produces_warningFutureWarningassert_series_equal)r/   s1s2resexpmsgr8   r8   r9   test_concat_empty_series6   s2   z(TestEmptyConcat.test_concat_empty_seriestzNUTCvaluesr;   c                 C   sx   t g ddj|}|rd ntj}t ||d}tt tjgt| ddj||d}t	||gdd}t
|| d S )NM8[ns]rA   )r   r    r    r!   )r   dttz_localizer"   r@   r   pdNaTlenr   r,   r-   )r/   rM   rO   firstrA   secondr6   r7   r8   r8   r9   !test_concat_empty_series_timelikeU   s   z1TestEmptyConcat.test_concat_empty_series_timelikezleft,right,expectedm8[ns]rP   )categoryr[   r[   )r[   r*   r*   c                 C   s*   t t|dt|dg}|j|ksJ d S NrQ   )r   r   rA   )r/   leftrightr6   r7   r8   r8   r9   test_concat_empty_series_dtypesg   s   z/TestEmptyConcat.test_concat_empty_series_dtypesrA   )r@   int8uint8boolrZ   rP   c                 C   sR   t |}tt|dg}|j|ksJ tt|dt|dg}|j|ks'J d S r\   )r"   rA   r   r   )r/   rA   r7   r8   r8   r9   0test_concat_empty_series_dtypes_match_roundtrips|   s
   
z@TestEmptyConcat.test_concat_empty_series_dtypes_match_roundtrips)r@   r`   ra   rZ   rP   dtype2c                    sz   ||kr	t d dd dd   fdd}t|}t|}|||}tt|dt|dgj}|j|ks;J d S )	Nz%same dtype is not applicable for testc                 S   s`   | j |j h}t|h d s| j dks|j dkrdS t|ddh s.| j dks,|j dkr.dS d S )N>   r   iure   rf   r   kindrV   rA   rd   typsr8   r8   r9   int_result_type   s   zSTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.int_result_typec                 S   s8   | j |j h}t|h d s| j dks|j dkrdS d S )N>   fre   rf   rl   rg   ri   r8   r8   r9   float_result_type   s
   zUTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.float_result_typec                    s0    | |}|d ur|S | |}|d ur|S dS )NOr8   )rA   rd   r7   rm   rk   r8   r9   get_result_type   s   

zSTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.get_result_typerQ   )pytestskipr"   rA   r   r   rh   )r/   rA   rd   rp   r6   r7   r8   ro   r9   *test_concat_empty_series_dtypes_roundtrips   s   

	

z:TestEmptyConcat.test_concat_empty_series_dtypes_roundtripsc                 C   s4   t tddttjdttjdgjtjksJ d S )NrP   rQ   )r   r   r"   bool_int64rA   object_r/   r8   r8   r9   &test_concat_empty_series_dtypes_triple   s   z6TestEmptyConcat.test_concat_empty_series_dtypes_triplec                 C   s.   t ttg ddtddgjdksJ d S )Nr[   rQ   r@   )r   r   r"   arrayrA   rw   r8   r8   r9   2test_concat_empty_series_dtype_category_with_array   s   zBTestEmptyConcat.test_concat_empty_series_dtype_category_with_arrayc                 C   s   t tdddtdddg}|jdksJ t tdddtddg}ttj}|j|ks5J t tdddtddg}td}|j|ksPJ d S )Nr@   rQ   SparsezSparse[float64]r*   )r   r   r)   rA   rT   SparseDtyper"   r@   )r/   r7   r6   r8   r8   r9   &test_concat_empty_series_dtypes_sparse   s    
z6TestEmptyConcat.test_concat_empty_series_dtypes_sparsec                 C   sP   t g dtjg dd}t |jd}t||gdd}|t}t|| d S )N)r   r    r    r;   )RowEmptyCol	NumberColr   r   r!   )	r   r"   rC   r   r   r)   r*   r,   r-   )r/   df_1df_2r7   r6   r8   r8   r9   !test_concat_empty_df_object_dtype   s
   
z1TestEmptyConcat.test_concat_empty_df_object_dtypec                 C   s   t tdd}|d tj|d< |d tj|d< |d tj|d< t||g}|d jtjks5J |d jtjks?J |d jtjksIJ t||tjg}|d jtj	ks]J |d jtjksgJ |d jtjksqJ d S )Nabcr   r   r   r   )
r   r&   r)   r"   rt   int32r@   r   rA   rv   )r/   r1   r7   r8   r8   r9   "test_concat_empty_dataframe_dtypes   s   z2TestEmptyConcat.test_concat_empty_dataframe_dtypeszTODO(infer_string))reasonc                 C   sv   t  }t dddgiddgdd}t dg itddd}t||gddd}t|| t||gdd	d}t|| d S )
Nr   r    r	   r   ru   )r   rA   inner)r   joinouter)r   r   r   r,   r-   )r/   df_emptydf_adf_expectedr7   r8   r8   r9   test_concat_inner_join_empty   s   z,TestEmptyConcat.test_concat_inner_join_emptyc                 C   s\   t dd gdd ggddgd}t dd gdd ggddgd}t||g}|j}t|j| d S )Nr    r	   r   r   )datar   r<   r   )r   r   dtypesr,   rF   )r/   df1df2r7   r6   r8   r8   r9   test_empty_dtype_coerce  s
   z'TestEmptyConcat.test_empty_dtype_coercec                 C   s   t ddgd}t ddgd}t|||g}t g dd}t|| t ddgd}t dgd}t||g}t ddgd}t|| d S )Nr   r   r   r   r   r   r   )r   r   r,   r-   )r/   r   r   r7   r6   df3df4r8   r8   r9   test_concat_empty_dataframe  s   z+TestEmptyConcat.test_concat_empty_dataframec                 C   sv   t g dg dd}t dg di}t|d d |d d g}|d jtjks+J |s7|d jtjks9J d S d S )Nr;   r   )r   r   r   r   r   )r   r   rA   r"   ru   rv   )r/   r0   r   r   r7   r8   r8   r9   ,test_concat_empty_dataframe_different_dtypes  s
    z<TestEmptyConcat.test_concat_empty_dataframe_different_dtypesc                 C   s\   t dtjg t di}t dtjg dt di}| }t||g}t|| dS )z<48510 `concat` to an empty EA should maintain type EA dtype.r   rQ   r;   N)r   rT   ry   
Int64Dtyper'   r   r,   r-   )r/   r   df_newr6   r7   r8   r8   r9   test_concat_to_empty_ea%  s
   z'TestEmptyConcat.test_concat_to_empty_ea)__name__
__module____qualname__r:   rL   rq   markparametrizerY   r"   rt   r   rv   float32ru   r_   rc   rs   rx   rz   r}   r   r   xfailr   r   r   r   r   r   r8   r8   r8   r9   r      sP    $

	(	
	r   )numpyr"   rq   pandas._configr   pandasrT   r   r   r   r   r   pandas._testing_testingr,   r   r8   r8   r8   r9   <module>   s    