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