Draft tests for new Example class

spacy/tests/test_new_example.py

@@ -4,12 +4,8 @@ from spacy.tokens import Doc
 from spacy.vocab import Vocab
 
 
-@pytest.fixture
+def test_Example_init_requires_doc_objects():
-def vocab():
+    vocab = Vocab()
-    return Vocab()
-
-
-def test_Example_init_requires_doc_objects(vocab):
     with pytest.raises(TypeError):
         eg = Example(None, None)
     with pytest.raises(TypeError):
@@ -19,12 +15,50 @@ def test_Example_init_requires_doc_objects(vocab):
 
 
 
-def test_Example_from_dict(vocab):
+def test_Example_from_dict_basic():
     eg = Example.from_dict(
-        Doc(vocab, words=["hello", "world"]),
+        Doc(Vocab(), words=["hello", "world"]),
         {
             "words": ["hello", "world"]
         }
     )
     assert isinstance(eg.x, Doc)
     assert isinstance(eg.y, Doc)
+
+
+@pytest.mark.parametrize("annots", [
+    {"words": ["ice", "cream"], "tags": ["NN", "NN"]},
+])
+def test_Example_from_dict_with_tags(annots):
+    vocab = Vocab()
+    predicted = Doc(vocab, words=annots["words"])
+    eg = Example.from_dict(predicted, annots)
+    for i, token in enumerate(eg.reference):
+        assert token.tag_ == annots["tags"][i]
+
+
+"""
+def test_Example_from_dict_with_entities(vocab):
+    # TODO
+    pass
+
+def test_Example_from_dict_with_parse(vocab):
+    # TODO
+    pass
+
+def test_Example_from_dict_with_morphology(vocab):
+    # TODO
+    pass
+
+def test_Example_from_dict_with_sent_start(vocab):
+    # TODO
+    pass
+
+def test_Example_from_dict_with_cats(vocab):
+    # TODO
+    pass
+
+def test_Example_from_dict_with_links(vocab):
+    # TODO
+    pass
+"""