B
    .(äb  ã               @   sJ   d dl mZ d dlZd dlmZmZ d dlmZ ejdd„ ƒZ	dd„ Z
dS )	é    )Úunicode_literalsN)ÚDocÚToken)ÚVocabc             C   s`   t jddd t jddd„ d t jdd	d„ d
 | dƒ}d|j_tjddd d|d j_|S )NÚ
_test_attrF)ÚdefaultÚ
_test_propc             S   s
   t | jƒS )N)ÚlenÚtext)Údoc© r   úc/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/serialize/test_serialize_extension_attrs.pyÚ<lambda>   ó    zdoc_w_attrs.<locals>.<lambda>)ÚgetterÚ_test_methodc             S   s   d  t| jƒ|¡S )Nz{}{})Úformatr	   r
   )r   Úargr   r   r   r      r   )ÚmethodzThis is a test.ÚtestÚ_test_tokenÚt0Út1é   )r   Zset_extensionÚ_r   r   r   )Zen_tokenizerr   r   r   r   Údoc_w_attrs	   s    r   c             C   s°   |   ¡ }ttƒ ƒ |¡}|j d¡s(t‚|jjdks8t‚|jjt	|j
ƒksNt‚|j d¡d t	|j
ƒd¡kspt‚|d jjdks„t‚|d jjdks˜t‚|d jjdks¬t‚d S )	Nr   r   z{}{}r   r   r   r   é   )Úto_bytesr   r   Ú
from_bytesr   ZhasÚAssertionErrorr   r   r	   r
   r   r   r   )r   Zdoc_br   r   r   r   Ú#test_serialize_ext_attrs_from_bytes   s    "r    )Ú
__future__r   ZpytestZspacy.tokensr   r   Zspacy.vocabr   Zfixturer   r    r   r   r   r   Ú<module>   s
   