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.
98 lines
3.5 KiB
98 lines
3.5 KiB
package poseidon
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/consensys/gnark/frontend"
|
|
"github.com/consensys/gnark/test"
|
|
"github.com/succinctlabs/gnark-plonky2-verifier/field"
|
|
"github.com/succinctlabs/gnark-plonky2-verifier/utils"
|
|
)
|
|
|
|
type TestPoseidonBN128Circuit struct {
|
|
In [spongeWidth]frontend.Variable
|
|
Out [spongeWidth]frontend.Variable
|
|
}
|
|
|
|
func (circuit *TestPoseidonBN128Circuit) Define(api frontend.API) error {
|
|
fieldAPI := field.NewFieldAPI(api)
|
|
poseidonChip := NewPoseidonBN128Chip(api, fieldAPI)
|
|
output := poseidonChip.Poseidon(circuit.In)
|
|
|
|
for i := 0; i < spongeWidth; i++ {
|
|
api.AssertIsEqual(
|
|
output[i],
|
|
circuit.Out[i],
|
|
)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestPoseidonBN128(t *testing.T) {
|
|
assert := test.NewAssert(t)
|
|
|
|
testCaseFn := func(in [spongeWidth]frontend.Variable, out [spongeWidth]frontend.Variable) {
|
|
circuit := TestPoseidonBN128Circuit{In: in, Out: out}
|
|
witness := TestPoseidonBN128Circuit{In: in, Out: out}
|
|
err := test.IsSolved(&circuit, &witness, field.TEST_CURVE.ScalarField())
|
|
assert.NoError(err)
|
|
}
|
|
|
|
testCases := [][2][]string{
|
|
{
|
|
{"0", "0", "0", "0"},
|
|
{
|
|
"5317387130258456662214331362918410991734007599705406860481038345552731150762",
|
|
"17768273200467269691696191901389126520069745877826494955630904743826040320364",
|
|
"19413739268543925182080121099097652227979760828059217876810647045303340666757",
|
|
"3717738800218482999400886888123026296874264026760636028937972004600663725187",
|
|
},
|
|
},
|
|
{
|
|
{"0", "1", "2", "3"},
|
|
{
|
|
"6542985608222806190361240322586112750744169038454362455181422643027100751666",
|
|
"3478427836468552423396868478117894008061261013954248157992395910462939736589",
|
|
"1904980799580062506738911865015687096398867595589699208837816975692422464009",
|
|
"11971464497515232077059236682405357499403220967704831154657374522418385384151",
|
|
},
|
|
},
|
|
{
|
|
{
|
|
"21888242871839275222246405745257275088548364400416034343698204186575808495616",
|
|
"21888242871839275222246405745257275088548364400416034343698204186575808495616",
|
|
"21888242871839275222246405745257275088548364400416034343698204186575808495616",
|
|
"21888242871839275222246405745257275088548364400416034343698204186575808495616",
|
|
},
|
|
{
|
|
"13055670547682322550638362580666986963569035646873545133474324633020685301274",
|
|
"19087936485076376314486368416882351797015004625427655501762827988254486144933",
|
|
"10391468779200270580383536396630001155994223659670674913170907401637624483385",
|
|
"17202557688472898583549180366140168198092766974201433936205272956998081177816",
|
|
},
|
|
},
|
|
{
|
|
{
|
|
"6542985608222806190361240322586112750744169038454362455181422643027100751666",
|
|
"3478427836468552423396868478117894008061261013954248157992395910462939736589",
|
|
"1904980799580062506738911865015687096398867595589699208837816975692422464009",
|
|
"11971464497515232077059236682405357499403220967704831154657374522418385384151",
|
|
},
|
|
{
|
|
"21792249080447013894140672594027696524030291802493510986509431008224624594361",
|
|
"3536096706123550619294332177231935214243656967137545251021848527424156573335",
|
|
"14869351042206255711434675256184369368509719143073814271302931417334356905217",
|
|
"5027523131326906886284185656868809493297314443444919363729302983434650240523",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
var in [spongeWidth]frontend.Variable
|
|
var out [spongeWidth]frontend.Variable
|
|
copy(in[:], utils.StrArrayToFrontendVariableArray(testCase[0]))
|
|
copy(out[:], utils.StrArrayToFrontendVariableArray(testCase[1]))
|
|
testCaseFn(in, out)
|
|
}
|
|
}
|