unittest代码举例
时间: 2024-09-08 20:04:34 浏览: 49
在Python中,unittest是一个用于编写和运行测试用例的框架。下面是一个简单的例子,展示了如何使用unittest来测试一个简单的函数。
假设我们有一个简单的函数,用于判断一个数字是否是偶数:
```python
# 文件名:math_functions.py
def is_even(number):
"""判断一个数是否是偶数"""
return number % 2 == 0
```
接下来,我们将使用unittest框架来测试这个函数:
```python
# 文件名:test_math_functions.py
import unittest
from math_functions import is_even
class TestIsEvenFunction(unittest.TestCase):
def test_is_even(self):
self.assertTrue(is_even(2), "2 应该是偶数")
self.assertTrue(is_even(0), "0 应该是偶数")
self.assertFalse(is_even(1), "1 不应该是偶数")
self.assertFalse(is_even(-1), "-1 不应该是偶数")
if __name__ == '__main__':
unittest.main()
```
在这个测试类`TestIsEvenFunction`中,我们定义了一个方法`test_is_even`,它使用`assertTrue`和`assertFalse`方法来断言`is_even`函数的返回值。如果`is_even`函数的返回值与预期不符,测试就会失败。
要运行这个测试,你可以将测试代码保存为`test_math_functions.py`,然后在命令行中执行:
```
python -m unittest test_math_functions.py
```
这将执行所有的测试方法,并且会输出测试结果。
阅读全文