From 4ec89788cacfd89b55f516f6d8596bc05c71055c Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal@gmail.com>
Date: Mon, 14 Sep 2015 17:48:13 +1000
Subject: [PATCH] * Add tests for new vectors functionality

---
 tests/vectors/test_vectors.py | 109 ++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 tests/vectors/test_vectors.py

diff --git a/tests/vectors/test_vectors.py b/tests/vectors/test_vectors.py
new file mode 100644
index 000000000..3c7539e6a
--- /dev/null
+++ b/tests/vectors/test_vectors.py
@@ -0,0 +1,109 @@
+import pytest
+
+@pytest.mark.models
+def test_token_vector(EN):
+    token = EN(u'Apples and oranges')[0]
+    token.vector
+    token.vector_norm
+
+@pytest.mark.models
+def test_lexeme_vector(EN):
+    lexeme = EN.vocab[u'apples']
+    lexeme.vector
+    lexeme.vector_norm
+
+
+@pytest.mark.models
+def test_doc_vector(EN):
+    doc = EN(u'Apples and oranges')
+    doc.vector
+    doc.vector_norm
+
+@pytest.mark.models
+def test_span_vector(EN):
+    span = EN(u'Apples and oranges')[0:2]
+    span.vector
+    span.vector_norm
+
+@pytest.mark.models
+def test_token_token_similarity(EN):
+    apples, oranges = EN(u'apples oranges')
+    assert apples.similarity(oranges) == oranges.similarity(apples)
+    assert 0.0 < apples.similarity(oranges) < 1.0
+    
+
+@pytest.mark.models
+def test_token_lexeme_similarity(EN):
+    apples = EN(u'apples')
+    oranges = EN.vocab[u'oranges']
+    assert apples.similarity(oranges) == oranges.similarity(apples)
+    assert 0.0 < apples.similarity(oranges) < 1.0
+ 
+
+@pytest.mark.models
+def test_token_span_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = doc[0]
+    oranges = doc[1:3]
+    assert apples.similarity(oranges) == oranges.similarity(apples)
+    assert 0.0 < apples.similarity(oranges) < 1.0
+ 
+
+@pytest.mark.models
+def test_token_doc_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = doc[0]
+    assert apples.similarity(doc) == doc.similarity(apples)
+    assert 0.0 < apples.similarity(doc) < 1.0
+ 
+
+@pytest.mark.models
+def test_lexeme_span_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = EN.vocab[u'apples']
+    span = doc[1:3]
+    assert apples.similarity(span) == span.similarity(apples)
+    assert 0.0 < apples.similarity(span) < 1.0
+
+
+@pytest.mark.models
+def test_lexeme_lexeme_similarity(EN):
+    apples = EN.vocab[u'apples']
+    oranges = EN.vocab[u'oranges']
+    assert apples.similarity(oranges) == oranges.similarity(apples)
+    assert 0.0 < apples.similarity(oranges) < 1.0
+ 
+
+@pytest.mark.models
+def test_lexeme_doc_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = EN.vocab[u'apples']
+    assert apples.similarity(doc) == doc.similarity(apples)
+    assert 0.0 < apples.similarity(doc) < 1.0
+ 
+
+@pytest.mark.models
+def test_span_span_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = doc[0:2]
+    oj = doc[1:3]
+    assert apples.similarity(oj) == oj.similarity(apples)
+    assert 0.0 < apples.similarity(oj) < 1.0
+ 
+
+@pytest.mark.models
+def test_span_doc_similarity(EN):
+    doc = EN(u'apples orange juice')
+    apples = doc[0:2]
+    oj = doc[1:3]
+    assert apples.similarity(doc) == doc.similarity(apples)
+    assert 0.0 < apples.similarity(doc) < 1.0
+ 
+
+@pytest.mark.models
+def test_doc_doc_similarity(EN):
+    apples = EN(u'apples and apple pie')
+    oranges = EN(u'orange juice')
+    assert apples.similarity(oranges) == apples.similarity(oranges)
+    assert 0.0 < apples.similarity(oranges) < 1.0
+