test: add circuit tests

Author: ivokub

prover/zkevm/prover/bls/circuits_add_test.go

@@ -0,0 +1,78 @@
+package bls
+
+import (
+	"testing"
+
+	"github.com/consensys/linea-monorepo/prover/protocol/compiler/dummy"
+	"github.com/consensys/linea-monorepo/prover/protocol/dedicated/plonk"
+	"github.com/consensys/linea-monorepo/prover/protocol/wizard"
+	"github.com/consensys/linea-monorepo/prover/utils/csvtraces"
+)
+
+func TestBlsG1Add(t *testing.T) {
+	limits := &Limits{
+		NbG1AddInputInstances:   16,
+		NbG1AddCircuitInstances: 1,
+	}
+	ct := csvtraces.MustOpenCsvFile("testdata/bls_g1_add_input.csv")
+	var blsAdd *BlsAdd
+	var blsAddSource *BlsAddDataSource
+	cmp := wizard.Compile(
+		func(b *wizard.Builder) {
+			blsAddSource = &BlsAddDataSource{
+				CsAdd:  ct.GetCommit(b, "CIRCUIT_SELECTOR_G1_ADD"),
+				Limb:   ct.GetCommit(b, "LIMB"),
+				Index:  ct.GetCommit(b, "INDEX"),
+				IsData: ct.GetCommit(b, "DATA_G1_ADD"),
+				IsRes:  ct.GetCommit(b, "RSLT_G1_ADD"),
+			}
+			blsAdd = newAdd(b.CompiledIOP, G1, limits, blsAddSource, []plonk.Option{plonk.WithRangecheck(16, 6, true)})
+		},
+		dummy.Compile,
+	)
+
+	proof := wizard.Prove(cmp,
+		func(run *wizard.ProverRuntime) {
+			ct.Assign(run, "CIRCUIT_SELECTOR_G1_ADD", "LIMB", "INDEX", "DATA_G1_ADD", "RSLT_G1_ADD")
+			blsAdd.Assign(run)
+		})
+
+	if err := wizard.Verify(cmp, proof); err != nil {
+		t.Fatal("proof failed", err)
+	}
+	t.Log("proof succeeded")
+}
+
+func TestBlsG2Add(t *testing.T) {
+	limits := &Limits{
+		NbG2AddInputInstances:   16,
+		NbG2AddCircuitInstances: 1,
+	}
+	ct := csvtraces.MustOpenCsvFile("testdata/bls_g2_add_input.csv")
+	var blsAdd *BlsAdd
+	var blsAddSource *BlsAddDataSource
+	cmp := wizard.Compile(
+		func(b *wizard.Builder) {
+			blsAddSource = &BlsAddDataSource{
+				CsAdd:  ct.GetCommit(b, "CIRCUIT_SELECTOR_G2_ADD"),
+				Limb:   ct.GetCommit(b, "LIMB"),
+				Index:  ct.GetCommit(b, "INDEX"),
+				IsData: ct.GetCommit(b, "DATA_G2_ADD"),
+				IsRes:  ct.GetCommit(b, "RSLT_G2_ADD"),
+			}
+			blsAdd = newAdd(b.CompiledIOP, G2, limits, blsAddSource, []plonk.Option{plonk.WithRangecheck(16, 6, true)})
+		},
+		dummy.Compile,
+	)
+
+	proof := wizard.Prove(cmp,
+		func(run *wizard.ProverRuntime) {
+			ct.Assign(run, "CIRCUIT_SELECTOR_G2_ADD", "LIMB", "INDEX", "DATA_G2_ADD", "RSLT_G2_ADD")
+			blsAdd.Assign(run)
+		})
+
+	if err := wizard.Verify(cmp, proof); err != nil {
+		t.Fatal("proof failed", err)
+	}
+	t.Log("proof succeeded")
+}