Test util.get_model_meta instead of util.load_model

This commit is contained in:
Adriane Boyd 2020-06-03 12:10:12 +02:00
parent 75f08ad62d
commit a57bdeecac

View File

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