githubmirror/LeetCode-Go

structures/Heap_test.go

30 строк511 B

halfrost

Fix go report

09 сен 2022, 00:55

09 сен 2022, 00:55e227edf

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))
	}
}