单元测试
单元测试是用来对一个模块、一个函数或者一个类来进行正确性检验的测试工作。
待测试用例
# 待测试用例
class MyDict(dict):
def __init__(self, **kw):
super().__init__(**kw)
def __getattr__(self, key):
try:
return self[key]
except KeyError:
# 主动抛出异常
raise AttributeError("has no attribute %s " % (key))
单元测试用例
import unittest
# 单元测试用例
class TestDict(unittest.TestCase):
# 每一类测试都需要满足test为开头 test_xxx()
def test_init(self):
# 输入测试数据
d = MyDict(a =1, b="test_init")
# unittest.TestCase的断言方法
self.assertEqual(d.a, 1)
self.assertEqual(d.b, "test_init")
self.assertTrue(isinstance(d, dict))
def test_get(self):
# 输入测试数据
d = MyDict(a = 123)
self.assertEqual(d.a, 123)
#self.assertEqual(d.a, 1111)
运行单元测试用例
# 运行方法
def runTest():
# 加上这2行,即单元测试用来运行
if __name__ == "__main__":
unittest.main()
# 运行
runTest()
运行结果
D:\PythonProject>python main.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
感觉很妙,单元测试确实很爽
数据库单元测试setUp()和tearDown()
import unittest
# 单元测试用例
class TestDict(unittest.TestCase):
def setUp(self):
# 测试数据库中,进行连接数据库
print("setUp")
def tearDown(self):
# 测试数据库中,关闭数据库
print("tearDown")
运行打印如下
D:\PythonProject>python main.py
setUp
tearDown
.setUp
tearDown
.