You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.7 KiB
59 lines
1.7 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCombinations(t *testing.T) {
|
|
input := map[string][]interface{}{
|
|
"bool": {false, true},
|
|
"int": {1, 2, 3},
|
|
"string": {"foo", "bar"},
|
|
}
|
|
|
|
c := combinations(input)
|
|
assert.Equal(t, []map[string]interface{}{
|
|
{"bool": false, "int": 1, "string": "foo"},
|
|
{"bool": false, "int": 1, "string": "bar"},
|
|
{"bool": false, "int": 2, "string": "foo"},
|
|
{"bool": false, "int": 2, "string": "bar"},
|
|
{"bool": false, "int": 3, "string": "foo"},
|
|
{"bool": false, "int": 3, "string": "bar"},
|
|
{"bool": true, "int": 1, "string": "foo"},
|
|
{"bool": true, "int": 1, "string": "bar"},
|
|
{"bool": true, "int": 2, "string": "foo"},
|
|
{"bool": true, "int": 2, "string": "bar"},
|
|
{"bool": true, "int": 3, "string": "foo"},
|
|
{"bool": true, "int": 3, "string": "bar"},
|
|
}, c)
|
|
}
|
|
|
|
func TestUniformSetChoice(t *testing.T) {
|
|
set := uniformSetChoice([]string{"a", "b", "c"})
|
|
r := rand.New(rand.NewSource(2384))
|
|
|
|
for i := 0; i < 100; i++ {
|
|
t.Run(fmt.Sprintf("Iteration%03d", i), func(t *testing.T) {
|
|
set = append(set, t.Name())
|
|
|
|
t.Run("ChooseAtLeastSubset", func(t *testing.T) {
|
|
require.True(t, len(set.ChooseAtLeast(r, 1)) >= 1)
|
|
require.True(t, len(set.ChooseAtLeast(r, 2)) >= 2)
|
|
require.True(t, len(set.ChooseAtLeast(r, len(set)/2)) >= len(set)/2)
|
|
})
|
|
t.Run("ChooseAtLeastEqualOrGreaterToLength", func(t *testing.T) {
|
|
require.Len(t, set.ChooseAtLeast(r, len(set)), len(set))
|
|
require.Len(t, set.ChooseAtLeast(r, len(set)+1), len(set))
|
|
require.Len(t, set.ChooseAtLeast(r, len(set)*10), len(set))
|
|
})
|
|
t.Run("ChooseSingle", func(t *testing.T) {
|
|
require.True(t, len(set.Choose(r)) >= 1)
|
|
})
|
|
})
|
|
}
|
|
}
|