def test_package_model(trained_rasa_model, parameters):
output_path = tempfile.mkdtemp()
train_path = unpack_model(trained_rasa_model)
model_path = _package_model(
_fingerprint(),
output_path,
train_path,
parameters["model_name"],
parameters["prefix"],
parameters["uncompress"],
)
assert os.path.exists(model_path)
file_name = os.path.basename(model_path)
if parameters["model_name"]:
assert parameters["model_name"] in file_name
if parameters["prefix"]:
assert parameters["prefix"] in file_name
if parameters["uncompress"]:
assert os.path.isdir(model_path)
assert not file_name.endswith(".tar.gz")
else:
assert file_name.endswith(".tar.gz")
def test_package_model(trained_rasa_model: Text, parameters: Dict):
output_path = tempfile.mkdtemp()
train_path = rasa.model.unpack_model(trained_rasa_model)
model_path = rasa.model.package_model(
_fingerprint(),
output_path,
train_path,
parameters["model_name"],
parameters["prefix"],
)
assert os.path.exists(model_path)
file_name = os.path.basename(model_path)
if parameters["model_name"]:
assert parameters["model_name"] in file_name
if parameters["prefix"]:
assert parameters["prefix"] in file_name
assert file_name.endswith(".tar.gz")