githubmirror/LeetCode-Go

structures/Heap_test.go

30 строк512 B

YDZ

添加 structures

06 авг 2020, 18:29

06 авг 2020, 18:291934e38

package structures

import (
	"container/heap"
	"fmt"
	"testing"

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

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

	ih := new(intHeap)
	heap.Init(ih)

	heap.Push(ih, 1)
	heap.Pop(ih)

	begin, end := 0, 10
	for i := begin; i < end; i++ {
		heap.Push(ih, i)
		ast.Equal(0, (*ih)[0], "插入 %d 后的最小值却是 %d，ih=%v", i, (*ih)[0], (*ih))
	}

	for i := begin; i < end; i++ {
		fmt.Println(i, *ih)
		ast.Equal(i, heap.Pop(ih), "Pop 后 ih=%v", (*ih))
	}
}