def test_mkdir_normal():
io.model_mkdir(save_dir)
assert os.path.isdir(save_dir) is True
assert os.path.isdir(os.path.join(save_dir, "estimator")) is True
shutil.rmtree(save_dir)
def test_model_loadobj_predictor_missing_param():
io.model_mkdir(save_dir)
err_msg = "Loading the predictor requires the dict from `param.pkl`."
with pytest.raises(RuntimeError, match=err_msg):
io.model_loadobj(save_dir, "predictor", None)
shutil.rmtree(save_dir)
def test_model_saveobj_type_unknown():
io.model_mkdir(save_dir)
err_msg = "Unknown object type: unknown."
with pytest.raises(ValueError, match=err_msg):
io.model_saveobj(save_dir, "unknown", None)
shutil.rmtree(save_dir)
def test_mkdir_already_exist():
os.mkdir(save_dir)
err_msg = "The directory to be created already exists ./tmp."
with pytest.raises(RuntimeError, match=err_msg):
io.model_mkdir(save_dir)
shutil.rmtree(save_dir)
def test_model_saveobj_layer_invalid_data_type():
io.model_mkdir(save_dir)
obj = ["layer_1", "layer_2"]
err_msg = "The layer to be saved should be in the form of dict."
with pytest.raises(RuntimeError, match=err_msg):
io.model_saveobj(save_dir, "layer", obj)
shutil.rmtree(save_dir)
def test_model_saveobj_layer_missing_estimator_dir():
io.model_mkdir(save_dir)
estimator_dir = os.path.join(save_dir, "estimator")
shutil.rmtree(estimator_dir)
obj = {1: "layer_1"}
with pytest.raises(RuntimeError) as execinfo:
io.model_saveobj(save_dir, "layer", obj)
assert "Cannot find the target directory" in str(execinfo.value)
shutil.rmtree(save_dir)
