mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 01:46:28 +03:00
Start adding tests for new pipeline management
This commit is contained in:
parent
e43530269c
commit
3a65a0c970
0
spacy/tests/pipeline/__init__.py
Normal file
0
spacy/tests/pipeline/__init__.py
Normal file
43
spacy/tests/pipeline/test_add_pipe.py
Normal file
43
spacy/tests/pipeline/test_add_pipe.py
Normal file
|
@ -0,0 +1,43 @@
|
|||
from __future__ import unicode_literals
|
||||
import pytest
|
||||
|
||||
from ... import language
|
||||
from ...language import Language
|
||||
|
||||
@pytest.fixture
|
||||
def nlp():
|
||||
return Language()
|
||||
|
||||
@pytest.fixture
|
||||
def name():
|
||||
return 'parser'
|
||||
|
||||
def new_pipe(doc):
|
||||
return doc
|
||||
|
||||
|
||||
def test_add_pipe_no_name(nlp):
|
||||
nlp.add_pipe(new_pipe)
|
||||
assert 'new_pipe' in nlp.pipe_names
|
||||
|
||||
def test_add_pipe_duplicate_name(nlp):
|
||||
nlp.add_pipe(new_pipe, name='duplicate_name')
|
||||
with pytest.raises(ValueError):
|
||||
nlp.add_pipe(new_pipe, name='duplicate_name')
|
||||
|
||||
|
||||
def test_add_pipe_first(nlp, name):
|
||||
nlp.add_pipe(new_pipe, name=name, first=True)
|
||||
assert nlp.pipeline[0][0] == name
|
||||
|
||||
|
||||
def test_add_pipe_last(nlp, name):
|
||||
nlp.add_pipe(lambda doc: doc, name='lambda_pipe')
|
||||
nlp.add_pipe(new_pipe, name=name, last=True)
|
||||
assert nlp.pipeline[0][0] != name
|
||||
assert nlp.pipeline[-1][0] == name
|
||||
|
||||
|
||||
def test_cant_add_pipe_first_and_last(nlp):
|
||||
with pytest.raises(ValueError):
|
||||
nlp.add_pipe(new_pipe, first=True, last=True)
|
Loading…
Reference in New Issue
Block a user