Set - make set from slice & add slice to set (#45)

datastructures/set.go

@@ -13,6 +13,14 @@ func MakeSet[T comparable]() *Set[T] {
 	}
 }
 
+func MakeSetFromElements[T comparable](elements ...T) *Set[T] {
+	set := MakeSet[T]()
+	for _, element := range elements {
+		set.Add(element)
+	}
+	return set
+}
+
 func (set *Set[T]) Exists(key T) bool {
 	_, exists := set.container[key]
 	return exists
@@ -22,6 +30,12 @@ func (set *Set[T]) Add(key T) {
 	set.container[key] = struct{}{}
 }
 
+func (set *Set[T]) AddElements(elements ...T) {
+	for _, element := range elements {
+		set.Add(element)
+	}
+}
+
 func (set *Set[T]) Remove(key T) error {
 	_, exists := set.container[key]
 	if !exists {

datastructures/set_test.go

@@ -35,3 +35,13 @@ func TestSetToSlice(t *testing.T) {
 	assert.Contains(t, slice, 5)
 	assert.Contains(t, slice, 7)
 }
+
+func TestMakeSetFromElements(t *testing.T) {
+	intSlice := []int{1, 2, 3}
+	intSet := MakeSetFromElements(intSlice...)
+	assert.ElementsMatch(t, intSet.ToSlice(), intSlice)
+
+	stringSlice := []string{"frog", "frogger", "froggy"}
+	stringSet := MakeSetFromElements(stringSlice...)
+	assert.ElementsMatch(t, stringSet.ToSlice(), stringSlice)
+}