def test_DotDict_keys(self):
"""Test DotDict get_keys method"""
res = {'a': [{'b': 1, 'c': 1}, {'c': 2}]}
mdict = DotDict(res)
expect = ['a.b', 'a.c']
result = [r for r in mdict.get_keys('a')]
self.assertEqual(set(expect), set(result))
res = {
'a': [{
'b': [{
'c': 2
}, {
'c': {
'd': 1
}
}]
}, {
'b': [{
'c': 4
}, {
'c': 5
}]
}]
}
mdict = DotDict(res)
expect = ['a.b', 'a.b.c', 'a.b.c.d']
result = [r for r in mdict.get_keys('a')]
self.assertEqual(set(expect), set(result))
def test_get_keys(self):
"""test get_keys method"""
rec = DotDict(self.rec1)
expect = ['a', 'a.b', 'a.c']
expect.sort()
result = rec.get_keys()
result.sort()
self.assertEqual(expect, result)
rec = DotDict(self.rec2)
result = rec.get_keys()
result.sort()
self.assertEqual(expect, result)
def test_DotDict_list(self):
"""Test DotDict class"""
res = {'a': [{'b': 1, 'c': 1}, {'c': 1}]}
mdict = DotDict(res)
expect = 1
result = mdict.get('a.b')
self.assertEqual(expect, result)
res = {'a': [{'c': 1}, {'b': 1, 'c': 1}]}
mdict = DotDict(res)
expect = 1
result = mdict.get('a.b')
self.assertEqual(expect, result)
def test_DotDict(self):
"""Test DotDict class"""
res = {u'zip': {u'code': u'14850'}}
mdict = DotDict(res)
mdict['zip.code'] = 14850
expect = {u'zip': {u'code': 14850}}
self.assertEqual(expect, mdict)
res = {'a': {'b': {'c': 10}, 'd': 10}}
mdict = DotDict(res)
mdict['x.y.z'] = 10
expect = {'a': {'b': {'c': 10}, 'd': 10}, 'x': {'y': {'z': 10}}}
self.assertEqual(expect, mdict)
mdict['a.b.k.m'] = 10
expect = {
'a': {
'b': {
'c': 10,
'k': {
'm': 10
}
},
'd': 10
},
'x': {
'y': {
'z': 10
}
}
}
self.assertEqual(expect, mdict)
expect = 10
result = mdict.get('a.b.k.m')
self.assertEqual(expect, result)
res = {'a': {'b': {'c': 10}, 'd': [{'x': 1}, {'x': 2}]}}
mdict = DotDict(res)
expect = 1
result = mdict.get('a.d.x')
self.assertEqual(expect, result)
expect = None
result = mdict.get('a.M.Z')
self.assertEqual(expect, result)
res = {'a': {'b': {'c': 1, 'd': 2}}}
mdict = DotDict(res)
expect = {'a': {'b': {'c': 1}}}
mdict.delete('a.b.d')
self.assertEqual(expect, mdict)
def test_get(self):
"""test get method"""
rec = DotDict(self.rec1)
expect = [1, 2]
result = rec.get('a.c')
self.assertEqual(expect, result)
self.assertEqual(expect, rec['a.c'])
def test_DotDict_values(self):
"""Test DotDict get_values method"""
res = {'a': [{'b': 1, 'c': 1}, {'c': 2}]}
mdict = DotDict(res)
expect = [1]
result = [r for r in mdict.get_values('a.b')]
self.assertEqual(expect, result)
expect = [1, 2]
result = [r for r in mdict.get_values('a.c')]
self.assertEqual(expect, result)
res = {'a': [{'b': [{'c': 2}, {'c': 3}]}, {'b': [{'c': 4}, {'c': 5}]}]}
mdict = DotDict(res)
expect = [2, 3, 4, 5]
result = [r for r in mdict.get_values('a.b.c')]
self.assertEqual(expect, result)
def test_delete(self):
"""test delete method"""
rec = DotDict(self.rec1)
rec.delete('a.c')
expect = {'a': {'b': 1}}
self.assertEqual(expect, rec)
def test_set(self):
"""test set method"""
rec = DotDict(self.rec1)
rec['a.d'] = 111
self.assertEqual(rec['a.d'], 111)
self.assertRaises(Exception, rec.set, ('a.c.d', 1))
def test_get_values(self):
"""test get_values method"""
rec = DotDict(self.rec2)
expect = [1, 2, 10, 20]
result = [o for o in rec.get_values('a.c')]
self.assertEqual(expect, result)
