mirror of
https://github.com/explosion/spaCy.git
synced 2025-05-30 10:43:18 +03:00
Switch from python 3 only TemporaryDirectory to pytest's tmpdir
This commit is contained in:
parent
f32c6630cb
commit
38021fbb00
|
@ -1,15 +1,15 @@
|
||||||
# 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 = [
|
||||||
{
|
{
|
||||||
|
|
Loading…
Reference in New Issue
Block a user