Broaden the early-out test for nsmallest and nlargest

diff --git a/Lib/heapq.py b/Lib/heapq.py
index d615239b94608a89ffd570724953347d7e0cd675..d52cd715e12d9fbeec70553e0e45c9f878c31503 100644 (file)
--- 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))