单元测试其实就是指自己测自己写的代码,用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。单元测试常用的单元测试框架有unittest、pytest。
一、unittest框架
编写测试用例步骤:
1.先导入unittest模块;
2.创建一个类(任意名)继承unittest.TestCase;
3.编写需要测试的代码对函数进行各方面的测试;类中的方法名不能随意取,必须得以test_ 开头,这样Python才能在运行的时候运行所有的以test_开头的方法。
import unittest
def add(a,b):
return a+b
class TestAdd(unittest.TestCase):
'''测试add方法'''
def test_add_normal(self):
'''正常测试加法的,by huoziyang'''
result = add(1,2)
self.assertEqual(3,result)
def test_add_error1(self):
'''测试失败使用'''
result = add(1,2)
self.assertEqual(4,result)
def test_add_error2(self):
'''测试失败有msg的'''
result = add(