o
    h                     @   sH   d dl Z d dlmZ d dlmZ d dlmZ e jje jj	gZ
dd ZdS )    N)Pipeline)CoNLL)TEST_MODELS_DIRc                  C   s   t dtddd} | d}dd |jD }|g dksJ d	|}tj|d
}t dtdddd} | | dd |jD }|g dksDJ dS )z
    If the doc does not have the entire text available, make sure it still safely processes the text

    Test case supplied from user - see issue #1428
    enztokenize,nerN)dir
processorsdownload_methodzDIn February, I traveled to Seattle.  Dr. Pritchett gave me a new hipc                 S      g | ]}|j qS  text.0xr
   r
   \/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/ner/test_from_conllu.py
<listcomp>       z$test_from_conllu.<locals>.<listcomp>)FebruarySeattle	Pritchettz{:C}

)	input_strT)r   r   tokenize_pretokenizedr   c                 S   r	   r
   r   r   r
   r
   r   r      r   )r   r   entsformatr   	conll2doc)pipedocr   
doc_conllur
   r
   r   test_from_conllu	   s   
r   )pyteststanzar   stanza.utils.conllr   stanza.testsr   markpipelinetravis
pytestmarkr   r
   r
   r
   r   <module>   s    