Reverse argument order for nsmallest() and nlargest().

Reads better when the iterable is a generator expression.
This commit is contained in:
Raymond Hettinger 2004-06-15 23:53:35 +00:00
parent 969297f488
commit aefde435ef
4 changed files with 9 additions and 9 deletions

View file

@ -92,13 +92,13 @@ class TestHeap(unittest.TestCase):
def test_nsmallest(self):
data = [random.randrange(2000) for i in range(1000)]
for i in (0, 1, 2, 10, 100, 400, 999, 1000, 1100):
self.assertEqual(nsmallest(data, i), sorted(data)[:i])
for n in (0, 1, 2, 10, 100, 400, 999, 1000, 1100):
self.assertEqual(nsmallest(n, data), sorted(data)[:n])
def test_largest(self):
data = [random.randrange(2000) for i in range(1000)]
for i in (0, 1, 2, 10, 100, 400, 999, 1000, 1100):
self.assertEqual(nlargest(data, i), sorted(data, reverse=True)[:i])
for n in (0, 1, 2, 10, 100, 400, 999, 1000, 1100):
self.assertEqual(nlargest(n, data), sorted(data, reverse=True)[:n])
def test_main(verbose=None):
test_classes = [TestHeap]