Tidy up and expect warning
# coding: utf-8
from __future__ import unicode_literals
+import pytest
from spacy.kb import KnowledgeBase
from spacy.util import ensure_path
-
from spacy.lang.en import English
-from spacy.tests.util import make_tempdir
+
+from ..tests.util import make_tempdir
def test_issue4674():
@@ -15,8 +16,11 @@ def test_issue4674():
vector1 = [0.9, 1.1, 1.01]
vector2 = [1.8, 2.25, 2.01]
+ with pytest.warns(UserWarning):
kb.set_entities(
- entity_list=["Q1", "Q1"], freq_list=[32, 111], vector_list=[vector1, vector2]
+ entity_list=["Q1", "Q1"],
+ freq_list=[32, 111],
+ vector_list=[vector1, vector2],
)
assert kb.get_size_entities() == 1
