[Python-checkins] python/dist/src/Lib/test test_heapq.py,1.7,1.8

Thu Jun 10 01:03:19 EDT 2004

Update of /cvsroot/python/python/dist/src/Lib/test
In directory sc8-pr-cvs1.sourceforge.net:/tmp/cvs-serv16530/Lib/test
Modified Files:
	test_heapq.py 
Log Message:
SF patch #969791: Add nlargest() and nsmallest() to heapq.
Index: test_heapq.py
===================================================================
RCS file: /cvsroot/python/python/dist/src/Lib/test/test_heapq.py,v
retrieving revision 1.7
retrieving revision 1.8
diff -C2 -d -r1.7 -r1.8
*** test_heapq.py	7 Dec 2002 10:33:42 -0000	1.7
--- test_heapq.py	10 Jun 2004 05:03:17 -0000	1.8
***************
*** 3,7 ****
 from test.test_support import verify, vereq, verbose, TestFailed

! from heapq import heappush, heappop, heapify, heapreplace
 import random

--- 3,7 ----
 from test.test_support import verify, vereq, verbose, TestFailed

! from heapq import heappush, heappop, heapify, heapreplace, nlargest, nsmallest
 import random

***************
*** 85,88 ****
--- 85,97 ----
 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: