test: half-GCD test with bigger integers

Consensys · Sep 21, 2024 · 6f69c7e · 6f69c7e
1 parent dda7a27
commit 6f69c7e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ecc/eisenstein_test.go b/ecc/eisenstein_test.go
@@ -1,8 +1,8 @@
 package ecc
 
 import (
+	"crypto/rand"
 	"math/big"
-	"math/rand"
 	"testing"
 
 	"github.com/leanovate/gopter"
@@ -245,8 +245,9 @@ func TestEisensteinHalfGCD(t *testing.T) {
 // GenNumber generates a random integer
 func GenNumber() gopter.Gen {
 	return func(genParams *gopter.GenParameters) *gopter.GenResult {
-		elmt := *big.NewInt(rand.Int63())
-		genResult := gopter.NewGenResult(elmt, gopter.NoShrinker)
+		var prime, _ = new(big.Int).SetString("7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffed", 16) // 2^255 - 19
+		elmt, _ := rand.Int(rand.Reader, prime)
+		genResult := gopter.NewGenResult(*elmt, gopter.NoShrinker)
 		return genResult
 	}
 }