o
    oi7                  	   @   sf   d dl Z d dlmZ ddlmZmZ z
d dlmZ dZW n ee	fy)   dZY nw G dd	 d	Z
dS )
    N)parameterized   )	CACHE_DIRparse_test_case_file)InverseNormalizerTFc                   @   s^   e Zd ZerededdndZee	de
jje dde
jde
jjd	d
 ZdS )TestTimeviF)lang	cache_diroverwrite_cacheNz6vi/data_inverse_text_normalization/test_cases_time.txtzQ`pynini` not installed, please install via nemo_text_processing/pynini_install.sh)reasonCPUc                 C   s    | j j|dd}||ksJ d S )NF)verbose)inverse_normalizerinverse_normalize)self
test_inputexpectedpred r   [/home/ubuntu/.local/lib/python3.10/site-packages/tests/nemo_text_processing/vi/test_time.pytest_denorm!   s   zTestTime.test_denorm)__name__
__module____qualname__PYNINI_AVAILABLEr   r   r   r   expandr   pytestmarkskipifrun_only_onunitr   r   r   r   r   r      s    
r   )r   r   utilsr   r   Anemo_text_processing.inverse_text_normalization.inverse_normalizer   r   ImportErrorModuleNotFoundErrorr   r   r   r   r   <module>   s   