math/big: add Floor and Ceil methods to Rat

Adds methods for floor and ceil of a rational number r, floor(r) is the
largest integer <= r and ceil(r) is the smallest integer >= r.

Author: Armin Günther

src/math/big/rat.go

@@ -559,3 +559,18 @@ func (z *Rat) Quo(x, y *Rat) *Rat {
 	z.a.neg = a.neg != b.neg
 	return z.norm()
 }
+
+// Floor returns the largest [Int] <= z.
+func (z *Rat) Floor() *Int {
+	// z.b is positive, so Euclidean division == floor division
+	return new(Int).Div(&z.a, &z.b)
+}
+
+// Ceil returns the smallest [Int] >= z.
+func (z *Rat) Ceil() *Int {
+	if z.IsInt() {
+		return new(Int).Set(&z.a)
+	}
+	f := z.Floor()
+	return f.Add(f, intOne)
+}

src/math/big/rat_test.go

@@ -744,3 +744,53 @@ func TestDenomRace(t *testing.T) {
 		<-c
 	}
 }
+
+var ratFloorTests = []struct {
+	rat string
+	out int64
+}{
+	{"123456.78", 123456},
+	{"100.5", 100},
+	{"5645.0222", 5645},
+	{"89898989", 89898989},
+	{"0", 0},
+	{"-123456.78", -123457},
+	{"-100.5", -101},
+	{"-5645.0222", -5646},
+	{"-89898989", -89898989},
+}
+
+func TestRatFloor(t *testing.T) {
+	for i, test := range ratFloorTests {
+		x, _ := new(Rat).SetString(test.rat)
+		out := x.Floor()
+		if out.Cmp(NewInt(test.out)) != 0 {
+			t.Errorf("#%d got out = %v; want %v", i, out, test.out)
+		}
+	}
+}
+
+var ratCeilTests = []struct {
+	rat string
+	out int64
+}{
+	{"123456.78", 123457},
+	{"100.5", 101},
+	{"5645.0222", 5646},
+	{"89898989", 89898989},
+	{"0", 0},
+	{"-123456.78", -123456},
+	{"-100.5", -100},
+	{"-5645.0222", -5645},
+	{"-89898989", -89898989},
+}
+
+func TestRatCeil(t *testing.T) {
+	for i, test := range ratCeilTests {
+		x, _ := new(Rat).SetString(test.rat)
+		out := x.Ceil()
+		if out.Cmp(NewInt(test.out)) != 0 {
+			t.Errorf("#%d got out = %v; want %v", i, out, test.out)
+		}
+	}
+}