o
    }oi                     @   s    d dl Z d dlmZ dd ZdS )    N)BLEUc                  C   sr   t dd} | jdddgddgd | jddgd	gd |  }|d
 dks'J |d dks/J |d dks7J d S )NF)verbose	dataset_1za b c d e f g h i j k lzm n o p r s t u v)namerefshyps	dataset_2za b cza b dtxt_bleu_dataset_1g      Y@txt_bleu_dataset_2g        txt_bleug      I@)r   updatecompute)metricans r   \/home/ubuntu/.local/lib/python3.10/site-packages/tests/collections/speechlm2/test_metrics.py	test_bleu   s   
r   )pytest(nemo.collections.speechlm2.parts.metricsr   r   r   r   r   r   <module>   s   