From c833ebe1ad72154bbac3213832a50bae0caa84f6 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 9 Jun 2020 15:29:05 +0200 Subject: [PATCH] Start tests for new example class --- spacy/tests/test_new_example.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 spacy/tests/test_new_example.py diff --git a/spacy/tests/test_new_example.py b/spacy/tests/test_new_example.py new file mode 100644 index 000000000..c481ae932 --- /dev/null +++ b/spacy/tests/test_new_example.py @@ -0,0 +1,30 @@ +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)