explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-29 01:13:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix test

Browse Source
This commit is contained in:
Matthew Honnibal 2020-09-14 21:21:13 +02:00
parent adf0bab23a
commit e8378b57bc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/tests/test_cli.py
View File

@ -137,7 +137,7 @@ def test_cli_converters_conllu2json_subtokens():
assert biluo_tags == ["O", "U-PER", "O", "O"] assert biluo_tags == ["O", "U-PER", "O", "O"]
def test_cli_converters_iob2json(en_vocab): def test_cli_converters_iob2json():
lines = [ lines = [
"I|O like|O London|I-GPE and|O New|B-GPE York|I-GPE City|I-GPE .|O", "I|O like|O London|I-GPE and|O New|B-GPE York|I-GPE City|I-GPE .|O",
"I|O like|O London|B-GPE and|O New|B-GPE York|I-GPE City|I-GPE .|O", "I|O like|O London|B-GPE and|O New|B-GPE York|I-GPE City|I-GPE .|O",
@ -145,7 +145,7 @@ def test_cli_converters_iob2json(en_vocab):
"I|PRP|O like|VBP|O London|NNP|B-GPE and|CC|O New|NNP|B-GPE York|NNP|I-GPE City|NNP|I-GPE .|.|O", "I|PRP|O like|VBP|O London|NNP|B-GPE and|CC|O New|NNP|B-GPE York|NNP|I-GPE City|NNP|I-GPE .|.|O",
] ]
input_data = "\n".join(lines) input_data = "\n".join(lines)
converted_docs = iob2docs(input_data, en_vocab, n_sents=10) converted_docs = iob2docs(input_data, n_sents=10)
assert len(converted_docs) == 1 assert len(converted_docs) == 1
converted = docs_to_json(converted_docs) converted = docs_to_json(converted_docs)
assert converted["id"] == 0 assert converted["id"] == 0