From 3fbc6f9398d6f2719091d96adb6bf624d3229649 Mon Sep 17 00:00:00 2001
From: David Mehren <dmehren1@gmail.com>
Date: Fri, 22 Jun 2018 09:19:26 +0200
Subject: [PATCH] [LOUDSTree/BPTree] Proper tests for fmt()

---
 src/trees/bp_tree.rs    | 13 ++++++++++++-
 src/trees/louds_tree.rs |  3 ++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/src/trees/bp_tree.rs b/src/trees/bp_tree.rs
index 9481cc8..f35255f 100644
--- a/src/trees/bp_tree.rs
+++ b/src/trees/bp_tree.rs
@@ -105,7 +105,7 @@ impl SuccinctTree<BPTree> for BPTree {
 
 impl Debug for BPTree {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
-        self.bits.fmt(f)
+        write!(f, "BPTree\n  {{ bits: {:?} }}", self.bits)
     }
 }
 
@@ -416,4 +416,15 @@ mod tests {
         let mut id_tree: Tree<i32> = TreeBuilder::new().with_node_capacity(5).build();
         assert_eq!(BPTree::from_id_tree(id_tree).unwrap_err(), EmptyTreeError)
     }
+
+    #[test]
+    fn print() {
+        let bitvec = bit_vec!(true, true, false, true, false, false);
+        let tree = BPTree::from_bitvec(bitvec.clone()).unwrap();
+        let str = format!("{:?}", tree);
+        assert_eq!(
+            str,
+            "BPTree\n  { bits: bit_vec![true, true, false, true, false, false] }"
+        )
+    }
 }
diff --git a/src/trees/louds_tree.rs b/src/trees/louds_tree.rs
index fccf53b..72efe09 100644
--- a/src/trees/louds_tree.rs
+++ b/src/trees/louds_tree.rs
@@ -262,7 +262,8 @@ mod tests {
         let bitvec =
             bit_vec![true, true, true, true, false, true, false, true, false, false, false, false];
         let tree = LOUDSTree::from_bitvec(bitvec.clone()).unwrap();
-        println!("{:?}", tree)
+        let str = format!("{:?}", tree);
+        assert_eq!(str, "LOUDSTree\n  { bits: bit_vec![true, true, true, true, false, true, false, true, false, false, false, false] }")
     }
 
     #[test]
-- 
GitLab