B
    .(äb´  ã               @   s8   d dl Z d dlmZ d dlmZ d dlmZ dd„ ZdS )é    N)ÚLanguage)ÚEnglish)Úmake_tempdirc           	      sŽ   G dd„ dt ƒ‰ ‡ fdd„tjd< tƒ } |  |  d¡¡ |  d¡jdksLt‚t	ƒ 2}|  
|¡ tj|dd}| d¡jdks€t‚W d Q R X d S )	Nc               @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )z#test_issue5137.<locals>.MyComponentÚmy_componentc             [   s   || _ | dd¡| _d S )NÚ
categoriesÚall_categories)ÚnlpÚgetr   )Úselfr   Úcfg© r   úT/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue5137.pyÚ__init__   s    z,test_issue5137.<locals>.MyComponent.__init__c             S   s   d S )Nr   )r
   Údocr   r   r   Ú__call__   s    z,test_issue5137.<locals>.MyComponent.__call__c             [   s   d S )Nr   )r
   ÚpathÚkwargsr   r   r   Úto_disk   s    z+test_issue5137.<locals>.MyComponent.to_diskc             [   s   d S )Nr   )r
   r   r   r   r   r   Ú	from_disk   s    z-test_issue5137.<locals>.MyComponent.from_diskN)Ú__name__Ú
__module__Ú__qualname__Únamer   r   r   r   r   r   r   r   ÚMyComponent   s
   r   c                s   ˆ | f|ŽS )Nr   )r   r   )r   r   r   Ú<lambda>   ó    z test_issue5137.<locals>.<lambda>r   r   Zmy_categories)r   )Úobjectr   Z	factoriesr   Zadd_pipeZcreate_pipeZget_piper   ÚAssertionErrorr   r   ÚspacyÚload)r   ZtmpdirZnlp2r   )r   r   Útest_issue5137   s    
r    )r   Zspacy.languager   Zspacy.lang.enr   Zspacy.tests.utilr   r    r   r   r   r   Ú<module>   s   