openstack
/
deb-python-multi-key-dict
Code Issues Proposed changes

Merge pull request #10 from StephenKappel/master 

Fix __delitem__ to remove reference from given key
This commit is contained in:
Lukasz Forynski 2015-04-23 21:56:09 +01:00
parent d71f2404c6 659ba81680
commit bfda636ce8
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
multi_key_dict.py
View File

@ -147,6 +147,10 @@ class multi_key_dict(object):
key_type = str(type(k))
if (key_type in self.__dict__ and k in self.__dict__[key_type]):
del self.__dict__[key_type][k]
# remove the reference from the given key
del self.__dict__[key_type][key]
else:
raise KeyError(key)
@ -464,6 +468,7 @@ def test_multi_key_dict():
curr_len = len(m)
del m[12]
assert( len(m) == curr_len - 1 ), 'expected len(m) == %d' % (curr_len - 1)
assert(not m.has_key(12)), 'expected deleted key to no longer be found!'
# try again
try: