《selenium2 python 自动化测试实战》（20）——unittest单元测试框架解析

孟船长

发布于 2018-05-18 18:25:34

6950

发布于 2018-05-18 18:25:34

文章被收录于专栏：自动化测试实战自动化测试实战

unittest是展开自动化测试的基础——这个框架很重要！

我们先自己写一个测试类：

1、被测试类 Widthget.py：

# coding: utf-8

class Widthget:

    def __init__(self, size=(10, 10)):
        self._size = size

    def getSize(self):
        return self._size
    
    def reSize(self, width, height):
        if width < 0 or height < 0:
            raise ValueError, 'illegal size'
        self._size = (width, height)

2、测试类：

# coding: utf-8

from Widthget import Widthget
import unittest


# 执行测试的类
class WidthgetTestCase(unittest.TestCase):

    def setUp(self):
        self.width = Widthget()

    def testSize(self):
        self.assertEqual(self.width.getSize(), (10, 10))
    
    def testResize(self):
        self.width.reSize(20, 20)
        self.assertEqual(self.width.getSize(), (20, 20))
    
    def tearDown(self):
        self.width = None

# 构造测试集
def suite():
    tc = unittest.TestSuite()
    tc.addTest(Widthget("testSize"))
    tc.addTest(Widthget("testResize"))
    return tc
    
if __name__ == '__main__':
    # unittest.main()
    runner = unittest.TextTestRunner()
    runner.run(suite())

运行测试文件可以看到：

优化1：

我们没必要单独写构造测试集的方法，也就是说我们可以直接将suite()方法放到if __name__=='__main__':里面：

if __name__ == '__main__':

    tc = unittest.TestSuite()
    tc.addTest(Widthget("testSize"))
    tc.addTest(Widthget("testResize"))

    runner = unittest.TextTestRunner()
    runner.run()

执行测试文件，可以看到用例也是可以通过的。