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