Switch from python 3 only TemporaryDirectory to pytest's tmpdir

This commit is contained in:
Isaac Sijaranamual 2017-12-11 00:16:04 +01:00
parent f32c6630cb
commit 38021fbb00

View File

@ -1,89 +1,89 @@
# coding: utf-8 # coding: utf-8
from __future__ import unicode_literals from __future__ import unicode_literals
import json import json
from tempfile import TemporaryDirectory, NamedTemporaryFile from tempfile import NamedTemporaryFile
from ...cli.train import train from ...cli.train import train
def test_cli_trained_model_can_be_saved(): def test_cli_trained_model_can_be_saved(tmpdir):
cmd = None cmd = None
lang = 'nl' lang = 'nl'
with TemporaryDirectory() as output_dir: output_dir = str(tmpdir)
train_file = NamedTemporaryFile('wb', dir=output_dir, delete=False) train_file = NamedTemporaryFile('wb', dir=output_dir, delete=False)
train_corpus = [ train_corpus = [
{ {
"id": "identifier_0", "id": "identifier_0",
"paragraphs": [ "paragraphs": [
{ {
"raw": "Jan houdt van Marie.\n", "raw": "Jan houdt van Marie.\n",
"sentences": [ "sentences": [
{ {
"tokens": [ "tokens": [
{ {
"id": 0, "id": 0,
"dep": "nsubj", "dep": "nsubj",
"head": 1, "head": 1,
"tag": "NOUN", "tag": "NOUN",
"orth": "Jan", "orth": "Jan",
"ner": "B-PER" "ner": "B-PER"
}, },
{ {
"id": 1, "id": 1,
"dep": "ROOT", "dep": "ROOT",
"head": 0, "head": 0,
"tag": "VERB", "tag": "VERB",
"orth": "houdt", "orth": "houdt",
"ner": "O" "ner": "O"
}, },
{ {
"id": 2, "id": 2,
"dep": "case", "dep": "case",
"head": 1, "head": 1,
"tag": "ADP", "tag": "ADP",
"orth": "van", "orth": "van",
"ner": "O" "ner": "O"
}, },
{ {
"id": 3, "id": 3,
"dep": "obj", "dep": "obj",
"head": -2, "head": -2,
"tag": "NOUN", "tag": "NOUN",
"orth": "Marie", "orth": "Marie",
"ner": "B-PER" "ner": "B-PER"
}, },
{ {
"id": 4, "id": 4,
"dep": "punct", "dep": "punct",
"head": -3, "head": -3,
"tag": "PUNCT", "tag": "PUNCT",
"orth": ".", "orth": ".",
"ner": "O" "ner": "O"
}, },
{ {
"id": 5, "id": 5,
"dep": "", "dep": "",
"head": -1, "head": -1,
"tag": "SPACE", "tag": "SPACE",
"orth": "\n", "orth": "\n",
"ner": "O" "ner": "O"
} }
], ],
"brackets": [] "brackets": []
} }
] ]
} }
] ]
} }
] ]
train_file.write(json.dumps(train_corpus).encode('utf-8')) train_file.write(json.dumps(train_corpus).encode('utf-8'))
train_file.close() train_file.close()
train_data = train_file.name train_data = train_file.name
dev_data = train_data dev_data = train_data
# spacy train -n 1 -g -1 nl output_nl training_corpus.json training \ # spacy train -n 1 -g -1 nl output_nl training_corpus.json training \
# corpus.json # corpus.json
train(cmd, lang, output_dir, train_data, dev_data, n_iter=1) train(cmd, lang, output_dir, train_data, dev_data, n_iter=1)
assert True assert True