这里是个简单的例子，用go testing 作单元测试压力测试，压力测试可用作性能测试，对比不同的实现方式的性能。
测试的包文件夹 mypack 共有三个文件 :
- mypack.go 单元功能文件
- mypack_test.go 单元测试文件
- mypack_benchmark_test.go 压力测试文件
mypack.go 文件的内容：
1
2
3
4
5
6
7
8
9
10
11
12
13
package mypack
import (
"errors"
)
func Division(a, b float64) (float64, error) {
if b == 0 {
return 0, errors.New("除数不能为0")
}
return a / b, nil
}
mypack_test.go 的文件内容：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package mypack
import (
"testing"
)
// 输入的参数列表
var fibTests = []struct {
a float64 // input
b float64 // input
expected float64 // expected result
}{
{4, 2, 2},
{6, 3, 2},
{100, 10, 10},
{8, 2, 4},
{15, 5, 3},
}
func Test_Division_1(t *testing.T) {
for _, fib := range fibTests {
if i, e := Division(fib.a, fib.b); i != fib.expected || e != nil { //try a unit test on function
t.Error("除法函数测试没通过") // 如果不是如预期的那么就报错
} else {
t.Log("第一个测试通过了") //记录一些你期望记录的信息
}
}
}
func Test_Division_2(t *testing.T) {
if _, e := Division(6, 0); e == nil { //try a unit test on function
t.Error("Division did not work as expected.") // 如果不是如预期的那么就报错
} else {
t.Log("one test passed.", e) //记录一些你期望记录的信息
}
}
mypack_benchmark_test.go 的文件内容：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package mypack
import (
"testing"
)
func Benchmark_Division(b *testing.B) {
for i := 0; i < b.N; i++ { //use b.N for looping
Division(4, 5)
}
}
func Benchmark_TimeConsumingFunction(b *testing.B) {
b.StopTimer() //调用该函数停止压力测试的时间计数
//做一些初始化的工作,例如读取文件数据,数据库连接之类的,
//这样这些时间不影响我们测试函数本身的性能
b.StartTimer() //重新开始时间
for i := 0; i < b.N; i++ {
Division(4, 5)
}
}
切换到这个文件夹下实行：
1go test -v -bench=".*"
输出：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
=== RUN Test_Division_1
--- PASS: Test_Division_1 (0.00s)
mypack_test.go:25: 第一个测试通过了
mypack_test.go:25: 第一个测试通过了
mypack_test.go:25: 第一个测试通过了
mypack_test.go:25: 第一个测试通过了
mypack_test.go:25: 第一个测试通过了
=== RUN Test_Division_2
--- PASS: Test_Division_2 (0.00s)
mypack_test.go:34: one test passed. 除数不能为0
PASS
Benchmark_Division 200000000 6.96 ns/op
Benchmark_TimeConsumingFunction 200000000 6.94 ns/op
ok _/home/ubuntu/tmp/mypack 4.203s