3
dE                 @   sB   d dl Z d dlmZ e jjdZdd Ze jjdddd ZdS )    N)Doczignore::DeprecationWarningc             C   s@   t | jdddgd}| jdks"t| | dd |D s<tdS )	z*Check that the default uk lemmatizer runs.abc)wordsZ	pymorphy3c             S   s   g | ]
}|j qS  )lemma).0tokenr   r   W/var/www/html/virt/lib64/python3.6/site-packages/spacy/tests/lang/uk/test_lemmatizer.py
<listcomp>   s    z&test_uk_lemmatizer.<locals>.<listcomp>N)r   vocabmodeAssertionError)Zuk_lemmatizerdocr   r   r   test_uk_lemmatizer   s    r   z
word,lemma   якийсь   розповідають   розповідати   розповіси   розповістиc             C   s8   | j dkstt| j|gd}| |d j|ks4td S )NZpymorphy3_lookup)r   r   )r   r   r   r   Zlemma_)Zuk_lookup_lemmatizerwordr   r   r   r   r   test_uk_lookup_lemmatizer   s    	r   r   r   r   r   r   r   )r   r   r   )	ZpytestZspacy.tokensr   markfilterwarningsZ
pytestmarkr   Zparametrizer   r   r   r   r   <module>   s     