U
    ÎZjµ  ã                   @   sd   d Z ddlZddlZddlmZ ddlmZ G dd„ deƒZ	ej
dd„ ƒZG dd	„ d	ƒZd
d„ ZdS )zB
Tests for behavior if an author does *not* implement EA methods.
é    N)ÚExtensionArrayc                   @   s   e Zd Zddœdd„ZdS )ÚMyEAN)Úreturnc                 C   s
   || _ d S ©N)Z_values)ÚselfÚvalues© r   úh/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/extension/test_extension.pyÚ__init__   s    zMyEA.__init__)Ú__name__Ú
__module__Ú__qualname__r
   r   r   r   r	   r      s   r   c                  C   s   t  d¡} t| ƒS )Né
   )ÚnpZaranger   )Zarrr   r   r	   Údata   s    
r   c                   @   s   e Zd Zdd„ ZdS )ÚTestExtensionArrayc              	   C   s(   |}t  t¡� t||ƒ W 5 Q R X d S r   )ÚpytestZraisesÚAttributeErrorÚgetattr)r   r   Zall_arithmetic_operatorsZop_namer   r   r	   Útest_errors   s    zTestExtensionArray.test_errorsN)r   r   r   r   r   r   r   r	   r      s   r   c               
   C   sN   d} t jt| d��2 G dd„ dtƒ}t  d ¡� |ƒ  W 5 Q R X W 5 Q R X d S )Nz<The `na_sentinel` argument of `MyEA.factorize` is deprecated)Úmatchc                   @   s   e Zd Zddd„ZdS )z#test_depr_na_sentinel.<locals>.MyEAéÿÿÿÿc                 S   s   d S r   r   )r   Zna_sentinelr   r   r	   Ú	factorize$   s    z-test_depr_na_sentinel.<locals>.MyEA.factorizeN)r   )r   r   r   r   r   r   r   r	   r   #   s   r   )ÚtmZassert_produces_warningÚDeprecationWarningr   )Úmsgr   r   r   r	   Útest_depr_na_sentinel   s
    r   )Ú__doc__Únumpyr   r   Zpandas._testingZ_testingr   Zpandas.core.arraysr   r   Zfixturer   r   r   r   r   r   r	   Ú<module>   s   
