Fix ntt batch test timeout in golang tests

Author: jeremyfelder

wrappers/golang/curves/bls12377/tests/ntt_test.go

@@ -151,11 +151,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := bls12_377.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/curves/bls12381/tests/ntt_test.go

@@ -151,11 +151,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := bls12_381.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/curves/bn254/tests/ntt_test.go

@@ -151,11 +151,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := bn254.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/curves/bw6761/tests/ntt_test.go

@@ -151,11 +151,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := bw6_761.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/fields/babybear/tests/ntt_no_domain_test.go

@@ -61,11 +61,12 @@ func TestNttDeviceAsyncNoDomain(t *testing.T) {
 
 func TestNttBatchNoDomain(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := babybear_extension.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/fields/babybear/tests/ntt_test.go

@@ -80,11 +80,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := babybear.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/internal/generator/ntt/templates/ntt_no_domain_test.go.tmpl

@@ -64,11 +64,12 @@ func TestNttDeviceAsyncNoDomain(t *testing.T) {
 
 func TestNttBatchNoDomain(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := {{.FieldNoDomain}}.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize
 

wrappers/golang/internal/generator/ntt/templates/ntt_test.go.tmpl

@@ -163,11 +163,12 @@ func TestNttDeviceAsync(t *testing.T) {
 
 func TestNttBatch(t *testing.T) {
 	cfg := ntt.GetDefaultNttConfig()
+	largestTestSize := 12
 	largestBatchSize := 100
 	scalars := {{.Field}}.GenerateScalars(1 << largestTestSize * largestBatchSize)
 
 	for _, size := range []int{4, largestTestSize} {
-		for _, batchSize := range []int{1, 16, largestBatchSize} {
+		for _, batchSize := range []int{2, 16, largestBatchSize} {
 			testSize := 1 << size
 			totalSize := testSize * batchSize