Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2026-01-06 15:32:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix argument order in pure python version of nsmallest() and nlargest().

Browse source
This commit is contained in:
Raymond Hettinger 2004-11-29 05:54:48 +00:00
parent f4c7c402d4
commit e1defa4175
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
Lib/heapq.py
View file

@ -175,7 +175,7 @@ def heapify(x):
for i in reversed(xrange(n//2)):
_siftup(x, i)
def nlargest(iterable, n):
def nlargest(n, iterable):
"""Find the n largest elements in a dataset.
Equivalent to: sorted(iterable, reverse=True)[:n]
@ -195,7 +195,7 @@ def nlargest(iterable, n):
result.sort(reverse=True)
return result
def nsmallest(iterable, n):
def nsmallest(n, iterable):
"""Find the n smallest elements in a dataset.
Equivalent to: sorted(iterable)[:n]

7
Lib/test/test_heapq.py
View file

@ -39,8 +39,11 @@ def test_push_pop(self):
self.check_invariant(results)
self.assertRaises(TypeError, heappush, [])
self.assertRaises(TypeError, heappush, None, None)
self.assertRaises(TypeError, heappop, None)
try:
self.assertRaises(TypeError, heappush, None, None)
self.assertRaises(TypeError, heappop, None)
except AttributeError:
pass
def check_invariant(self, heap):
# Check the heap invariant.