From 5b285470db4315c581f1ab11ea4853d46f03a9de Mon Sep 17 00:00:00 2001 From: Leo Chen Date: Fri, 8 May 2020 13:10:40 +0800 Subject: [PATCH] Fix error message of load_inference_model (#24314) * fix error message, test=develop * add unittest, test=develop --- python/paddle/fluid/io.py | 2 +- .../fluid/tests/unittests/test_inference_model_io.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index 5a75331e1d..d7e194c24d 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -1371,7 +1371,7 @@ def load_inference_model(dirname, if dirname is not None: load_dirname = os.path.normpath(dirname) if not os.path.isdir(load_dirname): - raise ValueError("There is no directory named '%s'", dirname) + raise ValueError("There is no directory named '%s'" % dirname) if model_filename is None: model_filename = '__model__' diff --git a/python/paddle/fluid/tests/unittests/test_inference_model_io.py b/python/paddle/fluid/tests/unittests/test_inference_model_io.py index f9408dfee5..5d3d536100 100644 --- a/python/paddle/fluid/tests/unittests/test_inference_model_io.py +++ b/python/paddle/fluid/tests/unittests/test_inference_model_io.py @@ -192,5 +192,13 @@ class TestInstance(unittest.TestCase): [MODEL_DIR, ["x", "y"], [avg_cost], [], cp_prog]) +class TestLoadInferenceModelError(unittest.TestCase): + def test_load_model_not_exist(self): + place = core.CPUPlace() + exe = executor.Executor(place) + self.assertRaises(ValueError, load_inference_model, + './test_not_exist_dir', exe) + + if __name__ == '__main__': unittest.main() -- GitLab