From bfd334a42d35e58369e3a53e16733234e2741975 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Sat, 22 Nov 2003 03:55:23 +0000 Subject: Extend temporary hashability to remove() and discard(). Brings the functionality back in line with sets.py. --- Lib/test/test_set.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'Lib/test/test_set.py') diff --git a/Lib/test/test_set.py b/Lib/test/test_set.py index 3203d51693..85f87f7d12 100644 --- a/Lib/test/test_set.py +++ b/Lib/test/test_set.py @@ -182,12 +182,22 @@ class TestSet(TestJointOps): self.assert_('a' not in self.s) self.assertRaises(KeyError, self.s.remove, 'Q') self.assertRaises(TypeError, self.s.remove, []) + s = self.thetype([frozenset(self.word)]) + self.assert_(self.thetype(self.word) in s) + s.remove(self.thetype(self.word)) + self.assert_(self.thetype(self.word) not in s) + self.assertRaises(KeyError, self.s.remove, self.thetype(self.word)) def test_discard(self): self.s.discard('a') self.assert_('a' not in self.s) self.s.discard('Q') self.assertRaises(TypeError, self.s.discard, []) + s = self.thetype([frozenset(self.word)]) + self.assert_(self.thetype(self.word) in s) + s.discard(self.thetype(self.word)) + self.assert_(self.thetype(self.word) not in s) + s.discard(self.thetype(self.word)) def test_pop(self): for i in xrange(len(self.s)): -- cgit v1.2.1