python 单元测试_python单元测试

个人在工作中用的不多。
编写单元测试，我们需要引入Python自带的unittest模块：
class Dict(dict):
def __init__(self, **kw):
super(Dict, self).__init__(**kw)
def __getattr__(self, key):
try:
return self[key]
except KeyError:
raise AttributeError(r"'Dict' object has no attribute '%s'" % key)
def __setattr__(self, key, value):
self[key] = value
编写单元测试类：
unittest中有很多方法测试类中的属性的方法。
# -*- coding: utf-8 -*-
import unittest
import Test
class TestDict(unittest.TestCase):
def test_init(self):
d = Test.Dict(a=1, b='test')
self.assertEqual(d['a'], 1)
self.assertEqual(d['b'], 'test')
self.assertTrue(isinstance(d, dict))
def test_key(self):
d = Test.Dict()
d['key'] = 'value'
self.assertEqual(d['key'], 'value')
def test_attr(self):
d = Test.Dict()
d.key = 'value'
self.assertTrue('key' in d)
self.assertEqual(d['key'], 'value')
def test_keyerror(self):
d = Test.Dict()
with self.assertRaises(KeyError):
value = d['empty']
def test_attrerror(self):
d = Test.Dict()
with self.assertRaises(AttributeError):
value = d.empty
if __name__=='__main__':
TestDict.test_attrerror()
setUp与tearDown
可以在单元测试中编写两个特殊的setUp()和tearDown()方法。这两个方法会分别在每调用一个测试方法的前后分别被执行。
可以将类中每一个方法都需要运行的代码放到其中。比如连接数据库等操作
def setUp(self):
print 'setUp...'
def tearDown(self):
print 'tearDown...'

原文作者：恋红尘，醉江山

原出处：网络

转载声明：以上内容来源于网络，版权归原作者所有，如来源信息有误或侵犯权益，请联系我们删除或授权事宜。本文仅作为交流使用,不涉及商用.