unittest是python中的单元测试模块,类似于java中的junit包,下面就简单介绍一下这个模块的使用方法
首先定义一个计算器类:calculator.py,其中有一个加法方法add
class Calculator:
def __init__(self, a, b):
self.a = int(a)
self.b = int(b)
def add(self):
return self.a + self.b
接下来我们编写另外一个文件(单元测试代码与被测试代码分离开):testadd.py
from calculator import Calculator
import unittest
class TestAdd(unittest.TestCase):
def setUp(self):
print "test case start"
def tearDown(self):
print "test case end"
def test_add(self):
s = Calculator(1, 2)
self.assertEqual(s.add(), 3)
def test_add2(self):
s = Calculator(10, 20)
self.assertEqual(s.add(), 30)
if __name__ == '__main__':
unittest.main()
执行上述文件就会自动运行TestAdd类中的test_add和test_add2方法,在Pycharm中执行的结果如下: