added String method to SetMap

Signed-off-by: Tim Henderson <[email protected]>

Author: timtadh

set/setmap.go

@@ -1,6 +1,9 @@
 package set
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/timtadh/data-structures/errors"
 	"github.com/timtadh/data-structures/types"
 )
@@ -13,6 +16,17 @@ func NewSetMap(m types.Map) *SetMap {
 	return &SetMap{m}
 }
 
+func (s *SetMap) String() string {
+	if s.Size() <= 0 {
+		return "{}"
+	}
+	items := make([]string, 0, s.Size())
+	for item, next := s.Items()(); next != nil; item, next = next() {
+		items = append(items, fmt.Sprintf("%v", item))
+	}
+	return "{" + strings.Join(items, ", ") + "}"
+}
+
 func (s *SetMap) Items() types.KIterator {
 	return s.Keys()
 }

set/sortedset_test.go

@@ -49,7 +49,7 @@ func TestAddMarshalUnmarshalHas(x *testing.T) {
 		t.assert_nil(set.Add(item))
 	}
 	for i, item := range items {
-		t.assert(fmt.Sprintf("!set.Has(item)", i), set.Has(item))
+		t.assert(fmt.Sprintf("!set.Has(%v) ", i), set.Has(item))
 	}
 	marshal, unmarshal := types.IntMarshals()
 	mset1 := NewMSortedSet(set, marshal, unmarshal)
@@ -59,7 +59,7 @@ func TestAddMarshalUnmarshalHas(x *testing.T) {
 	t.assert_nil(mset2.UnmarshalBinary(bytes))
 	set2 := mset2.SortedSet()
 	for i, item := range items {
-		t.assert(fmt.Sprintf("!set.Has(item)", i), set2.Has(item))
+		t.assert(fmt.Sprintf("!set.Has(%v)", i), set2.Has(item))
 	}
 }
 
@@ -82,7 +82,7 @@ func TestAddHasDeleteRandom(x *testing.T) {
 	for i := 0; i < 10; i++ {
 		item, err := set.Random()
 		t.assert_nil(err)
-		t.assert(fmt.Sprintf("!set.Has(item)", i), set.Has(item))
+		t.assert(fmt.Sprintf("!set.Has(%v)", i), set.Has(item))
 	}
 	for _, item := range items {
 		t.assert_nil(set.Delete(item))