函数测试用例(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方法,我们可以在测试方法中直接调用待测试的函数,并使用断言判断函数的输出是否符合预期。这样就可以有效地进行函数测试并提供可靠的结果。
