def _run_main(self, commandline='', status=0): with self.assertRaises(SystemExit) as context: main(*commandline.split()) self.assertEqual(context.exception.code, status)
from beastling.cli import main import beastling.cli class NoneException (Exception): pass beastling.cli.wrap_errors = NoneException if __name__ == "__main__": # pragma: no cover main()
# coding: utf8 from __future__ import unicode_literals from beastling.cli import main if __name__ == "__main__": main()
def _run_main(commandline='', status=0): with pytest.raises(SystemExit) as context: main(*commandline.split()) assert context.code == status