python unittest

直接上代码===

mathfunc类中有如下几个函数

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

test_mathfunc类如下：

#coding=utf-8
'''
Created on 2018年8月24日

@author: yanerfree
'''
import unittest
from mathfunc import *
import logging
import logging.config
import HTMLTestRunner
import time

conf_path = "../conf/log.conf"
logging.config.fileConfig(conf_path)
logger = logging.getLogger()

class TestMathFunc(unittest.TestCase):
    """Test mathfunc.py"""
      
    @classmethod
    def setUpClass(cls):
        print "This setUpClass() method only called once."

    @classmethod
    def tearDownClass(cls):
        print "This tearDownClass() method only called once too."
         
    def setUp(self):
        print "do something before test"
        
    def test_add(self):
        """test method add(a,b)"""
        logger.info("add")
        self.assertEqual(3, add(1,2), "1+2=3 ?")
        self.assertNotEqual(3, add(2,2), "2+2!=3 ?")

    def test_minus(self):
        """test method minus(a,b)"""
        logger.info("minus")
        self.assertEqual(1, minus(3,2), "3-2=1 ?")
        self.assertNotEqual(2, minus(3,2), "3-2!=2 ?")
        
    def tearDown(self):
        print "do something after test"

if __name__ == '__main__':
    #html Report
    tamp =int(time.strftime('%Y%m%d%H%M', time.localtime(time.time())))
    file_path = file(r'./%s_result.html'%str(tamp),'wb')
    runner = HTMLTestRunner.HTMLTestRunner(file_path)
    suite = unittest.TestSuite()
    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestMathFunc))

    runner.run(suite)
    '''
    #show on console
    suite = unittest.TestSuite()
    suite.addTest(TestMathFunc("test_add"))
    suite.addTest(TestMathFunc("test_minus"))
    
    unittest.TextTestRunner(verbosity=2).run(suite)
    '''
'''
if __name__ == '__main__':
    #save to file
    suite = unittest.TestSuite()
    suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestMathFunc))

    with open('UnittestTextReport.txt', 'a') as f:
        runner = unittest.TextTestRunner(stream=f, verbosity=2)
        runner.run(suite)
 '''