o
    iv                     @   sN   d dl Z d dlmZ d dlmZ d dlmZ g dZe j	dedd Z
dS )	    N)registry)Lookups)get_lang_class)	bncaelenfafrnbnlsvlangc                 C   s>  t ddd }t| }| }|jdddid}|jjrJ dd	dii|jd
 d d< tt	 |d W d    n1 s@w   Y  |
  |jjsOJ |d}| }|jr\J |d jdkseJ | }|jdddid}|j
| d |dd jdksJ dD ]}||\}	}
t|	tsJ t|
tsJ qd S )Nlemmatizer_init_lookupsc                  S   sV   t  } | dddd | dddi | ddd	d
ii | ddddggi | S )Nlemma_lookupcopey)r   xlemma_indexverb)r   cop	lemma_exccoping)r   lemma_rulesing )r   	add_tablelookups r   U/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/test_lemmatizers.pyr      s   z;test_lemmatizer_initialize.<locals>.lemmatizer_init_lookups
lemmatizermodelookup)configr   z@misc
initialize
componentsr   r   r   r   )ruler#   
pos_lookup)r   miscr   add_piper   tablesr$   pytestraises
ValueErrorr%   
readouterroutlemma_get_lookups_config
isinstancelist)r   capfdr   lang_clsnlpr!   doccapturedr"   requiredoptionalr   r   r    test_lemmatizer_initialize   s4   


r<   )r,   spacyr   spacy.lookupsr   
spacy.utilr   	LANGUAGESmarkparametrizer<   r   r   r   r    <module>   s    