Sped up TestIsPrimeAgainstBaseline

fxt · fxt · commit d77956c07e15 · 2015-08-19T16:38:41.000-04:00
diff --git a/benchmark_test.go b/benchmark_test.go
@@ -24,8 +24,8 @@ package primes_test
 
 import (
 	"math"
+	"math/rand"
 	"testing"
-	"testing/quick"
 
 	"github.com/fxtlabs/primes"
 )
@@ -88,11 +88,16 @@ func TestIsPrimeAgainstBaseline(t *testing.T) {
 		p := primes.IsPrime(n)
 		q := baselineIsPrime(n)
 		if p != q {
-			t.Errorf("ISPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)
+			t.Errorf("IsPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)
 		}
 	}
-	if err := quick.CheckEqual(primes.IsPrime, baselineIsPrime, &quick.Config{MaxCount: 50}); err != nil {
-		t.Error(err)
+	for i := 0; i < 10000; i++ {
+		n := rand.Intn(math.MaxInt32)
+		p := primes.IsPrime(n)
+		q := baselineIsPrime(n)
+		if p != q {
+			t.Errorf("IsPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)
+		}
 	}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -24,8 +24,8 @@ package primes_test`
`24`	`24`
`25`	`25`	`import (`
`26`	`26`	`"math"`
	`27`	`+ "math/rand"`
`27`	`28`	`"testing"`
`28`		`- "testing/quick"`
`29`	`29`
`30`	`30`	`"github.com/fxtlabs/primes"`
`31`	`31`	`)`
`@@ -88,11 +88,16 @@ func TestIsPrimeAgainstBaseline(t *testing.T) {`
`88`	`88`	`p := primes.IsPrime(n)`
`89`	`89`	`q := baselineIsPrime(n)`
`90`	`90`	`if p != q {`
`91`		`- t.Errorf("ISPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)`
	`91`	`+ t.Errorf("IsPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)`
`92`	`92`	`}`
`93`	`93`	`}`
`94`		`- if err := quick.CheckEqual(primes.IsPrime, baselineIsPrime, &quick.Config{MaxCount: 50}); err != nil {`
`95`		`- t.Error(err)`
	`94`	`+ for i := 0; i < 10000; i++ {`
	`95`	`+ n := rand.Intn(math.MaxInt32)`
	`96`	`+ p := primes.IsPrime(n)`
	`97`	`+ q := baselineIsPrime(n)`
	`98`	`+ if p != q {`
	`99`	`+ t.Errorf("IsPrimeAgainstBaseline(%d) == %v, want %v", n, p, q)`
	`100`	`+ }`
`96`	`101`	`}`
`97`	`102`	`}`
`98`	`103`