You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
2.6 KiB
124 lines
2.6 KiB
package poseidon
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const prime uint64 = 18446744069414584321
|
|
|
|
func compareHash(inpBI [NROUNDSF]uint64, capBI [CAPLEN]uint64) ([CAPLEN]uint64, error) {
|
|
h, err := Hash(inpBI, capBI)
|
|
h1, _ := NeptuneHash(inpBI, capBI)
|
|
|
|
if h != h1 {
|
|
return h, fmt.Errorf("Not same")
|
|
}
|
|
|
|
return h, err
|
|
}
|
|
|
|
func TestPoseidonHashCompare(t *testing.T) {
|
|
b0 := uint64(0)
|
|
b1 := uint64(1)
|
|
bm1 := prime - 1
|
|
bM := prime
|
|
|
|
h, err := compareHash([NROUNDSF]uint64{b0, b0, b0, b0, b0, b0, b0, b0}, [CAPLEN]uint64{b0, b0, b0, b0})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
4330397376401421145,
|
|
14124799381142128323,
|
|
8742572140681234676,
|
|
14345658006221440202,
|
|
}, h,
|
|
)
|
|
|
|
h, err = compareHash([NROUNDSF]uint64{b1, b1, b1, b1, b1, b1, b1, b1}, [CAPLEN]uint64{b1, b1, b1, b1})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
16428316519797902711,
|
|
13351830238340666928,
|
|
682362844289978626,
|
|
12150588177266359240,
|
|
}, h,
|
|
)
|
|
|
|
h, err = compareHash([NROUNDSF]uint64{b1, b1, b1, b1, b1, b1, b1, b1}, [CAPLEN]uint64{b1, b1, b1, b1})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
16428316519797902711,
|
|
13351830238340666928,
|
|
682362844289978626,
|
|
12150588177266359240,
|
|
}, h,
|
|
)
|
|
|
|
h, err = compareHash(
|
|
[NROUNDSF]uint64{bm1, bm1, bm1, bm1, bm1, bm1, bm1, bm1},
|
|
[CAPLEN]uint64{bm1, bm1, bm1, bm1},
|
|
)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
13691089994624172887,
|
|
15662102337790434313,
|
|
14940024623104903507,
|
|
10772674582659927682,
|
|
}, h,
|
|
)
|
|
|
|
h, err = compareHash([NROUNDSF]uint64{bM, bM, bM, bM, bM, bM, bM, bM}, [CAPLEN]uint64{b0, b0, b0, b0})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
4330397376401421145,
|
|
14124799381142128323,
|
|
8742572140681234676,
|
|
14345658006221440202,
|
|
}, h,
|
|
)
|
|
|
|
h, err = compareHash([NROUNDSF]uint64{
|
|
uint64(923978),
|
|
uint64(235763497586),
|
|
uint64(9827635653498),
|
|
uint64(112870),
|
|
uint64(289273673480943876),
|
|
uint64(230295874986745876),
|
|
uint64(6254867324987),
|
|
uint64(2087),
|
|
}, [CAPLEN]uint64{b0, b0, b0, b0})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t,
|
|
[CAPLEN]uint64{
|
|
1892171027578617759,
|
|
984732815927439256,
|
|
7866041765487844082,
|
|
8161503938059336191,
|
|
}, h,
|
|
)
|
|
}
|
|
|
|
func BenchmarkPoseidonHash(b *testing.B) {
|
|
inp := [NROUNDSF]uint64{1, 2, 3, 4, 5, 6, 7, 8}
|
|
cap := [CAPLEN]uint64{10, 11, 12, 13}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
Hash(inp, cap) //nolint:errcheck,gosec
|
|
}
|
|
}
|
|
|
|
func BenchmarkPoseidonNeptuneHash(b *testing.B) {
|
|
inp := [NROUNDSF]uint64{1, 2, 3, 4, 5, 6, 7, 8}
|
|
cap := [CAPLEN]uint64{10, 11, 12, 13}
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
Hash(inp, cap) //nolint:errcheck,gosec
|
|
}
|
|
}
|