如何在 Golang 中编写参数化的测试函数？

import (
    "testing"
)

// 定义一个测试函数
func TestMyFunction(t *testing.T) {
    // 使用 t.Run 创建参数化测试用例
    tests := []struct {
        input    int
        expected int
    }{
        {1, 1},
        {2, 4},
        {3, 9},
    }

    for _, tt := range tests {
        t.Run(fmt.Sprintf("input:%v", tt.input), func(t *testing.T) {
            // 使用输入值进行测试
            result := myFunction(tt.input)
            if result != tt.expected {
                t.Errorf("expected %v, got %v", tt.expected, result)
            }
        })
    }
}

// 待测试的函数
func myFunction(input int) int {
    return input * input
}

func TestCalculateDistance(t *testing.T) {
    tests := []struct {
        pt1, pt2  Point
        expected float64
    }{
        {Point{0, 0}, Point{1, 1}, 1.4142135623730951},
        {Point{-2, 3}, Point{-4, 9}, 8.06225774829855},
        {Point{5, -1}, Point{1, -3}, 5.0},
    }

    for _, tt := range tests {
        t.Run(fmt.Sprintf("input: (%v, %v)", tt.pt1, tt.pt2), func(t *testing.T) {
            result := calculateDistance(tt.pt1, tt.pt2)
            if math.Abs(result-tt.expected) > 0.00001 {
                t.Errorf("expected %v, got %v", tt.expected, result)
            }
        })
    }
}

type Point struct {
    x, y int
}

func calculateDistance(p1, p2 Point) float64 {
    dx := float64(p1.x - p2.x)
    dy := float64(p1.y - p2.y)
    return math.Sqrt(dx*dx + dy*dy)
}