326 lines
9.1 KiB
Go
326 lines
9.1 KiB
Go
|
package mathUtil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetRandRangeInt(t *testing.T) {
|
||
|
|
lower, upper := 10, 100
|
||
|
|
rand := GetRand().GetRandRangeInt(lower, upper)
|
||
|
|
if rand < lower || rand >= upper {
|
||
|
|
t.Errorf("Expected a num between %d and %d, but got %d", lower, upper, rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandInt(t *testing.T) {
|
||
|
|
var n int = 100
|
||
|
|
var rand int = GetRand().GetRandInt(n)
|
||
|
|
if rand >= n {
|
||
|
|
t.Errorf("Expected a num < %d, but got %d", n, rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandInt32(t *testing.T) {
|
||
|
|
var n int32 = 100
|
||
|
|
var rand int32 = GetRand().GetRandInt32(n)
|
||
|
|
if rand >= n {
|
||
|
|
t.Errorf("Expected a num < %d, but got %d", n, rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandInt64(t *testing.T) {
|
||
|
|
var n int64 = 100
|
||
|
|
var rand int64 = GetRand().GetRandInt64(n)
|
||
|
|
if rand >= n {
|
||
|
|
t.Errorf("Expected a num < %d, but got %d", n, rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandFloat32(t *testing.T) {
|
||
|
|
var rand float32 = GetRand().GetRandFloat32()
|
||
|
|
if rand >= 1 {
|
||
|
|
t.Errorf("Expected a num < 1, but got %f", rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandFloat64(t *testing.T) {
|
||
|
|
var rand float64 = GetRand().GetRandFloat64()
|
||
|
|
if rand >= 1 {
|
||
|
|
t.Errorf("Expected a num < 1, but got %f", rand)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandNumList(t *testing.T) {
|
||
|
|
if _, err := GetRand().GetRandNumList(11, 10, 11, false); err.Error() != "minValue can't be bigger than maxValue." {
|
||
|
|
t.Error("Expected err, but got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := GetRand().GetRandNumList(1, 10, 11, false); err.Error() != "随机的数量超过区间的元素数量" {
|
||
|
|
t.Error("Expected err, but got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := GetRand().GetRandNumList(1, 10001, 10, false); err.Error() != "随机数的区间不能大于10000" {
|
||
|
|
t.Error("Expected err, but got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
randNumList, _ := GetRand().GetRandNumList(1, 10, 1, false)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
randNumList, _ = GetRand().GetRandNumList(1, 10, 3, false)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
randNumList, _ = GetRand().GetRandNumList(1, 10, 5, false)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
randNumList, _ = GetRand().GetRandNumList(1, 10, 7, false)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
randNumList, _ = GetRand().GetRandNumList(1, 10, 9, false)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
randNumList, _ = GetRand().GetRandNumList(1, 10, 11, true)
|
||
|
|
fmt.Printf("randNumList:%v\n", randNumList)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandIntList(t *testing.T) {
|
||
|
|
source := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
|
||
|
|
|
||
|
|
if _, err := GetRand().GetRandIntList(source, 11, false); err.Error() != "随机的数量超过列表的元素数量" {
|
||
|
|
t.Error("Expected err, but got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
randIntList, _ := GetRand().GetRandIntList(source, 1, false)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
randIntList, _ = GetRand().GetRandIntList(source, 3, false)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
randIntList, _ = GetRand().GetRandIntList(source, 5, false)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
randIntList, _ = GetRand().GetRandIntList(source, 7, false)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
randIntList, _ = GetRand().GetRandIntList(source, 9, false)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
randIntList, _ = GetRand().GetRandIntList(source, 10, true)
|
||
|
|
fmt.Printf("randIntList:%v\n", randIntList)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandInterfaceList(t *testing.T) {
|
||
|
|
item1 := NewItem(1, "name1", 1)
|
||
|
|
item2 := NewItem(2, "name1", 1)
|
||
|
|
item3 := NewItem(3, "name1", 1)
|
||
|
|
item4 := NewItem(4, "name1", 1)
|
||
|
|
item5 := NewItem(5, "name1", 1)
|
||
|
|
item6 := NewItem(6, "name1", 1)
|
||
|
|
item7 := NewItem(7, "name1", 1)
|
||
|
|
item8 := NewItem(8, "name1", 1)
|
||
|
|
item9 := NewItem(9, "name1", 1)
|
||
|
|
item10 := NewItem(10, "name1", 1)
|
||
|
|
|
||
|
|
source := make([]interface{}, 0, 10)
|
||
|
|
source = append(source, item1)
|
||
|
|
source = append(source, item2)
|
||
|
|
source = append(source, item3)
|
||
|
|
source = append(source, item4)
|
||
|
|
source = append(source, item5)
|
||
|
|
source = append(source, item6)
|
||
|
|
source = append(source, item7)
|
||
|
|
source = append(source, item8)
|
||
|
|
source = append(source, item9)
|
||
|
|
source = append(source, item10)
|
||
|
|
|
||
|
|
if _, err := GetRand().GetRandInterfaceList(source, 11, false); err.Error() != "随机的数量超过列表的元素数量" {
|
||
|
|
t.Error("Expected err, but got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
randInterfaceList, _ := GetRand().GetRandInterfaceList(source, 1, false)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
randInterfaceList, _ = GetRand().GetRandInterfaceList(source, 3, false)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
randInterfaceList, _ = GetRand().GetRandInterfaceList(source, 5, false)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
randInterfaceList, _ = GetRand().GetRandInterfaceList(source, 7, false)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
randInterfaceList, _ = GetRand().GetRandInterfaceList(source, 9, false)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
randInterfaceList, _ = GetRand().GetRandInterfaceList(source, 10, true)
|
||
|
|
fmt.Printf("randInterfaceList:%v\n", randInterfaceList)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandWeight(t *testing.T) {
|
||
|
|
source := make([]IWeight, 0, 10)
|
||
|
|
if _, err := GetRand().GetRandWeight(source); err == nil {
|
||
|
|
t.Errorf("err should not be nil, but it's nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
item1 := NewItem(1, "name1", 1)
|
||
|
|
item2 := NewItem(2, "name2", 2)
|
||
|
|
item3 := NewItem(3, "name3", 3)
|
||
|
|
item4 := NewItem(4, "name4", 4)
|
||
|
|
item5 := NewItem(5, "name5", 5)
|
||
|
|
item6 := NewItem(6, "name6", 60)
|
||
|
|
item7 := NewItem(7, "name7", 70)
|
||
|
|
item8 := NewItem(8, "name8", 80)
|
||
|
|
item9 := NewItem(9, "name9", 90)
|
||
|
|
item10 := NewItem(10, "name10", 100)
|
||
|
|
|
||
|
|
source = append(source, item1)
|
||
|
|
source = append(source, item2)
|
||
|
|
source = append(source, item3)
|
||
|
|
source = append(source, item4)
|
||
|
|
source = append(source, item5)
|
||
|
|
source = append(source, item6)
|
||
|
|
source = append(source, item7)
|
||
|
|
source = append(source, item8)
|
||
|
|
source = append(source, item9)
|
||
|
|
source = append(source, item10)
|
||
|
|
|
||
|
|
data := make(map[int]int)
|
||
|
|
for i := 0; i < 10000; i++ {
|
||
|
|
if result, err := GetRand().GetRandWeight(source); err != nil {
|
||
|
|
t.Errorf("err should be nil, but it's not:%s", err)
|
||
|
|
} else {
|
||
|
|
if item, ok := result.(*Item); !ok {
|
||
|
|
t.Errorf("convert to Item failed")
|
||
|
|
} else {
|
||
|
|
if count, ok := data[item.Id]; ok {
|
||
|
|
data[item.Id] = count + 1
|
||
|
|
} else {
|
||
|
|
data[item.Id] = 1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
total := 0
|
||
|
|
for _, v := range data {
|
||
|
|
total += v
|
||
|
|
}
|
||
|
|
|
||
|
|
k := 1
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 2
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 3
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 4
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 5
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 6
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 7
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 8
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 9
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
k = 10
|
||
|
|
if v, ok := data[k]; ok {
|
||
|
|
fmt.Printf("%d:%d, ratio:%d\n", k, v, v*100/total)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRandWeight2(t *testing.T) {
|
||
|
|
source := make([]IWeight, 0, 10)
|
||
|
|
if _, err := GetRand().GetRandWeight(source); err == nil {
|
||
|
|
t.Errorf("err should not be nil, but it's nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
item1 := NewItem(1, "name1", 1)
|
||
|
|
item2 := NewItem(2, "name2", 2)
|
||
|
|
item3 := NewItem(3, "name3", 3)
|
||
|
|
item4 := NewItem(4, "name4", 4)
|
||
|
|
item5 := NewItem(5, "name5", 5)
|
||
|
|
item6 := NewItem(6, "name6", 60)
|
||
|
|
item7 := NewItem(7, "name7", 70)
|
||
|
|
item8 := NewItem(8, "name8", 80)
|
||
|
|
item9 := NewItem(9, "name9", 90)
|
||
|
|
item10 := NewItem(10, "name10", 100)
|
||
|
|
|
||
|
|
source = append(source, item1)
|
||
|
|
source = append(source, item2)
|
||
|
|
source = append(source, item3)
|
||
|
|
source = append(source, item4)
|
||
|
|
source = append(source, item5)
|
||
|
|
source = append(source, item6)
|
||
|
|
source = append(source, item7)
|
||
|
|
source = append(source, item8)
|
||
|
|
source = append(source, item9)
|
||
|
|
source = append(source, item10)
|
||
|
|
|
||
|
|
rindexs, err := GetRand().GetRandItemsByWeight(source, 2, false)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("GetRandItemsByWeight出错:%s", err)
|
||
|
|
}
|
||
|
|
if len(rindexs) != 2 || rindexs[0] == rindexs[1] {
|
||
|
|
t.Errorf("GetRandItemsByWeight随机内容出错")
|
||
|
|
}
|
||
|
|
|
||
|
|
rindexs, err = GetRand().GetRandItemsByWeight(source, 10, false)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("GetRandItemsByWeight出错:%s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(rindexs) != 10 {
|
||
|
|
t.Errorf("GetRandItemsByWeight随机内容出错")
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = GetRand().GetRandItemsByWeight(source, 12, false)
|
||
|
|
if err == nil {
|
||
|
|
t.Errorf("GetRandItemsByWeight出错:%s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
rindexs, err = GetRand().GetRandItemsByWeight(source, 15, true)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("GetRandItemsByWeight出错:%s", err)
|
||
|
|
}
|
||
|
|
if len(rindexs) != 15 {
|
||
|
|
t.Errorf("GetRandItemsByWeight随机内容出错")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
type Item struct {
|
||
|
|
Id int
|
||
|
|
Name string
|
||
|
|
Weight int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (item *Item) GetWeight() int {
|
||
|
|
return item.Weight
|
||
|
|
}
|
||
|
|
|
||
|
|
func (item *Item) String() string {
|
||
|
|
return fmt.Sprintf("Id:%d,Name:%s\n", item.Id, item.Name)
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewItem(id int, name string, weight int) *Item {
|
||
|
|
return &Item{
|
||
|
|
Id: id,
|
||
|
|
Name: name,
|
||
|
|
Weight: weight,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_GetRandom(t *testing.T) {
|
||
|
|
rand, err := GetRand().GetRandNumList(1, 10000, 10, true)
|
||
|
|
if err != nil {
|
||
|
|
t.Log(fmt.Errorf("获取随机数错误：%s", err))
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Log(rand)
|
||
|
|
}
|