多返回值函数的表测试



我在Go上咬紧牙关,在深入研究表驱动的测试后,我遇到了以下问题:

我有一个返回多个值的函数

// Halves an integer and and returns true if it was even or false if it was odd.
func half(n int) (int, bool) {
    h := n / 2
    e := n%2 == 0
    return h, e
}

我知道对于half(1),返回值应该是0, false,对于half(2),它应该匹配1, true,但我似乎不知道如何将其放在表上。

一个人怎么会有类似以下的东西呢?

var halfTests = []struct {
    in  int
    out string
}{
    {1, <0, false>},
    {3, <1, true>},
}

还有其他更惯用的方法吗?

作为参考,这里有一个类似FizzBuzz函数的测试,使用表格:

var fizzbuzzTests = []struct {
    in  int
    out string
}{
    {1, "1"},
    {3, "Fizz"},
    {5, "Buzz"},
    {75, "FizzBuzz"},
}
func TestFizzBuzz(t *testing.T) {
    for _, tt := range fizzbuzzTests {
        s := FizzBuzz(tt.in)
        if s != tt.out {
            t.Errorf("Fizzbuzz(%d) => %s, want %s", tt.in, s, tt.out)
        }
    }
}

只需在结构中添加另一个包含第二个返回值的字段。示例:

var halfTests = []struct {
    in   int
    out1 int
    out2 bool
}{
    {1, 0, false},
    {3, 1, true},
}

您的测试功能如下所示:

func TestHalf(t *testing.T) {
    for _, tt := range halfTests {
        s, t := half(tt.in)
        if s != tt.out1 || t != tt.out2 {
            t.Errorf("half(%d) => %d, %v, want %d, %v", tt.in, s, t, tt.out1, tt.out2)
        }
    }
}

相关内容

  • 没有找到相关文章