
    bMh                     `    d dl Zd dlZd dlmZmZmZmZmZm	Z	m
Z
 d dlmZ  G d d          ZdS )    N)	DataFrameDatetimeIndexIndex
MultiIndexSeriesconcat
date_rangec                       e Zd Zd Zd Zd Zd Zd Zd Zd Z	e
j                            d ej        d	          d
fdg          d             Zd Zd ZdS )TestSeriesConcatc                    t          t          j        dt          j                  t	          dd          d          }d|_        |d d         |dd         |dd          g}t          |          }t          j        ||           |j        |j        k    sJ t          |g d	
          }|	                                }t          t          j        |j        j        d                    |_        t          j        g d	d |D                       t          j        t          |                    g}t!          g d	|j        g|          }||_        t          j        ||           d S )N   dtype
2020-01-01periodsfooindexname      r         )keyszM8[ns]c                 ,    g | ]}t          |          S  )len).0xs     g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/reshape/concat/test_series.py
<listcomp>z7TestSeriesConcat.test_concat_series.<locals>.<listcomp>$   s    *B*B*Ba3q66*B*B*B    )levelscodes)r   nparangefloat64r	   r   r   tmassert_series_equalcopyr   arrayr   valuesrepeatr   r   )selftspiecesresultexpected	exp_codes	exp_indexs          r"   test_concat_seriesz#TestSeriesConcat.test_concat_series   sY   Ib
+++\2666
 
 

 RaR&"QrT(BrssG,
vr***{bg%%%%YYY///7799 "(/!J!J!JKKYyyy*B*B6*B*B*BCCRYsSUwwEWEWX	yyy"(&;9MMM	"
vx00000r$   c                    t          dg          }t          g t                    }|}d}t          j        t          |          5  t          ||g          }d d d            n# 1 swxY w Y   t          j        ||           d S )Nr   r   zDThe behavior of array concatenation with empty entries is deprecated)match)r   objectr*   assert_produces_warningFutureWarningr   r+   )r0   s1s2r4   msgr3   s         r"   1test_concat_empty_and_non_empty_series_regressionzBTestSeriesConcat.test_concat_empty_and_non_empty_series_regression)   s    QC[[Bf%%%T'SAAA 	& 	&RH%%F	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&
vx00000s   A$$A(+A(c                    t          t          j        dt          j                  t	          dd                    }|d d         |dd          |dd         g}t          |d	          }t          |          j        }t          j	        ||           t          |g d
d          }t          |g d
          j        }t          j	        ||           d S )N
   r   r   r   r   r   r   axis)ABC)r   rF   )
r   r'   r(   r)   r	   r   r   Tr*   assert_frame_equal)r0   r1   r2   r3   r4   s        r"   test_concat_series_axis1z)TestSeriesConcat.test_concat_series_axis14   s    Ib
+++:lTV3W3W3W
 
 
 SbS'2abb62ad8,Q'''V$$&
fh///___1===V???;;;=
fh/////r$   c                    t          t          j                            d                              d          d          }t          t          j                            d                              d          d          }t          ||gd          }t          ||d          }t          j        ||           d |_	        t          ||gd          }t          j
        |j        t          dd	gd
                     d S )Nr   r   rG   r   rH   r   rE   rG   rH   r   r:   r   )r   r'   randomdefault_rngstandard_normalr   r   r*   rK   r   assert_index_equalcolumnsr   r0   sr>   r3   r4   s        r"   /test_concat_series_axis1_preserves_series_namesz@TestSeriesConcat.test_concat_series_axis1_preserves_series_namesC   s    29((++;;A>>SIIIBI))!,,<<Q??cJJJBa(((12..//
fh///Ba(((
fneS!HH.M.M.MNNNNNr$   c                    t          t          j                            d                              d          g dd          }t          t          j                            d                              d          g dd          }t          ||gd	|
          }t          ||dg d          }|r|                                }t          j	        ||           d S )Nr      )cabrG   r      )dr[   r\   rZ   rH   r   )rF   sortrO   )rZ   r[   r\   r^   rC   )
r   r'   rP   rQ   rR   r   r   
sort_indexr*   rK   )r0   r_   rV   r>   r3   r4   s         r"   %test_concat_series_axis1_with_reindexz6TestSeriesConcat.test_concat_series_axis1_with_reindexP   s    I!!!$$44Q77UX
 
 
 I!!!$$44Q77&&&
 
 

 Bad33312..6J6J6JKKK 	-**,,H
fh/////r$   c                    t          g d          }t          g d          }t          ||gdddgdg          }t          ddgd	d
gddggt          ddgd                    }t	          j        ||           t          ||gdddgddg          }t          ddgd	d
gddggt          j        ddgddg                    }t	          j        ||           d S )N)r   r   rY   )r]   r      r   r[   r\   rG   )rF   r   namesr]   r   r   rY   rc   rN   rT   )r[   r   )r\   r   rH   )rd   )r   r   r   r   r*   rK   r   from_tuplesrU   s        r"   &test_concat_series_axis1_names_appliedz7TestSeriesConcat.test_concat_series_axis1_names_applied`   s   999IIIBasCjFFFVaVaV$eS#JS.I.I.I
 
 
 	fh///Bax.B3PS*UUUVaVaV$*Hh+?SzRRR
 
 
 	fh/////r$   c                    t          ddd          dd         }t          t          j                            d                              t          |                    |d	          }t          t          j                            d                              t          |                    |d	          }t          ||gd
