如何测试使用可变参数列表的 Golang 函数
在 Golang 中,可变参数列表允许函数接受任意数量的参数。这在需要在函数中处理数量不确定的值时非常有用。然而,测试使用可变参数列表的函数可能会有些棘手。
编写可变参数列表函数
让我们以一个计算切片中所有元素之和的函数为例:
func Sum(nums ...int) int { sum := 0 for _, num := range nums { sum += num } return sum }
编写测试
为了测试 Sum
函数,我们可以编写一个单元测试:
import ( "testing" ) func TestSum(t *testing.T) { tests := []struct { input []int want int }{ {[]int{1, 2, 3}, 6}, {[]int{4, 5, 6}, 15}, {[]int{}, 0}, } for _, test := range tests { got := Sum(test.input...) if got != test.want { t.Errorf("Sum(%v) = %d, want %d", test.input, got, test.want) } } }
在 TestSum
函数中:
tests
切片,其中包含了不同的输入值和期望输出。...
展开运算符将 input
切片传递给 Sum
函数,作为可变参数列表。got
(实际输出)等于 want
(期望输出)。实战案例
下面是一个使用可变参数列表函数的实际示例:
import ( "fmt" ) func PrintValues(values ...interface{}) { for _, value := range values { fmt.Println(value) } } func main() { PrintValues(1, "hello", 3.14) }
在 main
函数中,我们调用 PrintValues
函数并传递不同的值。该函数将打印每个值。
输出:
1 hello 3.14