使用Unittest框架对add函数进行测试
时间: 2024-05-13 21:19:35 浏览: 120
好的,假设我们有一个名为`add`的函数,它接受两个参数并返回它们的和。我们可以使用Unittest框架编写测试用例来测试这个函数。
```python
import unittest
def add(a, b):
return a + b
class TestAddFunction(unittest.TestCase):
def test_add_positive_numbers(self):
self.assertEqual(add(2, 3), 5)
def test_add_negative_numbers(self):
self.assertEqual(add(-2, -3), -5)
def test_add_zero(self):
self.assertEqual(add(0, 0), 0)
if __name__ == '__main__':
unittest.main()
```
在这个例子中,我们定义了一个名为`TestAddFunction`的测试类,它继承自`unittest.TestCase`。我们定义了三个测试方法`test_add_positive_numbers`,`test_add_negative_numbers`和`test_add_zero`来测试`add`函数的三种情况。
每个测试方法都使用`assertEqual`方法来检查`add`函数的返回值是否与预期值相同。在本例中,我们测试了正数相加、负数相加和零相加的情况。
最后,我们在脚本的末尾使用`unittest.main()`来运行测试用例。如果所有测试都通过,输出结果应该为OK;否则,会输出测试失败的详细信息。
阅读全文