Python + selenium之组织unittest单元测试用例

当增加被测功能和相应的测试用例之后unittest单元测试框架如何扩展和组织新增的测试用例的。

# coding =utf-8
# calculator

class Count ():
    def __init__(self, a, b):
        self.a = int (a)
        self.b = int (b)

    def add(self):
        return self.a + self.b

    def sub(self):
        return self.a - self.b

import unittest
from calculator import Count


class TestAdd (unittest.TestCase):
    def setUp(self):
        print("加法测试开始")

    def test_add(self):
        j = Count(2, 3)
        self.assertEqual (j.add (), 4, msg='加法不行等')

    def test_add2(self):
        j = Count(2, 4)
        self.assertEqual (j.add (), 5, msg='加法不行等')

    def tearDown(self):
        print("加法测试结束")


class TestSub (unittest.TestCase):
    def setup(self):
        print("减法测试开始")

    def test_sub(self):
        j = Count(10, 3)
        self.assertEqual (j.sub (), 18, msg='键法不行等')

    def test_sub2(self):
        j = Count(10, 2)
        self.assertEqual (j.sub (), 18, msg='键法不行等')

    def tearDown(self):
        print("减法测试结束")


if __name__ == '__main__':
    suite = unittest.TestSuite ()
    suite.addTest (TestAdd ('test_add'))
    suite.addTest (TestAdd ('test_add2'))
    suite.addTest (TestSub ('test_sub'))
    suite.addTest (TestSub ('test_sub2'))
    run = unittest.TextTestRunner ()
    run.run (suite)

 运行的结果如下：

上面中创建了TestAdd（）和TestSub（）两个类，分别测试calculator.py文件中的add（）和sub（）两个功能。通过TestSuite类的addTest（）方法把不同测试类中的测试方法组装到测试套件中。

通过。

从测试结果可以看出，setUp（）和tearDown（）方法分别作用于每个测试用例的开始和结束。如果每个类中的setUp和tearDown（）所做的事情是一样的，就可以将其封装再一个测试类中。

import unittest
from calculator import Count


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

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


class TestAdd (MyTest):
    def test_add(self):
        j = Count (2, 3)
        self.assertEqual (j.add (), 5, msg='加法不行等')

    def test_add2(self):
        j = Count (2, 4)
        self.assertEqual (j.add (), 6, msg='加法不行等')


class TestSub (MyTest):
    def test_sub(self):
        j = Count (10, 3)
        self.assertEqual (j.sub (), 7, msg='键法不行等')
        

    def test_sub2(self):
        j = Count (10, 2)
        self.assertEqual (j.sub (), 8, msg='键法不行等')


if __name__ == '__main__':
    suite = unittest.TestSuite ()
    suite.addTest (TestAdd ('test_add'))
    suite.addTest (TestAdd ('test_add2'))
    suite.addTest (TestSub ('test_sub'))
    suite.addTest (TestSub ('test_sub2'))
    run = unittest.TextTestRunner ()
    run.run (suite)
    # unittest.main()

创建MyTest（）类的好处显而易见，对于测试类和测试方法来说，应将注意力放在具体用例的编写商，无需关心SetUp()和TearDown（）所做的事情。前提条件是SetUp()和TearDown（）所做的事情是每个用例都需要的。