
    h                     :    S SK Jr  S SKJr  S rS rS rS rS rg)	    )pickle)Languagec                      [        5       n U " S5      n[        R                  " US5      n[        R                  " U5      nUR                  S:X  d   eg )Nzpickle roundtrip   )r   r   dumpsloadstext)nlpdocdatadoc2s       Y/home/james-whalen/.local/lib/python3.13/site-packages/spacy/tests/doc/test_pickle_doc.pytest_pickle_single_docr      sE    
*C
 
!C<<QD<<D99****    c                     [        5       n [        S5       H  nU R                  [        U5         nM     [        R
                  " U " S5      S5      n[        U R                  S [        S5       5       5      5      n[        R
                  " US5      n[        U5      [        U5      S-  :  d   e[        R                  " U5      nUS   R                  S:X  d   eUS   R                  S:X  d   e[        U5      S:X  d   eg )	Ni'  0c              3   8   #    U  H  n[        U5      v   M     g 7f)N)str).0is     r   	<genexpr>8test_list_of_docs_pickles_efficiently.<locals>.<genexpr>   s     4AQs   d      r   99)r   rangevocabr   r   r   listpipelenr   r	   )r
   r   _one_pickleddocsmany_pickledmany_unpickleds          r   %test_list_of_docs_pickles_efficientlyr'      s    
*C5\IIc!f ,,s3x,K4s445D<<b)L|K 01 4555\\,/N!!!S((("""d***~#%%%r   c                      [        5       n U " S5      nSUR                  S'   UR                  5       nUR                  UR                  5      R                  U5      nUR                  S   SL d   eg NHelloF)r   r   )r   	user_datato_bytes	__class__r   
from_bytesr
   r   br   s       r   test_user_data_from_diskr1      s_    
*C
g,C!CMM&A==#..q1D>>&!U***r   c                      [        5       n U " S5      nSUR                  S'   [        R                  " U5      n[        R                  " U5      nUR                  S   SL d   eg r)   )r   r+   r   r   r   r/   s       r   test_user_data_unpicklesr3   $   sR    
*C
g,C!CMM&SA<<?D>>&!U***r   c                      S n [        5       nU" S5      nXR                  S'   [        R                  " U5      n[        R                  " U5      nUR                  S 5      S:X  d   eg )Nc                     g)Nhello! )d1d2s     r   
inner_func'test_hooks_unpickle.<locals>.inner_func.   s    r   r*   
similarityr6   )r   
user_hooksr   r   r   r<   )r:   r
   r   r0   r   s        r   test_hooks_unpickler>   -   sW     *C
g,C#-NN< SA<<?D??4 H,,,r   N)	spacy.compatr   spacy.languager   r   r'   r1   r3   r>   r7   r   r   <module>rA      s$     #+&++	-r   