B
    .(b)                 @   s   d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ d d	lmZ ejd
d Zejdd Zdd Zdd Zdd Zdd ZdS )    )unicode_literalsN)Adam)NumpyOps)NORM)	GoldParse)Vocab)Doc)DependencyParserc               C   s   t tdd idS )Nc             S   s   | S )N )sr
   r
   Q/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/parser/test_preset_sbd.py<lambda>       zvocab.<locals>.<lambda>)Zlex_attr_getters)r   r   r
   r
   r
   r   vocab   s    r   c             C   s   t | }d|jd< d|jd< |d |jg f|j tt d}x\tdD ]P}i }t| dd	d
dgd}t|ddddgddddgd}|j	|g|g||d qLW |S )N   Ztoken_vector_width    Zhidden_widthleftgMbP?
   abcd)words      ROOT)Zheadsdeps)sgdlosses)
r	   cfgZ	add_labelZbegin_trainingr   r   ranger   r   update)r   parserr   ir   docZgoldr
   r
   r   r"      s    


r"   c             C   s8   t | jddddgd}| |}tt|jdks4td S )Nr   r   r   r   )r   r   )r   r   lenlistsentsAssertionError)r"   r$   r
   r
   r   test_no_sentences%   s    r)   c             C   s   t | jddddgd}d|d _| |}tt|jdks>tt | jddddgd}d|d	 _d|d _d|d
 _| |}tt|jdkstd S )Nr   r   r   r   )r   T   Fr   r   )r   r   
sent_startr%   r&   r'   r(   )r"   r$   r
   r
   r   test_sents_1+   s    



r,   c             C   sL   t | jddddgd}d|d _d|d _| |}tt|jd	ksHtd S )
Nr   r   r   r   )r   Tr   r*   r   )r   r   r+   r%   r&   r'   r(   )r"   r$   r
   r
   r   test_sents_1_28   s
    

r-   c             C   s   t | jddddgd}d|d _d|d _| |}tt|jdksHtt | jddddgd}d|d _d	|d
 _d|d _| |}tt|jdkstd S )Nr   r   r   r   )r   Tr   r   Fr*   )r   r   r+   r%   r&   r'   r(   )r"   r$   r
   r
   r   test_sents_1_3@   s    




r.   )
__future__r   ZpytestZthinc.neural.optimizersr   Zthinc.neural.opsr   Zspacy.attrsr   Z
spacy.goldr   Zspacy.vocabr   Zspacy.tokensr   Zspacy.pipeliner	   Zfixturer   r"   r)   r,   r-   r.   r
   r
   r
   r   <module>   s   