
    h                         S SK r S SKJr  S SKJr  S SKJr  S SKJr  S SK	J
r
  S SKJr  S SKJr  S	 rS /S
-  rS\S '   S\S'   S\S'   SS\04SS\04/rS rS rg)    N)assert_equal)util)
SENT_START)English)Language)make_tempdir)Examplec                      [        5       n U R                  S5      n[        R                  " [        5         UR                  S5        S S S 5        g ! , (       d  f       g = f)NsenterA)r   add_pipepytestraisesNotImplementedError	add_label)nlpr   s     Z/home/james-whalen/.local/lib/python3.13/site-packages/spacy/tests/pipeline/test_senter.pytest_label_typesr      s>    
*C\\(#F	*	+ 
,	+	+s   A
A         	   z4I like green eggs. Eat blue ham. I like purple eggs.sent_startsz;She likes purple eggs. They hate ham. You like yellow eggs.c            	      $  ^ [        5       n U R                  S5        / m[         H>  nTR                  [        R
                  " U R                  US   5      US   5      5        M@     U R                  5         U R                  U4S jS9  [        R                  " [        5         U R                  S S9  S S S 5        [        R                  " [        5         U R                  TS9  S S S 5        g ! , (       d  f       NA= f! , (       d  f       g = f)Nr   r   r   c                     > T $ N )train_exampless   r   <lambda>*test_initialize_examples.<locals>.<lambda>,   s        )get_examplesc                      g r   r   r   r!   r   r   r    .   s    Dr!   )r   r   
TRAIN_DATAappendr	   	from_dictmake_doc
initializer   r   	TypeError)r   tr   s     @r   test_initialize_examplesr+   $   s    
*CLLNg//QqT0BAaDIJ  NNNN 6N7	y	!L1 
"	y	!N3 
"	! 
"	!	!	!s   $C0D0
C>
Dc            	         [        5       n / n[         H>  nUR                  [        R                  " U R                  US   5      US   5      5        M@     SUS   R                  S   l        SUS   R                  S   l        SUS   R                  S   l        U R                  S5        U R                  5       n[        S5       H  n0 nU R                  XUS9  M     WS   S	:  d   e[        S   S   nU " U5      nS/S
-  nSUS'   SUS'   SUS'   U Vs/ s H  n[        UR                  5      PM     snU:X  d   e[        5        n	U R                  U	5        [        R                   " U	5      n
U
" U5      nU Vs/ s H  n[        UR                  5      PM     snU:X  d   e S S S 5        / SQnU R#                  U5       Vs/ s H  owR%                  [&        /5      PM     nnU R#                  U5       Vs/ s H  owR%                  [&        /5      PM     nnU Vs/ s H
  o" U5      PM     sn Vs/ s H  owR%                  [&        /5      PM     nn[)        X5        [)        UU5        U R+                  S5      R,                  S:X  d   eSU R.                  ;  d   eg s  snf s  snf ! , (       d  f       GN= fs  snf s  snf s  snf s  snf )Nr   r   F
      r      )sgdlossesgMbP?r   r   r   )zJust a sentence.z$Then one more sentence about London.zHere is another one.zI like London.)IS)r   r$   r%   r	   r&   r'   	referenceis_sent_startr   r(   rangeupdateintr   to_diskr   load_model_from_pathpipeto_arrayr   r   get_pipelabelspipe_labels)r   r   r*   	optimizerir1   	test_textdocgold_sent_startstmp_dirnlp2doc2textsbatch_deps_1batch_deps_2textno_batch_depss                    r   test_overfitting_IOrM   3   s   
)CNg//QqT0BAaDIJ  5:N1#138N1"049N1#1LL I3Z

>
@  (e### 1a I
i.CsRxQQQ*-.#QC #.2BBBB 
7G((1I.23dAOO$d37GGGG	 
E ;>((5/J/3LL*./LJ:=((5/J/3LL*./LJEJ.KUTs4yU.K.Ksj\".K   ,}- <<!((J6663??***5 / 4	 
 KJ.Ks<   J 	4J*=J%	J*J<9KK4K%J**
J9)r   numpy.testingr   spacyr   spacy.attrsr   spacy.lang.enr   spacy.languager   spacy.tests.utilr   spacy.trainingr	   r   SENT_STARTSr$   r+   rM   r   r!   r   <module>rV      s     &  " ! # ) " cBhAAA 	?	$
 	F	$	
44+r!   