
    h                     d    S SK r S SKJr  S SKJr  S SKJr  S SKJrJ	r	  S SK
Jr  S rS rS	 rS
 rg)    N)registry)EntityRecognizer)DEFAULT_NER_MODEL)DocSpan)Examplec                 f    [        U R                  / SQS9nSSS/0n[        R                  " X5      $ )N)JoelovesvisitingLondonduringtheweekendwordsentities)r      PERSON)      LOC)r   vocabr   	from_dict)nerdocgolds      [/home/james-whalen/.local/lib/python3.13/site-packages/spacy/tests/doc/test_add_entities.py_ner_exampler   
   s9    
		PC )?;<DS''    c                 t  ^ / SQn[        XS9nS[        0n[        R                  " USS9S   n[	        X5      mTR                  U4S j5        T" U5        S/Ul        U Vs/ s H  oUR                  PM     sn/ SQ:X  d   eS	/Ul        U Vs/ s H  oUR                  PM     sn/ S
Q:X  d   eg s  snf s  snf )NThisisalionr   modelTvalidatec                     > [        T 5      /$ Nr   r   s   r   <lambda>4test_doc_add_entities_set_ents_iob.<locals>.<lambda>       L-.r    )ANIMALr      )Or3   r3   B)WORDr      )r4   Ir3   r3   )r   r   r   resolver   
initializeentsent_iob_)en_vocabtextr   cfgr'   wr   s         @r   "test_doc_add_entities_set_ents_iobr@      s    &D
h
#C%
&CS409E
8
+CNN./H !CH #$1JJ$(<<<<CH #$1JJ$(<<<< % %s   #B0B5c                 r  ^ / SQn[        XS9nS[        0n[        R                  " USS9S   n[	        X5      mTR                  U4S j5        T" U5        U Vs/ s H  oUR                  PM     nn[        UR                  5      Ul        U Vs/ s H  oUR                  PM     snU:X  d   egs  snf s  snf )z7Ensure that resetting doc.ents does not change anythingr"   r   r'   Tr(   c                     > [        T 5      /$ r+   r,   r-   s   r   r.   !test_ents_reset.<locals>.<lambda>*   r0   r    N)	r   r   r   r8   r   r9   r;   listr:   )r<   r=   r   r>   r'   t	orig_iobsr   s          @r   test_ents_resetrG   #   s    &D
h
#C%
&CS409E
8
+CNN./H%()SSI)CHH~CH #$1JJ$	111 *$s   B/B4c                    / SQn[        XS9n[        USSSS9nU/Ul        [        USSSS9n[        R                  " [
        5         [        UR                  5      U/-   Ul        S S S 5        g ! , (       d  f       g = f)	N)	LouisianaOfficeofConservationr   r   r2   i  )label   i  )r   r   r:   pytestraises
ValueErrorrD   )r<   r=   r   entity
new_entitys        r   test_add_overlapping_entitiesrT   1   sh    8D
h
#C#q!3'FxCHc1as+J	z	">ZL0 
#	"	"s   A00
A>)rO   spacyr   spacy.pipeliner   spacy.pipeline.nerr   spacy.tokensr   r   spacy.trainingr   r   r@   rG   rT    r    r   <module>r[      s+      + 0 " "(= 21r    