def test_exit_msg_negative(self): with self.assertRaises(SystemExit) as cm: curator.exit_msg(False) self.assertEqual(cm.exception.code, 1)
def test_exit_msg_positive(self): with self.assertRaises(SystemExit) as cm: curator.exit_msg(True) self.assertEqual(cm.exception.code, 0)