orbweaver 0.18.2

use orbweaver::prelude::*;

const MEDIUM_TXT_PATH: &str = "assets/medium.txt";

fn get_medium_graph() -> DirectedAcyclicGraph {
    use std::io::BufRead;
    let mut builder = DirectedGraphBuilder::new();
    std::io::BufReader::new(
        std::fs::File::open(MEDIUM_TXT_PATH).expect("Unable to read medium.txt"),
    )
    .lines()
    .map_while(Result::ok)
    .for_each(|l| {
        if let Some((parent, child)) = l.split_once('\t') {
            builder.add_edge(parent, child);
        }
    });

    builder.build_acyclic().expect("This should work")
}

#[test]
fn get_all_leaves_equivalent_to_leaves_under_root() {
    let dag = get_medium_graph();
    let all_leaves = dag.get_all_leaves();
    let leaves_under_root = dag
        .get_leaves_under(["1781f676dedf5767f3243db0a9738b35"])
        .unwrap();
    assert_eq!(all_leaves.len(), leaves_under_root.len());
}

#[test]
fn get_all_leaves_equivalent_to_leaves_under_root_after_subset() {
    let dag = get_medium_graph();
    let dag_s = dag.subset("1781f676dedf5767f3243db0a9738b35").unwrap();
    let leaves_under_root = dag_s
        .get_leaves_under(["1781f676dedf5767f3243db0a9738b35"])
        .unwrap();
    assert_eq!(dag_s.get_all_leaves().len(), leaves_under_root.len());
    assert_eq!(dag.get_all_leaves().len(), leaves_under_root.len());
}

#[test]
fn get_all_leaves_no_mistakes_after_subset() {
    let dag = get_medium_graph();
    let dag_s = dag.subset("1781f676dedf5767f3243db0a9738b35").unwrap();
    let leaves_under_a84 = dag_s
        .get_leaves_under(["a840e87f74917054b4a91f034349de6b"])
        .unwrap();
    assert_eq!(leaves_under_a84.len(), 1);
}