100% coverage

Author: RealA10N

dsu_test.go

@@ -6,6 +6,14 @@ import (
 	"alon.kr/x/dsu"
 )
 
+func assertAllSame(t *testing.T, values ...uint) {
+	for i := 1; i < len(values); i++ {
+		if values[i] != values[0] {
+			t.Error("All values should be the same")
+		}
+	}
+}
+
 func TestSimpleUnion(t *testing.T) {
 	d := dsu.NewDsu(3)
 	d.Union(0, 1)
@@ -21,9 +29,7 @@ func TestDoubleUnion(t *testing.T) {
 	d := dsu.NewDsu(3)
 	d.Union(0, 1)
 	d.Union(1, 2)
-	if d.Find(0) != d.Find(1) || d.Find(1) != d.Find(2) || d.Find(0) != d.Find(2) {
-		t.Error("0, 1, and 2 should be in the same set")
-	}
+	assertAllSame(t, d.Find(0), d.Find(1), d.Find(2))
 }
 
 func TestUnionOfSameSet(t *testing.T) {
@@ -34,3 +40,10 @@ func TestUnionOfSameSet(t *testing.T) {
 		t.Error("0 and 1 should be in the same set")
 	}
 }
+
+func TestReverseOrderUnion(t *testing.T) {
+	d := dsu.NewDsu(3)
+	d.Union(1, 2)
+	d.Union(0, 1)
+	assertAllSame(t, d.Find(0), d.Find(1), d.Find(2))
+}