def testDiscardByKey(self): ks = KeyedSet(firstLetter, ["apple", "avocado", "banana"]) discarded = ks.discard_by_key("a") self.assertEqual(discarded, set(["apple", "avocado"])) self.assertEqual(ks.subset_by_key("a"), set()) self.assertEqual(ks, set(["banana"])) self.assertEqual(ks.discard_by_key("b"), set(["banana"])) self.assertTrue(not ks) try: self.assertEqual(ks.discard_by_key("a"), set()) except: self.assertTrue(False)
def testDiscardByKey(self): ks = KeyedSet(firstLetter, ["apple", "avocado", "banana"]) discarded = ks.discard_by_key('a') self.assertEqual(discarded, set(['apple', 'avocado'])) self.assertEqual(ks.subset_by_key('a'), set()) self.assertEqual(ks, set(['banana'])) self.assertEqual(ks.discard_by_key('b'), set(['banana'])) self.assertTrue(not ks) try: self.assertEqual(ks.discard_by_key('a'), set()) except: self.assertTrue(False)