
    h                         S SK r S SKrS SKJr  S SKJr  SSKJrJr  \R                  S 5       r
\R                  " 5       S 5       r\R                  R                  S5      S	 5       rS
 rS rS rS rS rS rS rS rg)    N)Doc)Vocab   )add_vecs_to_vocab
get_cosinec                      S/ SQ4S/ SQ4/$ )Napple   r      orange) r       c/home/james-whalen/.local/lib/python3.13/site-packages/spacy/tests/vocab_vectors/test_similarity.pyvectorsr   
   s    i 8\":;;r   c                     [        X5        U $ )N)r   )en_vocabr   s     r   vocabr      s    h(Or   i  c                     S/ SQ4S/ SQ4/n[        X5        Uu  u  p#u  pE[        XU/S9nUS   R                  US   5      US   R                  US   5      :X  d   eg)	zATest if indexing issue still occurs during Token-Token similarityar
   letter)         wordsr   r   N)r   r   
similarity)r   r   word1vec1word2vec2docs          r   test_issue2219r&      sp     Y(I!67Gh(%,"]eMU
hen
-Cq6SV$A(9(9#a&(AAAAr   c                 n   Uu  u  p#u  pEX   nX   nUR                   (       d   eUR                   (       d   eUR                  S:w  d   eUR                  S:w  d   eUR                  S   UR                  S   :w  a   UR                  S   UR                  S   :w  d   e[        UR	                  U5      [
        5      (       d   e[        R                  " UR	                  U5      [        X55      5      (       d   e[        R                  " UR	                  U5      UR	                  U5      5      (       d   eg )Nr   r   )	
has_vectorvector_normvector
isinstancer    floatnumpyiscloser   )r   r   r!   r"   r#   r$   lex1lex2s           r   test_vectors_similarity_LLr1      s    %,"]eMU<D<D??????q   q   ;;q>T[[^+A$++a.0PPPdood+U3333==.
40FGGGG==.0EFFFFr   c                    Uu  u  p#u  pE[        XU/S9nUS   R                  (       d   eUS   R                  (       d   eUS   R                  S:w  d   eUS   R                  S:w  d   eUS   R                  S   US   R                  S   :w  a&  US   R                  S   US   R                  S   :w  d   e[	        US   R                  US   5      [        5      (       d   e[        R                  " US   R                  US   5      [        X55      5      (       d   e[        R                  " US   R                  US   5      US   R                  US   5      5      (       d   eg )Nr   r   r   )
r   r(   r)   r*   r+   r    r,   r-   r.   r   r   r   r!   r"   r#   r$   r%   s          r   test_vectors_similarity_TTr4   -   sP   %,"]eMU
e5>
*Cq6q6q6"""q6"""q6==s1v}}Q//CFMM!4DAVWHX4XXXc!f''A/7777==Q**3q62Jt4JKKKK==Q**3q62CF4E4Ec!f4MNNNNr   c                     Uu  u  p#u  pE[        XU/S9n[        USS R                  USS 5      [        5      (       d   eUSS R                  USS 5      USS R                  USS 5      :X  d   eg )Nr   r   r   r   r   r+   r    r,   r3   s          r   test_vectors_similarity_SSr7   :   s    %,"]eMU
e5>
*Cc!Ah))#a(3U;;;;q8s1Qx(C!H,?,?Aa,IIIIr   c                     Uu  u  p#u  pE[        XU/S9n[        XU/S9n[        UR                  U5      [        5      (       d   eUR                  U5      UR                  U5      :X  d   eg )Nr   r6   )r   r   r!   r"   r#   r$   doc1doc2s           r   test_vectors_similarity_DDr;   A   sh    %,"]eMUuEN+DuEN+Ddood+U3333??4 DOOD$9999r   c                 (   Uu  u  p#u  pE[        XU/S9n[        UR                  US   5      [        5      (       d   e[        US   R                  U5      [        5      (       d   eUR                  US   5      US   R                  U5      :X  d   eg )Nr   r   r6   r3   s          r   test_vectors_similarity_TDr=   I   s    %,"]eMU
e5>
*CcnnSV,e4444c!f'',e4444>>#a&!SV%6%6s%;;;;r   c                 @   Uu  u  p#u  pE[        XU/S9n[        US S R                  US   5      [        5      (       d   e[        US   R                  US S 5      [        5      (       d   eUS S R                  US   5      US   R                  US S 5      :X  d   eg )Nr   r   r   r6   r3   s          r   test_vectors_similarity_TSr?   Q   s    %,"]eMU
e5>
*Cc"1g((Q0%8888c!f''BQ0%8888r7c!f%Q):):3r7)CCCCr   c                     Uu  u  p#u  pE[        XU/S9n[        UR                  US S 5      [        5      (       d   eUR                  US S 5      US S R                  U5      :X  d   eg )Nr   r   r6   r3   s          r   test_vectors_similarity_DSrA   Y   sn    %,"]eMU
e5>
*CcnnS!W-u5555>>#bq'"c"1g&8&8&====r   c                     [        5       n [        U SS/S9n[        U / SQS9n[        R                  " [        5         UR                  U5        S S S 5        [        R                  " [        5         UR                  US   5        S S S 5        [        R                  " [        5         UR                  US S 5        S S S 5        [        R                  " [        5         UR                  U5        S S S 5        [        R                  " [        5         US   R                  U5        S S S 5        [        R                  " [        5         US S R                  U5        S S S 5        g ! , (       d  f       GN = f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)Nr   br   )cder   r   )r   r   pytestwarnsUserWarningr    )r   r9   r:   s      r   "test_vectors_similarity_no_vectorsrJ   `   s   GEuS#J'DuO,D	k	" 
#	k	"Q  
#	k	"Ra! 
#	k	" 
#	k	"Q4  
#	k	"RaD! 
#	" 
#	"	"	"	"	"	"	"	"	"	"	"sG   E(1E:(FFF-
F>(
E7:
F
F
F*-
F;>
G)r-   rG   spacy.tokensr   spacy.vocabr   utilr   r   fixturer   r   markissuer&   r1   r4   r7   r;   r=   r?   rA   rJ   r   r   r   <module>rQ      s        0 < <  
 4B BG
OJ:<D>"r   