o
    i                     @   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 )zTest that noun_chunks raises Value Error for 'da' 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.
    u   Det er en sætningN)pytestraises
ValueErrorlistnoun_chunks)da_tokenizerdoc r
   X/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/da/test_noun_chunks.pytest_noun_chunks_is_parsed   s   "r   zHun elsker at plukker frugt.)PRONVERBPARTr   NOUNPUNCT)nsubjROOTmarkobjr   punct)   r   r   Hunfrugtu   Påfugle er de smukkeste fugle.)r   AUXDETADJr   r   )r   copdetamodr   r   )         r   r   r   u   Påfuglezde smukkeste fugleu9   Rikke og Jacob Jensen glæder sig til en hyggelig skovtur)
PROPNCCONJr&   r&   r   r   ADPr   r   r   )
r   ccconjflatr   r   caser!   r"   obl)
r#   r   r   r   r   r   r$   r%   r   )RikkezJacob Jensensigzen hyggelig skovturz(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_da_noun_chunks.<locals>.<listcomp>c                 S   s   g | ]\}}|| qS r
   r
   )r2   iheadr
   r
   r   r4   ?   s    )wordsheadsdepspos)lenr   vocab	enumerater   r   r1   )r   r1   r:   r9   r8   expected_noun_chunkstokensr	   r   r5   npr
   r
   r   test_da_noun_chunks5   s   
rA   )r   spacy.tokensr   r   DA_NP_TEST_EXAMPLESr   parametrizerA   r
   r
   r
   r   <module>   s4    $