B
    .(b'                 @   s,   d dl mZ d dlmZ d dlZdd ZdS )    )unicode_literals)DocNc          	   C   s(  ddddddg}t jddgd	d
gddgddgddgddggdd}t| |d}||_ddddddg}dd |D |ksztddddddg}dd |D |kstdd |D }t|t|kst| F}|d df|d g}ddgddgd }	|j|d d!d"g||	d# W d Q R X dddd!d"ddg}d$d |D |ks@tdddd!d"ddg}d%d |D |ksjtd&d |D }
t|
t|kst|d'  |
d'  kst|d  |
d  kst|d  |
d  kst|d(  |
d)  kst|d)  |
d*  ks$td S )+NIZliveinZNewYorkrightnowg      ?g?g       @g @g      @g@g      @gffffff@g      @gffffff@g      @gffffff@f)Zdtype)wordsc             S   s   g | ]
}|j qS  )text).0tokenr
   r
   T/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue3540.py
<listcomp>   s    z"test_issue3540.<locals>.<listcomp>c             S   s   g | ]
}|j qS r
   )lemma_)r   r   r
   r
   r   r      s    c             S   s   g | ]
}|j qS r
   )vector)r   r   r
   r
   r   r      s             ZPROPNZpobjZcompound)POSZDEPZNewZYork)headsattrsc             S   s   g | ]
}|j qS r
   )r   )r   r   r
   r
   r   r   "   s    c             S   s   g | ]
}|j qS r
   )r   )r   r   r
   r
   r   r   %   s    c             S   s   g | ]
}|j qS r
   )r   )r   r   r
   r
   r   r   '   s    r            )	npZasarrayr   tensorAssertionErrorlenZ
retokenizesplittolist)Zen_vocabr	   r   docZ	gold_textZ
gold_lemmaZ	vectors_1Zretokenizerr   r   Z	vectors_2r
   r
   r   test_issue3540	   s6    &
$r"   )
__future__r   Zspacy.tokensr   numpyr   r"   r
   r
   r
   r   <module>   s   