more on bst

1 files changed, 38 insertions, 0 deletions

diff --git a/set/set_test.go b/set/set_test.go
new file mode 100644
index 0000000..16d3e8c
--- /dev/null
+++ b/set/set_test.go
@@ -0,0 +1,38 @@
+package set
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/snonux/algorithms/ds"
+)
+
+const factor int = 10
+const minLength int = 1
+const maxLength int = 10
+
+func TestSet(t *testing.T) {
+	s := NewElementary()
+	for i := minLength; i <= maxLength; i *= factor {
+		test(s, i, t)
+	}
+}
+
+func test(s Set, l int, t *testing.T) {
+	cb := func(t *testing.T) {
+		list := ds.NewRandomArrayList(l, -1)
+		for i, a := range list {
+			s.Set(a, i)
+		}
+		for i, a := range list {
+			val, err := s.Get(a)
+			if err != nil {
+				t.Errorf("Element %v: %v", val, err)
+			}
+			if val != i {
+				t.Errorf("Element is %v but expected %v", val, i)
+			}
+		}
+	}
+	t.Run(fmt.Sprintf("%d", l), cb)
+}