exercism/go/grains/grains_test.go

62 lines
1.3 KiB
Go
Raw Normal View History

2016-08-13 23:20:14 +00:00
package grains
import (
"testing"
)
var squareTests = []struct {
input int
expectedVal uint64
expectError bool
}{
{1, 1, false},
{2, 2, false},
{3, 4, false},
{4, 8, false},
{16, 32768, false},
{32, 2147483648, false},
{64, 9223372036854775808, false},
{65, 0, true},
{0, 0, true},
{-1, 0, true},
}
func TestSquare(t *testing.T) {
for _, test := range squareTests {
actualVal, actualErr := Square(test.input)
if actualVal != test.expectedVal {
t.Errorf("Square(%d) expected %d, Actual %d", test.input, test.expectedVal, actualVal)
}
// if we expect an error and there isn't one
if test.expectError && actualErr == nil {
t.Errorf("Square(%d) expected an error, but error is nil", test.input)
}
// if we don't expect an error and there is one
if !test.expectError && actualErr != nil {
t.Errorf("Square(%d) expected no error, but error is: %s", test.input, actualErr)
}
}
}
func TestTotal(t *testing.T) {
var expected uint64 = 18446744073709551615
if actual := Total(); actual != expected {
t.Errorf("Total() expected %d, Actual %d", expected, actual)
}
}
func BenchmarkSquare(b *testing.B) {
b.StopTimer()
for _, test := range squareTests {
b.StartTimer()
for i := 0; i < b.N; i++ {
Square(test.input)
}
b.StopTimer()
}
}