diff options
author | Raymond Hettinger <python@rcn.com> | 2014-03-26 09:00:54 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2014-03-26 09:00:54 (GMT) |
commit | 8f2420c94b350cf37b0ac06bda5539f754c1b469 (patch) | |
tree | 41a9441a4b315dbeaa4401d4e0d979573002a594 /Lib/heapq.py | |
parent | b0e69511936997810a1591a1e894910eaa44e3e3 (diff) | |
download | cpython-8f2420c94b350cf37b0ac06bda5539f754c1b469.zip cpython-8f2420c94b350cf37b0ac06bda5539f754c1b469.tar.gz cpython-8f2420c94b350cf37b0ac06bda5539f754c1b469.tar.bz2 |
Broaden the early-out test for nsmallest and nlargest
Diffstat (limited to 'Lib/heapq.py')
-rw-r--r-- | Lib/heapq.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Lib/heapq.py b/Lib/heapq.py index d615239..d52cd71 100644 --- 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)) |