adding method to create set from keys of map of type comparable

set.go

@@ -194,3 +194,27 @@ func NewThreadUnsafeSet[T comparable](vals ...T) Set[T] {
 	}
 	return &s
 }
+
+// Creates and returns a new set with the given keys of the map.
+// Operations on the resulting set are thread-safe.
+func NewSetFromMapKeys[T comparable, V any](val map[T]V) Set[T] {
+	s := NewSet[T]()
+
+	for k := range val {
+		s.Add(k)
+	}
+
+	return s
+}
+
+// Creates and returns a new set with the given keys of the map.
+// Operations on the resulting set are not thread-safe.
+func NewThreadUnsafeSetFromMapKeys[T comparable, V any](val map[T]V) Set[T] {
+	s := NewThreadUnsafeSet[T]()
+
+	for k := range val {
+		s.Add(k)
+	}
+
+	return s
+}

set_test.go

@@ -25,7 +25,9 @@ SOFTWARE.
 
 package mapset
 
-import "testing"
+import (
+	"testing"
+)
 
 func makeSetInt(ints []int) Set[int] {
 	s := NewSet[int]()
@@ -1096,6 +1098,82 @@ func Test_ToSliceUnthreadsafe(t *testing.T) {
 	}
 }
 
+func Test_NewSetFromMapKey_Ints(t *testing.T) {
+	m := map[int]int{
+		5: 5,
+		2: 3,
+	}
+
+	s := NewSetFromMapKeys(m)
+
+	if len(m) != s.Cardinality() {
+		t.Errorf("Length of Set is not the same as the map. Expected: %d. Actual: %d", len(m), s.Cardinality())
+	}
+
+	for k := range m {
+		if !s.Contains(k) {
+			t.Errorf("Element %d not found in map: %v", k, m)
+		}
+	}
+}
+
+func Test_NewSetFromMapKey_Strings(t *testing.T) {
+	m := map[int]int{
+		5: 5,
+		2: 3,
+	}
+
+	s := NewSetFromMapKeys(m)
+
+	if len(m) != s.Cardinality() {
+		t.Errorf("Length of Set is not the same as the map. Expected: %d. Actual: %d", len(m), s.Cardinality())
+	}
+
+	for k := range m {
+		if !s.Contains(k) {
+			t.Errorf("Element %q not found in map: %v", k, m)
+		}
+	}
+}
+
+func Test_NewThreadUnsafeSetFromMapKey_Ints(t *testing.T) {
+	m := map[int]int{
+		5: 5,
+		2: 3,
+	}
+
+	s := NewThreadUnsafeSetFromMapKeys(m)
+
+	if len(m) != s.Cardinality() {
+		t.Errorf("Length of Set is not the same as the map. Expected: %d. Actual: %d", len(m), s.Cardinality())
+	}
+
+	for k := range m {
+		if !s.Contains(k) {
+			t.Errorf("Element %d not found in map: %v", k, m)
+		}
+	}
+}
+
+func Test_NewThreadUnsafeSetFromMapKey_Strings(t *testing.T) {
+	m := map[int]int{
+		5: 5,
+		2: 3,
+	}
+
+	s := NewThreadUnsafeSetFromMapKeys(m)
+
+	if len(m) != s.Cardinality() {
+		t.Errorf("Length of Set is not the same as the map. Expected: %d. Actual: %d", len(m), s.Cardinality())
+	}
+
+	for k := range m {
+		if !s.Contains(k) {
+			t.Errorf("Element %q not found in map: %v", k, m)
+		}
+	}
+}
+
 func Test_Example(t *testing.T) {
 	/*
 	   requiredClasses := NewSet()