o
    }o™in  ã                   @   s.   d dl mZ G dd„ deƒZG dd„ dƒZdS )é    )ÚPrettyStrEnumc                   @   s   e Zd ZdZdZdS )ÚASRModelTypeÚctcÚrnntN)Ú__name__Ú
__module__Ú__qualname__ÚCTCÚRNNT© r   r   úI/home/ubuntu/.local/lib/python3.10/site-packages/tests/utils/test_enum.pyr      s    r   c                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestPrettyStrEnumc              
   C   sF   zt dƒ W dS  ty" } zt|ƒdksJ ‚W Y d}~dS d}~ww )z+Test pretty error message for invalid valueÚ	incorrectzBincorrect is not a valid ASRModelType. Possible choices: ctc, rnntN)r   Ú
ValueErrorÚstr)ÚselfÚer   r   r   Útest_incorrect_value   s   €ÿz&TestPrettyStrEnum.test_incorrect_valuec                 C   s   t dƒt jks	J ‚dS )z#Test that correct value is acceptedr   N)r   r	   ©r   r   r   r   Útest_correct_value    s   z$TestPrettyStrEnum.test_correct_valuec                 C   s   t tdƒƒdks
J ‚dS )z†
        Test that str() returns the source value,
        useful for serialization/deserialization and user-friendly logging
        r   N)r   r   r   r   r   r   Útest_str$   s   zTestPrettyStrEnum.test_strN)r   r   r   r   r   r   r   r   r   r   r      s    r   N)Únemo.utils.enumr   r   r   r   r   r   r   Ú<module>   s   