a
    oÝEb©  ã                   @   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|v sDJ ‚dt|ƒ› �|v sZJ ‚t|jƒ|v slJ ‚|dkr€d|v s€J ‚d S )Nr   é   zLength: r   z...)ÚtypeZ_concat_same_typeÚreprÚ__name__ÚlenÚstrÚdtype)ÚselfÚdatar   Úresult© r   úd/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/extension/base/printing.pyÚtest_array_repr   s    z!BasePrintingTests.test_array_reprc                 C   s   t |ƒ}t|t ƒsJ ‚d S ©N)r   Ú
isinstance)r   r   r   r   r   r   Útest_array_repr_unicode   s    z)BasePrintingTests.test_array_repr_unicodec                 C   s"   t  |¡}|jjt|ƒv sJ ‚d S r   )ÚpdZSeriesr   Únamer	   )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|v s6J ‚d S )Nr   )Úbuf)ÚioÚStringIOr   r   ÚinfoÚgetvaluer   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   