[b0043] python 归纳 (二八)_python测试使用快速上手

参考

Python必会的单元测试框架 —— unittest

 # -*- coding: utf-8 -*-
 """
 测试代码组织,整理( 非可执行)

 """

 import unittest
 from test_mathfunc import TestMathFunc
 from HTMLTestRunner import HTMLTestRunner

 ### 1

 def add(a, b):
     return a+b

 def minus(a, b):
     return a-b

 def multi(a, b):
     return a*b

 def divide(a, b):
     return a/b

 ### 2
 class TestMathFunc(unittest.TestCase):
     """Test mathfuc.py"""

     ## 1

     @classmethod
     def setUpClass(cls):
         print "This setUpClass() method only called once."

     @classmethod
     def tearDownClass(cls):
         print "This tearDownClass() method only called once too."

     ## 2
     def setUp(self):
         print "do something before test.Prepare environment."

     def tearDown(self):
         print "do something after test.Clean up."

     ## 3
     def test_add(self):
         """Test method add(a, b)"""
         print "add"
         self.assertEqual(3, add(1, 2))
         self.assertNotEqual(3, add(2, 2))

     def test_minus(self):
         """Test method minus(a, b)"""
         print "minus"
         self.assertEqual(1, minus(3, 2))

     def test_multi(self):
         """Test method multi(a, b)"""
         print "multi"
         self.assertEqual(6, multi(2, 3))

     @unittest.skip("I don't want to run this case.")
     def test_divide(self):
         """Test method divide(a, b)"""
         print "divide"
         self.assertEqual(2, divide(6, 3))
         self.assertEqual(2.5, divide(5, 2))

 if __name__ == '__main__':
     suite = unittest.TestSuite()
     suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestMathFunc))
     # suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestMathFunc2))
     # suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestMathFunc3))

     # tests = [TestMathFunc("test_add"), TestMathFunc("test_minus"), TestMathFunc("test_divide")]
     # suite.addTests(tests)

     with open('HTMLReport.html', 'w') as f:
         # runner = unittest.TextTestRunner(stream=f, verbosity=2)
         runner = HTMLTestRunner(stream=f,
                                 title='MathFunc Test Report',
                                 description='generated by HTMLTestRunner.',
                                 verbosity=2
                                 )
         runner.run(suite)