def test_six(self, stdout):
     main()
     self.assertEqual(stdout.getvalue(), "\n".join([
         'NO',
     ]) + '\n')
 def test_four(self, stdout):
     main()
     self.assertEqual(stdout.getvalue(), "\n".join([
         'YES',
     ]) + '\n')