Update test

This commit is contained in:
Matthew Honnibal 2020-06-22 14:29:52 +02:00
parent 792bf9476c
commit f73fa77bb9

View File

@ -54,7 +54,8 @@ def test_add_label(parser):
losses = {} losses = {}
doc = Doc(parser.vocab, words=["a", "b", "c", "d"]) doc = Doc(parser.vocab, words=["a", "b", "c", "d"])
gold = {"heads": [1, 1, 3, 3], "deps": ["right", "ROOT", "left", "ROOT"]} gold = {"heads": [1, 1, 3, 3], "deps": ["right", "ROOT", "left", "ROOT"]}
parser.update((doc, gold), sgd=sgd, losses=losses) example = Example.from_dict(doc, gold)
parser.update([example], sgd=sgd, losses=losses)
doc = Doc(parser.vocab, words=["a", "b", "c", "d"]) doc = Doc(parser.vocab, words=["a", "b", "c", "d"])
doc = parser(doc) doc = parser(doc)
assert doc[0].dep_ == "right" assert doc[0].dep_ == "right"