Add BitArray.Toggle()

Author: andreas19

bitarray.go

@@ -112,6 +112,18 @@ func (ba *BitArray) unset(idx int) {
 	ba.data[n] &^= 1 << i
 }
 
+// Toggle toggles the state of the bit at index idx and reports whether it is set after being toggled.
+func (ba *BitArray) Toggle(idx int) bool {
+	ba.checkIdx(idx)
+	b := ba.get(idx)
+	if b {
+		ba.unset(idx)
+	} else {
+		ba.set(idx)
+	}
+	return !b
+}
+
 // Get reports whether the bit at index idx is set.
 func (ba *BitArray) Get(idx int) bool {
 	ba.checkIdx(idx)

bitarray_test.go

@@ -185,6 +185,16 @@ func TestUnsetIdxSize(t *testing.T) {
 	t.Error("did not panic")
 }
 
+func TestToggle(t *testing.T) {
+	want := "1000000010"
+	ba := New(10, 9, 0)
+	ba.Toggle(0)
+	ba.Toggle(1)
+	if got := ba.String(); got != want {
+		t.Errorf("got %q, want %q", got, want)
+	}
+}
+
 func TestGet(t *testing.T) {
 	s := "0100110101"
 	tests := []struct {