def test_rename(self):
d = {
'a': {
'x': 1,
'y': 1,
},
'b': {
'x': 2,
'y': 2,
},
'c': {
'x': 3,
'y': 3,
},
}
_rename(d, 'a', 'd')
r = {
'b': {
'x': 2,
'y': 2,
},
'c': {
'x': 3,
'y': 3,
},
'd': {
'x': 1,
'y': 1,
},
}
self.assertEqual(d, r)
def test_rename_to_existing_name(self):
d = {
'a': 1,
'b': 2,
'c': 3,
}
with self.assertRaises(KeyError):
_rename(d, 'a', 'c')
def test_rename_with_same_key(self):
d = {
'a': 1,
'b': 2,
}
_rename(d, 'a', 'a')
r = {
'a': 1,
'b': 2,
}
self.assertEqual(d, r)
def rename(self, key, key_new):
"""
Rename a dict item key from 'key' to 'key_new'.
If key_new exists, a KeyError will be raised.
"""
_rename(self, key, key_new)