B
    .(b                 @   sL   d dl mZ d dlZd dlmZ d dlmZmZmZ ddl	m
Z
 dd ZdS )	    )unicode_literalsN)Doc)DEPPOSTAG   )get_docc                s  ddddg}ddddg}dddd	g}d
dddg}ddddg}| j  x|D ]} | qHW  fdd|D } fdd|D } fdd|D }	tttg}
tjtt|||	dd}t	| ||d}|
|
| dd |D }t| ||||d}dd |D }||ks
td S )NZThisisaZsentenceZDETZVERBZNOUN  depZadjnnZatmZDTZVBZZNNc                s   g | ]}  |qS  )add).0d)stringsr   T/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue5048.py
<listcomp>   s    z"test_issue5048.<locals>.<listcomp>c                s   g | ]}  |qS r   )r   )r   p)r   r   r   r      s    c                s   g | ]}  |qS r   )r   )r   t)r   r   r   r      s    Zuint64)Zdtype)wordsspacesc             S   s   g | ]}|j |j|jfqS r   )textpos_tag_)r   tokenr   r   r   r      s    )r   posdepstagsc             S   s   g | ]}|j |j|jfqS r   )r   r   r   )r   r   r   r   r   r   "   s    )r   r   r   r   r   numpyarraylistzipr   Z
from_arrayr   AssertionError)Zen_vocabr   Zpos_sr   Zdeps_sZtags_swr   r   r    attrsr"   docZv1Zdoc2Zv2r   )r   r   test_issue5048   s&    

r)   )
__future__r   r!   Zspacy.tokensr   Zspacy.attrsr   r   r   utilr   r)   r   r   r   r   <module>   s
   