diff --git a/Lib/test/test_heapq.py b/Lib/test/test_heapq.py --- a/Lib/test/test_heapq.py +++ b/Lib/test/test_heapq.py @@ -357,13 +357,11 @@ def test_get_only(self): for f in (self.module.heapify, self.module.heappop): - self.assertRaises(TypeError, f, GetOnly()) + self.assertRaises((TypeError, AttributeError), f, GetOnly()) for f in (self.module.heappush, self.module.heapreplace): - self.assertRaises(TypeError, f, GetOnly(), 10) - for f in (self.module.nlargest, self.module.nsmallest): - self.assertRaises(TypeError, f, 2, GetOnly()) + self.assertRaises((TypeError, AttributeError), f, GetOnly(), 10) - def test_get_only(self): + def test_cmp_err(self): seq = [CmpErr(), CmpErr(), CmpErr()] for f in (self.module.heapify, self.module.heappop): self.assertRaises(ZeroDivisionError, f, seq)