Skip to content

Commit 9b18aa1

Browse files
committed
add mermaid tests
1 parent 26e3344 commit 9b18aa1

2 files changed

Lines changed: 19 additions & 0 deletions

File tree

hugr-core/src/hugr/views/render.rs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -417,5 +417,16 @@ mod tests {
417417
{
418418
assert!(RenderConfig::try_from(config).is_err());
419419
}
420+
421+
#[allow(deprecated)]
422+
let config = {
423+
let mut config = RenderConfig::default();
424+
config.entrypoint = Some(h.entrypoint());
425+
config
426+
};
427+
assert_eq!(
428+
MermaidFormatter::from_render_config(config, &h),
429+
h.mermaid_format()
430+
)
420431
}
421432
}

hugr-core/src/hugr/views/rerooted.rs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,4 +233,12 @@ mod test {
233233
);
234234
assert!(extracted_hugr.get_optype(extracted_bb).is_dataflow_block());
235235
}
236+
237+
#[test]
238+
fn mermaid_format() {
239+
let h = simple_cfg_hugr();
240+
let rerooted = h.with_entrypoint(h.entrypoint());
241+
let mermaid_str = rerooted.mermaid_format().finish();
242+
assert_eq!(mermaid_str, h.mermaid_format().finish());
243+
}
236244
}

0 commit comments

Comments
 (0)