B
    .(bp                 @   s|   d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl	m
Z
 ejddgdd	 Zejdd
gdd ZdS )    )unicode_literalsN)StringStore)Vocab)NORMztext1,text2)ZhelloZbyec             C   sf   t  }||  }|| }tj|dd}t|}||  |ks>t|| |ksNtt|t|ksbtd S )N)protocol)r   srslypickle_dumpspickle_loadsAssertionErrorlen)text1text2ZstringstoreZstore1Zstore2data	unpickled r   G/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/test_pickles.pytest_pickle_string_store   s    
r   )dogcatc             C   s  t ttdd id}|dtjddd ||  }|| }|j| d d ksRt|j|d d kshtt	|}t
|}||  j|jkst|| j|jkst||  j|jkst|| j|jkst||  j|| jkst|jd k	stt|d jd	d	d	d	d	gkstd S )
Nc             S   s   | d d S )Nr   r   )stringr   r   r   <lambda>       z#test_pickle_vocab.<locals>.<lambda>)Zlex_attr_gettersr   )   f)Zdtyper   g      ?)r   intr   Z
set_vectornumpyZonesZnorm_r   r   r	   r
   ZorthZnormZvectorslistZvector)r   r   ZvocabZlex1Zlex2r   r   r   r   r   test_pickle_vocab   s    

r   )
__future__r   Zpytestr   r   Zspacy.stringsr   Zspacy.vocabr   Zspacy.attrsr   markZparametrizer   r   r   r   r   r   <module>   s   