diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/etc/libimagutil/src/tree.rs | 55 |
1 files changed, 38 insertions, 17 deletions
diff --git a/lib/etc/libimagutil/src/tree.rs b/lib/etc/libimagutil/src/tree.rs index 10b10874..c5284522 100644 --- a/lib/etc/libimagutil/src/tree.rs +++ b/lib/etc/libimagutil/src/tree.rs @@ -144,10 +144,31 @@ impl<'a, T, ID, IG> TreeBuilder<'a, T, ID, IG> } } +pub struct Traverse<'a, T: Sized>(&'a Tree<T>); + + #[cfg(test)] mod tests { use super::*; + struct IdGetterImpl; + impl IdGetter<usize> for IdGetterImpl { + type ID = usize; + + fn get_id_for_node(&self, node: &usize) -> Result<Self::ID> { + Ok(*node) + } + + fn get_id_for_parent_of(&self, node: &usize) -> Result<Option<Self::ID>> { + if *node > 1 { + Ok(Some(*node - 1)) + } else { + Ok(None) + } + } + } + + #[test] fn test_insertion() { let mut tree = Tree::new(); @@ -163,28 +184,28 @@ mod tests { let mut tree = Tree::new(); (1..100).for_each(|i| tree.add_node(i)); - struct IdGetterImpl; - impl IdGetter<usize> for IdGetterImpl { - type ID = usize; + let r = tree.build_tree::<usize, IdGetterImpl>(IdGetterImpl).build(); + assert!(r.is_ok()); + } - fn get_id_for_node(&self, node: &usize) -> Result<Self::ID> { - Ok(*node) - } - fn get_id_for_parent_of(&self, node: &usize) -> Result<Option<Self::ID>> { - if *node > 1 { - Ok(Some(*node - 1)) - } else { - Ok(None) - } - } - } + #[test] + fn test_linking_leaves_none_unlinked() { + let mut tree = Tree::new(); + (1..100).for_each(|i| tree.add_node(i)); + let r = tree.build_tree::<usize, IdGetterImpl>(IdGetterImpl) .build(); + assert!(r.is_ok()); - let r = tree.build_tree::<usize, IdGetterImpl>(IdGetterImpl) - .build(); + let c = tree.elements.iter() + .filter(|element| element.parent_idx.is_none()) + .count(); + assert_eq!(c, 1); - assert!(r.is_ok()); + let c = tree.elements.iter() + .filter(|element| element.childs_indexes.is_empty()) + .count(); + assert_eq!(c, 1); } } |