mirror of
https://github.com/explosion/spaCy.git
synced 2025-02-06 06:30:35 +03:00
31 lines
721 B
Python
31 lines
721 B
Python
import pytest
|
|
from spacy.gold.new_example import NewExample as Example
|
|
from spacy.tokens import Doc
|
|
from spacy.vocab import Vocab
|
|
|
|
|
|
@pytest.fixture
|
|
def vocab():
|
|
return Vocab()
|
|
|
|
|
|
def test_Example_init_requires_doc_objects(vocab):
|
|
with pytest.raises(TypeError):
|
|
eg = Example(None, None)
|
|
with pytest.raises(TypeError):
|
|
eg = Example(Doc(vocab, words=["hi"]), None)
|
|
with pytest.raises(TypeError):
|
|
eg = Example(None, Doc(vocab, words=["hi"]))
|
|
|
|
|
|
|
|
def test_Example_from_dict(vocab):
|
|
eg = Example.from_dict(
|
|
Doc(vocab, words=["hello", "world"]),
|
|
{
|
|
"words": ["hello", "world"]
|
|
}
|
|
)
|
|
assert isinstance(eg.x, Doc)
|
|
assert isinstance(eg.y, Doc)
|