php中文网

如何测试使用可变参数列表的 Golang 函数?

php中文网

如何测试使用可变参数列表的 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

以上就是如何测试使用可变参数列表的 Golang 函数?的详细内容,更多请关注php中文网其它相关文章!