o
    h$                     @   s   d Z ddlZddlZddlZddlmZ ddlmZ ejj	ejj
gZdZdd Zeddd	edd
dedddeddd	edddeddd	edddeddd
edddg	Zdd Zdd Zdd Zdd ZdS )zM
Test some of the functions used for converting an AMT json to a Stanza json
    N)convert_amt)TEST_MODELS_DIRz&Jennifer Sh'reyan has lovely antennae.c                 C   s   | ||dS )N)labelstartOffset	endOffset )r   
start_charend_charr   r   \/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/ner/test_convert_amt.py
fake_label   s   r   Person   	      Andorian	Appendage   %         &   c                  G   s   dd | D S )Nc                 S   s   g | ]}t | qS r   )LABELS.0xr   r   r
   
<listcomp>%       zfake_labels.<locals>.<listcomp>r   indicesr   r   r
   fake_labels$      r   c                  G   s   t t|  fgS )N)TEXTr   r   r   r   r
   	fake_docs'   r    r"   c                  C   sT  t tdd} | tddksJ t tddd} | tdddks$J t tdd} | tdks4J t tdd} | tdksDJ t tdd} | tdksTJ t tdd} | tdksdJ t tddd} | tddksvJ t tddd} | tddksJ t tdd} | tdksJ t tdd} | tdksJ dS )z\
    Test a few orders on nested items to make sure the desired results are coming back
    r   r               N)r   remove_nestingr"   )resultr   r   r
   test_remove_nesting*   s(   r)   c                  C   s   t jtdd d} dd }tttdd| }||ddd	d	d
d	 tttddd| }||ddd	d	d
d	 tttddd| }||ddd	d	d
d	 tttdd| }||dd	d	d	d
d	 tttdd| }||dd	d	d	d
d	 d S )Ntokenize)dir
processorsdownload_methodc                 W   s&   dd | d D }|t |ksJ d S )Nc                 S   s   g | ]}|d  qS )r   r   r   r   r   r
   r   Q   r   z;test_process_doc.<locals>.check_results.<locals>.<listcomp>r   )list)docexpectednerr   r   r
   check_resultsP   s   z'test_process_doc.<locals>.check_resultsr$   r#   zB-PersonzI-PersonOzB-Appendager%   r      r   r   )stanzaPipeliner   r   process_docr!   r   )nlpr2   r/   r   r   r
   test_process_docM   s   r9   )__doc__ospytestr5   stanza.utils.datasets.nerr   stanza.testsr   markpipelinetravis
pytestmarkr!   r   r   r   r"   r)   r9   r   r   r   r
   <module>   s.    








#