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)
            }
        })
    }
}

引用

https://blog.youkuaiyun.com/x356982611/article/details/80443252