Broaden the early-out test for nsmallest and nlargest

8f2420c9 · Raymond Hettinger · b0e69511 · 8f2420c9
Kaydet (Commit) 8f2420c9 authored Mar 26, 2014 tarafından Raymond Hettinger
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

heapq.py Lib/heapq.py +2 -2

No files found.
--- a/Lib/heapq.py
+++ b/Lib/heapq.py
@@ -197,7 +197,7 @@ def nlargest(n, iterable):
    Equivalent to:  sorted(iterable, reverse=True)[:n]
    """
-    if n < 0:
+    if n <= 0:
        return []
    it = iter(iterable)
    result = list(islice(it, n))
@@ -215,7 +215,7 @@ def nsmallest(n, iterable):
    Equivalent to:  sorted(iterable)[:n]
    """
-    if n < 0:
+    if n <= 0:
        return []
    it = iter(iterable)
    result = list(islice(it, n))