libgo/go/math/big/gcd_test.go

   1 // Copyright 2012 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 // This file implements a GCD benchmark.
   6 // Usage: go test math/big -test.bench GCD
   7
   8 package big
   9
  10 import (
  11         "math/rand"
  12         "testing"
  13 )
  14
  15 // randInt returns a pseudo-random Int in the range [1<<(size-1), (1<<size) - 1]
  16 func randInt(r *rand.Rand, size uint) *Int {
  17         n := new(Int).Lsh(intOne, size-1)
  18         x := new(Int).Rand(r, n)
  19         return x.Add(x, n) // make sure result > 1<<(size-1)
  20 }
  21
  22 func runGCD(b *testing.B, aSize, bSize uint) {
  23         b.StopTimer()
  24         var r = rand.New(rand.NewSource(1234))
  25         aa := randInt(r, aSize)
  26         bb := randInt(r, bSize)
  27         b.StartTimer()
  28         for i := 0; i < b.N; i++ {
  29                 new(Int).GCD(nil, nil, aa, bb)
  30         }
  31 }
  32
  33 func BenchmarkGCD10x10(b *testing.B)         { runGCD(b, 10, 10) }
  34 func BenchmarkGCD10x100(b *testing.B)        { runGCD(b, 10, 100) }
  35 func BenchmarkGCD10x1000(b *testing.B)       { runGCD(b, 10, 1000) }
  36 func BenchmarkGCD10x10000(b *testing.B)      { runGCD(b, 10, 10000) }
  37 func BenchmarkGCD10x100000(b *testing.B)     { runGCD(b, 10, 100000) }
  38 func BenchmarkGCD100x100(b *testing.B)       { runGCD(b, 100, 100) }
  39 func BenchmarkGCD100x1000(b *testing.B)      { runGCD(b, 100, 1000) }
  40 func BenchmarkGCD100x10000(b *testing.B)     { runGCD(b, 100, 10000) }
  41 func BenchmarkGCD100x100000(b *testing.B)    { runGCD(b, 100, 100000) }
  42 func BenchmarkGCD1000x1000(b *testing.B)     { runGCD(b, 1000, 1000) }
  43 func BenchmarkGCD1000x10000(b *testing.B)    { runGCD(b, 1000, 10000) }
  44 func BenchmarkGCD1000x100000(b *testing.B)   { runGCD(b, 1000, 100000) }
  45 func BenchmarkGCD10000x10000(b *testing.B)   { runGCD(b, 10000, 10000) }
  46 func BenchmarkGCD10000x100000(b *testing.B)  { runGCD(b, 10000, 100000) }
  47 func BenchmarkGCD100000x100000(b *testing.B) { runGCD(b, 100000, 100000) }