]> granicus.if.org Git - python/commitdiff
Add test for merge stability
authorRaymond Hettinger <python@rcn.com>
Mon, 19 Feb 2007 07:30:21 +0000 (07:30 +0000)
committerRaymond Hettinger <python@rcn.com>
Mon, 19 Feb 2007 07:30:21 +0000 (07:30 +0000)
Lib/test/test_heapq.py

index b7c3ab2f12d5c968040cb86c2c167c8efb487131..b2bbfab0fa1586391091386b2240945a98b018f6 100644 (file)
@@ -111,6 +111,21 @@ class TestHeap(unittest.TestCase):
         self.assertEqual(sorted(chain(*inputs)), list(merge(*inputs)))
         self.assertEqual(list(merge()), [])
 
+    def test_merge_stability(self):
+        class Int(int):
+            pass
+        inputs = [[], [], [], []]
+        for i in range(20000):
+            stream = random.randrange(4)
+            x = random.randrange(500)
+            obj = Int(x)
+            obj.pair = (x, stream)
+            inputs[stream].append(obj)
+        for stream in inputs:
+            stream.sort()
+        result = [i.pair for i in merge(*inputs)]
+        self.assertEqual(result, sorted(result))
+
     def test_nsmallest(self):
         data = [(random.randrange(2000), i) for i in range(1000)]
         for f in (None, lambda x:  x[0] * 547 % 2000):