测试函数是一种特殊的函数,它的主要目的是用来检查其他函数的正确性。测试函数通常会调用被测试的函数,并使用断言语句来检查函数的输出是否符合预期。
下面是一个简单的 Python 测试函数的例子,它测试了一个名为 square
的函数,该函数接受一个数字作为输入,并返回该数字的平方。
deftest_square():
assert square(2) == 4
assert square(-2) == 4
assert square(0) == 0
def square(x):
return x ** 2
在这个例子中,测试函数 test_square
调用了被测试的函数 square
并使用断言语句来检查其输出是否符合预期。
要运行测试函数,可以使用 Python 的 unittest
模块。例如,可以使用以下代码来运行上述测试函数:
import unittestclass TestSquare(unittest.TestCase):
def test_square(self):
self.assertEqual(square(2), 4)
self.assertEqual(square(-2), 4)
self.assertEqual(square(0), 0)
if __name__ == '__main__':
unittest.main()
在运行时,unittest
会自动检测测试函数并运行它们,并在测试失败时报告错误。