mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-27 13:58:05 +00:00
36 lines
647 B
Go
36 lines
647 B
Go
|
package stack
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/big"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
// helper functions
|
||
|
|
func testPeakInteger(t *testing.T, tStack *RandomAccess, n uint16) *Int {
|
||
|
|
stackElement, err := tStack.Peek(n)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
item, err := stackElement.Integer()
|
||
|
|
if err != nil {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
return item
|
||
|
|
}
|
||
|
|
|
||
|
|
func testPopInteger(t *testing.T, tStack *RandomAccess) *Int {
|
||
|
|
stackElement, err := tStack.Pop()
|
||
|
|
assert.Nil(t, err)
|
||
|
|
item, err := stackElement.Integer()
|
||
|
|
if err != nil {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
return item
|
||
|
|
}
|
||
|
|
|
||
|
|
func testMakeStackInt(t *testing.T, num int64) *Int {
|
||
|
|
a, err := NewInt(big.NewInt(num))
|
||
|
|
assert.Nil(t, err)
|
||
|
|
return a
|
||
|
|
}
|