1package plonky2_verifier
 2
 3import (
 4	"errors"
 5	"fmt"
 6	. "gnark-plonky2-verifier/field"
 7	"testing"
 8
 9	"github.com/consensys/gnark/frontend"
10	"github.com/consensys/gnark/test"
11)
12
13type TestArithmeticGateCircuit struct{}
14
15func (circuit *TestArithmeticGateCircuit) Define(api frontend.API) error {
16	commonCircuitData := DeserializeCommonCircuitData("./data/step/common_circuit_data.json")
17	numSelectors := len(commonCircuitData.SelectorsInfo.groups)
18
19	fieldAPI := NewFieldAPI(api)
20	qeAPI := NewQuadraticExtensionAPI(fieldAPI, commonCircuitData.DegreeBits)
21	plonkChip := NewPlonkChip(api, qeAPI, commonCircuitData)
22
23	arithmeticGate := ArithmeticGate{numOps: 20}
24	vars := EvaluationVars{localConstants: localConstants[numSelectors:], localWires: localWires, publicInputsHash: publicInputsHash}
25
26	constraints := arithmeticGate.EvalUnfiltered(plonkChip, vars)
27
28	if len(constraints) != len(arithmeticGateExpectedConstraints) {
29		return errors.New("arithmetic gate constraints length mismatch")
30	}
31
32	for i := 0; i < len(constraints); i++ {
33		fmt.Printf("constraints[%d] = %v\n", i, constraints[i])
34	}
35
36	for i := 0; i < len(constraints); i++ {
37		qeAPI.AssertIsEqual(constraints[i], arithmeticGateExpectedConstraints[i])
38	}
39
40	return nil
41}
42
43func TestArithmeticGate(t *testing.T) {
44	assert := test.NewAssert(t)
45
46	testCase := func() {
47		circuit := TestArithmeticGateCircuit{}
48		witness := TestArithmeticGateCircuit{}
49		err := test.IsSolved(&circuit, &witness, TEST_CURVE.ScalarField())
50		assert.NoError(err)
51	}
52
53	testCase()
54}