def test_select_all(self, input): self.assertEqual(prompt(9), list(range(9)))
def test_complex_selection(self, input): self.assertEqual(prompt(9), [0, 2, 3, 4, 9])
def test_select_two(self, input): self.assertEqual(prompt(9), [0, 4])
def test_select_range(self, input): self.assertEqual(prompt(9), [0, 1, 2, 3, 4])
def test_select_one(self, input): self.assertEqual(prompt(9), [2])