a
    j=icµ  ã                   @   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   úv/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/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 d   ƒ n1 s.0    Y  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   sv   d} t jt| dP G dd„ dtƒ}t  d ¡ |ƒ  W d   ƒ n1 sJ0    Y  W d   ƒ n1 sh0    Y  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   
