o
    iE                     @   sB   d dl Z d dlmZ e jdZdd Ze jdddd	 ZdS )
    N)Doczignore::DeprecationWarningc                 C   s>   t | jg dd}| jdksJ | | dd |D sJ dS )z*Check that the default uk lemmatizer runs.)abcwords	pymorphy3c                 S   s   g | ]}|j qS  )lemma).0tokenr	   r	   W/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/uk/test_lemmatizer.py
<listcomp>   s    z&test_uk_lemmatizer.<locals>.<listcomp>N)r   vocabmode)uk_lemmatizerdocr	   r	   r   test_uk_lemmatizer   s   r   z
word,lemma))   якийсьr   )u   розповідаютьu   розповідати)u   розповісиu   розповістиc                 C   s8   | j dksJ t| j|gd}| |d j|ksJ d S )Npymorphy3_lookupr   r   )r   r   r   lemma_)uk_lookup_lemmatizerwordr
   r   r	   r	   r   test_uk_lookup_lemmatizer   s   	r   )	pytestspacy.tokensr   markfilterwarnings
pytestmarkr   parametrizer   r	   r	   r	   r   <module>   s    