U
    sVc                     @   s8   d dl Z d dlZd dlZd dlmZ G dd deZdS )    N)BaseExtensionTestsc                   @   sJ   e Zd ZdZejdddgdd Zdd Zd	d
 Z	dd Z
dd ZdS )BasePrintingTestsz6Tests checking the formatting of your EA when printed.sizebigsmallc                 C   s   |dkr|d d }nt ||gd }t|}t |j|ksDtdt| |ksZtt|j|kslt|dkrd|kstd S )Nr      zLength: r   z...)typeZ_concat_same_typerepr__name__AssertionErrorlenstrdtype)selfdatar   result r   H/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/extension/base/printing.pytest_array_repr   s    z!BasePrintingTests.test_array_reprc                 C   s   t |}t|t std S N)r   
isinstancer   )r   r   r   r   r   r   test_array_repr_unicode   s    z)BasePrintingTests.test_array_repr_unicodec                 C   s"   t |}|jjt|kstd S r   )pdZSeriesr   namer	   r   )r   r   Zserr   r   r   test_series_repr   s    
z"BasePrintingTests.test_series_reprc                 C   s   t d|i}t| d S )NA)r   	DataFramer	   )r   r   Zdfr   r   r   test_dataframe_repr"   s    z%BasePrintingTests.test_dataframe_reprc                 C   s:   t  }td|ij|d | }|jj|ks6td S )Nr   )buf)	ioStringIOr   r   infogetvaluer   r   r   )r   r   r   r   r   r   r   test_dtype_name_in_info&   s    z)BasePrintingTests.test_dtype_name_in_infoN)r
   
__module____qualname____doc__pytestmarkZparametrizer   r   r   r   r#   r   r   r   r   r   	   s   
r   )r   r'   Zpandasr   Z pandas.tests.extension.base.baser   r   r   r   r   r   <module>   s   