test: include minimal number of test files

Author: ivokub

prover/zkevm/prover/bls/module_map_test.go

@@ -1,6 +1,7 @@
 package bls
 
 import (
+	"errors"
 	"os"
 	"testing"
 
@@ -10,33 +11,39 @@ import (
 	"github.com/consensys/linea-monorepo/prover/utils/csvtraces"
 )
 
-func testBlsG1Map(t *testing.T, withCircuit bool) {
-	limits := &Limits{
-		NbG1MapToInputInstances:   16,
-		NbG1MapToCircuitInstances: 1,
+func testBlsMap(t *testing.T, withCircuit bool, g group, path string, limits *Limits) {
+	f, err := os.Open(path)
+	if errors.Is(err, os.ErrNotExist) {
+		t.Fatal("csv file does not exist, please run `go generate` to generate the test data")
 	}
-	f, err := os.Open("testdata/bls_g1_map_inputs.csv")
 	if err != nil {
-		t.Fatal(err)
+		t.Fatal("failed to open csv file", err)
 	}
 	defer f.Close()
 	ct, err := csvtraces.NewCsvTrace(f)
 	if err != nil {
 		t.Fatal("failed to create csv trace", err)
 	}
+	var mapString string
+	if g == G1 {
+		mapString = "MAP_FP_TO_G1"
+	} else {
+		mapString = "MAP_FP2_TO_G2"
+	}
 	var blsMap *BlsMap
+	var blsMapSource *blsMapDataSource
 	cmp := wizard.Compile(
 		func(b *wizard.Builder) {
-			blsMapSource := &blsMapDataSource{
+			blsMapSource = &blsMapDataSource{
 				ID:      ct.GetCommit(b, "ID"),
-				CsMap:   ct.GetCommit(b, "CIRCUIT_SELECTOR_MAP_FP_TO_G1"),
+				CsMap:   ct.GetCommit(b, "CIRCUIT_SELECTOR_"+mapString),
 				Index:   ct.GetCommit(b, "INDEX"),
 				Counter: ct.GetCommit(b, "CT"),
 				Limb:    ct.GetCommit(b, "LIMB"),
-				IsData:  ct.GetCommit(b, "DATA_MAP_FP_TO_G1"),
-				IsRes:   ct.GetCommit(b, "RSLT_MAP_FP_TO_G1"),
+				IsData:  ct.GetCommit(b, "DATA_"+mapString),
+				IsRes:   ct.GetCommit(b, "RSLT_"+mapString),
 			}
-			blsMap = newMap(b.CompiledIOP, G1, limits, blsMapSource)
+			blsMap = newMap(b.CompiledIOP, g, limits, blsMapSource)
 			if withCircuit {
 				blsMap = blsMap.WithMapCircuit(b.CompiledIOP, query.PlonkRangeCheckOption(16, 6, true))
 			}
@@ -46,7 +53,7 @@ func testBlsG1Map(t *testing.T, withCircuit bool) {
 
 	proof := wizard.Prove(cmp,
 		func(run *wizard.ProverRuntime) {
-			ct.Assign(run, "ID", "CIRCUIT_SELECTOR_MAP_FP_TO_G1", "INDEX", "CT", "LIMB", "DATA_MAP_FP_TO_G1", "RSLT_MAP_FP_TO_G1")
+			ct.Assign(run, "ID", "CIRCUIT_SELECTOR_"+mapString, "INDEX", "CT", "LIMB", "DATA_"+mapString, "RSLT_"+mapString)
 			blsMap.Assign(run)
 		})
 
@@ -57,63 +64,33 @@ func testBlsG1Map(t *testing.T, withCircuit bool) {
 }
 
 func TestBlsMapG1NoCircuit(t *testing.T) {
-	testBlsG1Map(t, false)
+	limits := &Limits{
+		NbG1MapToInputInstances:   16,
+		NbG1MapToCircuitInstances: 1,
+	}
+	testBlsMap(t, false, G1, "testdata/bls_g1_map_inputs.csv", limits)
 }
 
 func TestBlsMapG1WithCircuit(t *testing.T) {
-	testBlsG1Map(t, true)
+	limits := &Limits{
+		NbG1MapToInputInstances:   16,
+		NbG1MapToCircuitInstances: 1,
+	}
+	testBlsMap(t, true, G1, "testdata/bls_g1_map_inputs.csv", limits)
 }
 
-func testBlsG2Map(t *testing.T, withCircuit bool) {
+func TestBlsMapG2NoCircuit(t *testing.T) {
 	limits := &Limits{
 		NbG2MapToInputInstances:   4,
 		NbG2MapToCircuitInstances: 1,
 	}
-	f, err := os.Open("testdata/bls_g2_map_inputs.csv")
-	if err != nil {
-		t.Fatal(err)
-	}
-	defer f.Close()
-	ct, err := csvtraces.NewCsvTrace(f)
-	if err != nil {
-		t.Fatal("failed to create csv trace", err)
-	}
-	var blsMap *BlsMap
-	cmp := wizard.Compile(
-		func(b *wizard.Builder) {
-			blsMapSource := &blsMapDataSource{
-				ID:      ct.GetCommit(b, "ID"),
-				CsMap:   ct.GetCommit(b, "CIRCUIT_SELECTOR_MAP_FP2_TO_G2"),
-				Index:   ct.GetCommit(b, "INDEX"),
-				Counter: ct.GetCommit(b, "CT"),
-				Limb:    ct.GetCommit(b, "LIMB"),
-				IsData:  ct.GetCommit(b, "DATA_MAP_FP2_TO_G2"),
-				IsRes:   ct.GetCommit(b, "RSLT_MAP_FP2_TO_G2"),
-			}
-			blsMap = newMap(b.CompiledIOP, G2, limits, blsMapSource)
-			if withCircuit {
-				blsMap = blsMap.WithMapCircuit(b.CompiledIOP, query.PlonkRangeCheckOption(16, 6, true))
-			}
-		},
-		dummy.Compile,
-	)
-
-	proof := wizard.Prove(cmp,
-		func(run *wizard.ProverRuntime) {
-			ct.Assign(run, "ID", "CIRCUIT_SELECTOR_MAP_FP2_TO_G2", "INDEX", "CT", "LIMB", "DATA_MAP_FP2_TO_G2", "RSLT_MAP_FP2_TO_G2")
-			blsMap.Assign(run)
-		})
-
-	if err := wizard.Verify(cmp, proof); err != nil {
-		t.Fatal("proof failed", err)
-	}
-	t.Log("proof succeeded")
-}
-
-func TestBlsMapG2NoCircuit(t *testing.T) {
-	testBlsG2Map(t, false)
+	testBlsMap(t, false, G2, "testdata/bls_g2_map_inputs.csv", limits)
 }
 
 func TestBlsMapG2WithCircuit(t *testing.T) {
-	testBlsG2Map(t, true)
+	limits := &Limits{
+		NbG2MapToInputInstances:   4,
+		NbG2MapToCircuitInstances: 1,
+	}
+	testBlsMap(t, true, G2, "testdata/bls_g2_map_inputs.csv", limits)
 }

prover/zkevm/prover/bls/module_msm_test.go

@@ -16,8 +16,11 @@ func testBlsMsm(t *testing.T, withCircuit bool, g group, path string, limits *Li
 	if err != nil {
 		t.Fatal(err)
 	}
-	if len(files) == 0 {
+	switch len(files) {
+	case 0:
 		t.Fatal("no csv files found, please run `go generate` to generate the test data")
+	case 1:
+		t.Log("single CSV file found. For complete testing, generate all test files with `go generate`")
 	}
 	// we test all files found
 	var cmp *wizard.CompiledIOP

prover/zkevm/prover/bls/module_pair_test.go

@@ -26,8 +26,11 @@ func testBlsPair(t *testing.T, withCircuit bool) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	if len(files) == 0 {
+	switch len(files) {
+	case 0:
 		t.Fatal("no csv files found, please run `go generate` to generate the test data")
+	case 1:
+		t.Log("single CSV file found. For complete testing, generate all test files with `go generate`")
 	}
 	// we test all files found
 	var cmp *wizard.CompiledIOP

prover/zkevm/prover/bls/module_pointeval_test.go

@@ -22,8 +22,11 @@ func testPointEval(t *testing.T, withCircuit bool) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	if len(files) == 0 {
+	switch len(files) {
+	case 0:
 		t.Fatal("no csv files found, please run `go generate` to generate the test data")
+	case 1:
+		t.Log("single CSV file found. For complete testing, generate all test files with `go generate`")
 	}
 	// we test all files found
 	var cmp *wizard.CompiledIOP