欢迎访问宙启技术站
智能推送

函数测试用例(FunctionTestCase())的使用方法

发布时间:2023-12-26 05:41:50

FunctionTestCase()是Python标准库unittest中的一个测试用例类,可以用于测试函数的输入和输出是否符合预期。

用法:

1. 导入相关的模块和类:

import unittest
from unittest import FunctionTestCase

2. 定义待测试的函数:

def add(a, b):
    return a + b

3. 定义一个测试类,并继承自FunctionTestCase类,重写runTest方法:

class AddTestCase(unittest.FunctionTestCase):
    def runTest(self):
        result = add(2, 3)
        self.assertEqual(result, 5, "Addition failed")

4. 如果需要测试多个函数,可以定义多个测试类,并在测试方法中分别调用不同的函数。

5. 运行测试:

if __name__ == '__main__':
    unittest.main()

具体的使用例子如下:

import unittest
from unittest import FunctionTestCase

def add(a, b):
    return a + b

def subtract(a, b):
    return a - b

class AddTestCase(unittest.FunctionTestCase):
    def runTest(self):
        result = add(2, 3)
        self.assertEqual(result, 5, "Addition failed")

class SubtractTestCase(unittest.FunctionTestCase):
    def runTest(self):
        result = subtract(5, 2)
        self.assertEqual(result, 3, "Subtraction failed")

if __name__ == '__main__':
    unittest.main()

在上面的例子中,我们定义了两个待测试的函数add()和subtract(),分别用于做加法和减法。

然后我们定义了两个测试类AddTestCase和SubtractTestCase,它们都继承自FunctionTestCase类,并重写了runTest方法,在测试方法中调用对应的函数,并使用assertEqual方法进行断言,判断函数的输出是否符合预期。

最后,我们运行unittest.main()来执行测试。

总结:

FunctionTestCase类可以简化函数的测试,通过继承并重写runTest方法,我们可以在测试方法中直接调用待测试的函数,并使用断言判断函数的输出是否符合预期。这样就可以有效地进行函数测试并提供可靠的结果。