Python + selenium之组织unittest单元测试用例
当增加被测功能和相应的测试用例之后unittest单元测试框架如何扩展和组织新增的测试用例的。
1 # coding =utf-8 2 # calculator 3 4 class Count (): 5 def __init__(self, a, b): 6 self.a = int (a) 7 self.b = int (b) 8 9 def add(self): 10 return self.a + self.b 11 12 def sub(self): 13 return self.a - self.b
1 import unittest 2 from calculator import Count 3 4 5 class TestAdd (unittest.TestCase): 6 def setUp(self): 7 print("加法测试开始") 8 9 def test_add(self): 10 j = Count(2, 3) 11 self.assertEqual (j.add (), 4, msg='加法不行等') 12 13 def test_add2(self): 14 j = Count(2, 4) 15 self.assertEqual (j.add (), 5, msg='加法不行等') 16 17 def tearDown(self): 18 print("加法测试结束") 19 20 21 class TestSub (unittest.TestCase): 22 def setup(self): 23 print("减法测试开始") 24 25 def test_sub(self): 26 j = Count(10, 3) 27 self.assertEqual (j.sub (), 18, msg='键法不行等') 28 29 def test_sub2(self): 30 j = Count(10, 2) 31 self.assertEqual (j.sub (), 18, msg='键法不行等') 32 33 def tearDown(self): 34 print("减法测试结束") 35 36 37 if __name__ == '__main__': 38 suite = unittest.TestSuite () 39 suite.addTest (TestAdd ('test_add')) 40 suite.addTest (TestAdd ('test_add2')) 41 suite.addTest (TestSub ('test_sub')) 42 suite.addTest (TestSub ('test_sub2')) 43 run = unittest.TextTestRunner () 44 run.run (suite)
运行的结果如下：
上面中创建了TestAdd（）和TestSub（）两个类，分别测试calculator.py文件中的add（）和sub（）两个功能。通过TestSuite类的addTest（）方法把不同测试类中的测试方法组装到测试套件中。
通过。
从测试结果可以看出，setUp（）和tearDown（）方法分别作用于每个测试用例的开始和结束。如果每个类中的setUp和tearDown（）所做的事情是一样的，就可以将其封装再一个测试类中。
1 import unittest 2 from calculator import Count 3 4 5 class MyTest (unittest.TestCase): 6 def setUp(self): 7 print("test case start") 8 9 def tearDown(self): 10 print("test case end") 11 12 13 class TestAdd (MyTest): 14 def test_add(self): 15 j = Count (2, 3) 16 self.assertEqual (j.add (), 5, msg='加法不行等') 17 18 def test_add2(self): 19 j = Count (2, 4) 20 self.assertEqual (j.add (), 6, msg='加法不行等') 21 22 23 class TestSub (MyTest): 24 def test_sub(self): 25 j = Count (10, 3) 26 self.assertEqual (j.sub (), 7, msg='键法不行等') 27 28 29 def test_sub2(self): 30 j = Count (10, 2) 31 self.assertEqual (j.sub (), 8, msg='键法不行等') 32 33 34 if __name__ == '__main__': 35 suite = unittest.TestSuite () 36 suite.addTest (TestAdd ('test_add')) 37 suite.addTest (TestAdd ('test_add2')) 38 suite.addTest (TestSub ('test_sub')) 39 suite.addTest (TestSub ('test_sub2')) 40 run = unittest.TextTestRunner () 41 run.run (suite) 42 # unittest.main()
创建MyTest（）类的好处显而易见，对于测试类和测试方法来说，应将注意力放在具体用例的编写商，无需关心SetUp()和TearDown（）所做的事情。前提条件是SetUp()和TearDown（）所做的事情是每个用例都需要的。
