def test_log_dict_with_invalid_format():
data = {"a": 0}
fmt = "abc"
path = "test.{}".format(fmt)
with mlflow.start_run():
with pytest.raises(ValueError, match="Invalid file format: {}.".format(fmt)):
lg.log_dict(data, path, fmt)
def test_log_dict(path: str) -> None:
data = {"a": 0}
with mlflow.start_run() as run:
lg.log_dict(data, path)
assert_file_exists_in_artifacts(run, path)
artifacts_dir = run.info.artifact_uri.replace("file://", "")
loaded_data = _read_data(os.path.join(artifacts_dir, path))
assert loaded_data == data
def test_log_dict_with_fmt(fmt: str) -> None:
data = {"a": 0}
path = "test.{}".format(fmt.lstrip("."))
with mlflow.start_run() as run:
lg.log_dict(data, path, fmt)
assert_file_exists_in_artifacts(run, path)
artifacts_dir = run.info.artifact_uri.replace("file://", "")
loaded_data = _read_data(os.path.join(artifacts_dir, path))
assert loaded_data == data