math/bigrat_test.go at master · pkg/math · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package math

import (
	"math/big"
	"testing"
)

func r(n, d int64) *big.Rat { return big.NewRat(n, d) }

var equalBigRatTests = []struct {
	a, b *big.Rat
	want bool
}{
	{r(1, 2), r(1, 2), true},
	{r(1, 1), r(1, 1), true},
	{r(1, 2), r(2, 4), true},
	{r(1, 2), r(2, 3), false},
}

func TestEqualBigRat(t *testing.T) {
	for i, tt := range equalBigRatTests {
		got := EqualBigRat(tt.a, tt.b)
		if got != tt.want {
			t.Errorf("%d: EqualBigRat(%v, %v) = %v, want %v", i+1, tt.a, tt.b, got, tt.want)
		}
	}
}

var maxBigRatTests = []struct {
	a, b *big.Rat
	want *big.Rat
}{
	{r(1, 2), r(1, 3), r(1, 2)},
	{r(1, 3), r(1, 2), r(1, 2)},
	{r(1, 2), r(1, 4), r(1, 2)},
	{r(1, 4), r(1, 2), r(1, 2)},
}

func TestMaxBigRat(t *testing.T) {
	for i, tt := range maxBigRatTests {
		got := MaxBigRat(tt.a, tt.b)
		if !EqualBigRat(tt.want, got) {
			t.Errorf("%d: MaxBigRat(%v, %v) = %v, want %v", i+1, tt.a, tt.b, got, tt.want)
		}
	}
}

var minBigRatTests = []struct {
	a, b *big.Rat
	want *big.Rat
}{
	{r(1, 2), r(1, 3), r(1, 3)},
	{r(1, 3), r(1, 2), r(1, 3)},
	{r(1, 2), r(1, 4), r(1, 4)},
	{r(1, 4), r(1, 2), r(1, 4)},
}

func TestMinBigRat(t *testing.T) {
	for i, tt := range minBigRatTests {
		got := MinBigRat(tt.a, tt.b)
		if !EqualBigRat(tt.want, got) {
			t.Errorf("%d: MinBigRat(%v, %v) = %v, want %v", i+1, tt.a, tt.b, got, tt.want)
		}
	}
}