python单元测试怎么写
时间: 2023-10-04 21:12:34 浏览: 43
在 Python 中,可以使用内置的 unittest 模块来编写单元测试。下面是一个简单的示例:
```python
import unittest
def add(x, y):
return x + y
class TestAdd(unittest.TestCase):
def test_add(self):
self.assertEqual(add(1, 2), 3)
self.assertEqual(add(-1, 1), 0)
self.assertEqual(add(0, 0), 0)
if __name__ == '__main__':
unittest.main()
```
上面的代码定义了一个名为 add 的函数,并编写了一个名为 TestAdd 的测试类,它继承自 unittest.TestCase 类。在 TestAdd 类中,定义了一个名为 test_add 的测试方法,它使用 assertEqual 方法来断言 add 函数的输出是否符合预期。最后,使用 unittest.main() 函数来运行测试。
你可以在命令行中运行此文件,或使用 IDE(如 PyCharm)中的测试运行器来运行单元测试。
当你运行此文件时,你应该会看到类似于以下内容的输出:
```
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
```
这表示测试通过了,如果有错误或失败,它们将在输出中显示。
阅读全文