62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
|
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()
|
||
|
|
}
|
||
|
|
}
|