insert-delete-getrandom-o1-duplicates-allowed_test.go

package problem0381

import (
	"fmt"
	"testing"

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

type question struct {
	para
	ans
}

// para 是参数
type para struct {
	orders []string
	paras  []int
}

// ans 是答案
type ans struct {
	solutions []interface{}
}

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

	qs := []question{

		question{
			para{
				[]string{"RandomizedCollection", "insert", "insert", "insert", "insert", "insert", "insert", "remove", "remove", "remove", "insert", "remove"},
				[]int{0, 9, 9, 1, 1, 2, 1, 2, 1, 1, 9, 1},
			},
			ans{
				[]interface{}{nil, true, false, true, false, true, false, true, true, true, false, true},
			},
		},

		question{
			para{
				[]string{"RandomizedCollection", "insert", "insert", "insert", "insert", "insert", "remove", "remove", "remove", "insert", "remove", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "remove", "remove"},
				[]int{0, 1, 1, 2, 2, 2, 1, 1, 2, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1},
			},
			ans{
				[]interface{}{nil, true, false, true, false, false, true, true, true, true, true, 1, 2, 2, 1, 2, 2, 1, 1, 2, 2, true, false},
			},
		},
		// 如需多个测试，可以复制上方元素。
	}

	for _, q := range qs {
		sol, ord, par := q.ans.solutions, q.para.orders, q.para.paras

		ast.Equal(len(sol), len(ord))
		ast.Equal(len(sol), len(par))

		rs := Constructor()
		for i := 1; i < len(ord); i++ {
			fmt.Println(ord[i], par[i])
			switch ord[i] {
			case "insert":
				ast.Equal(sol[i], rs.Insert(par[i]), "Insert %d", par[i])
			case "remove":
				ast.Equal(sol[i], rs.Remove(par[i]), "Remove %d", par[i])
			case "getRandom":
				r := rs.GetRandom()
				ast.Contains([]int{1, 2}, r, "GetRandom %d", r)
			default:
				ast.Fail("无法处理的命令", "%s", ord[i])
			}
		}
	}
}