写一个unittest测试
时间: 2024-05-13 19:20:54 浏览: 13
这是一个简单的unittest测试,测试一个函数是否能正确计算两个数的和。
```python
import unittest
def add_numbers(a, b):
return a + b
class TestAddNumbers(unittest.TestCase):
def test_add_positive_numbers(self):
result = add_numbers(3, 4)
self.assertEqual(result, 7)
def test_add_negative_numbers(self):
result = add_numbers(-3, -4)
self.assertEqual(result, -7)
def test_add_zero_to_number(self):
result = add_numbers(0, 5)
self.assertEqual(result, 5)
def test_add_float_numbers(self):
result = add_numbers(3.5, 4.5)
self.assertEqual(result, 8.0)
if __name__ == '__main__':
unittest.main()
```
这个测试包含了四个测试用例:
- test_add_positive_numbers:测试正数相加是否正确计算
- test_add_negative_numbers:测试负数相加是否正确计算
- test_add_zero_to_number:测试0与数字相加是否正确计算
- test_add_float_numbers:测试浮点数相加是否正确计算
使用unittest.TestCase的assertEqual方法来验证计算结果是否符合预期。运行测试时,可以使用unittest.main()方法。