多返回值函数的表格测试

huangapple go评论90阅读模式
英文:

Table tests for multi-return value function

问题

我正在学习Go语言,并且在研究表驱动测试时遇到了以下问题:

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

// 将整数除以2,并返回是否为偶数的布尔值。
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)
        }
    }
}
英文:

I'm cutting my teeth on Go and after digging into table driven tests I ran into the following problem:

I have a function that returns multiple values

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

I know that for half(1) the return value should be 0, false and for half(2) it should match 1, true, but I can't seem to figure out how to put this on a table.

How would one go to have something that resembles the following?

var halfTests = []struct {
	in  int
	out string
}{
	{1, &lt;0, false&gt;},
	{3, &lt;1, true&gt;},
}

Is there any other, more idiomatic way of doing this?

For reference, here's a test for something that resembles a FizzBuzz function, using tables:

var fizzbuzzTests = []struct {
	in  int
	out string
}{
	{1, &quot;1&quot;},
	{3, &quot;Fizz&quot;},
	{5, &quot;Buzz&quot;},
	{75, &quot;FizzBuzz&quot;},
}

func TestFizzBuzz(t *testing.T) {
	for _, tt := range fizzbuzzTests {
		s := FizzBuzz(tt.in)
		if s != tt.out {
			t.Errorf(&quot;Fizzbuzz(%d) =&gt; %s, want %s&quot;, tt.in, s, tt.out)
		}
	}
}

答案1

得分: 11

只需在结构体中添加另一个字段来保存第二个返回值。示例代码如下:

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

请注意,这里将第二个返回值命名为t,以避免与测试函数中的t变量冲突。

英文:

Just add another field to your struct that holds the second return value. Example:

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

Your testing function would look like the following:

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

huangapple
  • 本文由 发表于 2015年12月17日 06:33:05
  • 转载请务必保留本文链接:https://go.coder-hub.com/34323408.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定