|
|
@@ -137,6 +137,7 @@ impl<T: Ord> Node<T> {
|
|
|
#[cfg(test)]
|
|
|
mod tests {
|
|
|
use super::Node;
|
|
|
+ use std::cmp::max;
|
|
|
|
|
|
#[test]
|
|
|
fn is_sorted() {
|
|
|
@@ -178,4 +179,24 @@ mod tests {
|
|
|
}
|
|
|
traverse(&root);
|
|
|
}
|
|
|
+
|
|
|
+ #[test]
|
|
|
+ fn test_height() {
|
|
|
+ let values = vec![5, 3, 7, 1, 4, 6, 9];
|
|
|
+ let mut root = Node::new(values[0]);
|
|
|
+ for v in values[1..].iter() {
|
|
|
+ root.insert(*v);
|
|
|
+ }
|
|
|
+ fn traverse(node: &Node<i32>) -> i32 {
|
|
|
+ let lheight = node.left.as_ref().map_or(0, |left| traverse(left.as_ref()));
|
|
|
+ let rheight = node
|
|
|
+ .right
|
|
|
+ .as_ref()
|
|
|
+ .map_or(0, |right| traverse(right.as_ref()));
|
|
|
+ let height = max(lheight, rheight) + 1;
|
|
|
+ assert_eq!(node.height, height);
|
|
|
+ height
|
|
|
+ }
|
|
|
+ traverse(&root);
|
|
|
+ }
|
|
|
}
|
