math/big: panic if n <= 0 for ProbablyPrime
Fixes #9509
Change-Id: I3b86745d38e09093fe2f4b918d774bd6608727d7
Reviewed-on: https://go-review.googlesource.com/2313
Reviewed-by: Robert Griesemer <gri@golang.org>
diff --git a/src/math/big/int_test.go b/src/math/big/int_test.go
index 6070cf3..af3af91 100644
--- a/src/math/big/int_test.go
+++ b/src/math/big/int_test.go
@@ -989,6 +989,21 @@
break
}
}
+
+ // check that ProbablyPrime panics if n <= 0
+ c := NewInt(11) // a prime
+ for _, n := range []int{-1, 0, 1} {
+ func() {
+ defer func() {
+ if n <= 0 && recover() == nil {
+ t.Fatalf("expected panic from ProbablyPrime(%d)", n)
+ }
+ }()
+ if !c.ProbablyPrime(n) {
+ t.Fatalf("%v should be a prime", c)
+ }
+ }()
+ }
}
type intShiftTest struct {
