mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-03 08:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

SF patch #969791: Add nlargest() and nsmallest() to heapq.

Browse source
This commit is contained in:
Raymond Hettinger 2004-06-10 05:03:17 +00:00
parent 7d019664d7
commit 33ecffb65a
5 changed files with 75 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Lib/test/test_heapq.py
View file

@ -2,7 +2,7 @@
from test.test_support import verify, vereq, verbose, TestFailed
from heapq import heappush, heappop, heapify, heapreplace
from heapq import heappush, heappop, heapify, heapreplace, nlargest, nsmallest
import random
def check_invariant(heap):
@ -84,6 +84,15 @@ def test_main():
data.sort()
sorted = [heappop(heap) for i in range(size)]
vereq(data, sorted)
# 7) Check nlargest() and nsmallest()
data = [random.randrange(2000) for i in range(1000)]
copy = data[:]
copy.sort(reverse=True)
vereq(nlargest(data, 400), copy[:400])
copy.sort()
vereq(nsmallest(data, 400), copy[:400])
# Make user happy
if verbose:
print "All OK"