o
    oi0                     @   sB   d dl 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)
Normalizer   )	CACHE_DIRparse_test_case_filec                   @   sr   e Zd ZeddedddZeede	j
de	j
jdd	 Zeed
e	j
de	j
jdd ZdS )TestPunctuationcasedenFT)
input_caselang	cache_diroverwrite_cachepost_processz5en/data_text_normalization/test_cases_punctuation.txtCPUc                 C   s2   | j j|ddd}||ksJ d| d| d S )NFverbosepunct_post_processzinput: z != normalizer_en	normalizeself
test_inputexpectedpred r   b/home/ubuntu/.local/lib/python3.10/site-packages/tests/nemo_text_processing/en/test_punctuation.py	test_norm   s    zTestPunctuation.test_normzAen/data_text_normalization/test_cases_punctuation_match_input.txtc                 C   s:   | j j|ddd}||ksJ d| d| d| dd S )NTr   zfor input |z
|: pred: |z| != expected: ||r   r   r   r   r   #test_norm_python_punct_post_process$   s   (z3TestPunctuation.test_norm_python_punct_post_processN)__name__
__module____qualname__r   r   r   r   expandr   pytestmarkrun_only_onunitr   r   r   r   r   r   r      s    


r   )r$   r   1nemo_text_processing.text_normalization.normalizer   utilsr   r   r   r   r   r   r   <module>   s
   