From 972bd98987bba683e5bae521522a105a2e1ae9b0 Mon Sep 17 00:00:00 2001
From: Marco Molteni <marco.molteni@pix4d.com>
Date: Wed, 25 Sep 2024 09:58:43 +0200
Subject: [PATCH] sets: add Union()

---
 sets/sets.go      | 12 ++++++++++
 sets/sets_test.go | 59 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 71 insertions(+)

diff --git a/sets/sets.go b/sets/sets.go
index 3c892f8..658b407 100644
--- a/sets/sets.go
+++ b/sets/sets.go
@@ -105,3 +105,15 @@ func (s *Set[T]) Intersection(x *Set[T]) *Set[T] {
 	}
 	return result
 }
+
+// Union returns a set containing all the elements of s and x.
+func (s *Set[T]) Union(x *Set[T]) *Set[T] {
+	result := New[T](max(s.Size(), x.Size()))
+	for item := range s.items {
+		result.items[item] = struct{}{}
+	}
+	for item := range x.items {
+		result.items[item] = struct{}{}
+	}
+	return result
+}
diff --git a/sets/sets_test.go b/sets/sets_test.go
index 32c3fa5..9f3cfc0 100644
--- a/sets/sets_test.go
+++ b/sets/sets_test.go
@@ -201,6 +201,65 @@ func TestIntersection(t *testing.T) {
 	}
 }
 
+func TestUnion(t *testing.T) {
+	type testCase struct {
+		name     string
+		s        *sets.Set[int]
+		x        *sets.Set[int]
+		wantList []int
+	}
+
+	test := func(t *testing.T, tc testCase) {
+		result := tc.s.Union(tc.x)
+		sorted := result.OrderedList()
+
+		qt.Assert(t, qt.DeepEquals(sorted, tc.wantList))
+	}
+
+	testCases := []testCase{
+		{
+			name:     "both empty",
+			s:        sets.From[int](),
+			x:        sets.From[int](),
+			wantList: []int{},
+		},
+		{
+			name:     "empty x",
+			s:        sets.From(1, 2),
+			x:        sets.From[int](),
+			wantList: []int{1, 2},
+		},
+		{
+			name:     "empty s",
+			s:        sets.From[int](),
+			x:        sets.From(1, 2),
+			wantList: []int{1, 2},
+		},
+		{
+			name:     "identical",
+			s:        sets.From(1, 2),
+			x:        sets.From(1, 2),
+			wantList: []int{1, 2},
+		},
+		{
+			name:     "all different",
+			s:        sets.From(1, 3),
+			x:        sets.From(2, 4),
+			wantList: []int{1, 2, 3, 4},
+		},
+		{
+			name:     "partial overlap",
+			s:        sets.From(1, 3),
+			x:        sets.From(3, 5),
+			wantList: []int{1, 3, 5},
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) { test(t, tc) })
+	}
+}
+
 func TestRemoveFound(t *testing.T) {
 	type testCase struct {
 		name     string