candy_test.go

package problem0135

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
)

type Result struct {
	Input    []int
	Expected int
}

var Results = []Result{
	{[]int{1, 0, 2}, 5},
	{[]int{1, 2, 2}, 4},
	{[]int{4, 5, 18, 16, 15, 11, 14, 11, 1, 4, 15, 15, 4, 15, 11, 1, 2, 15, 19, 5, 17, 4, 5, 8, 20, 12, 11, 7, 0, 13, 18, 17, 17, 4, 14, 16, 0, 15, 13, 13, 15, 19, 16, 1, 13, 11, 2, 17, 11, 17, 12, 13, 12, 7, 11, 9, 4, 16, 18, 15, 5, 20, 15, 20, 9, 2, 9, 3, 8, 1, 20, 4, 19, 7, 20, 20, 1, 7, 0, 11, 5, 19, 11, 16, 11, 12, 12, 19, 14, 7, 20, 5, 12, 2, 14, 12, 12, 4, 19, 10}, 187},
	{[]int{20, 44, 100, 98, 4, 39, 26, 77, 61, 43, 46, 10, 11, 96, 75, 98, 72, 63, 88, 63, 17, 19, 30, 33, 9, 9, 67, 100, 94, 66, 89, 98, 17, 21, 83, 50, 82, 8, 53, 20, 2, 94, 28, 74, 0, 28, 10, 26, 11, 70, 19, 40, 6, 71, 100, 51, 60, 79, 19, 53, 95, 95, 75, 30, 1, 69, 49, 77, 38, 71, 3, 17, 38, 6, 62, 17, 28, 36, 54, 12, 11, 13, 37, 71, 68, 99, 82, 36, 10, 12, 85, 23, 31, 9, 68, 89, 0, 80, 34, 58}, 194},
}

func TestCandy(t *testing.T) {
	assert := assert.New(t)

	for _, res := range Results {
		want := res.Expected
		got := candy(res.Input)
		assert.Equal(want, got, fmt.Sprintf("%+v", res))
	}
}