From 18a7345a0b2514e26bef935da366b3cc631115be Mon Sep 17 00:00:00 2001
From: mo khan <mo.khan@gmail.com>
Date: Sun, 16 Aug 2020 16:50:09 -0600
Subject: Add spec to ensure that the tree is balanced

---
 src/02/03/btree_test.c | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/02/03/btree_test.c b/src/02/03/btree_test.c
index f3697db..24e7ea9 100644
--- a/src/02/03/btree_test.c
+++ b/src/02/03/btree_test.c
@@ -87,8 +87,15 @@ Ensure(
   tree = btree_insert(tree, 4);
   tree = btree_insert(tree, 3);
 
-  assert_that(tree, is_not_equal_to(NULL));
   btree_inspect(tree);
+
+  assert_that(tree, is_not_equal_to(NULL));
+  assert_that(tree->data, is_equal_to(3));
+  assert_that(tree->left->data, is_equal_to(2));
+  assert_that(tree->left->left->data, is_equal_to(1));
+
+  assert_that(tree->right->data, is_equal_to(4));
+  assert_that(tree->right->right->data, is_equal_to(5));
 }
 
 TestSuite *binary_search_tree_tests() {
-- 
cgit v1.2.3