Python main示例

示例#1

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

示例#2

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