def test_input_seven(self, stdout): main() self.assertEqual(stdout.getvalue(), "\n".join([ '1', ]) + '\n')
def test_input_eight(self, stdout): main() self.assertEqual(stdout.getvalue(), "\n".join([ '5', ]) + '\n')
def test_input_seven(self, stdout): main() self.assertEqual(stdout.getvalue(), "\n".join([ '1', ]) + '\n')
def test_input_eight(self, stdout): main() self.assertEqual(stdout.getvalue(), "\n".join([ '5', ]) + '\n')