先来看一下单元测试的概念:
单元测试:单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。
单元测试:单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。
一、对函数进行测试
# 要测试的函数:
import unittest
def sum(x, y):
return x + y
def sub(x, y):
return x - y
# 测试该函数的测试类
class mytest(unittest.TestCase): # setUp()初始化, def setUp(self): pass # 销毁 def tearDown(self): pass # 具体的测试用例,以test开头 def testsum(self): self.assertEqual(sum(1, 2), 3, 'test sum fail') def testsub(self): self.assertEqual(sub(2, 1), 1, 'test sub fail') if __name__ == '__main__': unittest.main(2) 二、对类进行测试
#要测试的类
class Person(object): def __init__(self, name): self.name = name def my_add(self, a, b): return a+b
#对上面的类进行测试(测试脚本)
import unittest import person class PersonTest(unittest.TestCase): # 初始化 def setUp(self): self.p = person.Person('tom') def test_my_add(self): inlist = [(1, 2), (-1, -2), (0, 0)] qw = [3, -3, 0] for i, j in zip(inlist, qw): self.assertEqual(self.p.my_add(i[0], i[1]), j, 'test add fail') if __name__ == '__main__': unittest.main()