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（）所做的事情是每个用例都需要的。