From e29a10306c146247cd66f9a77c93817d50d93ca0 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Wed, 11 Jun 2008 13:14:50 +0000 Subject: [PATCH] Add test for heapq using both __lt__ and __le__. --- Lib/test/test_heapq.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/Lib/test/test_heapq.py b/Lib/test/test_heapq.py index fec027e3b6..6f05e171fa 100644 --- a/Lib/test/test_heapq.py +++ b/Lib/test/test_heapq.py @@ -196,6 +196,27 @@ class TestHeapPython(TestHeap): class TestHeapC(TestHeap): module = c_heapq + def test_comparison_operator(self): + # Issue 3501: Make sure heapq works with both __lt__ and __le__ + def hsort(data, comp): + data = map(comp, data) + self.module.heapify(data) + return [self.module.heappop(data).x for i in range(len(data))] + class LT: + def __init__(self, x): + self.x = x + def __lt__(self, other): + return self.x > other.x + class LE: + def __init__(self, x): + self.x = x + def __lt__(self, other): + return self.x >= other.x + data = [random.random() for i in range(100)] + target = sorted(data, reverse=True) + self.assertEqual(hsort(data, LT), target) + self.assertEqual(hsort(data, LE), target) + #============================================================================== -- 2.50.0