o
    if                     @   s|   d dl 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d Zdd ZdS )    N)
StateClassDocVocabc                   C   s   t  S )Nr    r   r   Q/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/parser/test_state.pyvocab   s   r	   c                 C   s   t | g ddS )N)abcd)wordsr   )r	   r   r   r   doc   s   r   c                 C   sP   t | }|jg ksJ |jttt| ksJ | rJ | dks&J d S )N   )r   stackqueuelistrangelenis_finalbuffer_lengthr   stater   r   r   test_init_state   s
   r   c                 C   s   t | }|  | dksJ |jdgksJ d|jvsJ |  |jddgks,J d|jvs3J | dks;J |  |jdgksGJ d|jvsNJ d S )N   r         )r   pushr   r   r   popr   r   r   r   test_push_pop   s   r    c                 C   sX   t | }| dksJ | t| ksJ |  | dks"J | dks*J d S )Nr   r   r   )r   stack_depthr   r   r   r   r   r   r   test_stack_depth)   s   r"   c                 C   sv   t | }|ddksJ |ddd |jddddgks J |ddks)J |ddd |ddks9J d S )Nr   r   headchildlabelr   )r   Hadd_arcarcsr   r   r   r   test_H2   s   r+   c                 C   s   t | }|dddksJ |ddd |jddddgks!J |dddks+J |ddd |dddks<J |ddksEJ d S )Nr   r   r#   r   r$   )r   Lr)   r*   n_Lr   r   r   r   test_L<      r.   c                 C   s   t | }|dddksJ |ddd |jddddgks!J |dddks+J |ddd |dddks<J |ddksEJ d S )Nr   r   r#   r$   r   )r   Rr)   r*   n_Rr   r   r   r   test_RG   r/   r2   )pytest+spacy.pipeline._parser_internals.stateclassr   spacy.tokens.docr   spacy.vocabr   fixturer	   r   r   r    r"   r+   r.   r2   r   r   r   r   <module>   s    

	
