测试函数,python代码
时间: 2023-02-06 12:18:26 浏览: 136
测试函数是一种特殊的函数,它的主要目的是用来检查其他函数的正确性。测试函数通常会调用被测试的函数,并使用断言语句来检查函数的输出是否符合预期。
下面是一个简单的 Python 测试函数的例子,它测试了一个名为 `square` 的函数,该函数接受一个数字作为输入,并返回该数字的平方。
```
def test_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 unittest
class 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` 会自动检测测试函数并运行它们,并在测试失败时报告错误。