d          }t          t          d                    }t          j        |j        |d           d S )Nz01-Jan-2013z01-Jan-2014MS)freqr   r   valuer   r   TrF   ignore_index)exact)r	   r   r'   rP   rQ   rR   r   r   r   ranger*   rS   rT   )r0   datesr=   r>   r3   r4   s         r"   0test_concat_series_axis1_same_names_ignore_indexzATestSeriesConcat.test_concat_series_axis1_same_names_ignore_indexq   s    =-dCCCAbDII!!!$$44SZZ@@
 
 

 I!!!$$44SZZ@@
 
 
 Rqt<<<q??
fnhdCCCCCCr$   zs1name,s2name   +   r   )rs   rt   c                     t          ddd|          }t          ddd|          }t          ||g          }t          ddddd          }t          j        ||           d S )	Nr   r   )r[   r\   rN   r   rc   )rZ   r^   )r[   r\   rZ   r^   )r   r   r*   r+   )r0   s1names2namer=   r>   r3   r4   s          r"   &test_concat_series_name_npscalar_tuplez7TestSeriesConcat.test_concat_series_name_npscalar_tuple   s{    
 !!$$6222!!$$6222R!!::;;
vx00000r$   c                    t          ddgd          }t          ddg          }t          ddg          }t          |||gd          }t          ddgddgddgdg d	          }t          j        ||           t          |||gdg d
          }t          ddgddgddgd
g d
	          }t          j        ||           t          |||gdd          }t          ddgddgddgd          }t          j        ||           d S )Nr   r   r   rN   r]   r   rE   )r   r   r   re   )redblueyellow)rF   r   Trm   r   )r   r   r   r*   rK   )r0   named_seriesunnamed_series1unnamed_series2r3   r4   s         r"   (test_concat_series_partial_columns_namesz9TestSeriesConcat.test_concat_series_partial_columns_names   so   q!f5111 !Q.. !Q..HqQQQF1v1a&11===
 
 
 	fh///?O<***
 
 

 FQFq!f==---
 
 
 	fh///?O<1SW
 
 
 !QQF1v>>??
fh/////r$   c                      |dg          }t          |j        d d d         g          }t          j        ||           d S )Nd   rk   )r   ilocr*   assert_equal)r0   frame_or_seriesobjr3   s       r"   &test_concat_series_length_one_reversedz7TestSeriesConcat.test_concat_series_length_one_reversed   sG    ose$$$$B$())
$$$$$r$   N)__name__
__module____qualname__r7   r@   rL   rW   ra   rg   rr   pytestmarkparametrizer'   int64ry   r   r   r   r$   r"   r   r      s        1 1 10	1 	1 	10 0 0O O O0 0 0 0 0 0"D D D$ [828C=='2NC 1 1 10 0 0:% % % % %r$   r   )numpyr'   r   pandasr   r   r   r   r   r   r	   pandas._testing_testingr*   r   r   r$   r"   <module>r      s                               _% _% _% _% _% _% _% _% _% _%r$   