pub(super) const GRAPH_LABEL_CKPT_STATE: &str = "STATE";
pub(super) fn graph_label_ckpt_dir(
data_dir: &std::path::Path,
core_id: usize,
) -> std::path::PathBuf {
data_dir
.join("graph-label-ckpt")
.join(format!("core-{core_id}"))
}
pub(super) fn graph_label_ckpt_state_path(ckpt_dir: &std::path::Path) -> std::path::PathBuf {
ckpt_dir.join(GRAPH_LABEL_CKPT_STATE)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn per_core_dirs_are_distinct() {
let base = std::path::Path::new("/data");
let d0 = graph_label_ckpt_dir(base, 0);
let d1 = graph_label_ckpt_dir(base, 1);
assert_ne!(d0, d1);
assert!(d0.to_str().expect("utf8 path").contains("core-0"));
assert!(d1.to_str().expect("utf8 path").contains("core-1"));
}
#[test]
fn state_path_lives_under_the_core_dir() {
let dir = graph_label_ckpt_dir(std::path::Path::new("/data"), 3);
assert_eq!(
graph_label_ckpt_state_path(&dir),
dir.join(GRAPH_LABEL_CKPT_STATE)
);
}
}