def testRemove(self): ks = KeyedSet(firstLetter, ["apple", "avocado", "banana"]) for item in ["apple", "banana"]: ks.remove(item) for item in ["acacia", "cheese"]: try: ks.remove(item) except KeyError: self.assertTrue(True) except: self.assertTrue(False) else: self.assertTrue(False) self.assertEqual(ks, set(["avocado"])) # Proper key sets self.assertEqual(ks.keys(), ["a"]) self.assertEqual(ks.subset_by_key("a"), set(["avocado"])) self.assertEqual(ks.subset_by_key("b"), set()) self.assertEqual(ks.subset_by_key("c"), set())
def testRemove(self): ks = KeyedSet(firstLetter, ["apple", "avocado", "banana"]) for item in ["apple", "banana"]: ks.remove(item) for item in ["acacia", "cheese"]: try: ks.remove(item) except KeyError: self.assertTrue(True) except: self.assertTrue(False) else: self.assertTrue(False) self.assertEqual(ks, set(["avocado"])) # Proper key sets self.assertEqual(ks.keys(), ['a']) self.assertEqual(ks.subset_by_key("a"), set(["avocado"])) self.assertEqual(ks.subset_by_key("b"), set()) self.assertEqual(ks.subset_by_key("c"), set())