B
    .(b                 @   sf   d dl mZ d dlZddlmZmZ ejddgejddd	d
dgdd Zejj	dd Z
dS )    )unicode_literalsN   )get_docapply_transition_sequencetextzA test sentencepunct.!? c             C   s   |rddddgndddg}| || }t |jdd |D |d}|rRt|dksZnd	sZttt|jdkspttd
d |jD t|kstd S )N      r   c             S   s   g | ]
}|j qS  )r   ).0tr   r   K/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/lang/en/test_sbd.py
<listcomp>   s    z,test_en_sbd_single_punct.<locals>.<listcomp>)wordsheads   r   c             s   s   | ]}t |V  qd S )N)len)r   sentr   r   r   	<genexpr>   s    z+test_en_sbd_single_punct.<locals>.<genexpr>)r   vocabr   AssertionErrorlistsentssum)en_tokenizerr   r   r   tokensdocr   r   r   test_en_sbd_single_punct	   s    r"   c       	      C   s   d}ddddddddddg
}dddd	d
dddd	d
g
}dddddddddddddg}| |}t |jdd |D ||d}t||| tt|jdkstx |D ]}|jdks|jstqW dd |D ddddddddddg
kstd S )Nz*This is a sentence . This is another one .r   r   ZnsubjROOTZdetattrr   zL-nsubjSzL-detzR-attrDzR-punctzB-ROOTzL-attrc             S   s   g | ]
}|j qS r   )r   )r   r   r   r   r   r      s    z+test_en_sentence_breaks.<locals>.<listcomp>)r   r   depsr   c             S   s   g | ]}|j jqS r   )headi)r   tokenr   r   r   r   $   s    r         )	r   r   r   r   r   r   r   depZis_space)	r   Z	en_parserr   r   r)   Z
transitionr    r!   r,   r   r   r   test_en_sentence_breaks   s    
r0   )
__future__r   Zpytestutilr   r   markZparametrizer"   Zxfailr0   r   r   r   r   <module>   s
    
