o
    –h0  ã                   @   s8   d Z ddlZddlmZ ddlmZmZmZ dd„ ZdS )z3
Test the utils file of the NER dataset processing
é    N)Úlist_doc_entities)ÚBIO_1ÚBIO_2Úwrite_and_convertc                 C   s°   t | tƒ}t|ƒ}dg}||ksJ ‚t | tƒ}t|ƒ}g d¢}||ks%J ‚t | d ttg¡ƒ}t|ƒ}g d¢}||ks=J ‚t | d tttg¡ƒ}t|ƒ}g d¢}||ksVJ ‚dS )zD
    Test the function which lists all of the entities in a doc
    ©)ÚJenniferzSh'reyanÚPERSON)©)r   r   ©)ÚBeckettr   ©)ÚCerritosÚLOCATIONz

)r   r	   r
   r   )r   r   r	   r
   r   N)r   r   r   r   Újoin)Útmp_pathÚdocÚentitiesÚexpected© r   ú_/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/datasets/ner/test_utils.pyÚtest_list_doc_entities
   s    

r   )	Ú__doc__ÚpytestÚstanza.utils.datasets.ner.utilsr   Ú/stanza.tests.datasets.ner.test_prepare_ner_filer   r   r   r   r   r   r   r   Ú<module>   s
    