package pqueue

import (
	"testing"

	"github.com/Pallinder/go-randomdata"

	"474420502.top/eson/structure/compare"
)

func TestPush(t *testing.T) {
	pq := New(compare.Int)
	for i := 0; i < 10; i++ {
		v := randomdata.Number(0, 100)
		pq.Push(v)
		t.Error(v)
	}
	t.Error(pq.Pop())
	t.Error(pq.Pop())
	t.Error(pq.Pop())
	t.Error(pq.Pop())
	t.Error(pq.Pop())
}

func TestPop(t *testing.T) {
	pq := New(compare.Int)
	for i := 0; i < 10; i++ {
		v := randomdata.Number(0, 100)
		pq.Push(v)
		t.Error(v)
	}

	for i := 0; i < 10; i++ {
		t.Error(pq.Pop())
	}
}

func BenchmarkPriorityPush(b *testing.B) {

	l := loadTestData()
	execCount := 5
	b.N = len(l) * execCount

	b.ResetTimer()
	b.StartTimer()

	for i := 0; i < execCount; i++ {
		pq := New(compare.Int)
		for _, v := range l {
			pq.Push(v)
		}
	}
}

func BenchmarkPriorityPop(b *testing.B) {

	l := loadTestData()

	pq := New(compare.Int)
	for _, v := range l {
		pq.Push(v)
	}

	b.N = len(l) / 1000
	b.ResetTimer()
	b.StartTimer()

	for i := 0; i < b.N; i++ {
		pq.Pop()
	}
}