python单元测试--这个必须有

最新推荐文章于 2026-04-01 09:08:25 发布

原创最新推荐文章于 2026-04-01 09:08:25 发布 · 1.1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #单元测试

Python学习总结专栏收录该内容

10 篇文章

订阅专栏

本文强调了单元测试在软件开发中的重要性，指出单元测试能确保代码模块正确行为，并为重构提供信心。介绍了测试用例应覆盖各种输入组合、边界条件和异常情况。文章通过一个Dict类的实现及测试代码示例，展示了如何编写简洁的测试代码，并讲解了unittest.TestCase的使用，包括测试方法命名规则、断言方法如assertEquals()以及如何期待抛出错误。此外，还提及了setUp()和tearDown()方法在管理测试环境中的作用，如在测试前后初始化和清理资源。

工作后忽然发现，对于一个开发者，自己的代码一定要做好充分的单元测试才能提交代码。

单元测试可以有效地测试某个程序模块的行为，是未来重构代码的信心保证。
单元测试的测试用例要覆盖常用的输入组合、边界条件和异常。
单元测试代码要非常简单，如果测试代码太复杂，那么测试代码本身就可能有bug。
单元测试通过了并不意味着程序就没有bug了，但是不通过程序肯定有bug。

其他的不多说，直接看一个测试：

我们来编写一个Dict类，这个类的行为和dict一致，但是可以通过属性来访问，用起来就像下面这样：

>>> d = Dict(a=1, b=2)
>>> d['a']
1
>>> d.a
1

mydict.py的实现代码和测试代码如下：

class Dict(dict):
    def __init__(self, **kw):
        super(Dict, self).__init__(**kw)

    def __getattr__(self, key):
        try:
            return self[key]
        except KeyError:
            raise AttributeError(r"'Dict' object has no attribute '%s' " % key)

    def __setattr__(self, key, value):
        self[key] = value

if __name__ == "__main__":


    import unittest

    class TestDict(unittest.TestCase):
        """
        Unit test case for class Dict
        """

        def setUp(self):
            print 'setUp...'

        def tearDown(self):
            print 'tearDown...'

        def test_init(self):
            d = Dict(a=1, b='test')
            self.assertEquals(d.a, 1)
            self.assertEquals(d.b, 'test')
            self.assertTrue(isinstance(d, dict))

        def test_key(self):
            d = Dict()
            d['key'] = 'value'

            self.assertEquals(d.key, 'value')

        def test_attr(self):
            d = Dict()
            d.key = 'value'

            self.assertTrue('key' in d)
            self.assertEquals(d['key'], 'value')

        def test_keyerror(self):
            d = Dict()
            with self.assertRaises(KeyError):
                value = d['empty']

        def test_attrerror(self):
            d = Dict()
            with self.assertRaises(AttributeError):
                value = d.empty

    unittest.main()

有的一些地方，把测试用例代码和被测试的代码分开写，个人觉得不好管理。

编写单元测试时，我们需要编写一个测试类，从unittest.TestCase继承。
以test开头的方法就是测试方法，不以test开头的方法不被认为是测试方法，测试的时候不会被执行。
对每一类测试都需要编写一个test_xxx()方法。由于unittest.TestCase提供了很多内置的条件判断，我们只需要调用这些方法就可以断言输出是否是我们所期望的。最常用的断言就是assertEquals()。
另一种重要的断言就是期待抛出指定类型的Error，比如通过d[‘empty’]访问不存在的key时，断言会抛出KeyError。

通过在类最后调用unittest.main() 就可以执行测试用例代码。

可以在单元测试中编写两个特殊的setUp()和tearDown()方法。这两个方法会分别在每调用一个测试方法的前后分别被执行。

setUp()和tearDown()方法有什么用呢？设想你的测试需要启动一个数据库，这时，就可以在setUp()方法中连接数据库，在tearDown()方法中关闭数据库，这样，不必在每个测试方法中重复相同的代码。