o
    hx                     @   sV   d dl Z d dlZd dlT d dlmZ d dlm  mZ e jj	gZ
dddZdd ZdS )	    N)*)Documentc                    sH    fdd| j D }t||ksJ |dkr |d j|ks"J d S d S )Nc                    s$   g | ]}|j  kr|jkr|qS  sourcetarget).0edger   r   _/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/server/test_ud_enhancer.py
<listcomp>   s   $ zcheck_edges.<locals>.<listcomp>   r   )r	   lenisExtra)graphr   r   numr   edgesr   r   r
   check_edges
   s   r   c                  C   s   t jtdd} | d}tj|dd d}t|jdksJ |jd }|j}t|jdks-J t|j	d	ks6J t
|d
ddd t
|dd
d |j}t|jdksQJ t|j	dksZJ t
|d
ddd t
|dd
dd d S )Nztokenize,pos,lemma,depparse)dir
processorszThis is the car that I boughten)languagepronouns_patternr   r            FT)stanzaPipelineTEST_MODELS_DIRud_enhancerprocess_docr   sentencebasicDependenciesnoder	   r   enhancedDependencies)nlpdocresultr    basicenhancedr   r   r
   test_one_sentence   s   
r)   )N)pytestr   stanza.testsstanza.models.common.docr   stanza.server.ud_enhancerserverr   markpipeline
pytestmarkr   r)   r   r   r   r
   <module>   s    

