pymox是一套用于python代码unit test的框架. 官方链接如下:
https://code.google.com/p/pymox/
但是从官方的wiki上看到pymox的使用案例较少,而且比较简略.
这里按照通常使用情况整理下.
首先从最基本的框架说起.
创建一个unit test的test case, 在这里需要先import两个包
import mox
import unittest
然后创建一个对应的测试类.
class TestZoo(unittest.TestCase):
def setUp(self):
self.m = mox.Mox()
def tearDown(self):
self.m.UnsetStubs()
现在设定被测程序存放于zoo.py, 提供函数如下:
from datetime import date
def get_most_popular_animal():
month = get_month()
if month in [11,12,1]:
return 'penguin'
elif month in [6,7,8]:
return 'swan