go 单元测试和benchmark测试-蒲公英云

go 单元测试和benchmark测试

Go语言的表格驱动测试

main.go

package main
import (
    "fmt"
)
func isOneBitCharacter(bits []int) bool {
    var result bool
    var idx int
    for idx = 0; idx < len(bits); {
        if bits[idx] == 0 {
            idx++
            result = true
        } else {
            idx = idx + 2
            result = false
        }
    }
    return result
}
func main() {
    var bits = []int{1, 1, 1, 0}
    fmt.Println(isOneBitCharacter(bits))
}

main_test.go

package main
import "testing"
func Benchmark_isOneBitCharacter(b *testing.B) {
    type args struct {
        bits []int
    }
    tests := struct {
        name string
        args args
        want bool /*预期结果*/
    }{
   "1", args{[]int{1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0}}, false}
    for idx := 0; idx < b.N; idx++ {
        b.Run(tests.name, func(b *testing.B) {
            if got := isOneBitCharacter(tests.args.bits); got != tests.want {
                b.Errorf("isOneBitCharacter() = %v, want %v", got, tests.want)
            }
        })
    }
}
func Test_isOneBitCharacter(t *testing.T) {
    type args struct {
        bits []int
    }
    tests := []struct {
        name string
        args args
        want bool /*预期结果*/
    }{
        // TODO: Add test cases.
        {
   "1", args{[]int{1, 1, 1, 0}}, false},
        {
   "2", args{[]int{1, 1, 0, 0}}, true},
        {
   "3", args{[]int{1, 0, 0}}, true},
        {
   "4", args{[]int{}}, false},
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            if got := isOneBitCharacter(tt.args.bits); got != tt.want {
                t.Errorf("isOneBitCharacter() = %v, want %v", got, tt.want)
            }
        })
    }
}