
    hf                         S SK r S SKJr  S SKJr  S SKJr  \ R                  S 5       r\ R                  S 5       r	S r
S rS	 rS
 rS rS rg)    N)
StateClassDocVocabc                      [        5       $ )Nr        W/home/james-whalen/.local/lib/python3.13/site-packages/spacy/tests/parser/test_state.pyvocabr      s	    7Nr
   c                     [        U / SQS9$ )N)abcd)wordsr   )r   s    r   docr      s    u011r
   c                     [        U 5      nUR                  / :X  d   eUR                  [        [	        [        U 5      5      5      :X  d   eUR                  5       (       a   eUR                  5       S:X  d   eg )N   )r   stackqueuelistrangelenis_finalbuffer_lengthr   states     r   test_init_stater      sf    sOE;;";;$uSX////~~ A%%%r
   c                    [        U 5      nUR                  5         UR                  5       S:X  d   eUR                  S/:X  d   eSUR                  ;  d   eUR                  5         UR                  SS/:X  d   eSUR                  ;  d   eUR                  5       S:X  d   eUR                  5         UR                  S/:X  d   eSUR                  ;  d   eg )N   r         )r   pushr   r   r   popr   s     r   test_push_popr&      s    sOE	JJL A%%%;;1#EKK	JJL;;1a&   EKK A%%%	IIK;;1#EKKr
   c                     [        U 5      nUR                  5       S:X  d   eUR                  5       [        U 5      :X  d   eUR	                  5         UR                  5       S:X  d   eUR                  5       S:X  d   eg )Nr   r!   r"   )r   stack_depthr   r   r$   r   s     r   test_stack_depthr)   )   st    sOE!### CH,,,	JJL A%%%!###r
   c                    [        U 5      nUR                  S5      S:X  d   eUR                  SSS5        UR                  SSSS./:X  d   eUR                  S5      S:X  d   eUR                  SSS5        UR                  S5      S:X  d   eg )Nr   r"   headchildlabelr!   )r   Hadd_arcarcsr   s     r   test_Hr3   2   s    sOE771:	MM!Q::1q1=>>>>771:??	MM!Q771:??r
   c                 R   [        U 5      nUR                  SS5      S:X  d   eUR                  SSS5        UR                  SSSS./:X  d   eUR                  SS5      S:X  d   eUR                  SSS5        UR                  SS5      S:X  d   eUR	                  S5      S:X  d   eg )Nr#   r"   r+   r   r,   )r   Lr1   r2   n_Lr   s     r   test_Lr7   <       sOE771a=B	MM!Q::1q1=>>>>771a=A	MM!Q771a=A99Q<1r
   c                 R   [        U 5      nUR                  SS5      S:X  d   eUR                  SSS5        UR                  SSSS./:X  d   eUR                  SS5      S:X  d   eUR                  SSS5        UR                  SS5      S:X  d   eUR	                  S5      S:X  d   eg )Nr   r"   r+   r,   r#   )r   Rr1   r2   n_Rr   s     r   test_Rr<   G   r8   r
   )pytest+spacy.pipeline._parser_internals.stateclassr   spacy.tokens.docr   spacy.vocabr   fixturer   r   r   r&   r)   r3   r7   r<   r	   r
   r   <module>rB      s_     B      2 2& $r
   