a
    PfH                     @   sN   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 dZdS )    )StringION)	DataFrameconcatread_csvc                   @   sF   e Zd Zejddi ddgdgdd Zdd Zd	d
 Zdd Z	dS )TestInvalidConcatobj      )r   r	   c                 C   sZ   t td}dt| d}tjt|d t||g W d    n1 sL0    Y  d S )Nr	   z#cannot concatenate object of type 'z+'; only Series and DataFrame objs are validmatch)r   rangetypepytestraises	TypeErrorr   )selfr   df1msg r   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/reshape/concat/test_invalid.pytest_concat_invalid   s
    z%TestInvalidConcat.test_concat_invalidc                 C   sJ   t td}d}tjt|d t| W d    n1 s<0    Y  d S )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   r   r   r   "test_concat_invalid_first_argument   s
    z4TestInvalidConcat.test_concat_invalid_first_argumentc                 C   s   t dd tdD  d S )Nc                 s   s$   | ]}t tjd dV  qdS )r	   )   r   N)r   nprandomZdefault_rng).0_r   r   r   	<genexpr>%       z>TestInvalidConcat.test_concat_generator_obj.<locals>.<genexpr>   )r   r   )r   r   r   r   test_concat_generator_obj#   s    z+TestInvalidConcat.test_concat_generator_objc                 C   s\   d}t t|dd}t|dd}W d    n1 s60    Y  t t|}t|| d S )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)Zignore_index)r   r   r   tmZassert_frame_equal)r   datareaderresultexpectedr   r   r   test_concat_textreader_obj'   s
    	*z,TestInvalidConcat.test_concat_textreader_objN)
__name__
__module____qualname__r   markZparametrizer   r   r    r'   r   r   r   r   r      s
   

	r   )ior   numpyr   r   Zpandasr   r   r   Zpandas._testingZ_testingr"   r   r   r   r   r   <module>   s
   