explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 01:46:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Simplify underscore tests

Browse Source
This commit is contained in:
ines 2018-04-28 23:30:33 +02:00
parent 6fb6371670
commit abdb853ebf
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/tests/test_underscore.py
View File

@ -5,7 +5,7 @@ import pytest
from mock import Mock
from ..vocab import Vocab
from ..tokens.doc import Doc
from ..tokens import Doc, Span, Token
from ..tokens.underscore import Underscore
@ -60,10 +60,7 @@ def test_token_underscore_method():
assert token._.hello() == 'cheese'
@pytest.mark.parametrize('obj', [
Doc(Vocab(), words=['hello', 'world']),
Doc(Vocab(), words=['hello', 'world'])[1],
Doc(Vocab(), words=['hello', 'world'])[0:2]])
@pytest.mark.parametrize('obj', [Doc, Span, Token])
def test_underscore_raises_for_dup(obj):
obj.set_extension('test', default=None)
with pytest.raises(ValueError):
@ -78,9 +75,8 @@ def test_underscore_raises_for_dup(obj):
{'getter': True}])
def test_underscore_raises_for_invalid(invalid_kwargs):
invalid_kwargs['force'] = True
doc = Doc(Vocab(), words=['hello', 'world'])
with pytest.raises(ValueError):
doc.set_extension('test', **invalid_kwargs)
Doc.set_extension('test', **invalid_kwargs)
@pytest.mark.parametrize('valid_kwargs', [
@ -91,5 +87,4 @@ def test_underscore_raises_for_invalid(invalid_kwargs):
{'method': lambda: None}])
def test_underscore_accepts_valid(valid_kwargs):
valid_kwargs['force'] = True
doc = Doc(Vocab(), words=['hello', 'world'])
doc.set_extension('test', **valid_kwargs)
Doc.set_extension('test', **valid_kwargs)