
    Mh"                     T    S SK rS SKrS SKrS SKJrJrJrJr  S SK	J
r   " S S5      rg)    N)	DataFrameIndexSeriesconcatc                   (   \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 r\R                  R                  SSS/5      \R                  R                  SSS/5      \R                  R                  SSS/5      S 5       5       5       rS rS rS rSrg)TestDataFrameConcat   c                    [        [        R                  " S5      SS/[        R                  S9n[        [        R                  " S5      [        R                  S9n[        X4SS9R                  n[        [        R                  " S5      /S	-  [        R                  " S
5      /S	-  -   / SQS9n[        R                  " X45        g )N)
      foobar)datacolumnsdtype)r   r      axisfloat64r   float32)r   r   r   r   index)r   nponesr   r   r   dtypesr   r   tmassert_series_equalselfdf1df2resultsexpecteds        \/var/www/html/env/lib/python3.13/site-packages/pandas/tests/reshape/concat/test_dataframe.py"test_concat_multiple_frames_dtypes6TestDataFrameConcat.test_concat_multiple_frames_dtypes   s    RWWW-u~RZZXRWWW-RZZ@#!,33XXi !A%))<(=(AA&
 	w1    c           	          [        [        R                  " S5      [        S5      S9n[        [        R                  " S5      S-  [        S5      S9n[	        X4SS/S9n[        S	S	S
S
S
S.S	S	S
S
S
S.S.5      n[
        R                  " X45        g )N)r   r   ABr   )   r   r   )beebah)r,   bookeys      ?g       @))r,   r-   r   )r,   r-   r   )r,   r.   r   )r,   r.   r   )r,   r.   r   )AB)r   r   r   listr   r   assert_frame_equalr   s        r$   test_concat_tuple_keys*TestDataFrameConcat.test_concat_tuple_keys   s    d<!+T$Z@#>>*JK (+'*'*'*'* (+'*'*'*'*
$ 	g0r'   c                    [        SS/SS/S.5      n[        SS/SS	9n[        X/US
9n[        / SQ/ SQS.[        R                  R                  SS/SS/4SS /S9S9n[        R                  " X45        [        SS/S S	9n[        X/US/S9n[        R                  " Xd5        [        X/US
9n[        / SQ/ SQS.[        R                  R                  SS/SS/4S S /S9S9n[        R                  " Xx5        g )Nr   r   皙?皙?)r   r   abbaznamer/   )r   r   r   r   )r9   r:   r9   r:   r   namesr   r0   rA   )r   r   r   pd
MultiIndexfrom_productr   r5   )	r   dfr   concatted_named_from_keysexpected_namedindex_no_nameconcatted_named_from_namesconcatted_unnamedexpected_unnameds	            r$   test_concat_named_keys*TestDataFrameConcat.test_concat_named_keys3   s   1vsCj9:sCju-$*B8%$@!" )=>--,,sCj1a&-A%QU,W
 	7HsCjt4%+RH=QVPW%X"
8I"B8-@$ )=>--,,sCj1a&-A$PT,V
 	/Br'   c                 8   [        SSS/0[        S5      S9n[        SSS/0[        S5      S9n[        S/ SQ0/ S	QS9n[        X/S
S9n[        R                  " XC5        [        X/SS9n[        R                  " XS5        [        X/SS9n[        R                  " Xc5        [        SS/SS//SS/SS/S9n[        X/SS9n[        R                  " X5        [        X/SS9n	[        R                  " X5        [        SS/5      n
[        SS/5      n[        / SQ/ S	QS9n[        X/S
S9n[        R                  " X5        [        X/SS9n[        R                  " X5        [        X/SS9n[        R                  " X5        [        SS/SS//SS/SS/S9n[        X/SS9n[        R                  " UU5        [        X/SS9n[        R                  " UU5        [        R                  " [        SS9   [        X/SS9  S S S 5        g ! , (       d  f       g = f)Nr2   r9   r:   r   r   333333?皙?)r9   r:   rP   rQ   )r   r   r   r   r   r   rowsr   r   )r   r   r   zNo axis named)match	something)
r   ranger   r   r5   r   r   pytestraises
ValueError)r   r    r!   expected_indexconcatted_indexconcatted_rowconcatted_0expected_columnsconcatted_columnsconcatted_1series1series2expected_index_seriesconcatted_index_seriesconcatted_row_seriesconcatted_0_seriesexpected_columns_seriesconcatted_columns_seriesconcatted_1_seriess                      r$   test_concat_axis_parameter.TestDataFrameConcat.test_concat_axis_parameterI   s"   sCj)q:sCj)q: #C)=#>lS #':
o>z7
m<cZa0
k: %3Z#s$QFS#J
 #C:I>
/BcZa0
k<#s$#s$ !'';< P!'(:!I
5M%w&8vF
3K#W$6Q?
1I #,3Z#s$QFQF#
 $*7*<9#M 
68OP#W$6Q?
02IJ ]]:_=G%K8 >==s   6H
Hc           	      h   [        S[        S5      0S[        R                  R	                  / SQ/ SQ/SS/S9S	9n[        UR                  S S2S S 24   UR                  S
S 2S S 24   45      n[        S/ SQ0S[        R                  R                  / SQSS/S9S	9n[        R                  " X#5        g )Ncol	   int32)A0A1A2)B0B1B2r   r   r@   )r   r   )r   r         ))ro   rr   )ro   rs   )rq   rs   )rq   rt   )
r   rU   rC   rD   rE   r   ilocfrom_tuplesr   r5   r   rF   resultr#   s       r$   test_concat_numerical_names/TestDataFrameConcat.test_concat_numerical_names   s    E!H**');<QF + 	
 !Qa9:L!--++HQRTUPV , 
 	f/r'   c                    [        SS0/5      n[        X/SS9nUR                  S5      n[        [        R                  " SS/5      R                  SS5      SS/S9R                  S5      n[        R                  " X#5        g )Nr;   r<   r   r   categoryr   r*   )r   r   astyper   arrayreshaper   r5   rz   s       r$   test_concat_astype_dup_col.TestDataFrameConcat.test_concat_astype_dup_col   sx    c
|$RH1%:&HHc3Z ((A.c


&
 	 	f/r'   c                     [        S[        / SQ[        / SQSS9S905      n[        S[        SS/[        S	S
/SS9S905      n[        X#/SSS/US9n[	        UR
                  5      SS/:X  d   eg )Nvaluer   r   r+   )r;   r<   cidr>   r   rv   rw   r;   r<   r   t1t2)r   r0   sort)r   r   )r   r   )r   r   r   r   r4   r   )r   r   r   r   r{   s        r$   test_concat_dataframe_keys_bug2TestDataFrameConcat.test_concat_dataframe_keys_bug   s}    fYeO$.OPQ
 AeS#JT6R!STU qd|$GFNN#'IIIIr'   c                     [        [        / SQSS95      n[        [        / SQSS95      n[        X/5      n[        UR                  S5      U/5      n[        R
                  " X45        g )N)TFTTbool)r   )r   r   r   int64)r   r   r   r   r   r5   r   r    r!   r{   r#   s        r$   test_concat_bool_with_int-TestDataFrameConcat.test_concat_bool_with_int   sY     8GHy89
#3::g.45
f/r'   c                 8   / SQn/ SQn[        X!S9n[        U/S/SS/S9n[        R                  R	                  S/U/SS/S9n[        X%S9n[
        R                  " XF5        [
        R                  " UR                  R                  S	   [        S	S
/SS95        g )N)r   r   r+   r   )r   r   r2   IDdaterB   r@   r   r+   r>   )r   r   rC   rD   rE   r   r5   assert_index_equalr   levelsr   )r   r   r   rF   r{   mir#   s          r$   )test_concat_duplicates_in_index_with_keys=TestDataFrameConcat.test_concat_duplicates_in_index_with_keys   s    D.C5v?]]''#tVn'M$1
f/
fll11!4eQF6PQr'   ignore_indexTForderCFr   r   r   c                 "   [        [        R                  " S[        R                  US95      n[	        U/S-  XSS9nU(       dL  UR                  5        H7  nUR                  5        H   n[        R                  " Xx5      (       d  M    e   M9     g g )N)r      )r   r   r   T)r   r   copy)r   r   zerosr   r   _iter_column_arraysshares_memory)	r   r   r   r   using_copy_on_writerF   resarrarr2s	            r$   test_concat_copies&TestDataFrameConcat.test_concat_copies   sx    
 rxxrzzGHbTAXD$O"..0224D!//:::: 5 1 #r'   c                     [        S/S/SS.5      n[        SS/05      n[        X/SSSS9n[        S	[        R                  /SS/S	[        R                  /S
.5      n[        R
                  " X45        g )Nr   r   r2   r3   r   r2   d   Touterr   joinr   r1   )r   r2   r3   )r   r   r   nanr   r5   r   s        r$   test_outer_sort_columns+TestDataFrameConcat.test_outer_sort_columns   sl    qc23se%
G$O#rvvaXS"&&MRS
f/r'   c                     [        S/S/SS.5      n[        S/SS.5      n[        X/SSSS	9n[        SS/SS/S
.5      n[        R                  " X45        g )Nr   r   r   r   r   r2   r   Tinnerr   r   r2   r   r   r   r5   r   s        r$   test_inner_sort_columns+TestDataFrameConcat.test_inner_sort_columns   s_    qc23se*+
G$O!Qq#h78
f/r'   c                     [        S/SS.5      n[        U/SSSS9n[        S/S/S.5      n[        R                  " X#5        g )Nr   r   r   Tr   r   r   r   )r   r    r{   r#   s       r$   test_sort_columns_one_df,TestDataFrameConcat.test_sort_columns_one_df   sF    se*+DwTJ!C512
f/r'    N)__name__
__module____qualname____firstlineno__r%   r6   rM   ri   r|   r   r   r   r   rV   markparametrizer   r   r   r   __static_attributes__r   r'   r$   r   r      s    	212C,89t0*	0J0
R [[^dE];[[WsCj1[[VaV,	; - 2 <	;000r'   r   )numpyr   rV   pandasrC   r   r   r   r   pandas._testing_testingr   r   r   r'   r$   <module>r      s)        X0 X0r'   