如何测试使用可变参数列表的 Golang 函数
在 Golang 中，可变参数列表允许函数接受任意数量的参数。这在需要在函数中处理数量不确定的值时非常有用。然而，测试使用可变参数列表的函数可能会有些棘手。
编写可变参数列表函数
让我们以一个计算切片中所有元素之和的函数为例：
立即学习“go语言免费学习笔记（深入）”；
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
