[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)