forked from openlp/openlp
raise exception when item not in list
This commit is contained in:
parent
aa899bfde0
commit
6bdadff334
@ -69,17 +69,14 @@ class CategoryActionList(object):
|
||||
"""
|
||||
Implement the __getitem__() method to make this class a dictionary type
|
||||
"""
|
||||
try:
|
||||
return self.actions[key][1]
|
||||
except:
|
||||
raise KeyError('Action "%s" does not exist.' % key)
|
||||
return self.actions[key][1]
|
||||
|
||||
def __contains__(self, item):
|
||||
def __contains__(self, key):
|
||||
"""
|
||||
Implement the __contains__() method to make this class a dictionary type
|
||||
"""
|
||||
for weight, action in self.actions:
|
||||
if action == item:
|
||||
if action == key:
|
||||
return True
|
||||
return False
|
||||
|
||||
@ -106,12 +103,6 @@ class CategoryActionList(object):
|
||||
self.index += 1
|
||||
return self.actions[self.index - 1][1]
|
||||
|
||||
def has_key(self, key):
|
||||
"""
|
||||
Implement the has_key() method to make this class a dictionary type
|
||||
"""
|
||||
return key in self
|
||||
|
||||
def append(self, action):
|
||||
"""
|
||||
Append an action
|
||||
@ -136,6 +127,7 @@ class CategoryActionList(object):
|
||||
if action[1] == remove_action:
|
||||
self.actions.remove(action)
|
||||
return
|
||||
raise ValueError('Action "%s" does not exist.' % remove_action)
|
||||
|
||||
|
||||
class CategoryList(object):
|
||||
|
@ -72,7 +72,7 @@ class TestCategoryActionList(TestCase):
|
||||
self.assertEqual(self.action1, returned_action)
|
||||
|
||||
# THEN: Test if an exception is raised when trying to access a non existing item.
|
||||
self.assertRaises(KeyError, self.list.__getitem__, 1)
|
||||
self.assertRaises(IndexError, self.list.__getitem__, 1)
|
||||
|
||||
def contains_test(self):
|
||||
"""
|
||||
@ -141,12 +141,17 @@ class TestCategoryActionList(TestCase):
|
||||
Test the remove() method
|
||||
"""
|
||||
# GIVEN: The list
|
||||
self.list.append(self.action1)
|
||||
|
||||
# WHEN: Delete an item from the list.
|
||||
self.list.remove(self.action1)
|
||||
|
||||
# THEN: Now the element should not be in the list anymore.
|
||||
self.assertFalse(self.action1 in self.list)
|
||||
|
||||
# THEN: Check if an exception is raised when trying to remove a not present action.
|
||||
self.assertRaises(ValueError, self.list.remove, self.action2)
|
||||
|
||||
|
||||
class TestActionList(TestCase, TestMixin):
|
||||
"""
|
||||
|
Loading…
Reference in New Issue
Block a user