Golangでのベンチマークテストの書き方

2017-11-29 Go

「プログラミング言語Go」の練習問題1.3の解答を作成してみました。スライスに含まれる文字列（strings）を連結する方法として、ループによる方法と strings.Join を使う方法のそれぞれの実行時間を計測するため、ベンチマークテストを作成します。コードは以下の通りです。（main.go の方の main関数は、ベンチマークテストには不要です。）

// Exercise 1.3 (p.9)
// for benchmark:
// $ go test -bench=.
// $ go test -bench=SlicePrintLoop
// $ go test -bench=SlicePrintJoin
package main

import (
        "strings"
)

func SlicePrintLoop(argslice []string) {
        s, sep := "", ""
        for _, arg := range argslice {
                s += sep + arg
                sep = " "
        }
}

func SlicePrintJoin(argslice []string) {
        strings.Join(argslice, " ")
}

func main() {
        slc := []string{"hello", "world"}

        SlicePrintLoop(slc)
        SlicePrintJoin(slc)
}
</pre> 
 
 
<pre class="lang:go decode:true " title="main_test.go" >package main

import "testing"

var slc []string

func init() {
        for i := 0; i &lt; 1000; i++ {
                slc = append(slc, "a")
        }
}

func BenchmarkSlicePrintLoop(b *testing.B) {
        for i := 0; i &lt; b.N; i++ {
                SlicePrintLoop(slc)
        }
}

func BenchmarkSlicePrintJoin(b *testing.B) {
        for i := 0; i &lt; b.N; i++ {
                SlicePrintJoin(slc)
        }
}

// Exercise 1.3 (p.9)

// for benchmark:

// $ go test -bench=.

// $ go test -bench=SlicePrintLoop

// $ go test -bench=SlicePrintJoin

package main

import (

"strings"

)

func SlicePrintLoop(argslice []string) {

s, sep := "", ""

for _, arg := range argslice {

s += sep + arg

sep = " "

}

func SlicePrintJoin(argslice []string) {

strings.Join(argslice, " ")

}

func main() {

slc := []string{"hello", "world"}

SlicePrintLoop(slc)

SlicePrintJoin(slc)

}

</pre>

<pre class="lang:go decode:true " title="main_test.go" >package main

import "testing"

var slc []string

func init() {

for i := 0; i < 1000; i++ {

slc = append(slc, "a")

}

func BenchmarkSlicePrintLoop(b *testing.B) {

for i := 0; i < b.N; i++ {

SlicePrintLoop(slc)

}

func BenchmarkSlicePrintJoin(b *testing.B) {

for i := 0; i < b.N; i++ {

SlicePrintJoin(slc)

}

ベンチマークテストを実行するには、これらのファイルのあるディレクトリで、以下のコマンドを実行します。

$ go test -bench=.

1	$ go test -bench=.

しばらくすると、以下のような結果が出力されます。

BenchmarkSlicePrintLoop-2               5000            287222 ns/op
BenchmarkSlicePrintJoin-2             200000             10510 ns/op
PASS
ok          exercises-gopl/1.3        3.691s

BenchmarkSlicePrintLoop-2 5000 287222 ns/op

BenchmarkSlicePrintJoin-2 200000 10510 ns/op

PASS

ok exercises-gopl/1.3 3.691s

今回は、1000個の文字列からなるスライスでベンチマークテストを行いましたが、ループによる方法よりも strings.Join を使う方法の方が27倍ほど速いことがわかります。

以上、簡単な例でしたが、ベンチマークテストのわかりやすい例だと思います。

参考文献

「プログラミング言語Go」 Alan A.A. Donovan (著),‎ Brian W. Kernighan (著),‎ 柴田芳樹 (翻訳)