o
    i\                     @   sr   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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 )zTest that noun_chunks raises Value Error for 'la' language if Doc is not parsed.
    To check this test, we're constructing a Doc
    with a new Vocab here and forcing is_parsed to 'False'
    to make sure the noun chunks don't run.
    zHaec est sententia.N)pytestraises
ValueErrorlistnoun_chunks)la_tokenizerdoc r
   X/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/la/test_noun_chunks.pytest_noun_chunks_is_parsed   s   "r   z"Haec narrantur a poetis de Perseo.)DETVERBADPNOUNr   PROPNPUNCT)z
nsubj:passROOTcaseoblr   r   punct)   r   r   r   poetisPerseoz'Perseus autem in sinu matris dormiebat.)r   ADVr   r   r   r   r   )nsubj	discourser   r   nmodr   r   )         r   r   r   r   Perseuszsinu matrisz(text,pos,deps,heads,expected_noun_chunksc                 C   s   | |}t |t |ksJ t|jdd |D dd t|D ||d}t|j}t |t |ks3J t|D ]\}	}
|
j||	 ksDJ q7d S )Nc                 S   s   g | ]}|j qS r
   )text).0tr
   r
   r   
<listcomp>,   s    z'test_la_noun_chunks.<locals>.<listcomp>c                 S   s   g | ]\}}|| qS r
   r
   )r&   iheadr
   r
   r   r(   -   s    )wordsheadsdepspos)lenr   vocab	enumerater   r   r%   )r   r%   r.   r-   r,   expected_noun_chunkstokensr	   r   r)   npr
   r
   r   test_la_noun_chunks#   s   
r5   )r   spacy.tokensr   r   LA_NP_TEST_EXAMPLESmarkparametrizer5   r
   r
   r
   r   <module>   s(    