B
    .(b                  @   s   d dl mZ d dlZd dlZd dlmZ ddlmZmZ ej	dd Z
e	 dd	 Zd
d Zdd Zdd Zdd Zdd ZdS )    )unicode_literalsN)Doc   )
get_cosineadd_vecs_to_vocabc               C   s   ddddgfddddgfgS )	NZapple   r      Zorange r   r   r   X/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/vocab_vectors/test_similarity.pyvectors   s    r   c             C   s   t | | | S )N)r   )Zen_vocabr   r   r   r   vocab   s    
r   c             C   s   |\\}}\}}| | }| | }|j s*t|j s4t|jdksBt|jdksPt|jd |jd krx|jd |jd ks|tt||t||stt||||std S )Nr   r   )
has_vectorAssertionErrorvector_normvectornumpyisclose
similarityr   )r   r   word1vec1word2vec2Zlex1Zlex2r   r   r   test_vectors_similarity_LL   s    

,r   c             C   s   |\\}}\}}t | ||gd}|d js.t|d js<t|d jdksNt|d jdks`t|d jd |d jd kr|d jd |d jd kstt|d |d t||stt|d |d |d |d std S )N)wordsr   r   )	r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   docr   r   r   test_vectors_similarity_TT#   s    <$r   c          	   C   sZ   |\\}}\}}t | ||gd}tt& ||d |d |ksLtW d Q R X d S )N)r   r   )r   pytestwarnsUserWarningr   r   )r   r   r   r   r   r   r   r   r   r   test_vectors_similarity_TD/   s    r"   c             C   sL   |\\}}\}}t | ||gd}||d d |d d |ksHtd S )N)r   r   )r   r   r   )r   r   r   r   r   r   r   r   r   r   test_vectors_similarity_DS6   s    r#   c          	   C   sj   |\\}}\}}t | ||gd}tt6 |d d |d |d |d d ks\tW d Q R X d S )N)r   r   r   )r   r   r    r!   r   r   )r   r   r   r   r   r   r   r   r   r   test_vectors_similarity_TS<   s    r$   )
__future__r   r   r   Zspacy.tokensr   utilr   r   Zfixturer   r   r   r   r"   r#   r$   r   r   r   r   <module>   s   