From bd9e2bddbf9f9c096d02008738ea8fd5b519272e Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Fri, 9 Dec 2022 14:38:04 +0100 Subject: [PATCH] Modify similarity tests to avoid spurious warnings --- spacy/tests/vocab_vectors/test_similarity.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/spacy/tests/vocab_vectors/test_similarity.py b/spacy/tests/vocab_vectors/test_similarity.py index b5f7303b5..91b2f5b05 100644 --- a/spacy/tests/vocab_vectors/test_similarity.py +++ b/spacy/tests/vocab_vectors/test_similarity.py @@ -7,7 +7,7 @@ from ..util import get_cosine, add_vecs_to_vocab @pytest.fixture def vectors(): - return [("apple", [1, 2, 3]), ("orange", [-1, -2, -3])] + return [("apple", [1, 2, 3]), ("orange", [-1, -2, -5])] @pytest.fixture() @@ -44,8 +44,7 @@ def test_vectors_similarity_TT(vocab, vectors): def test_vectors_similarity_TD(vocab, vectors): [(word1, vec1), (word2, vec2)] = vectors doc = Doc(vocab, words=[word1, word2]) - with pytest.warns(UserWarning): - assert doc.similarity(doc[0]) == doc[0].similarity(doc) + assert doc.similarity(doc[0]) == doc[0].similarity(doc) def test_vectors_similarity_DS(vocab, vectors): @@ -57,5 +56,4 @@ def test_vectors_similarity_DS(vocab, vectors): def test_vectors_similarity_TS(vocab, vectors): [(word1, vec1), (word2, vec2)] = vectors doc = Doc(vocab, words=[word1, word2]) - with pytest.warns(UserWarning): - assert doc[:2].similarity(doc[0]) == doc[0].similarity(doc[:2]) + assert doc[:2].similarity(doc[0]) == doc[0].similarity(doc[:2])