a
    Pfh                     @   sT   d dl Zd dlZd dlmZ d dlmZ d dlm	Z	 d dl
mZ dd Zdd ZdS )	    N)DtypeWarning)ArrowExtensionArray)_concatenate_chunksc                  C   sd   t d} dt| ddgidt| ddgig}t|}t| g d}t|d | d S )Npyarrowr         ?      @      )r   r   g      ?g       @)pytestimportorskipr   arrayr   tmZassert_extension_array_equalpachunksresultexpected r   g/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/parser/test_concatenate_chunks.pytest_concatenate_chunks_pyarrow   s    
r   c                  C   s   t d} dt| ddgidt| ddgig}tjtdd t|}W d    n1 sb0    Y  t	tjddgt
d	tddgg}t|d | d S )
Nr   r   r   r   abzhave mixed types)match)Zdtype)r
   r   r   r   r   Zassert_produces_warningr   r   npZconcatenateobjectZassert_numpy_array_equalr   r   r   r   'test_concatenate_chunks_pyarrow_strings   s    
&r   )numpyr   r
   Zpandas.errorsr   Zpandas._testingZ_testingr   Zpandas.core.arraysr   Z"pandas.io.parsers.c_parser_wrapperr   r   r   r   r   r   r   <module>   s   