B
    .(äb  ã               @   s   d dl mZ d dlZd dlmZ d dlmZ d dlm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dd„ Zdd„ ZdS )é    )Úunicode_literalsN)ÚVocab)ÚDoc)Ú
Lemmatizer)ÚLookups)Úutilc              C   s"   t ƒ } |  dddddœ¡ t| ƒS )NZlemma_lookupÚdogZboxZmouse)ÚdogsZboxenZmice)r   Z	add_tabler   )Zlookups© r
   úL/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/doc/test_creation.pyÚ
lemmatizer   s    r   c             C   s
   t | dS )N)r   )r   )r   r
   r
   r   Úvocab   s    r   c             C   s   t | ƒ}t|ƒdkst‚d S )Nr   )r   ÚlenÚAssertionError)r   Údocr
   r
   r   Útest_empty_doc   s    r   c             C   s@   t | dgd}|jdkst‚t | dgdgd}|jdks<t‚d S )NÚa)Úwordsza F)r   Úspaces)r   Útextr   )r   r   r
   r
   r   Útest_single_word   s    r   c             C   s\   t | ddgd}|d jdks"t‚|d jdks4t‚|d jdksFt‚|d jdksXt‚d S )Nr	   Zdogses)r   r   r   é   )r   r   r   Zlemma_)r   r   r
   r
   r   Útest_lookup_lemmatization$   s
    r   c          	   C   sL  ddddg}d}t  ||¡\}}t| ||d}dd„ |D ƒddddd	dd
gksRt‚dd„ |D ƒdddddd
dgksvt‚|j|ks„t‚dd„ |D ƒdd„ |D ƒks¤t‚ddddd	dd
g}d}t  ||¡\}}t| ||d}dd„ |D ƒddddd	dd
gksüt‚dd„ |D ƒdddddd
dgks"t‚|j|ks2t‚dd„ |D ƒdd„ |D ƒksTt‚d
d
dddd	dg}d}t  ||¡\}}t| ||d}dd„ |D ƒddddd	dd
gks®t‚dd„ |D ƒdddddd
dgksÔt‚|j|ksät‚dd„ |D ƒdd„ |D ƒkst‚t t¡2 d
d
dddd	dg}d}t  |dg |¡\}}W d Q R X d S )Nú'r	   Úrunz  'dogs'

run  )r   r   c             S   s   g | ]
}|j ‘qS r
   )r   )Ú.0Útr
   r
   r   ú
<listcomp>2   s    z3test_create_from_words_and_text.<locals>.<listcomp>z  z

ú c             S   s   g | ]
}|j ‘qS r
   )Úwhitespace_)r   r   r
   r
   r   r   3   s    Ú c             S   s   g | ]}|j  ¡ s|j ‘qS r
   )r   Úisspace)r   r   r
   r
   r   r   5   s    c             S   s   g | ]}|  ¡ s|‘qS r
   )r!   )r   Úwordr
   r
   r   r   6   s    c             S   s   g | ]
}|j ‘qS r
   )r   )r   r   r
   r
   r   r   >   s    c             S   s   g | ]
}|j ‘qS r
   )r   )r   r   r
   r
   r   r   ?   s    c             S   s   g | ]}|j  ¡ s|j ‘qS r
   )r   r!   )r   r   r
   r
   r   r   A   s    c             S   s   g | ]}|  ¡ s|‘qS r
   )r!   )r   r"   r
   r
   r   r   B   s    c             S   s   g | ]
}|j ‘qS r
   )r   )r   r   r
   r
   r   r   J   s    c             S   s   g | ]
}|j ‘qS r
   )r   )r   r   r
   r
   r   r   K   s    c             S   s   g | ]}|j  ¡ s|j ‘qS r
   )r   r!   )r   r   r
   r
   r   r   M   s    c             S   s   g | ]}|  ¡ s|‘qS r
   )r!   )r   r"   r
   r
   r   r   N   s    Zaway)r   Zget_words_and_spacesr   r   r   ÚpytestZraisesÚ
ValueError)r   r   r   r   r   r
   r
   r   Útest_create_from_words_and_text,   s>    $$$&&&r%   )Ú
__future__r   r#   Zspacy.vocabr   Zspacy.tokensr   Zspacy.lemmatizerr   Zspacy.lookupsr   Zspacyr   Zfixturer   r   r   r   r   r%   r
   r
   r
   r   Ú<module>   s   