unittest--python单元测试模块（一）

Hound1211

于 2018-07-18 10:57:00 发布

阅读量965

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： unittest Python 单元测试

本文链接：https://blog.youkuaiyun.com/ManyPeng/article/details/78059274

本文介绍了Python的unittest模块，用于进行单元测试。通过一个计算器类的加法方法为例，展示了如何编写测试代码并执行。unittest会在每个测试用例前后调用setUp和tearDown方法，确保测试环境的准备和清理。在Pycharm中，测试结果会清晰地显示测试用例的数量和是否通过。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

unittest是python中的单元测试模块，类似于java中的junit包，下面就简单介绍一下这个模块的使用方法

首先定义一个计算器类：calculator.py，其中有一个加法方法add

class Calculator:
    def __init__(self, a, b):
        self.a = int(a)
        self.b = int(b)
    
    def add(self):
        return self.a + self.b

接下来我们编写另外一个文件（单元测试代码与被测试代码分离开）：testadd.py

from calculator import Calculator
import unittest


class TestAdd(unittest.TestCase):
    def setUp(self):
        print "test case start"

    def tearDown(self):
        print "test case end"

    def test_add(self):
        s = Calculator(1, 2)
        self.assertEqual(s.add(), 3)

    def test_add2(self):
        s = Calculator(10, 20)
        self.assertEqual(s.add(), 30)


if __name__ == '__main__':
    unittest.main()

执行上述文件就会自动运行TestAdd类中的test_add和test_add2方法，在Pycharm中执行的结果如下：