B
    .(b                 @   sB   d dl mZ d dlmZ d dlmZmZ G dd dZdd ZdS )	    )unicode_literals)English)SpanDocc               @   s,   e Zd ZdZdd Zdd Zedd ZdS )	
CustomPipeZmy_pipec             C   s"   t jd| jd tjdd d d S )Nmy_ext)getter)default)r   Zset_extension_get_my_extr   )self r   T/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue4903.py__init__   s    zCustomPipe.__init__c             C   sN   g }x0|j D ]&}| |}|jd| || qW |jdd| |S )Nr   
)Zsentsr
   _setappendjoin)r   docZgathered_extsentZsent_extr   r   r   __call__   s    
zCustomPipe.__call__c             C   s
   t | jS )N)strend)spanr   r   r   r
      s    zCustomPipe._get_my_extN)__name__
__module____qualname__namer   r   staticmethodr
   r   r   r   r   r      s   r   c              C   s   t  } t }| | d | j|dd dddg}t| j|dd}|d jdksXt|d	 jdksjt|d jdks|td S )
NZsentencizer)afterzI like bananas.zDo you like them?zNo, I prefer wasabi.   )Z	n_processr      )r   r   Zadd_pipeZcreate_pipelistpipetextAssertionError)ZnlpZcustom_componentr$   Zdocsr   r   r   test_issue4903   s    
r&   N)	
__future__r   Zspacy.lang.enr   Zspacy.tokensr   r   r   r&   r   r   r   r   <module>   s   