Issue 13274: Make the pure python code for heapq more closely match the C implementation for an undefined corner case.

rhettinger · rhettinger · commit e584457e2460 · 2011-10-30T14:32:54.000-07:00
diff --git a/Lib/heapq.py b/Lib/heapq.py
@@ -185,6 +185,8 @@ def nlargest(n, iterable):
 
     Equivalent to:  sorted(iterable, reverse=True)[:n]
     """
+    if n < 0:
+        return []
     it = iter(iterable)
     result = list(islice(it, n))
     if not result:
@@ -201,6 +203,8 @@ def nsmallest(n, iterable):
 
     Equivalent to:  sorted(iterable)[:n]
     """
+    if n < 0:
+        return []
     if hasattr(iterable, '__len__') and n * 10 <= len(iterable):
         # For smaller values of n, the bisect method is faster than a minheap.
         # It is also memory efficient, consuming only n elements of space.
