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')