o
    iI                     @   sB   d Z ddlZddlZddlmZ ddlmZ G dd dejZdS )z.
Tests for NIST translation evaluation metric
    N)find)corpus_nistc                   @   s   e Zd Zdd ZdS )TestNISTc              	   C   s*  t d}t d}t d}t|}tt| d  dd }W d    n1 s*w   Y  t|ddV}t|dd7}ttd	d
 |}ttdd
 |}	ttdd|D ]\}
}t	|	||
}t
|| dk skJ qWW d    n1 svw   Y  W d    d S W d    d S 1 sw   Y  d S )Nzmodels/wmt15_eval/ref.ruzmodels/wmt15_eval/google.ruz#models/wmt15_eval/mteval-13a.output   utf8)encodingc                 S   s   |   S Nsplitx r   ]/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/nltk/test/unit/translate/test_nist.py<lambda>   s    z-TestNIST.test_sentence_nist.<locals>.<lambda>c                 S   s
   |   gS r
   r   r   r   r   r   r      s   
 
   g?)r   openmapfloat	readlinesr   listzipranger   abs)selfref_filehyp_filemteval_output_file
mteval_finmteval_nist_scoresref_finhyp_fin
hypotheses
referencesimteval_nist	nltk_nistr   r   r   test_sentence_nist   s$   
 "zTestNIST.test_sentence_nistN)__name__
__module____qualname__r(   r   r   r   r   r      s    r   )	__doc__iounittest	nltk.datar   nltk.translate.nist_scorer   TestCaser   r   r   r   r   <module>   s    