B
    .(b                 @   s8   d dl mZ d dlZd dlmZ d dlmZ dd ZdS )    )unicode_literalsN)English)EntityRulerc              C   s  t  } | j}tjdddgtjd}tjdddgtjd}tjdd	d
gtjd}tjdddgtjd}tjdddgtjd}|d| |d| |d| |d| d}t| }dddiddigdg}	||	 | | dd | |D }
t	|
dkst
tj|
d | tj|
d | tj|
d | tj|
d | | d}| | dd | |D }t	|dkst
tj|d | tj|d | tj|d | d S ) Ng?g      ?g?)Zdtypegɿg333333gg333333?ggffffff?r   g?gIlikeZDavidZBowiezI like David BowieZPERSONZLOWERZdavidZbowie)labelpatternc             S   s   g | ]
}|j qS  )vector).0tr	   r	   T/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue5082.py
<listcomp>    s    z"test_issue5082.<locals>.<listcomp>            Zmerge_entitiesc             S   s   g | ]
}|j qS r	   )r
   )r   r   r	   r	   r   r   *   s    )r   vocabnpZasarrayfloat32Z
set_vectorr   Zadd_patternsZadd_pipelenAssertionErrorZtestingZassert_array_equalZcreate_pipe)Znlpr   Zarray1Zarray2Zarray3Zarray4Zarray34textZrulerpatternsZparsed_vectors_1Z
merge_entsZparsed_vectors_2r	   r	   r   test_issue5082	   s:    



r   )	
__future__r   numpyr   Zspacy.lang.enr   Zspacy.pipeliner   r   r	   r	   r	   r   <module>   s   