
    OwgH                     T    d dl mZ d dlZd dlZd dlmZmZmZ d dl	m
Z  G d d      Zy)    )StringION)	DataFrameconcatread_csvc                   n    e Zd Zej                  j                  ddi ddgdg      d        Zd Zd Zd Z	y	)
TestInvalidConcatobj      )r
   r   c                     t        t        d            }dt        |       d}t        j                  t
        |      5  t        ||g       d d d        y # 1 sw Y   y xY w)Nr   z#cannot concatenate object of type 'z+'; only Series and DataFrame objs are validmatch)r   rangetypepytestraises	TypeErrorr   )selfr	   df1msgs       _/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/reshape/concat/test_invalid.pytest_concat_invalidz%TestInvalidConcat.test_concat_invalid   s]     a!1$s) =7 7 	 ]]9C0 	C:	 	 	s   AAc                     t        t        d            }d}t        j                  t        |      5  t        |       d d d        y # 1 sw Y   y xY w)Nr   z^first argument must be an iterable of pandas objects, you passed an object of type "DataFrame"r   )r   r   r   r   r   r   )r   r   r   s      r   "test_concat_invalid_first_argumentz4TestInvalidConcat.test_concat_invalid_first_argument   sG    a!@ 	 ]]9C0 	3K	 	 	s   AAc                 :    t        d t        d      D               y )Nc              3      K   | ];  }t        t        j                  j                  d       j                  d             = yw)r   )   r   N)r   nprandomdefault_rng).0_s     r   	<genexpr>z>TestInvalidConcat.test_concat_generator_obj.<locals>.<genexpr>%   s/     Tay..q188@ATs   AA   )r   r   )r   s    r   test_concat_generator_objz+TestInvalidConcat.test_concat_generator_obj#   s    T5QR8TT    c                     d}t        t        |      d      5 }t        |d      }d d d        t        t        |            }t        j                  |       y # 1 sw Y   4xY w)Nzindex,A,B,C,D
                  foo,2,3,4,5
                  bar,7,8,9,10
                  baz,12,13,14,15
                  qux,12,13,14,15
                  foo2,12,13,14,15
                  bar2,12,13,14,15
               r
   )	chunksizeT)ignore_index)r   r   r   tmassert_frame_equal)r   datareaderresultexpecteds        r   test_concat_textreader_objz,TestInvalidConcat.test_concat_textreader_obj'   sZ     htn2 	7fF6F	7HTN+
fh/	7 	7s   AA#N)
__name__
__module____qualname__r   markparametrizer   r   r%   r0    r&   r   r   r      sB    [[UQQFF$;< =U0r&   r   )ior   numpyr   r   pandasr   r   r   pandas._testing_testingr*   r   r6   r&   r   <module>r<      s'       
 (0 (0r&   