o
    i                     @   s|   d dl Z d dlmZ ddlmZ e jddd Ze jdg d	ge jd
g ddd Z	e jj
dddd ZdS )    N)Doc   )apply_transition_sequencei5  c                 C   sD   t | dgdgdgd}t|dksJ t|j}t|dks J dS )z*Test Issue #309: SBD fails on empty string r   ROOTwordsheadsdeps   N)r   lenlistsents)en_vocabdocr    r   P/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/lang/en/test_sbd.pytest_issue309   s   
r   r   )Atestsentencepunct).!? c                 C   s   |rg dng d}dgt | }|rg ||n|}t| |||d}|r-t |dks.J 	 t t|jdks9J tdd	 |jD t |ksIJ d S )
N)   r   r   r   )r   r   r   depr      r   r   c                 s   s    | ]}t |V  qd S )N)r   ).0sentr   r   r   	<genexpr>   s    z+test_en_sbd_single_punct.<locals>.<genexpr>)r   r   r   r   sum)r   r   r   r	   r
   r   r   r   r   test_en_sbd_single_punct   s   $r#   z=The step_through API was removed (but should be brought back))reasonc                 C   s   g d}g d}g d}g d}t | |||d}t||| tt|jdks)J |D ]}|jdks7|js7J q+dd	 |D g dksEJ d S )
N)
Thisisar   r   r%   r&   anotheroner   )
r   r   r   r   r      r*      r*   r*   )
nsubjr   detattrr   r,   r   r-   r.   r   )L-nsubjSzL-detR-attrDR-punctzB-ROOTr/   r0   zL-attrr1   r2   r3   r   r   r   c                 S   s   g | ]}|j jqS r   )headi)r   tokenr   r   r   
<listcomp>.   s    z+test_en_sentence_breaks.<locals>.<listcomp>)r   r   r   r   r   r   is_space)r   	en_parserr   r	   r
   
transitionr   r6   r   r   r   test_en_sentence_breaks   s   r;   )pytestspacy.tokensr   utilr   markissuer   parametrizer#   skipr;   r   r   r   r   <module>   s    


