o
    h
                     @   s   d Z ddlZddlZddlmZ ddlmZ ddlT ejj	Z
dZd Zd Zejd	d
dd Zdd Zdd Zdd ZdS )z5
Basic tests of the depparse processor boolean flags
    N)PipelineRequirementsException)CoNLL)*z\Barack Obama was born in Hawaii.  He was elected president in 2008.  Obama attended Harvard.aw  
1	Barack	Barack	PROPN	NNP	Number=Sing	0	_	_	_
2	Obama	Obama	PROPN	NNP	Number=Sing	1	_	_	_
3	was	be	AUX	VBD	Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin	2	_	_	_
4	born	bear	VERB	VBN	Tense=Past|VerbForm=Part|Voice=Pass	3	_	_	_
5	in	in	ADP	IN	_	4	_	_	_
6	Hawaii	Hawaii	PROPN	NNP	Number=Sing	5	_	_	_
7	.	.	PUNCT	.	_	6	_	_	_

1	He	he	PRON	PRP	Case=Nom|Gender=Masc|Number=Sing|Person=3|PronType=Prs	0	_	_	_
2	was	be	AUX	VBD	Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin	1	_	_	_
3	elected	elect	VERB	VBN	Tense=Past|VerbForm=Part|Voice=Pass	2	_	_	_
4	president	president	PROPN	NNP	Number=Sing	3	_	_	_
5	in	in	ADP	IN	_	4	_	_	_
6	2008	2008	NUM	CD	NumType=Card	5	_	_	_
7	.	.	PUNCT	.	_	6	_	_	_

1	Obama	Obama	PROPN	NNP	Number=Sing	0	_	_	_
2	attended	attend	VERB	VBD	Mood=Ind|Tense=Past|VerbForm=Fin	1	_	_	_
3	Harvard	Harvard	PROPN	NNP	Number=Sing	2	_	_	_
4	.	.	PUNCT	.	_	3	_	_	_


a  
('Barack', 4, 'nsubj:pass')
('Obama', 1, 'flat')
('was', 4, 'aux:pass')
('born', 0, 'root')
('in', 6, 'case')
('Hawaii', 4, 'obl')
('.', 4, 'punct')

('He', 3, 'nsubj:pass')
('was', 3, 'aux:pass')
('elected', 0, 'root')
('president', 3, 'xcomp')
('in', 6, 'case')
('2008', 3, 'obl')
('.', 3, 'punct')

('Obama', 2, 'nsubj')
('attended', 0, 'root')
('Harvard', 2, 'obj')
('.', 2, 'punct')
module)scopec                  C   s   t jtddd} | S )Nenztokenize,pos,lemma,depparse)dirlang
processors)stanzaPipelineTEST_MODELS_DIR)nlp r   ^/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/pipeline/test_depparse.pyen_depparse_pipeline@   s   r   c                 C   s*   | t }tddd |jD ksJ d S )N

c                 S      g | ]}|  qS r   dependencies_string.0sentr   r   r   
<listcomp>G       z!test_depparse.<locals>.<listcomp>)EN_DOCEN_DOC_DEPENDENCY_PARSES_GOLDjoin	sentences)r   docr   r   r   test_depparseE   s   "r    c                  C   sP   t jd	i dtddd} tjtd}| |}tddd |jD ks&J d S )
Ndepparser   T)r
   r   r	   depparse_pretagged)	input_strr   c                 S   r   r   r   r   r   r   r   r   R   r   z4test_depparse_with_pretagged_doc.<locals>.<listcomp>r   )	r   r   r   r   	conll2docEN_DOC_CONLLU_PRETAGGEDr   r   r   )r   r   processed_docr   r   r    test_depparse_with_pretagged_docJ   s   
r'   c                   C   sH   t t tjdi dtdd W d    d S 1 sw   Y  d S )Nr!   r   )r
   r   r	   r   )pytestraisesr   r   r   r   r   r   r   r   :test_raises_requirements_exception_if_pretagged_not_passedU   s   "r*   )__doc__r(   r   stanza.pipeline.corer   stanza.utils.conllr   stanza.testsmarkpipeline
pytestmarkr   lstripr%   stripr   fixturer   r    r'   r*   r   r   r   r   <module>   s&    

