mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-21 11:29:13 +03:00
Test util.get_model_meta instead of util.load_model
This commit is contained in:
parent
75f08ad62d
commit
a57bdeecac
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user