o
    oi                     @   sN   d dl Z d dlmZ d dlmZ d dlmZ ddlmZmZ G dd dZ	dS )	    N)parameterized)InverseNormalizer)
Normalizer   )	CACHE_DIRparse_test_case_filec                   @   s~   e Zd ZededddZeede	j
de	j
jdd Zededd	Zeed
e	j
de	j
jdd ZdS )TestCardinalzhFcased)lang	cache_diroverwrite_cache
input_casez2zh/data_text_normalization/test_cases_cardinal.txtCPUc                 C   s   | j |}||ksJ d S )N)normalizer_zh	normalize)self
test_inputexpectedpreds r   _/home/ubuntu/.local/lib/python3.10/site-packages/tests/nemo_text_processing/zh/test_cardinal.pytest_norm_date   s   zTestCardinal.test_norm_date)r   r   r   z6zh/data_inverse_text_normalization/test_cases_date.txtc                 C   s    | j j|dd}||ksJ d S )NF)verbose)inverse_normalizerinverse_normalize)r   r   r   predr   r   r   test_denorm$   s   zTestCardinal.test_denormN)__name__
__module____qualname__r   r   r   r   expandr   pytestmarkrun_only_onunitr   r   r   r   r   r   r   r   r      s    

r   )
r"   r   Anemo_text_processing.inverse_text_normalization.inverse_normalizer   1nemo_text_processing.text_normalization.normalizer   utilsr   r   r   r   r   r   r   <module>   s   