o
    Tib'                     @   sR   d dl Zd dl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)	DataFrame
RangeIndexSeriesconcat
date_rangec                   @   sH  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d%d& Zd'd( Zd)d* Zd+d, Zd-d. ZdS )/TestEmptyConcatc           	      C   s4  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 	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   Or
   Ai'  Z20130101s)Zperiodsfreqindex   r   )r   nprandomZdefault_rngZstandard_normallistcopyr   Zreindexastypeloctmassert_frame_equalranger   )	selfr   dfZdfcopyemptyframesZ	concattedexpectedresult r.   f/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/reshape/concat/test_empty.pytest_handle_empty_objects   s0   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)r4   dtyper   r   )r3   r5   r2   r   zDThe behavior of array concatenation with empty entries is deprecated)matchr   )r3   r   )r   r   )r   r   r   r   nanr   r%   r&   Zassert_produces_warningFutureWarningassert_series_equal)r(   s1s2resexpmsgr.   r.   r/   test_concat_empty_series2   s2   z(TestEmptyConcat.test_concat_empty_seriestzNUTCvaluesr1   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]r7   )r   r   r   r   )r   dtZtz_localizer   r6   r   pdZNaTlenr   r%   r&   )r(   rB   rD   firstr7   secondr,   r-   r.   r.   r/   !test_concat_empty_series_timelikeQ   s   z1TestEmptyConcat.test_concat_empty_series_timelikezleft,right,expectedm8[ns]rE   )categoryrN   rN   )rN   objectrO   c                 C   s*   t t|dt|dg}|j|ksJ d S NrF   )r   r   r7   )r(   leftrightr,   r-   r.   r.   r/   test_concat_empty_series_dtypesc   s   z/TestEmptyConcat.test_concat_empty_series_dtypesr7   )r6   int8uint8boolrM   rE   c                 C   sR   t |}tt|dg}|j|ksJ tt|dt|dg}|j|ks'J d S rP   )r   r7   r   r   )r(   r7   r-   r.   r.   r/   0test_concat_empty_series_dtypes_match_roundtripsx   s
   
z@TestEmptyConcat.test_concat_empty_series_dtypes_match_roundtrips)r6   rT   rU   rM   rE   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   uirZ   rY   r   kindrI   r7   rX   Ztypsr.   r.   r/   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>   rZ   rY   fr_   r[   r]   r.   r.   r/   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 )Nr   r.   )r7   rX   r-   r`   r^   r.   r/   get_result_type   s   

zSTestEmptyConcat.test_concat_empty_series_dtypes_roundtrips.<locals>.get_result_typerF   )pytestskipr   r7   r   r   r\   )r(   r7   rX   rb   r,   r-   r.   ra   r/   *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 )NrE   rF   )r   r   r   bool_int64r7   object_r(   r.   r.   r/   &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 )NrN   rF   r6   )r   r   r   arrayr7   ri   r.   r.   r/   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 )Nr6   rF   ZSparsezSparse[float64]rO   )r   r   r#   r7   rH   ZSparseDtyper   r6   )r(   r-   r,   r.   r.   r/   &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   r1   )RowZEmptyColZ	NumberColr   r   r   )	r   r   r9   r   r   r#   rO   r%   r&   )r(   Zdf_1Zdf_2r-   r,   r.   r.   r/   !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   rf   int32r6   r   r7   rh   )r(   r)   r-   r.   r.   r/   "test_concat_empty_dataframe_dtypes   s   z2TestEmptyConcat.test_concat_empty_dataframe_dtypesc                 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   rg   )r   r7   inner)r   joinouter)r   r   r   r%   r&   )r(   df_emptyZdf_aZdf_expectedr-   r.   r.   r/   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   r2   r
   )r   r   Zdtypesr%   r;   )r(   df1df2r-   r,   r.   r.   r/   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(   ry   rz   r-   r,   Zdf3Zdf4r.   r.   r/   test_concat_empty_dataframe  s   z+TestEmptyConcat.test_concat_empty_dataframec                 C   sn   t g dg dd}t dg di}t|d d |d d g}|d jtjks+J |d jtjks5J d S )Nr1   r|   )r   r   r   r   r   )r   r   r7   r   rg   rh   )r(   ry   rz   r-   r.   r.   r/   ,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   rF   r1   N)r   rH   rk   Z
Int64Dtyper"   r   r%   r&   )r(   rv   Zdf_newr,   r-   r.   r.   r/   test_concat_to_empty_ea  s
   z'TestEmptyConcat.test_concat_to_empty_ea)__name__
__module____qualname__r0   rA   rc   markZparametrizerL   r   rf   rq   rh   Zfloat32rg   rS   rW   re   rj   rl   rm   ro   rr   rw   r{   r}   r~   r   r.   r.   r.   r/   r      sN    "

	(		r   )numpyr   rc   ZpandasrH   r   r   r   r   r   Zpandas._testingZ_testingr%   r   r.   r.   r.   r/   <module>   s    