def test_long_random(self): a = random.sample(range(100000), 20000) r1 = lab9.part1(a, 10000) r2 = lab9.part2(a, 10000) r = sorted(a)[-10000:] self.assertEqual(r, sorted(r1)) self.assertEqual(r, sorted(r2))
def test_random(self): a = random.sample(range(100), 50) + random.sample(range(100), 50) r1 = lab9.part1(a, 40) r2 = lab9.part2(a, 40) r = sorted(a)[-40:] self.assertEqual(r, sorted(r1)) self.assertEqual(r, sorted(r2))
def test_ones(self): length = 100000 k = 10000 a = [1 for i in range(length)] r1 = lab9.part1(a, k) r2 = lab9.part2(a, k) r = a[-k:] self.assertEqual(r, r1) self.assertEqual(r, r2)