写一段unittest的代码
时间: 2023-07-10 20:19:07 浏览: 142
好的,以下是一个简单的unittest代码示例:
```python
import unittest
def add_numbers(a, b):
return a + b
class TestAddNumbers(unittest.TestCase):
def test_add_positive_numbers(self):
self.assertEqual(add_numbers(2, 3), 5)
def test_add_negative_numbers(self):
self.assertEqual(add_numbers(-2, -3), -5)
def test_add_mixed_numbers(self):
self.assertEqual(add_numbers(2, -3), -1)
self.assertEqual(add_numbers(-2, 3), 1)
if __name__ == '__main__':
unittest.main()
```
这个示例定义了一个`add_numbers`函数,用于计算两个数字的和。然后,通过继承`unittest.TestCase`类来创建一个测试类,其中定义了三个测试方法,分别测试了两个正数相加、两个负数相加以及一个正数和一个负数相加的情况。
在`main`函数中,我们调用`unittest.main()`来运行所有的测试方法,并检查它们是否通过了断言。如果在测试过程中发现了任何错误,unittest会抛出一个异常,并显示错误的详细信息。
阅读全文