o
    iB                     @   s   d dl Z d dlmZ dd Zdg dg dg dd	d
gfdg dg dg dddgfdg dg dg dg dfgZe jdedd ZdS )    N)Docc                 C   sB   | d}t t t|j W d   dS 1 sw   Y  dS )zPTest that noun_chunks raises Value Error for 'sv' language if Doc is not parsed.u!   Studenten läste den bästa bokenN)pytestraises
ValueErrorlistnoun_chunks)sv_tokenizerdoc r
   X/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/sv/test_noun_chunks.pytest_noun_chunks_is_parsed_sv   s   "r   u   En student läste en bok)DETNOUNVERBr   r   )detnsubjROOTr   dobj)      r      r   z
En studentzen boku"   Studenten läste den bästa boken.)r   r   r   ADJr   PUNCT)r   r   r   amodr   punct)r   r   r   r   r   r   	Studentenu   den bästa bokenuI   De samvetslösa skurkarna hade stulit de största juvelerna på söndagen)
r   r   r   r   r   r   r   r   ADPr   )
r   r   r   auxrootr   r   r   casenmod)
r   r   r   r   r      r!   r   	   r   )u   De samvetslösa skurkarnau   de största juvelernau   på söndagenz(text,pos,deps,heads,expected_noun_chunksc                 C   s   | |}t |t |ksJ dd |D }t|j||||d}t|j}	t |	t |ks.J t|	D ]\}
}|j||
 ks?J q2d S )Nc                 S   s   g | ]}|j qS r
   )text).0tr
   r
   r   
<listcomp>,   s    z'test_sv_noun_chunks.<locals>.<listcomp>)wordsheadsdepspos)lenr   vocabr   r   	enumerater#   )r   r#   r*   r)   r(   expected_noun_chunkstokensr'   r	   r   inpr
   r
   r   test_sv_noun_chunks&   s   
r2   )r   spacy.tokensr   r   SV_NP_TEST_EXAMPLESmarkparametrizer2   r
   r
   r
   r   <module>   s4    	