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')