Python使用unittest实现简单的单元测试实例

如果项目复杂,进行单元测试是保证降低出错率的好方法,Python提供的unittest可以很方便的实现单元测试,从而可以替换掉繁琐杂乱的main函数测试的方法,将测试用例、测试方法进行统一的管理和维护。

本文给出一个实例,很简单,看一下就明白了。

首先给出一个要测试的Python模块,代码如下:

# coding:utf8
'''
日期常用类

@author: www.crazyant.net
'''

def get_date_year_month(pm_date):
    """获取参数pm_date对应的年份和月份
    """
    if not pm_date:
        raise Exception("get_curr_year_month: pm_date can not be None")

    # get date's yyyymmddHHMMSS pattern
    str_date = str(pm_date).replace("-", "").replace(" ", "").replace(":", "")

    year = str_date[:4]
    month = str_date[4:6]
    return year, month

然后就可以编写测试脚本,代码如下:

# coding: utf8

"""
测试date_service.py

@author: peishuaishuai
"""

import unittest

from service import date_service

class DateServiceTest(unittest.TestCase):
    """
    test clean_tb_async_src_acct.py
    """

    def setup(self):
        """在这里做资源的初始化 """
        pass

    def tearDown(self):
        """在这里做资源的释放 """
        pass

    def test_get_date_year_month_1(self):
        """ 测试方法1,测试方法应该以test_开头 """

        pm_date = "2015-11-25 14:40:52"
        year, month = date_service.get_date_year_month(pm_date)
        self.assertEqual(year, "2015", "year not equal")
        self.assertEqual(month, "11", "month not equal")

    def test_get_date_year_month_2(self):
        """ 测试方法1,测试方法应该以test_开头 """
        pm_date = "20161225144052"
        year, month = date_service.get_date_year_month(pm_date)
        self.assertEqual(year, "2016", "year not equal")
        self.assertEqual(month, "12", "month not equal")

# test main
if __name__ == "__main__":
    unittest.main()

运行这个test_date_service.py,就会打印出如下信息:

..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

这里的每一个点,就代表运行成功了一个测试,最后会给出运行成功了全部的多少个测试以及测试的时间。

之前的很多时间,我一直不知道写单测有什么用,因为单测只是把写好的程序运行了一遍,并没有创建新的逻辑,我心里在疑惑“我已经将程序按照我的想法写好了,它就会按照我的设计来运行,为什么要用单测重新走一遍呢?”,后来出了一个事情,代码出了BUG,我调试了好久,才发现问题出在"obja.equals(objb)",因为obja和objb一个是Long一个是Integer,所以即使数值相同,也不会相等。

从那一刻,我发现单测做的事情,其实就是“验证程序是否按照我的想法在运行”,这才是它的终极目的,但是,这却是很关键的事情,设计往往没有错,但是写出来的代码却经常并不是按照我们所想的去运行的。

单测,就是验证代码是不是按照我们想象的在运行,这也是单测这个技术的意义所在。

本文地址:http://crazyant.net/1890.html,转载请注明来源。

相关推荐

Leave a Comment