Go单元测试

简介

单元测试是go语言级别提供的完整功能,测试代码以*_test.go命名,单元测试的case以Test开头,性能测试case以Benchmark开头,运行测试命令:go test <test_file_list>

例子

实现排列组合函数以及对应的单元测试和性能测试

创建工程目录结构

目录结构说明参见规范-项目

└── src
 └── hmath
 ├── hmath.go
 └── hmath_test.go

实现排列组合函数

// src/hmath/hmath.go
package hmath
func combination(m, n int) int {
 if n > m-n {
 n = m - n
 }
 c := 1
 for i := 0; i < n; i++ {
 c *= m - i
 c /= i + 1
 }
 return c
}

实现单元测试和性能测试

// src/hmath/hmath_test.go
package hmath
import (
 "math/rand"
 "testing"
)
// 单元测试
// 测试全局函数,以TestFunction命名
// 测试类成员函数,以TestClass_Function命名
func TestCombination(t *testing.T) {
 // 这里定义一个临时的结构体来存储测试case的参数以及期望的返回值
 for _, unit := range []struct {
 m int
 n int
 expected int
 }{
 {1, 0, 1},
 {4, 1, 4},
 {4, 2, 6},
 {4, 3, 4},
 {4, 4, 1},
 {10, 1, 10},
 {10, 3, 120},
 {10, 7, 120},
 } {
 // 调用排列组合函数,与期望的结果比对,如果不一致输出错误
 if actually := combination(unit.m, unit.n); actually != unit.expected {
 t.Errorf("combination: [%v], actually: [%v]", unit, actually)
 }
 }
}
// 性能测试
func BenchmarkCombination(b *testing.B) {
 // b.N会根据函数的运行时间取一个合适的值
 for i := 0; i < b.N; i++ {
 combination(i+1, rand.Intn(i+1))
 }
}
// 并发测试
func BenchmarkCombinationParallel(b *testing.B) {
 // 测试一个对象或者函数在多线程的场景下面是否安全
 b.RunParallel(func(pb *testing.PB) {
 for pb.Next() {
 m := rand.Intn(100) + 1
 n := rand.Intn(m)
 combination(m, n)
 }
 })
}

运行单元测试和性能测试

export GOPATH=$(pwd)
go test src/hmath/*.go # 单元测试
go test --cover src/hmath/*.go # 单元测试覆盖率
go test -bench=. src/hmath/*.go # 性能测试

上面命令的输出如下:

hatlonely@localhost: ~/hatlonely/github/tmp $ go test src/hmath/hmath*.go
ok command-line-arguments 0.005s
hatlonely@localhost: ~/hatlonely/github/tmp $ go test --cover src/hmath/hmath*.go
ok command-line-arguments 0.005s coverage: 100.0% of statements
hatlonely@localhost: ~/hatlonely/github/tmp $ go test -bench=. src/hmath/*.go
BenchmarkCombination-8 100000 217618 ns/op
BenchmarkCombinationParallel-8 3000000 401 ns/op
PASS
ok command-line-arguments 23.599s

本文来自:Segmentfault

感谢作者:hatlonely

查看原文:Go单元测试

• 编辑
• 预览