B
    .(bC	                 @   s   d dl mZ d dlZddlmZ dd Zejddd	d
gfddgfddgfddgfddgfgdd Zejddddddddddidfddddidfdd d!g
d"d# Z	ejd$dd%d&id
fdd%d'id	fd(d%d&id(fd(d%d)id*fgd+d, Z
d-d. ZdS )/    )unicode_literalsN   )get_docc             C   sH   dddg}dddg}t | j||d}dd	 |D }|dd
dgksDtd S )Nu   мамаu   мылаu   рамуz2NOUN__Animacy=Anim|Case=Nom|Gender=Fem|Number=SingzRVERB__Aspect=Imp|Gender=Fem|Mood=Ind|Number=Sing|Tense=Past|VerbForm=Fin|Voice=Actz2NOUN__Animacy=Anim|Case=Acc|Gender=Fem|Number=Sing)wordstagsc             S   s   g | ]
}|j qS  )Zlemma_).0tokenr   r   R/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/lang/ru/test_lemmatizer.py
<listcomp>   s    z-test_ru_doc_lemmatization.<locals>.<listcomp>u   мытьu   рама)r   ZvocabAssertionError)Zru_tokenizerr   r   doclemmasr   r   r
   test_ru_doc_lemmatization	   s    
r   ztext,lemmasu   гвоздикиu   гвоздикu   гвоздикаu   людиu   человекu   рекиu   рекаu   кольцоu   пепперониc             C   s   t | ||kstd S )N)sortednounr   )ru_lemmatizertextr   r   r   r
   test_ru_lemmatizer_noun_lemmas   s    r   ztext,pos,morphology,lemma)u   ройNOUNNu   рой)u   ройVERBNu   рыть)u   клейr   Nu   клей)u   клейr   Nu   клеить)u   триZNUMNu   триu   косr   NumberZSingZPluru   коса)u   косZADJNu
   косой)u
   потомr   Nu   пот)u
   потомZADVNu
   потомc             C   s   | ||||gkst d S )N)r   )r   r   pos
morphologylemmar   r   r
   4test_ru_lemmatizer_works_with_different_pos_homonyms#   s    r   ztext,morphology,lemmaZGenderZFemZMascu   винаZNeutu   виноc             C   s   |  |||gkstd S )N)r   r   )r   r   r   r   r   r   r
   +test_ru_lemmatizer_works_with_noun_homonyms8   s    
r   c             C   s,   |  ddgkst|  ddgks(td S )N   «"   »)punctr   )r   r   r   r
   test_ru_lemmatizer_punctE   s    r!   )
__future__r   Zpytestutilr   r   markZparametrizer   r   r   r!   r   r   r   r
   <module>   s:   
	