Go语言单元测试：初学者入门指南

2023-12-01 21:33:52

在 Go 语言中进行单元测试和基准测试

单元测试：确保代码的可靠性

单元测试是一种至关重要的测试技术，用于检查代码的最小独立单元，例如函数或方法。通过识别代码中的逻辑缺陷和错误，单元测试可确保代码在不同场景下都能正常运行。在 Go 语言中，单元测试主要使用 testing 包实现。

一个单元测试示例如下：

package main

import "testing"

func TestAdd(t *testing.T) {
    tests := []struct {
        a, b, expected int
    }{
        {1, 2, 3},
        {3, 4, 7},
        {-1, -2, -3},
    }

    for _, tt := range tests {
        result := Add(tt.a, tt.b)
        if result != tt.expected {
            t.Errorf("Add(%d, %d) = %d, want %d", tt.a, tt.b, result, tt.expected)
        }
    }
}

func Add(a, b int) int {
    return a + b
}

在这个示例中，TestAdd 函数是一个单元测试函数，它包含了多个测试用例。每个测试用例都检查了 Add 函数在不同输入下的期望输出。如果实际输出与期望输出不一致，就会报告错误。

基准测试：评估代码的性能

基准测试是一种测试技术，用于衡量代码的性能，包括执行时间和内存使用情况。通过识别代码中的性能瓶颈，基准测试可以帮助你优化代码以获得最佳性能。Go 语言中的基准测试通常使用 testing 包的 B 类型实现。

一个基准测试示例如下：

package main

import "testing"

func BenchmarkAdd(b *testing.B) {
    for i := 0; i < b.N; i++ {
        _ = Add(1, 2)
    }
}

func Add(a, b int) int {
    return a + b
}