From: Raymond Hettinger <python@rcn.com>
Date: Wed, 11 Jun 2008 13:14:50 +0000 (+0000)
Subject: Add test for heapq using both __lt__ and __le__.
X-Git-Tag: v2.6b1~83
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=e29a10306c146247cd66f9a77c93817d50d93ca0;p=python

Add test for heapq using both __lt__ and __le__.
---

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)
+
 
 #==============================================================================