def test_remove_model_invalid(empty_model_dir):
test_file = "something.else"
test_content = "Some other stuff"
test_file_path = os.path.join(empty_model_dir, test_file)
utils.write_to_file(test_file_path, test_content)
with pytest.raises(RasaException):
utils.remove_model(empty_model_dir)
os.remove(test_file_path)
def test_remove_model_invalid(empty_model_dir):
test_file = "something.else"
test_content = "Some other stuff"
test_file_path = os.path.join(empty_model_dir, test_file)
write_to_file(test_file_path, test_content)
with pytest.raises(ValueError) as e:
remove_model(empty_model_dir)
os.remove(test_file_path)
def test_remove_model_with_files(empty_model_dir):
metadata_file = "metadata.json"
metadata_content = {"pipeline": "pretrained_embeddings_spacy", "language": "en"}
metadata_path = os.path.join(empty_model_dir, metadata_file)
write_json_to_file(metadata_path, metadata_content)
fake_obj = {"Fake", "model"}
fake_obj_path = os.path.join(empty_model_dir, "component.pkl")
with io.open(fake_obj_path, "wb") as f:
pickle.dump(fake_obj, f)
assert remove_model(empty_model_dir)
def test_remove_model_empty(empty_model_dir):
assert remove_model(empty_model_dir)
def test_remove_model_with_files(fake_model_dir):
assert utils.remove_model(fake_model_dir)
