explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 17:36:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix formatting on serializer tests

Browse Source
This commit is contained in:
Matthew Honnibal 2016-05-02 16:07:21 +02:00
parent 6e1f1c4b9e
commit c1c11a8ae0
1 changed files with 38 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
spacy/tests/serialize/test_serialization.py
View File

@ -14,7 +14,6 @@ def equal(doc1, doc2):
# parse # parse
assert [ t.head.i for t in doc1 ] == [ t.head.i for t in doc2 ] assert [ t.head.i for t in doc1 ] == [ t.head.i for t in doc2 ]
assert [ t.dep for t in doc1 ] == [ t.dep for t in doc2 ] assert [ t.dep for t in doc1 ] == [ t.dep for t in doc2 ]
if doc1.is_parsed and doc2.is_parsed: if doc1.is_parsed and doc2.is_parsed:
assert [ s for s in doc1.sents ] == [ s for s in doc2.sents ] assert [ s for s in doc1.sents ] == [ s for s in doc2.sents ]
@ -26,63 +25,62 @@ def equal(doc1, doc2):
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens(EN): def test_serialize_tokens(EN):
doc1 = EN(u'This is a test sentence.',tag=False,parse=False,entity=False) doc1 = EN(u'This is a test sentence.',tag=False, parse=False, entity=False)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_tags(EN): def test_serialize_tokens_tags(EN):
doc1 = EN(u'This is a test sentence.',tag=True,parse=False,entity=False) doc1 = EN(u'This is a test sentence.',tag=True, parse=False, entity=False)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_parse(EN): def test_serialize_tokens_parse(EN):
doc1 = EN(u'This is a test sentence.',tag=False,parse=True,entity=False) doc1 = EN(u'This is a test sentence.',tag=False, parse=True, entity=False)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_ner(EN): def test_serialize_tokens_ner(EN):
doc1 = EN(u'This is a test sentence.',tag=False,parse=False,entity=True) doc1 = EN(u'This is a test sentence.', tag=False, parse=False, entity=True)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_tags_parse(EN): def test_serialize_tokens_tags_parse(EN):
doc1 = EN(u'This is a test sentence.',tag=True,parse=True,entity=True) doc1 = EN(u'This is a test sentence.', tag=True, parse=True, entity=False)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_tags_ner(EN): def test_serialize_tokens_tags_ner(EN):
doc1 = EN(u'This is a test sentence.',tag=True,parse=False,entity=True) doc1 = EN(u'This is a test sentence.', tag=True, parse=False, entity=True)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_ner_parse(EN): def test_serialize_tokens_ner_parse(EN):
doc1 = EN(u'This is a test sentence.',tag=False,parse=True,entity=True) doc1 = EN(u'This is a test sentence.', tag=False, parse=True, entity=True)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)
@pytest.mark.models @pytest.mark.models
def test_serialize_tokens_tags_parse_ner(EN): def test_serialize_tokens_tags_parse_ner(EN):
doc1 = EN(u'This is a test sentence.',tag=True,parse=True,entity=True) doc1 = EN(u'This is a test sentence.', tag=True, parse=True, entity=True)
doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes()) doc2 = Doc(EN.vocab).from_bytes(doc1.to_bytes())
equal(doc1,doc2) equal(doc1, doc2)