mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-27 10:26:35 +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