dynamic/rodcutting_test.go

package dynamic_test

import (
	"testing"

	"github.com/TheAlgorithms/Go/dynamic"
)

type rodCuttingTestCase struct {
	price    []int
	length   int
	expected int
}

func getRodCuttingTestCases() []rodCuttingTestCase {
	return []rodCuttingTestCase{
		{[]int{0, 1, 5, 8, 9}, 4, 10},
		{[]int{0, 2, 5, 7, 8, 0}, 5, 12},
		{[]int{0, 1, 5, 8, 9, 10, 17, 17, 20}, 8, 22},
		{[]int{0, 3, 5, 8, 9, 10, 17, 17, 20}, 8, 24},
	}
}

func cutRodSolTestFunc(t *testing.T, cutRodSolFunc func([]int, int) int) {
	for _, tc := range getRodCuttingTestCases() {
		actual := cutRodSolFunc(tc.price, tc.length)
		if actual != tc.expected {
			t.Errorf("expected: %d, got: %d", tc.expected, actual)
		}
	}
}

func TestCutRodRec(t *testing.T) {
	cutRodSolTestFunc(t, dynamic.CutRodRec)
}

func TestCutRodDp(t *testing.T) {
	cutRodSolTestFunc(t, dynamic.CutRodDp)
}