B
    .(äb<  ã               @   sZ   d dl mZ d dlZd dlmZ d dlmZ ddlmZ ej	dd„ ƒZ
d	d
„ Zdd„ ZdS )é    )Úunicode_literalsN)ÚLanguage)ÚSpané   )Úget_docc             C   sž   d}dddddddg}dd	d
d
d	d
dg}dddddddg}dddddddg}| |ƒ}t |jdd„ |D ƒ||||d}t|dd|jjd ƒg|_d|_d|_|S )NzI like New York in Autumn.é   r   éþÿÿÿéýÿÿÿéÿÿÿÿéûÿÿÿZPRPÚINZNNPÚ.ZPRONZVERBZPROPNZADPZPUNCTÚROOTÚprepZcompoundZpobjÚpunctc             S   s   g | ]
}|j ‘qS © )Útext)Ú.0Útr   r   úR/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/pipeline/test_factories.pyú
<listcomp>   s    zdoc.<locals>.<listcomp>)ÚwordsÚheadsÚtagsÚposÚdepsr   é   ZGPET)r   Zvocabr   ÚstringsÚentsZ	is_parsedZ	is_tagged)Zen_tokenizerr   r   r   r   r   ÚtokensÚdocr   r   r   r       s"    r    c             C   sN   t | ƒdkst‚tƒ }| d¡}|| ƒ t | ƒdks8t‚| d jdksJt‚d S )Né   Úmerge_noun_chunksé   r   zNew York)ÚlenÚAssertionErrorr   Úcreate_piper   )r    Únlpr"   r   r   r   Ú test_factories_merge_noun_chunks!   s    
r(   c             C   sz   t | ƒdkst‚t t| jƒƒdks&t‚tƒ }| d¡}|| ƒ t | ƒdksNt‚t t| jƒƒdksdt‚| d jdksvt‚d S )Nr!   r   Úmerge_entitiesr#   r   zNew York)r$   r%   Úlistr   r   r&   r   )r    r'   r)   r   r   r   Útest_factories_merge_ents*   s    
r+   )Ú
__future__r   ZpytestZspacy.languager   Zspacy.tokensr   Úutilr   Zfixturer    r(   r+   r   r   r   r   Ú<module>   s   	