python 测试用例

mydict.py代码如下：

 class Dict(dict):

     def __init__(self, **kw):
         super().__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

编写mydict_test.py如下：

 import unittest

 from Dict import Dict

 class TestDict(unittest.TestCase):

     def setUp(self):
         print('setUp...')

     def tearDown(self):
         print('tearDown...')

     def test_init(self):
         d = 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 = Dict()
         d['key'] = 'value'
         self.assertEqual(d.key, 'value')

     def test_attr(self):
         d = Dict()
         d.key = 'value'
         self.assertTrue('key' in d)
         self.assertEqual(d['key'], 'value')

     def test_keyerror(self):
         d = Dict()
         with self.assertRaises(KeyError):
             value = d['empty']
             print(value)

     def test_attrerror(self):
         d = Dict()
         with self.assertRaises(AttributeError):
             value = d.empty
             print(value)

 if __name__ == '__main__':
     unittest.main()