forests 0.0.2

Forest types and operations in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

extern crate forests;
use forests::*;

#[test]
fn basic_forest_001() {
    let /*mut */ forest: Forest<usize> = Forest::new();

    let expected_output: Vec<(_, _)> = vec![];
    let actual_output: Vec<(_, _)> = forest.iter().collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());
}

#[test]
fn basic_forest_002() {
    let mut forest: Forest<usize> = Forest::new();
    forest.create_node(1usize);
    forest.create_node(2usize);
    forest.create_node(3usize);

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(0), &1usize),
        (IterMovement::Right, &2usize),
        (IterMovement::Right, &3usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());
}

#[test]
fn basic_forest_003() {
    let mut forest: Forest<usize> = Forest::new();
    forest.create_node(7usize);
    forest.create_node(8usize);
    forest.create_node(9usize);

    for (idx, r) in forest.iter_mut().values().enumerate() {
        *r = idx;
    }

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(0), &0usize),
        (IterMovement::Right, &1usize),
        (IterMovement::Right, &2usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());
}

#[test]
fn basic_forest_004() {
    let mut forest: Forest<usize> = Forest::new();
    let node1 = forest.create_node(1usize);
    let node2 = forest.create_node(2usize);
    let node3 = forest.create_node(3usize);
    let node4 = forest.create_node(4usize);
    let _node5 = forest.create_node(5usize);
    forest.append_node_child(node2, node1);
    forest.append_node_child(node2, node3);
    forest.prepend_node_child(node3, node4);

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(0), &2usize),
        (IterMovement::DownFirst(1), &1usize),
        (IterMovement::Right, &3usize),
        (IterMovement::DownFirst(1), &4usize),
        (IterMovement::UpNRight(2), &5usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(1), &1usize),
        (IterMovement::RightDownFirstN(1), &4usize),
        (IterMovement::Up(1), &3usize),
        (IterMovement::Up(1), &2usize),
        (IterMovement::Right, &5usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().mode(IterMode::PostOrder).collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());
}

#[test]
fn basic_forest_005() {
    let mut forest: Forest<usize> = Forest::new();
    let node1 = forest.create_node(1usize);
    let node2 = forest.create_node(2usize);
    let node3 = forest.create_node(3usize);
    let node4 = forest.create_node(4usize);
    let _node5 = forest.create_node(5usize);
    forest.append_node_child(node2, node1);
    forest.insert_node_child_before(node2, node3, node1);
    forest.insert_node_child_after(node2, node4, node1);

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(0), &2usize),
        (IterMovement::DownFirst(1), &3usize),
        (IterMovement::Right, &1usize),
        (IterMovement::Right, &4usize),
        (IterMovement::UpNRight(1), &5usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());

    let expected_output: Vec<(_, _)> = vec![
        (IterMovement::DownFirst(1), &3usize),
        (IterMovement::Right, &1usize),
        (IterMovement::Right, &4usize),
        (IterMovement::Up(1), &2usize),
        (IterMovement::Right, &5usize),
    ];
    let actual_output: Vec<(_, _)> = forest.iter().mode(IterMode::PostOrder).collect::<Vec<_>>();
    assert_eq!(expected_output.as_slice(), actual_output.as_slice());
}