首页  >  文章  >  后端开发  >  如何在单元测试中比较 Go 切片和 NaN 元素?

如何在单元测试中比较 Go 切片和 NaN 元素?

WBOY
WBOY转载
2024-02-12 13:00:09964浏览

如何在单元测试中比较 Go 切片和 NaN 元素?

php小编百草在单元测试中比较 Go 切片和 NaN 元素是一个常见的问题。在处理切片时,我们经常需要比较两个切片是否相等,但当切片包含 NaN 元素时,比较会变得复杂。NaN 是一个特殊的浮点数,它代表不是一个数值。在 Go 中,使用 math包中的 IsNaN 函数来判断一个浮点数是否为 NaN。通过对切片中的每个元素进行遍历,并使用 IsNaN 函数判断是否为 NaN,我们可以实现切片的比较操作。

问题内容

我需要在单元测试中比较 2 个切片,我认为 assert 包应该解决这个问题,但它不适用于 nan:

import (
    "github.com/stretchr/testify/assert"
)

func callme() []float64 {
    return []float64{math.nan()}
}

func testcallme(t *testing.t) {
    assert.equal(t, []float64{math.nan()}, callme())
}

输出为:

Error:          Not equal: 
                            expected: []float64{NaN}
                            actual  : []float64{NaN}
                            
                            Diff:
            Test:           TestCallMe

我知道 nan 的一个属性不等于其自身,但另一方面我从函数中收到了预期的结果。

我想知道如何解决这个问题 - 获得简洁的单元测试断言和清晰的输出?

作为替代方案,我可以避免使用 assert 并在两个切片的每个元素上调用 math.isnan 并在两个切片的每个元素上调用 math.isnan ,但这在单元测试中看起来会格外冗长。

解决方法

github.com/google/go-cmp/cmp 包进行更复杂的比较。 cmpopts.equatenans 可以用于轻松比较可能包含 nan 的数据结构。

package calc_test

import (
    "math"
    "testing"

    "github.com/google/go-cmp/cmp"
    "github.com/google/go-cmp/cmp/cmpopts"
)

func calc(n float64) []float64 {
    return []float64{n, math.NaN()}
}

func TestCalc(t *testing.T) {
    want := []float64{1, math.NaN()}

    got := calc(1)  // PASS.
    if !cmp.Equal(got, want, cmpopts.EquateNaNs()) {
        t.Errorf("calc: got = %v; want = %v", got, want)
    }

    got = calc(2)   // FAIL with differences.
    if diff := cmp.Diff(want, got, cmpopts.EquateNaNs()); diff != "" {
        t.Errorf("calc: diff (-want +got) = \n%s", diff)
    }
}

以上是如何在单元测试中比较 Go 切片和 NaN 元素?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除