From e04668c20d4ef26ba01f4e34ae264079c47f38ba Mon Sep 17 00:00:00 2001
From: Emir Pasic <emirpasic@hotmail.com>
Date: Thu, 5 Mar 2015 05:20:39 +0100
Subject: [PATCH] - adding Size() method to the redblacktree (tests)

---
 trees/redblacktree/redblacktree_test.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/trees/redblacktree/redblacktree_test.go b/trees/redblacktree/redblacktree_test.go
index 07472d6..2685768 100644
--- a/trees/redblacktree/redblacktree_test.go
+++ b/trees/redblacktree/redblacktree_test.go
@@ -36,6 +36,10 @@ func TestRedBlackTree(t *testing.T) {
 	tree.Put(2, "b")
 	tree.Put(1, "a") //overwrite
 
+	if actualValue := tree.Size(); actualValue != 7 {
+		t.Errorf("Got %v expected %v", actualValue, 7)
+	}
+
 	// key,expectedValue,expectedFound
 
 	tests1 := [][]interface{}{
@@ -62,6 +66,11 @@ func TestRedBlackTree(t *testing.T) {
 	tree.Remove(6)
 	tree.Remove(7)
 	tree.Remove(8)
+	tree.Remove(5)
+
+	if actualValue := tree.Size(); actualValue != 4 {
+		t.Errorf("Got %v expected %v", actualValue, 7)
+	}
 
 	tests2 := [][]interface{}{
 		{1, "a", true},