mirror of
https://github.com/luxfi/util.git
synced 2026-10-30 15:13:56 +00:00
zeekayandHanzo Dev
b45161c43f
These three sit in luxfi/node under utils/, and P-Chain and X-Chain reach for them. Both chains have to leave that repository before it can be archived, and a chain should not have to depend on a node to measure a quantity in nLUX, keep a heap, or walk a sequence. units imports nothing at all; heap wanted only Zero, which is already here; iterator wanted heap and Zero. Three of iterator's tests exercise it against platformvm's state iterators. They stay with the node, and travel with platformvm, because a generic sequence should not be tested through one virtual machine's idea of state. What remains here covers the package on its own terms. Co-authored-by: Hanzo Dev <[email protected]>
73 lines
1.2 KiB
Go
73 lines
1.2 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package heap
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSet(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
setup func(h Set[int])
|
|
expected []int
|
|
}{
|
|
{
|
|
name: "only push",
|
|
setup: func(h Set[int]) {
|
|
h.Push(1)
|
|
h.Push(2)
|
|
h.Push(3)
|
|
},
|
|
expected: []int{1, 2, 3},
|
|
},
|
|
{
|
|
name: "out of order pushes",
|
|
setup: func(h Set[int]) {
|
|
h.Push(1)
|
|
h.Push(5)
|
|
h.Push(2)
|
|
h.Push(4)
|
|
h.Push(3)
|
|
},
|
|
expected: []int{1, 2, 3, 4, 5},
|
|
},
|
|
{
|
|
name: "push and pop",
|
|
setup: func(h Set[int]) {
|
|
h.Push(1)
|
|
h.Push(5)
|
|
h.Push(2)
|
|
h.Push(4)
|
|
h.Push(3)
|
|
h.Pop()
|
|
h.Pop()
|
|
h.Pop()
|
|
},
|
|
expected: []int{4, 5},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
h := NewSet[int](func(a, b int) bool {
|
|
return a < b
|
|
})
|
|
|
|
tt.setup(h)
|
|
|
|
require.Equal(len(tt.expected), h.Len())
|
|
for _, expected := range tt.expected {
|
|
got, ok := h.Pop()
|
|
require.True(ok)
|
|
require.Equal(expected, got)
|
|
}
|
|
})
|
|
}
|
|
}
|