explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 11:29:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test util.get_model_meta instead of util.load_model

Browse Source
This commit is contained in:
Adriane Boyd 2020-06-03 12:10:12 +02:00
parent 75f08ad62d
commit a57bdeecac
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
spacy/tests/test_misc.py
View File

@ -4,6 +4,7 @@ from __future__ import unicode_literals
import pytest
import os
import ctypes
import srsly
from pathlib import Path
from spacy import util
from spacy import prefer_gpu, require_gpu
@ -158,21 +159,22 @@ def test_load_model_version_compat():
with make_tempdir() as d:
# no change: compatible
nlp.to_disk(d)
nlp2 = util.load_model(d)
meta_path = Path(d / "meta.json")
util.get_model_meta(d)
# additional compatible upper pin
nlp.meta["spacy_version"] = ">=2.3.0,<2.4.0"
nlp.to_disk(d)
nlp2 = util.load_model(d)
srsly.write_json(Path(d / "meta.json"), nlp.meta)
util.get_model_meta(d)
# incompatible older version
nlp.meta["spacy_version"] = ">=2.2.5"
nlp.to_disk(d)
srsly.write_json(Path(d / "meta.json"), nlp.meta)
with pytest.warns(UserWarning):
nlp_reloaded = util.load_model(d)
util.get_model_meta(d)
# invalid version specification
nlp.meta["spacy_version"] = ">@#$%_invalid_version"
nlp.to_disk(d)
srsly.write_json(Path(d / "meta.json"), nlp.meta)
with pytest.warns(UserWarning):
nlp_reloaded = util.load_model(d)
util.get_model_meta(d)