def test_elements(self):
"""Check if a hashset of items in the HashSet can be retrieved."""
hashset = HashSet()
assert hashset.elements() == []
hashset.add('A')
assert hashset.elements() == ['A']
hashset.add('tiny')
self.assertCountEqual(hashset.elements(), ['A', 'tiny'])
hashset.add('hashset')
self.assertCountEqual(hashset.elements(), ['A', 'tiny', 'hashset'])
def test_remove(self):
"""Check to see if items can be removed from the HashSet."""
hashset = HashSet()
hashset.add('Please')
hashset.add('dont')
hashset.add('remove')
hashset.add('me')
assert hashset.size == 4
hashset.remove('Please')
hashset.remove('dont')
assert hashset.size == 2
self.assertCountEqual(hashset.elements(), ['remove', 'me'])
hashset.remove('me')
assert hashset.size == 1
with self.assertRaises(KeyError):
hashset.remove('NOOOO')
