go语言单元测试demo

最新推荐文章于 2024-03-11 19:44:17 发布

weixin_33806300

最新推荐文章于 2024-03-11 19:44:17 发布

阅读量112

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/code-style/archive/2013/06/03/3115795.html

本文介绍了一个简单的Go语言数学包，包含加法和求最大数两个功能，并通过测试文件验证了其正确性。

package mymath

func Add(a, b int) int {
    return a + b
}
func Max(a, b int) (ret int) {
    ret = a
    if b > a {
        ret = b
    }
    return
}

package mymath_test

import (
    "mymath"
    "testing"
)

type mathTest struct {
    a, b, ret int
}

var addTest = []mathTest{
    mathTest{4, 6, 10},
    mathTest{5, 6, 11},
    mathTest{2, -6, -4},
}

var maxTest = []mathTest{
    mathTest{3, 5, 5},
    mathTest{-3, 5, 5},
    mathTest{-3, -5, -3},
}

func TestAdd(t *testing.T) {
    for _, v := range addTest {
        ret := mymath.Add(v.a, v.b)
        if ret != v.ret {
            t.Errorf("%d add %d, want %d, but get %d", v.a, v.b, v.ret, ret)
        }
    }
}
func TestMax(t *testing.T) {
    for _, v := range maxTest {
        ret := mymath.Max(v.a, v.b)
        if ret != v.ret {
            t.Errorf("the max number between %d and %d is want %d, but get %d", v.a, v.b, v.ret, ret)
        }
    }
}