o
    i                     @   sd   d dl Z d dlmZ d dlmZ e jdd Ze jdd Zdd	 Zd
d Z	e j
ddd ZdS )    NDoc)filter_spansc                 C   s2   g d}g d}g d}g d}t | ||||dS )N)&Haarvriendlachtluid.Wekregenalweerruzietoenwede
supermarktingingenr	   Aanhetbeginvanr   r   isalr   fruitenr   groentesr	   Uiteindelijkhebbenr   danookgeen	avondetengekochtr	   )&      r$   r$   r$      r%   r%   r%      r&      r&   r%   r%      r(         r*   r(   r)   r)   r)   r)      r+   r)   r)   $   r,   r,   r,   r,   #   r,   r,   r,   )&z	nmod:possnsubjROOTadvmodpunctr.   r/   r0   objmarkr.   detr2   advclr1   caser4   oblr6   r4   nmodcopr0   r4   r/   ccr4   conjr1   r0   auxr.   r0   r0   r4   r2   r/   r1   )&PRONNOUNVERBADJPUNCTr=   r?   ADVr>   SCONJr=   DETr>   r>   rA   ADPrD   r>   rE   rD   r>   AUXrB   rD   r>   CCONJrD   r>   rA   r@   rF   r=   rB   rB   rD   r>   r?   rA   )wordsheadsdepsposr   )nl_vocabrH   rI   rJ   rK    rM   X/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/nl/test_noun_chunks.py	nl_sample   s
   ((((rO   c                   C   s   g dS )N)zhaar vriendr   r   r   de supermarktz	het beginrP   z	het fruitzde groentesr   zgeen avondetenrM   rM   rM   rM   rN   nl_reference_chunking   s   rQ   c                 C   sF   d}| |}t t t|j W d   dS 1 sw   Y  dS )zZ
    Test that noun_chunks raises Value Error for 'nl' language if Doc is not parsed.
    zHaar vriend lacht luid.N)pytestraises
ValueErrorlistnoun_chunks)nl_tokenizertxtdocrM   rM   rN   test_need_dep   s
   "rZ   c                 C   s    dd | j D }||ksJ dS )z
    Test the noun chunks of a sample text. Uses a sample.
    The sample text simulates a Doc object as would be produced by nl_core_news_md.
    c                 S   s   g | ]}|j  qS rM   )textlower).0srM   rM   rN   
<listcomp>   s    z!test_chunking.<locals>.<listcomp>N)rV   )rO   rQ   chunksrM   rM   rN   test_chunking   s   ra   i^*  c                 C   s@   t | g dg dg dg dd}t|j}t||ksJ d S )N)	Dit	programmawordt	beschouwdalsz'swereldseerstecomputerprogramma)	r4   z
nsubj:passzaux:passr/   r3   r4   fixedamodxcomp)	r#      rm   rm      rn      rn   rm   )	rD   r>   rF   r?   rC   rD   r>   r@   r>   )rH   rJ   rI   rK   )r   rU   rV   r   )rL   rY   r`   rM   rM   rN   test_no_overlapping_chunks   s   
rp   )rR   spacy.tokensr   
spacy.utilr   fixturerO   rQ   rZ   ra   r3   issuerp   rM   rM   rM   rN   <module>   s    
 )

	