在 golang 中為泛型函數編寫單元測試

php小編香蕉為您帶來了一篇關於在Golang中為泛型函數編寫單元測試的文章。 Golang是一種強型別的程式語言，然而，它在泛型方面的支援卻相對較弱。因此，為泛型函數編寫單元測試可能會有一些挑戰。本文將向您介紹如何在Golang中有效地為泛型函數編寫單元測試，以確保程式碼的品質和可靠性。無論您是初學者還是有經驗的開發者，本文都將為您提供實用的技巧和方法，幫助您輕鬆應對泛型函數的單元測試。讓我們一起來看看吧！

問題內容

我有這個簡單的通用函數，可以從 map# 檢索金鑰

// getmapkeys returns the keys of a map
func getmapkeys[t comparable, u any](m map[t]u) []t {
    keys := make([]t, len(m))
    i := 0
    for k := range m {
        keys[i] = k
        i++
    }
    return keys
}

我正在嘗試為其編寫表驅動的單元測試，如下所示：

var testunitgetmapkeys = []struct {
    name     string
    inputmap interface{}
    expected interface{}
}{
    {
        name:     "string keys",
        inputmap: map[string]int{"foo": 1, "bar": 2, "baz": 3},
        expected: []string{"foo", "bar", "baz"},
    },
    {
        name:     "int keys",
        inputmap: map[int]string{1: "foo", 2: "bar", 3: "baz"},
        expected: []int{1, 2, 3},
    },
    {
        name:     "float64 keys",
        inputmap: map[float64]bool{1.0: true, 2.5: false, 3.1415: true},
        expected: []float64{1.0, 2.5, 3.1415},
    },
}

但是，以下程式碼失敗

func (us *unitutilsuite) testunitgetmapkeys() {
    for i := range testunitgetmapkeys {
        us.t().run(testunitgetmapkeys[i].name, func(t *testing.t) {
            gotkeys := getmapkeys(testunitgetmapkeys[i].inputmap)
        })
    }
}

與

type interface{} of testunitgetmapkeys[i].inputmap does not match map[t]u (cannot infer t and u)

這已透過明確轉換修復

gotKeys := getMapKeys(testUnitGetMapKeys[i].inputMap.(map[string]string))

有沒有辦法自動化這些測試，而不必為每個輸入測試變數執行明確轉換？

解決方法

請注意，除非您的泛型函數除了泛型邏輯之外還執行一些特定於類型的邏輯，否則透過針對不同類型測試該函數您將一無所獲。此函數的通用邏輯對於類型參數的類型集中的所有類型都是相同的，因此可以使用單一類型完全執行。

但是如果您想針對不同類型執行測試，您可以簡單地執行以下操作：

var testUnitGetMapKeys = []struct {
    name string
    got  any
    want any
}{
    {
        name: "string keys",
        got:  getMapKeys(map[string]int{"foo": 1, "bar": 2, "baz": 3}),
        want: []string{"foo", "bar", "baz"},
    },
    {
        name: "int keys",
        got:  getMapKeys(map[int]string{1: "foo", 2: "bar", 3: "baz"}),
        want: []int{1, 2, 3},
    },
    {
        name: "float64 keys",
        got:  getMapKeys(map[float64]bool{1.0: true, 2.5: false, 3.1415: true}),
        want: []float64{1.0, 2.5, 3.1415},
    },
}

// ...

func (us *UnitUtilSuite) TestUnitGetMapKeys() {
    for _, tt := range testUnitGetMapKeys {
        us.T().Run(tt.name, func(t *testing.T) {
            if !reflect.DeepEqual(tt.got, tt.want) {
                t.Errorf("got=%v; want=%v", tt.got, tt.want)
            }
        })
    }
}

以上是在 golang 中為泛型函數編寫單元測試的詳細內容。更多資訊請關注PHP中文網其他相關文章！