explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-03 21:24:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

issue5230: added unittest test case for completion

Browse Source
This commit is contained in:
lfiedler 2020-04-06 21:36:02 +02:00
parent b63871ceff
commit e1e25c7e30
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
spacy/tests/regression/test_issue5230.py
View File

@ -1,5 +1,6 @@
# coding: utf8 # coding: utf8
import warnings import warnings
from unittest import TestCase
import pytest import pytest
import srsly import srsly
@ -80,14 +81,31 @@ def entity_linker():
return entity_linker return entity_linker
@pytest.mark.parametrize( objects_to_test = (
"obj",
[nlp(), vectors(), custom_pipe(), tagger(), entity_linker()], [nlp(), vectors(), custom_pipe(), tagger(), entity_linker()],
ids=["nlp", "vectors", "custom_pipe", "tagger", "entity_linker"], ["nlp", "vectors", "custom_pipe", "tagger", "entity_linker"],
) )
def test_to_disk_resource_warning(obj):
def write_obj_and_catch_warnings(obj):
with make_tempdir() as d: with make_tempdir() as d:
with warnings.catch_warnings(record=True) as warnings_list: with warnings.catch_warnings(record=True) as warnings_list:
warnings.filterwarnings("always", category=ResourceWarning) warnings.filterwarnings("always", category=ResourceWarning)
obj.to_disk(d) obj.to_disk(d)
assert len(warnings_list) == 0 return warnings_list
@pytest.mark.parametrize("obj", objects_to_test[0], ids=objects_to_test[1])
def test_to_disk_resource_warning(obj):
warnings_list = write_obj_and_catch_warnings(obj)
assert len(warnings_list) == 0
class TestToDiskResourceWarningUnittest(TestCase):
def test_resource_warning(self):
scenarios = zip(*objects_to_test)
for scenario in scenarios:
with self.subTest(msg=scenario[1]):
warnings_list = write_obj_and_catch_warnings(scenario[0])
self.assertEqual(len(warnings_list), 0)